dsh-cite 0.1.1 → 0.2.0

package/README.md

@@ -11,7 +11,7 @@ DeepSeek Harness 参考文献工具插件：通过 Crossref API 查询文献元
 | `cite_lookup` | 查文献元数据（DOI 精确查询 / 题录检索） | `doi` 或 `query` 至少一个；`limit` 1-10 默认 5 |
 | `cite_format` | 生成规范引文 | `doi` 必填；`style`：gb-t-7714 / apa / mla / chicago |
 | `cite_bibtex` | 生成 BibTeX 条目 | `doi` 必填；`key` 可选 |
-| `cite_check` | 从文本提取 DOI 并校验是否存在 | `text` 必填；`maxChecks` 1-50 默认 10 |
+| `cite_check` | 从文本提取 DOI 并并发校验是否存在（并发 3，保持输入顺序） | `text` 必填；`maxChecks` 1-50 默认 10 |
 
 ## 安装
 
@@ -38,7 +38,8 @@ Agent：
 - id: cite
   name: 'dsh-cite'
   config:
-    timeoutMs: 15000   # Crossref 请求超时
+    timeoutMs: 15000   # Crossref 请求超时（也可用 DSH_CITE_TIMEOUT_MS）
+    # userAgent: ...   # 自定义 User-Agent（也可用 DSH_CITE_USER_AGENT）
     userAgent: ''      # 自定义 UA（建议带上可联系邮箱）
 ```
 
@@ -51,7 +52,7 @@ Agent：
 ## 开发
 
 ```bash
