@a9i5k4/dsh-literature 0.2.0 → 0.2.2

package/lib/index.js

@@ -134,7 +134,12 @@ var init_config = __esm({
       entryMode: "auto",
       /** Watched folder: new PDFs here are auto-imported into the built-in library. */
       importDir: "",
-      watchImport: false
+      watchImport: false,
+      /** User-configured custom download sources (mirrors / institutional proxies).
+       *  Each: { id, label, urlTemplate, headers?, enabled, order }. URL templates
+       *  may use {doi} {arxiv} {isbn} {title} {url}. These are the user's own
+       *  configuration — the plugin never ships any such source itself. */
+      customSources: []
     };
     cached = null;
   }
@@ -1184,11 +1189,31 @@ function looksLikePdf(buffer) {
   if (!buffer || buffer.length < 8) return false;
   return buffer.subarray(0, 5).equals(PDF_MAGIC);
 }
-async function candidates(record, { unpaywallEmail } = {}) {
+function renderSourceTemplate(template, record) {
+  if (!template || !/^https?:\/\//i.test(String(template))) return "";
+  const vars = {
+    doi: record.doi ? encodeURIComponent(record.doi) : "",
+    arxiv: record.arxiv ? encodeURIComponent(record.arxiv) : "",
+    isbn: record.isbn ? encodeURIComponent(record.isbn) : "",
+    title: record.title ? encodeURIComponent(record.title) : "",
+    url: record.url ? encodeURIComponent(record.url) : ""
+  };
+  for (const [k, v] of Object.entries(vars)) {
+    if (!v && String(template).includes(`{${k}}`)) return "";
+  }
+  for (const [k, v] of Object.entries(vars)) {
+    if (!v && String(template).includes(`{${k}}`)) return "";
+  }
+  let out = String(template);
+  for (const [k, v] of Object.entries(vars)) out = out.split(`{${k}}`).join(v);
+  if (out.includes("{") || !/^https?:\/\//i.test(out)) return "";
+  return out;
+}
+async function candidates(record, { unpaywallEmail, customSources = [] } = {}) {
   const out = [];
-  const push2 = (url, source, kind = "pdf") => {
+  const push2 = (url, source, kind = "pdf", headers) => {
     if (!url) return;
-    out.push({ url: String(url), source, kind });
+    out.push({ url: String(url), source, kind, headers });
   };
   if (record.arxiv) {
     const base = String(record.arxiv).replace(/v\d+$/, "");
@@ -1231,6 +1256,10 @@ async function candidates(record, { unpaywallEmail } = {}) {
       warn("crossref link lookup failed:", e.message);
     }
   }
+  for (const src of [...customSources].filter((x) => x && x.enabled).sort((a, b) => (a.order ?? 100) - (b.order ?? 100))) {
+    const url = renderSourceTemplate(src.urlTemplate, record);
+    if (url) push2(url, src.label || src.id || "custom", "pdf", src.headers);
+  }
   const seen = /* @__PURE__ */ new Set();
   return out.filter((c) => {
     const k = c.url.toLowerCase();
@@ -1256,8 +1285,8 @@ function extractPdfUrlFromHtml(html, baseUrl2) {
   }
   return null;
 }
-async function fetchPdf(record, { timeoutMs = 3e4, unpaywallEmail = "" } = {}) {
-  const list = await candidates(record, { unpaywallEmail });
+async function fetchPdf(record, { timeoutMs = 3e4, unpaywallEmail = "", customSources = [] } = {}) {
+  const list = await candidates(record, { unpaywallEmail, customSources });
   if (!list.length) {
     const bookLike = ["book", "bookSection", "report", "thesis"].includes(record.itemType);
     throw new PdfFailure(
@@ -1273,7 +1302,8 @@ async function fetchPdf(record, { timeoutMs = 3e4, unpaywallEmail = "" } = {}) {
     try {
       const { buffer, contentType, finalUrl } = await httpGetBuffer(cand.url, {
         timeoutMs,
-        accept: "application/pdf,*/*;q=0.8"
+        accept: "application/pdf,*/*;q=0.8",
+        ...cand.headers && typeof cand.headers === "object" ? { headers: cand.headers } : {}
       });
       if (looksLikePdf(buffer)) {
         log(`pdf ok: ${cand.url} (${(buffer.length / 1024).toFixed(0)} KB via ${cand.source})`);
@@ -1966,7 +1996,7 @@ async function fetchItemPdf(key) {
     const result = await withRetry(
       (attempt) => {
         if (attempt > 1) emitTask({ id: task.id, state: "running", progress: 10, message: `\u7B2C ${attempt} \u6B21\u5C1D\u8BD5`, updatedAt: Date.now() });
-        return fetchPdf(current.record, { timeoutMs: config.fetchTimeoutMs, unpaywallEmail: config.unpaywallEmail });
+        return fetchPdf(current.record, { timeoutMs: config.fetchTimeoutMs, unpaywallEmail: config.unpaywallEmail, customSources: config.customSources });
       },
       { ...config.retry, label: `fetch ${key}` }
     );

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@a9i5k4/dsh-literature",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "description": "DSH Literature 文献侧窗：在 DeepSeek Harness 侧边栏识别 DOI/arXiv/标题、抓取元数据与全文、写入本地文献库或导出目录，并提供内置 PDF 阅读器（缩放/翻页/目录/搜索/高亮/笔记）。",
   "type": "module",
   "main": "lib/index.js",
@@ -13,6 +13,7 @@
     "lib",
     "cordis.patch.yml",
     "README.md",
+    "README_EN.md",
     "LICENSE"
   ],
   "scripts": {