add: api.convert.item2citation

windingwind · Oct 26, 2023 · b35d450 · b35d450
1 parent ce8a130
commit b35d450
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 4 deletions.
diff --git a/src/api.ts b/src/api.ts
@@ -45,6 +45,7 @@ import {
 } from "./modules/template/data";
 import { renderTemplatePreview } from "./modules/template/preview";
 import { getWorkspaceEditor } from "./modules/workspace/content";
+import { parseCitationHTML } from "./utils/citation";
 import {
   getEditorInstance,
   insert,
@@ -98,6 +99,7 @@ const convert = {
   html2md,
   annotations2html,
   note2html,
+  item2citation: parseCitationHTML,
 };
 
 const template = {

diff --git a/src/modules/convert/api.ts b/src/modules/convert/api.ts
@@ -27,7 +27,6 @@ import {
   jointPath,
   randomString,
 } from "../../utils/str";
-import { parseCitationHTML } from "../../utils/citation";
 import {
   copyEmbeddedImagesInHTML,
   importImageToNote,
@@ -1079,7 +1078,7 @@ async function processM2NRehypeCitationNodes(
         const ids = dataCitation.citationItems.map((c: { uris: string[] }) =>
           Zotero.URI.getURIItemID(c.uris[0]),
         );
-        const html = await parseCitationHTML(ids);
+        const html = await addon.api.convert.item2citation(ids);
         if (html) {
           const newNode = note2rehype(html);
           // root -> p -> span(cite, this is what we actually want)

diff --git a/src/modules/template/api.ts b/src/modules/template/api.ts
@@ -97,7 +97,7 @@ async function runTemplate(
   } catch (e) {
     ztoolkit.log(e);
     if (options.dryRun) {
-      return String(e);
+      return "Template Preview Error: " + String(e);
     }
     window.alert(`Template ${key} Error: ${e}`);
     return "";

diff --git a/src/utils/citation.ts b/src/utils/citation.ts
@@ -1,4 +1,12 @@
-export async function parseCitationHTML(citationIds: number[]) {
+export async function parseCitationHTML(
+  citationIds: number[],
+  args: {
+    locator?: string;
+    label?: string;
+    prefix?: string;
+    suffix?: string;
+  }[] = [],
+) {
   let html = "";
   if (citationIds.length === 0 || !citationIds.every((id) => id)) {
     return null;
@@ -14,6 +22,7 @@ export async function parseCitationHTML(citationIds: number[]) {
     }
   }
 
+  let i = 0;
   for (const item of items) {
     if (item.isRegularItem()) {
       // @ts-ignore
@@ -22,6 +31,12 @@ export async function parseCitationHTML(citationIds: number[]) {
         citationItems: [
           {
             uris: [Zotero.URI.getItemURI(item)],
+            ...(args[i++] || {
+              locator: "",
+              label: "",
+              prefix: "",
+              suffix: "",
+            }),
             itemData,
           },
         ],