-pnpm test       # 构建 + 14 个测试
+pnpm test       # 构建 + 18 个测试
 ```
 
 MIT

package/lib/config.d.ts

@@ -11,7 +11,9 @@ export interface ResolvedCiteConfig {
     timeoutMs: number;
     userAgent: string;
 }
-export declare function resolveConfig(config: CiteConfig | undefined | null): ResolvedCiteConfig;
+export declare const CITE_TIMEOUT_ENV = "DSH_CITE_TIMEOUT_MS";
+export declare const CITE_USER_AGENT_ENV = "DSH_CITE_USER_AGENT";
+export declare function resolveConfig(config: CiteConfig | undefined | null, env?: NodeJS.ProcessEnv): ResolvedCiteConfig;
 export declare function optionalString(args: Record<string, unknown>, key: string): string | undefined;
 export declare function requiredString(args: Record<string, unknown>, key: string, label: string): string;
 export declare function optionalInteger(args: Record<string, unknown>, key: string, label: string, lo: number, hi: number, fallback: number): number;

package/lib/config.js

@@ -4,8 +4,10 @@
  * @module dsh-cite/config
  */
 const DEFAULT_TIMEOUT_MS = 15000;
-const DEFAULT_USER_AGENT = 'dsh-cite/0.1.0 (DeepSeek Harness citation plugin; mailto:STARDUSTLC666@users.noreply.github.com)';
-export function resolveConfig(config) {
+const DEFAULT_USER_AGENT = 'dsh-cite/0.2.0 (DeepSeek Harness citation plugin; mailto:STARDUSTLC666@users.noreply.github.com)';
+export const CITE_TIMEOUT_ENV = 'DSH_CITE_TIMEOUT_MS';
+export const CITE_USER_AGENT_ENV = 'DSH_CITE_USER_AGENT';
+export function resolveConfig(config, env = process.env) {
     const cfg = config ?? {};
     let timeoutMs = DEFAULT_TIMEOUT_MS;
     if (cfg.timeoutMs !== undefined) {
@@ -14,7 +16,13 @@ export function resolveConfig(config) {
         }
         timeoutMs = Math.min(120000, Math.max(2000, Math.round(cfg.timeoutMs)));
     }
-    const userAgent = typeof cfg.userAgent === 'string' && cfg.userAgent.trim() !== '' ? cfg.userAgent.trim() : DEFAULT_USER_AGENT;
+    else if (env[CITE_TIMEOUT_ENV]?.trim()) {
+        const parsed = Number(env[CITE_TIMEOUT_ENV].trim());
+        if (!Number.isFinite(parsed) || parsed <= 0)
+            throw new Error(CITE_TIMEOUT_ENV + ' 必须是大于 0 的数字（毫秒）。');
+        timeoutMs = Math.min(120000, Math.max(2000, Math.round(parsed)));
+    }
+    const userAgent = typeof cfg.userAgent === 'string' && cfg.userAgent.trim() !== '' ? cfg.userAgent.trim() : (env[CITE_USER_AGENT_ENV]?.trim() || DEFAULT_USER_AGENT);
     return { timeoutMs, userAgent };
 }
 export function optionalString(args, key) {

package/lib/crossref.d.ts

@@ -17,6 +17,7 @@ export interface Work {
     containerTitle: string;
     publisher: string;
     year: number;
+    language: string;
     volume: string;
     issue: string;
     page: string;

package/lib/crossref.js

@@ -52,6 +52,7 @@ function normalizeWork(raw) {
         containerTitle: firstText(raw['container-title']),
         publisher: typeof raw.publisher === 'string' ? raw.publisher.trim() : '',
         year: year || 0,
+        language: typeof raw.language === 'string' ? raw.language.trim().toLowerCase() : '',
         volume: typeof raw.volume === 'string' ? raw.volume.trim() : '',
         issue: typeof raw.issue === 'string' ? raw.issue.trim() : '',
         page: typeof raw.page === 'string' ? raw.page.trim() : '',

package/lib/format.js

@@ -203,6 +203,11 @@ export function buildCitation(work, style, lang = 'zh') {
 function escapeLatex(value) {
     return value.replace(/([{}])/g, '\\$1').replace(/&/g, '\\&').replace(/%/g, '\\%').replace(/\$/g, '\\$');
 }
+/** BibTeX 引用键只允许安全字符，空格/逗号/花括号等替换为下划线。 */
+function sanitizeBibtexKey(value) {
+    const cleaned = value.replace(/[^A-Za-z0-9_.:\/-]+/g, '_').replace(/^_+|_+$/g, '');
+    return cleaned === '' ? 'work' : cleaned;
+}
 function bibtexType(work) {
     switch (work.type) {
         case 'journal-article': return 'article';
@@ -220,7 +225,7 @@ function bibtexType(work) {
 export function buildBibtex(work, key) {
     const first = work.authors[0];
     const firstWord = titleOf(work).split(/[^\p{L}\p{N}]+/u).find((part) => part !== '') ?? 'work';
-    const baseKey = key !== undefined && key.trim() !== '' ? key.trim() : ((first?.family ?? first?.name ?? 'author') + yearText(work) + firstWord).toLowerCase();
+    const baseKey = key !== undefined && key.trim() !== '' ? sanitizeBibtexKey(key.trim()) : ((first?.family ?? first?.name ?? 'author') + yearText(work) + firstWord).toLowerCase();
     const type = bibtexType(work);
     const fields = [];
     const authors = work.authors.map((author) => fullName(author)).join(' and ');

package/lib/tools.js

@@ -39,6 +39,7 @@ const workSchema = {
         containerTitle: { type: 'string' },
         publisher: { type: 'string' },
         year: { type: 'integer' },
+        language: { type: 'string' },
         volume: { type: 'string' },
         issue: { type: 'string' },
         page: { type: 'string' },
@@ -72,6 +73,22 @@ const checkSchema = {
     properties: { count: { type: 'integer' }, results: { type: 'array', items: checkItemSchema } },
     additionalProperties: true,
 };
+/** 以固定并发度执行异步任务并保持原顺序。 */
+async function mapWithConcurrency(items, concurrency, fn) {
+    const results = new Array(items.length);
+    let next = 0;
+    const workers = Array.from({ length: Math.min(concurrency, items.length) }, async () => {
+        while (true) {
+            const index = next;
+            next += 1;
+            if (index >= items.length)
+                return;
+            results[index] = await fn(items[index], index);
+        }
+    });
+    await Promise.all(workers);
+    return results;
+}
 function workLabel(work) {
     const title = work.title !== '' ? work.title : '(无标题)';
     return title + '（' + (work.year > 0 ? work.year + ', ' : '') + (work.containerTitle !== '' ? work.containerTitle : work.publisher) + '）';
@@ -100,6 +117,11 @@ export function buildCiteTools(cfg, fetchImpl) {
                 const work = await lookupDoi(doi, cfg, fetchImpl);
                 return { count: 1, works: [work] };
             }
+            const queryDoi = extractDois(query)[0];
+            if (queryDoi !== undefined) {
+                const work = await lookupDoi(queryDoi, cfg, fetchImpl);
+                return { count: 1, works: [work], matchedDoi: queryDoi };
+            }
             const limit = optionalInteger(args, 'limit', '返回条数', 1, 10, 5);
             const works = await searchWorks(query, limit, cfg, fetchImpl);
             if (works.length === 0)
@@ -172,16 +194,15 @@ export function buildCiteTools(cfg, fetchImpl) {
             const text = requiredString(args, 'text', '待检查文本');
             const maxChecks = optionalInteger(args, 'maxChecks', '校验数量', 1, 50, 10);
             const dois = extractDois(text).slice(0, maxChecks);
-            const results = [];
-            for (const doi of dois) {
+            const results = await mapWithConcurrency(dois, 3, async (doi) => {
                 try {
                     const work = await lookupDoi(doi, cfg, fetchImpl);
-                    results.push({ doi, ok: true, title: work.title, url: work.url });
+                    return { doi, ok: true, title: work.title, url: work.url };
                 }
                 catch (error) {
-                    results.push({ doi, ok: false, title: '', url: '', error: error instanceof Error ? error.message : String(error) });
+                    return { doi, ok: false, title: '', url: '', error: error instanceof Error ? error.message : String(error) };
                 }
-            }
+            });
             return { count: results.length, results };
         },
         timeoutMs: cfg.timeoutMs * 3 + 5000,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "dsh-cite",
-  "version": "0.1.1",
-  "description": "DSH 参考文献工具插件：cite_lookup/cite_format/cite_bibtex/cite_check 四工具，Crossref 元数据查询，GB/T 7714 / APA / MLA / Chicago 格式化，零运行时依赖。",
+  "version": "0.2.0",
+  "description": "DSH 参考文献工具插件：cite_lookup/cite_format/cite_bibtex/cite_check 四工具，Crossref 元数据查询（含 DOI 精确解析与并发校验），GB/T 7714 / APA / MLA / Chicago 格式化，零运行时依赖。",
   "type": "module",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",