Merge pull request #10 from taweili/master

Use closure to make creating new menu event easier and added a way to add extra block menu commands through reading the props of ollama-logseq-page

1 files changed, 22 insertions, 7 deletions

diff --git a/src/App.tsx b/src/App.tsx
index 6ab84a8..b6b6918 100644
--- a/src/App.tsx
+++ b/src/App.tsx
@@ -4,9 +4,7 @@ import {
   convertToFlashCardFromEvent,
   DivideTaskIntoSubTasksFromEvent,
   ollamaUI,
-  summarizeBlockFromEvent,
-  promptFromBlockEvent,
-  expandBlockEvent
+  promptFromBlockEventClosure
 } from "./ollama";
 import { useAppVisible } from "./utils";
 
@@ -46,16 +44,33 @@ function App() {
     if (!logseq.settings) {
       return
     }
+
+logseq.Editor.getPageBlocksTree("ollama-logseq-config").then((blocks) => {
+  blocks!.forEach((block) => {
+    logseq.Editor.getBlockProperty(block.uuid, "ollama-context-menu-title").then((title) => {
+      logseq.Editor.getBlockProperty(block.uuid, "ollama-prompt-prefix").then((prompt_prefix) => {
+        logseq.Editor.registerBlockContextMenuItem(title, promptFromBlockEventClosure(prompt_prefix))
+      })
+    }).catch((reason) => {
+    })
+  })
+}).catch((reason) => {
+  console.log("Can not find the configuration page named 'ollama-logseq-config'")
+  console.log(reason)
+})
+    
+
     logseq.Editor.registerSlashCommand("ollama", ollamaUI)
     logseq.Editor.registerBlockContextMenuItem("Ollama: Create a flash card", convertToFlashCardFromEvent)
-    logseq.Editor.registerBlockContextMenuItem("Ollama: Summarize block", summarizeBlockFromEvent)
     logseq.Editor.registerBlockContextMenuItem("Ollama: Divide into subtasks", DivideTaskIntoSubTasksFromEvent)
-    logseq.Editor.registerBlockContextMenuItem("Ollama: Prompt from Block", promptFromBlockEvent)
-    logseq.Editor.registerBlockContextMenuItem("Ollama: Expand Block", expandBlockEvent)
+    logseq.Editor.registerBlockContextMenuItem("Ollama: Prompt from Block", promptFromBlockEventClosure())
+    logseq.Editor.registerBlockContextMenuItem("Ollama: Summarize block", promptFromBlockEventClosure("Summarize: "))
+    logseq.Editor.registerBlockContextMenuItem("Ollama: Expand Block", promptFromBlockEventClosure("Expand: "))
+
     logseq.App.registerCommandShortcut(
       { "binding": logseq.settings.shortcut },
       ollamaUI
-    );
+    ); 
   }, [])
 
   if (visible) {