From fcd9d0288c1b946953efed68d38e3ad88c7ff454 Mon Sep 17 00:00:00 2001
From: windingwind <33902321+windingwind@users.noreply.github.com>
Date: Fri, 26 Jul 2024 10:22:13 +0800
Subject: [PATCH] fix: inject to note editor

---
 src/elements/workspace/workspace.ts | 1 +
 src/modules/editor/toolbar.ts       | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/src/elements/workspace/workspace.ts b/src/elements/workspace/workspace.ts
index 6f9bb3b..1e5303a 100644
--- a/src/elements/workspace/workspace.ts
+++ b/src/elements/workspace/workspace.ts
@@ -187,6 +187,7 @@ export class Workspace extends PluginCEBase {
     await editor._initPromise;
 
     const _document = editor._iframeWindow.document;
+    await waitUtilAsync(() => !!_document.querySelector(".toolbar"));
     const toolbar = _document.querySelector(".toolbar") as HTMLDivElement;
 
     const toggleOutline = this._addon.data.ztoolkit.UI.createElement(
diff --git a/src/modules/editor/toolbar.ts b/src/modules/editor/toolbar.ts
index 7e2f742..008f359 100644
--- a/src/modules/editor/toolbar.ts
+++ b/src/modules/editor/toolbar.ts
@@ -6,11 +6,13 @@ import { getNoteLink } from "../../utils/link";
 import { getString } from "../../utils/locale";
 import { openLinkCreator } from "../../utils/linkCreator";
 import { slice } from "../../utils/str";
+import { waitUtilAsync } from "../../utils/wait";
 
 export async function initEditorToolbar(editor: Zotero.EditorInstance) {
   const noteItem = editor._item;
 
   const _document = editor._iframeWindow.document;
+  await waitUtilAsync(() => !!_document.querySelector(".toolbar"));
   const toolbar = _document.querySelector(".toolbar") as HTMLDivElement;
   // Link creator
   registerEditorToolbarElement(