From 4cdc4d4ada6f4882597c6fdcb5e04b76b124e950 Mon Sep 17 00:00:00 2001
From: xiangyu <3170102889@zju.edu.cn>
Date: Fri, 30 Sep 2022 11:06:34 +0800
Subject: [PATCH] fix: DOMParse bug

---
 src/events.ts    |  7 -------
 src/knowledge.ts |  4 ----
 src/parse.ts     | 48 ++++++++++++++++++++++++++++--------------------
 3 files changed, 28 insertions(+), 31 deletions(-)

diff --git a/src/events.ts b/src/events.ts
index 0ad1a13..372275f 100644
--- a/src/events.ts
+++ b/src/events.ts
@@ -5,12 +5,6 @@ import AddonBase from "./module";
 class AddonEvents extends AddonBase {
   notifierCallback: any;
   notifierCbkDict: any;
-  // Important!
-  // Due to unknown reasons, the DOMParser constructor fails after the tab is opened.
-  // We restore it from the preserved object constructor.
-  // We init this object when the addon is initialized for later use.
-  // See src/knowledge.ts
-  _DOMParser: DOMParser;
   constructor(parent: Knowledge4Zotero) {
     super(parent);
     this.notifierCallback = {
@@ -177,7 +171,6 @@ class AddonEvents extends AddonBase {
       },
     };
     this.notifierCbkDict = {};
-    this._DOMParser = new DOMParser();
   }
 
   public async onInit() {
diff --git a/src/knowledge.ts b/src/knowledge.ts
index 5081634..f2b8722 100644
--- a/src/knowledge.ts
+++ b/src/knowledge.ts
@@ -111,10 +111,6 @@ class Knowledge extends AddonBase {
       this._Addon.views.switchView(OutlineType.treeView);
       this._Addon.views.updateOutline();
     }
-    // Important!
-    // Due to unknown reasons, the DOMParser constructor fails after the tab is opened.
-    // We restore it from the preserved object constructor.
-    DOMParser = this._Addon.events._DOMParser.__proto__.constructor;
   }
 
   closeWorkspaceWindow() {
diff --git a/src/parse.ts b/src/parse.ts
index 6e07b71..7c541a2 100644
--- a/src/parse.ts
+++ b/src/parse.ts
@@ -7,6 +7,16 @@ const asciidoctor = require("asciidoctor")();
 const seedrandom = require("seedrandom");
 
 class AddonParse extends AddonBase {
+  private getDOMParser(): DOMParser {
+    if (Zotero.platformMajorVersion > 60) {
+      return new DOMParser();
+    } else {
+      return Components.classes[
+        "@mozilla.org/xmlextras/domparser;1"
+      ].createInstance(Components.interfaces.nsIDOMParser);
+    }
+  }
+
   // A seedable version of Zotero.Utilities.randomString
   private randomString(len: number, chars: string, seed: string) {
     if (!chars) {
@@ -23,6 +33,7 @@ class AddonParse extends AddonBase {
     }
     return randomstring;
   }
+
   public parseNoteTree(note: Zotero.Item): TreeModel.Node<object> {
     const noteLines = this._Addon.knowledge.getLinesInNote(note);
     let tree = new TreeModel();
@@ -297,9 +308,7 @@ class AddonParse extends AddonBase {
       .join("\n")}</div>`;
     console.log(this.parseHTMLLines(item.getNote()).slice(0, lineCount));
 
-    let parser = Components.classes[
-      "@mozilla.org/xmlextras/domparser;1"
-    ].createInstance(Components.interfaces.nsIDOMParser);
+    let parser = this.getDOMParser();
     let doc = parser.parseFromString(note, "text/html");
 
     // Make sure this is the new note
@@ -308,12 +317,14 @@ class AddonParse extends AddonBase {
     );
     if (metadataContainer) {
       // Load base64 image data into src
-      let nodes = doc.querySelectorAll("img[data-attachment-key]");
+      let nodes = doc.querySelectorAll(
+        "img[data-attachment-key]"
+      ) as NodeListOf<HTMLElement>;
       for (let node of nodes) {
         node.remove();
       }
 
-      nodes = doc.querySelectorAll("span[style]");
+      nodes = doc.querySelectorAll("span[style]") as NodeListOf<HTMLElement>;
       for (let node of nodes) {
         // Browser converts #RRGGBBAA hex color to rgba function, and we convert it to rgb function,
         // because word processors don't understand colors with alpha channel
@@ -411,9 +422,7 @@ class AddonParse extends AddonBase {
     if (noteText.search(/data-schema-version/g) === -1) {
       noteText = `<div data-schema-version="8">${noteText}\n</div>`;
     }
-    let parser = Components.classes[
-      "@mozilla.org/xmlextras/domparser;1"
-    ].createInstance(Components.interfaces.nsIDOMParser);
+    let parser = this.getDOMParser();
     let doc = parser.parseFromString(noteText, "text/html");
 
     let metadataContainer: Element = doc.querySelector(
@@ -423,17 +432,16 @@ class AddonParse extends AddonBase {
   }
 
   parseLineText(line: string): string {
-    const parser = Components.classes[
-      "@mozilla.org/xmlextras/domparser;1"
-    ].createInstance(Components.interfaces.nsIDOMParser);
+    const parser = this.getDOMParser();
     try {
       if (line.search(/data-schema-version/g) === -1) {
         line = `<div data-schema-version="8">${line}</div>`;
       }
-      return parser
-        .parseFromString(line, "text/html")
-        .querySelector("body > div[data-schema-version]")
-        .innerText.trim();
+      return (
+        parser
+          .parseFromString(line, "text/html")
+          .querySelector("body > div[data-schema-version]") as HTMLElement
+      ).innerText.trim();
     } catch (e) {
       return "";
     }
@@ -453,10 +461,8 @@ class AddonParse extends AddonBase {
 
   // A realization of Markdown Note.js translator
   async parseNoteToMD(noteItem: Zotero.Item, options: any = {}) {
-    const doc = new DOMParser().parseFromString(
-      noteItem.getNote() || "",
-      "text/html"
-    );
+    const parser = this.getDOMParser();
+    const doc = parser.parseFromString(noteItem.getNote() || "", "text/html");
     Components.utils.import("resource://gre/modules/osfile.jsm");
     doc.querySelectorAll("span").forEach(function (span) {
       if (span.style.textDecoration === "line-through") {
@@ -714,7 +720,9 @@ class AddonParse extends AddonBase {
         return "[" + content + "](" + href + title + ")";
       },
     });
-    return turndownService.turndown(doc.body);
+    const parsedMD = turndownService.turndown(doc.body);
+    console.log(parsedMD);
+    return parsedMD;
   }
 }