plugin.ts

/**
 * Worktrunk OpenCode Plugin
 * Bridges OpenCode/Kilo Code lifecycle with Worktrunk status markers.
 */

export default async ({ $, worktree }) => {
  return {
    hooks: {
      /**
       * Injects current Worktrunk status into the agent's prompt context.
       */
      "tui.prompt.append": async () => {
        try {
          const info = await $`wt info --json`.json();
          return `\n\n[CONTEXT: ACTIVE WORKTREE]\nPath: ${worktree}\nBranch: ${info.branch}\nStatus: ${info.status}`;
        } catch (e) {
          return "";
        }
      },

      /**
       * Lifecycle: Update Worktrunk marker to '🤖' (active) when a task starts.
       */
      "session.status": async ({ status }) => {
        if (status === "running") {
          await $`wt config state marker set 🤖`.quiet();
        }
      },

      /**
       * Lifecycle: Revert Worktrunk marker to '💬' (idle) when the agent finishes.
       */
      "session.idle": async () => {
        await $`wt config state marker set 💬`.quiet();
      },

      /**
       * Lifecycle: Clear marker on error.
       */
      "session.error": async () => {
        await $`wt config state marker clear`.quiet();
      }
    },

    /**
     * Expose Worktrunk CLI as tools for the AI agent.
     */
    tools: {
      wt_create: {
        description: "Create a new Worktrunk worktree for a branch.",
        args: { branch: "string", base: "string?" },
        execute: async ({ branch, base }) => {
          return await $`wt create ${branch} ${base || ""}`.text();
        }
      },
      wt_switch: {
        description: "Switch to an existing Worktrunk worktree.",
        args: { target: "string" },
        execute: async ({ target }) => {
          return await $`wt switch ${target}`.text();
        }
      },
      wt_list: {
        description: "List all active Worktrunk worktrees.",
        execute: async () => {
          return await $`wt list`.text();
        }
      }
    }
  };
};