Initial release

2026-07-22 23:35:04 +08:00
commit 281f3c1615
17 changed files with 5001 additions and 0 deletions
@@ -0,0 +1,48 @@
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import { clearCurrentTask, reportCurrentTask } from "./src/herdr.ts";
+import { TaskState } from "./src/state.ts";
+import { registerTaskTools } from "./src/tools.ts";
+import { clearTaskWidget, updateTaskWidget } from "./src/widget.ts";
+
+export default function piTodoHerdr(pi: ExtensionAPI): void {
+  const state = new TaskState();
+  let widgetVisible = true;
+
+  const refresh = (ctx: ExtensionContext) => {
+    updateTaskWidget(state, ctx, widgetVisible);
+    void reportCurrentTask(state.list());
+  };
+
+  const restore = (ctx: ExtensionContext) => {
+    state.loadFromSession(ctx);
+    refresh(ctx);
+  };
+
+  pi.on("session_start", async (_event, ctx) => restore(ctx));
+  pi.on("session_tree", async (_event, ctx) => restore(ctx));
+  pi.on("session_shutdown", async (event) => {
+    if (event.reason === "quit") await clearCurrentTask();
+  });
+
+  registerTaskTools(pi, state, refresh);
+
+  pi.registerCommand("tasks", {
+    description: "Show or hide the task tree widget",
+    handler: async (_args, ctx) => {
+      if (state.list().length === 0) {
+        clearTaskWidget(ctx);
+        ctx.ui.notify("No tasks yet.", "info");
+        return;
+      }
+
+      widgetVisible = !widgetVisible;
+      if (widgetVisible) {
+        updateTaskWidget(state, ctx, true);
+        ctx.ui.notify("Task widget shown.", "info");
+      } else {
+        clearTaskWidget(ctx);
+        ctx.ui.notify("Task widget hidden.", "info");
+      }
+    },
+  });
+}