Update libs/src/LSPlugin.user.ts

Co-authored-by: Peng Xiao <pengxiao@outlook.com>
logseq · Apr 29, 2022 · 9885dc6 · 9885dc6
1 parent fc0f2b7
commit 9885dc6
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/libs/src/LSPlugin.user.ts b/libs/src/LSPlugin.user.ts
@@ -586,15 +586,21 @@ export class LSPluginUser
               const handler = args[0]
               caller[f](type, handler)
 
-              if (!isOff) {
-                that.App._installPluginHook(pid, type)
-              }
+  const unlisten = () => {
+    caller.off(type, handler)
+    if (!caller.listenerCount(type)) {
+      that.App._uninstallPluginHook(pid, type)
+    }
+  }
+
+  if (!isOff) {
+    that.App._installPluginHook(pid, type)
+  } else {
+    unlisten()
+    return
+  }
 
-              return () => {
-                caller.off(type, handler)
-                if (!caller.listenerCount(type)) {
-                  that.App._uninstallPluginHook(pid, type)
-                }
+  return unlisten
               }
             }
           }