@op1/threads 0.2.0 → 0.2.2

package/tui.ts

@@ -1,240 +0,0 @@
-import { Plugin } from "@opencode/plugin/tui";
-import { getComponentCatalogue } from "@opentui/solid/components";
-import { createEffect, createSignal } from "solid-js";
-import { z } from "zod";
-import { ThreadsRpc } from "./src/rpc";
-import { activity } from "./src/activity";
-import { workflowUI } from "./src/workflow-ui";
-import { cleanRoleTitle } from "./src/activity-model";
-import {
-  BoxRenderable,
-  ScrollBoxRenderable,
-  TextRenderable,
-  TextAttributes,
-  RGBA,
-} from "@opentui/core";
-
-const CoordinatorRef = z.object({ coordinatorID: z.string() });
-
-export default Plugin.define({
-  id: "op-threads",
-  setup(ctx) {
-    const stopWorkflows = workflowUI(ctx);
-    const rpc = ctx.client.rpc(ThreadsRpc);
-    const sidebar = activity(
-      ctx,
-      { BoxRenderable, ScrollBoxRenderable, TextRenderable, TextAttributes, RGBA },
-      { createEffect, createSignal },
-      getComponentCatalogue().spinner,
-    );
-    const [cleaned, saveCleaned] = ctx.storage.store("role-title-cleanup", {
-      initial: { ids: [] as string[] },
-    });
-    const initial: { workerIDs: string[] } = { workerIDs: [] };
-    const [seen, updateSeen] = ctx.storage.memory("seen-workers", { initial });
-    const closing = new Set<string>();
-    const abort = new AbortController();
-    let stopped = false;
-    let running = false;
-    let reopenPending = false;
-    let lastError: string | undefined;
-    let movingFrom: string | undefined;
-    function groupTabs() {
-      if (sidebar.mounted()) return;
-      const tabs = ctx.ui.tabs.list().map((tab) => {
-        const projectID = ctx.data.session.get(tab.sessionID)?.projectID;
-        return {
-          sessionID: tab.sessionID,
-          priority: tab.busy || tab.attention,
-          projectID:
-            typeof projectID === "string" && projectID.length > 0
-              ? projectID
-              : undefined,
-        };
-      });
-      const groups = new Map<string, typeof tabs>();
-      for (const tab of tabs) {
-        const key =
-          tab.projectID === undefined
-            ? `session:${tab.sessionID}`
-            : `project:${tab.projectID}`;
-        const group = groups.get(key);
-        if (group) group.push(tab);
-        else groups.set(key, [tab]);
-      }
-      const ordered = [...groups.values()].flatMap((group) =>
-        group
-          .sort((left, right) => Number(right.priority) - Number(left.priority))
-          .map((tab) => tab.sessionID),
-      );
-      const current = tabs.map((tab) => tab.sessionID);
-      const stamp = JSON.stringify(current);
-      if (movingFrom === stamp) return;
-      movingFrom = undefined;
-      for (const [index, sessionID] of ordered.entries()) {
-        if (current[index] === sessionID) continue;
-        if (ctx.ui.tabs.move(sessionID, index)) movingFrom = stamp;
-        break;
-      }
-    }
-    async function reconcile(reopen = false) {
-      if (stopped || !ctx.ui.tabs.enabled()) return;
-      if (running) {
-        reopenPending ||= reopen;
-        return;
-      }
-      running = true;
-      try {
-        groupTabs();
-        const route = ctx.ui.router.current();
-        const coordinatorIDs = [
-          ...new Set([
-            ...ctx.ui.tabs.list().flatMap((tab) => {
-              const link = CoordinatorRef.safeParse(
-                ctx.data.session.get(tab.sessionID)?.metadata?.opThreads,
-              );
-              return link.success
-                ? [tab.sessionID, link.data.coordinatorID]
-                : [tab.sessionID];
-            }),
-            ...(route.type === "session" ? [route.sessionID] : []),
-          ]),
-        ].slice(0, 100);
-        if (!coordinatorIDs.length) return;
-        const { workers } = await (reopen ? rpc.restore : rpc.snapshot)(
-          { coordinatorIDs },
-          {
-            location: ctx.location ?? ctx.data.location.default(),
-            signal: abort.signal,
-          },
-        );
-        if (reopen)
-          await sidebar.restore(workers.map((worker) => worker.workerID));
-        sidebar.updateWorkers(workers);
-        for (const worker of workers) {
-          if (stopped) return;
-          if (!reopen && sidebar.isDismissed(worker.workerID)) continue;
-          const tab = ctx.ui.tabs
-            .list()
-            .find((tab) => tab.sessionID === worker.workerID);
-          if (closing.has(worker.workerID) && tab) continue;
-          const closed = closing.delete(worker.workerID);
-          if (
-            worker.hidden &&
-            !tab?.active &&
-            !tab?.busy &&
-            !tab?.attention &&
-            ctx.data.session.status(worker.workerID) !== "running"
-          ) {
-            if (tab) {
-              if (!ctx.ui.tabs.close(worker.workerID)) continue;
-              closing.add(worker.workerID);
-            }
-            if (seen.workerIDs.includes(worker.workerID)) {
-              updateSeen((draft) => {
-                draft.workerIDs = draft.workerIDs.filter(
-                  (id) => id !== worker.workerID,
-                );
-              });
-            }
-            continue;
-          }
-          if (!reopen && !closed && seen.workerIDs.includes(worker.workerID))
-            continue;
-          await ctx.data.session.sync(worker.workerID);
-          if (
-            !stopped &&
-            ctx.ui.tabs.open(worker.workerID) &&
-            !seen.workerIDs.includes(worker.workerID)
-          ) {
-            updateSeen((draft) => {
-              draft.workerIDs.push(worker.workerID);
-            });
-          }
-        }
-        if (!stopped && ctx.ui.tabs.enabled()) {
-          const roles = new Map<string, "Main" | "Worker">();
-          for (const worker of workers) {
-            roles.set(worker.coordinatorID, "Main");
-            roles.set(worker.workerID, "Worker");
-          }
-          for (const tab of ctx.ui.tabs.list()) {
-            if (stopped) return;
-            const role = roles.get(tab.sessionID);
-            const session = ctx.data.session.get(tab.sessionID);
-            if (!role || !session || cleaned.ids.includes(tab.sessionID))
-              continue;
-            const fresh = await ctx.client.session.get(
-              { sessionID: tab.sessionID },
-              { signal: abort.signal },
-            );
-            if (stopped) return;
-            const title = cleanRoleTitle(fresh.title ?? "", true);
-            if (title && title !== fresh.title)
-              await ctx.client.session.update(
-                { sessionID: tab.sessionID, title },
-                { signal: abort.signal },
-              );
-            if (stopped) return;
-            await saveCleaned((draft) => {
-              if (!draft.ids.includes(tab.sessionID))
-                draft.ids.push(tab.sessionID);
-            });
-          }
-          groupTabs();
-        }
-        lastError = undefined;
-      } catch (error) {
-        const message = `Managed worker tabs: ${String(error)}`;
-        if (!stopped && message !== lastError)
-          ctx.ui.toast.show({ message, variant: "error" });
-        lastError = message;
-      } finally {
-        running = false;
-        if (reopenPending) {
-          reopenPending = false;
-          void reconcile(true);
-        }
-      }
-    }
-    const refresh = () => {
-      void reconcile();
-    };
-    const stopEvents = ctx.data.listen(({ details }) => {
-      if (details.type.startsWith("session.")) refresh();
-    });
-    const timer = setInterval(refresh, 3000);
-    const removeSlot = ctx.ui.slot({
-      append: "app",
-      render: () => {
-        createEffect(() => {
-          ctx.ui.tabs.list();
-          refresh();
-        });
-        ctx.keymap.layer(() => ({
-          mode: "global",
-          commands: [
-            {
-              id: "threads.reopen",
-              title: "Reopen managed worker tabs",
-              palette: true,
-              slash: { name: "threads" },
-              run: () => reconcile(true),
-            },
-          ],
-        }));
-        return null;
-      },
-    });
-    refresh();
-    return () => {
-      stopWorkflows();
-      stopped = true;
-      abort.abort();
-      sidebar.dispose();
-      clearInterval(timer);
-      stopEvents();
-      removeSlot();
-    };
-  },
-});