Support Command Registration (Closes #26)

RoamJS · Jul 14, 2021 · 1352f23 · 1352f23
1 parent 353ad62
commit 1352f23
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -26,7 +26,12 @@ import differenceInMilliseconds from "date-fns/differenceInMilliseconds";
 import { render } from "./SmartblocksMenu";
 import { render as renderStore } from "./SmartblocksStore";
 import lego from "./img/lego3blocks.png";
-import { getCustomWorkflows, sbBomb } from "./smartblocks";
+import {
+  CommandHandler,
+  getCustomWorkflows,
+  handlerByCommand,
+  sbBomb,
+} from "./smartblocks";
 
 addStyle(`.rm-page-ref--tag[data-tag="42SmartBlock"]:before, .rm-page-ref--tag[data-tag="SmartBlock"]:before {
   display:inline-block;
@@ -135,7 +140,16 @@ runExtension("smartblocks", () => {
   const isCustomOnly = tree.some((t) =>
     toFlexRegex("custom only").test(t.text)
   );
-
+  window.roamjs.extension.smartblocks = {};
+  window.roamjs.extension.smartblocks.registerCommand = ({
+    text,
+    handler,
+  }: {
+    text: string;
+    handler: CommandHandler;
+  }) => {
+    handlerByCommand[text] = handler;
+  };
   document.addEventListener("input", (e) => {
     const target = e.target as HTMLElement;
     if (

diff --git a/src/smartblocks.ts b/src/smartblocks.ts
@@ -148,12 +148,15 @@ const outputTodoBlocks = (
     }));
 
 type CommandOutput = string | string[] | InputTextNode[];
+export type CommandHandler = (
+  ...args: string[]
+) => CommandOutput | Promise<CommandOutput>;
 const COMMAND_REGEX = /<%([A-Z0-9]*)(?::(.*?))?%>/g;
 const COMMANDS: {
   text: string;
   help: string;
   args?: true;
-  handler: (...args: string[]) => CommandOutput | Promise<CommandOutput>;
+  handler: CommandHandler;
 }[] = [
   {
     text: "DATE",
@@ -367,7 +370,7 @@ const COMMANDS: {
     },
   },
 ];
-const handlerByCommand = Object.fromEntries(
+export const handlerByCommand = Object.fromEntries(
   COMMANDS.map((c) => [c.text, c.handler])
 );