From ebf34d72185c6eae35760f110c119736064a6c6e Mon Sep 17 00:00:00 2001
From: Jambo <bxtx1992@gmail.com>
Date: Sat, 23 Sep 2023 17:13:00 +0800
Subject: [PATCH] fix: init the dom after turning on the plugin

---
 main.ts | 90 ++++++++++++++++++++++++++++-----------------------------
 1 file changed, 45 insertions(+), 45 deletions(-)

diff --git a/main.ts b/main.ts
index 6bef998..443be25 100644
--- a/main.ts
+++ b/main.ts
@@ -162,7 +162,7 @@ export default class TypingAsstPlugin extends Plugin {
 						break;
 					}
 				}
-				this.btns.display(lineStyle);
+				this.btns?.display(lineStyle);
 			}
 		};
 
@@ -208,11 +208,9 @@ export default class TypingAsstPlugin extends Plugin {
 		});
 
 		this.registerDomEvent(document, "click", (evt: MouseEvent) => {
-			if (this.commands.isVisible()) {
-				this.commands.hide();
-			}
+			this.commands?.hide();
 			const selection = document.getSelection()?.toString();
-			if (!selection && this.btns.isVisible()) {
+			if (!selection && this.btns?.isVisible()) {
 				this.btns.hide();
 			}
 		});
@@ -223,7 +221,7 @@ export default class TypingAsstPlugin extends Plugin {
 
 		this.registerDomEvent(document, "keydown", (evt: KeyboardEvent) => {
 			renderEmptyText();
-			if (this.commands.isVisible()) {
+			if (this.commands?.isVisible()) {
 				const { key } = evt;
 				if (
 					key !== "ArrowUp" &&
@@ -239,48 +237,52 @@ export default class TypingAsstPlugin extends Plugin {
 					}
 				}
 			}
-			if (this.btns.isVisible()) {
-				this.btns.hide();
-			}
+			this.btns?.hide();
 		});
 		const scrollEvent = () => {
-			if (this.btns.isVisible()) {
+			if (this.btns?.isVisible()) {
 				handleSelection();
 			}
 		};
-		this.registerEvent(
-			this.app.workspace.on("active-leaf-change", () => {
-				if (this.scrollArea) {
-					this.scrollArea.removeEventListener("scroll", scrollEvent);
-				}
-				const view =
-					this.app.workspace.getActiveViewOfType(MarkdownView);
-				if (!view) return;
 
-				this.scrollArea =
-					view.containerEl.querySelector(".cm-scroller") ?? undefined;
-				const appHeader = document.querySelector(".titlebar");
-				const viewHeader =
-					view.containerEl.querySelector(".view-header");
-				const headerHeight =
-					(appHeader?.clientHeight ?? 0) +
-					(viewHeader?.clientHeight ?? 0);
+		const renderPlugin = () => {
+			if (this.scrollArea) {
+				this.scrollArea.removeEventListener("scroll", scrollEvent);
+			}
+			const view = this.app.workspace.getActiveViewOfType(MarkdownView);
+			if (!view) return;
+			this.scrollArea =
+				view.containerEl.querySelector(".cm-scroller") ?? undefined;
+			const appHeader = document.querySelector(".titlebar");
+			const viewHeader = view.containerEl.querySelector(".view-header");
+			const headerHeight =
+				(appHeader?.clientHeight ?? 0) +
+				(viewHeader?.clientHeight ?? 0);
 
-				if (!this.scrollArea) return;
-				const scrollArea = this.scrollArea;
-				this.commands?.remove();
-				this.commands = new CommandMenu({
-					scrollArea,
-					onMenu: onMenuClick,
-				});
-				this.btns?.remove();
-				this.btns = new SelectionBtns({
-					scrollArea,
-					headerHeight,
-					onAction: onSelectionAction,
-				});
-				scrollArea?.addEventListener("scroll", scrollEvent);
-			})
+			if (!this.scrollArea) return;
+			const scrollArea = this.scrollArea;
+
+			this.commands?.remove();
+			this.commands = new CommandMenu({
+				scrollArea,
+				onMenu: onMenuClick,
+			});
+
+			this.btns?.remove();
+			this.btns = new SelectionBtns({
+				scrollArea,
+				headerHeight,
+				onAction: onSelectionAction,
+			});
+
+			scrollArea?.addEventListener("scroll", scrollEvent);
+		};
+
+		/**Ensure that the plugin can be loaded and used immediately after it is turned on */
+		renderPlugin();
+
+		this.registerEvent(
+			this.app.workspace.on("active-leaf-change", renderPlugin)
 		);
 
 		this.registerDomEvent(document, "input", (evt: InputEvent) => {
@@ -292,11 +294,9 @@ export default class TypingAsstPlugin extends Plugin {
 				const cursor = view.editor.getCursor();
 				const editLine = view.editor.getLine(cursor.line);
 				if (editLine.replace(/[\s]*$/, "").length <= cursor.ch) {
-					this.commands.display();
+					this.commands?.display();
 				} else {
-					if (this.commands.isVisible()) {
-						this.commands.hide();
-					}
+					this.commands?.hide();
 				}
 			}
 		});