@flyingrobots/graft 0.4.0 → 0.5.0

package/src/mcp/tools/workspace-authorize.ts

@@ -0,0 +1,20 @@
+import { z } from "zod";
+import type { ToolDefinition, ToolContext, ToolHandler } from "../context.js";
+
+export const workspaceAuthorizeTool: ToolDefinition = {
+  name: "workspace_authorize",
+  description:
+    "Authorize a workspace for daemon binding and optionally change its daemon capability posture.",
+  schema: {
+    cwd: z.string(),
+    runCapture: z.boolean().optional(),
+  },
+  createHandler(ctx: ToolContext): ToolHandler {
+    return async (args) => {
+      return ctx.respond("workspace_authorize", { ...await ctx.authorizeWorkspace({
+        cwd: args["cwd"] as string,
+        runCapture: args["runCapture"] as boolean | undefined,
+      }) });
+    };
+  },
+};

package/src/mcp/tools/workspace-bind.ts

@@ -0,0 +1,25 @@
+import { z } from "zod";
+import type { ToolDefinition, ToolContext, ToolHandler } from "../context.js";
+
+export const workspaceBindTool: ToolDefinition = {
+  name: "workspace_bind",
+  description:
+    "Bind the current daemon session to a workspace by resolving repo and worktree identity server-side.",
+  schema: {
+    cwd: z.string(),
+    worktreeRoot: z.string().optional(),
+    gitCommonDir: z.string().optional(),
+    repoId: z.string().optional(),
+  },
+  createHandler(ctx: ToolContext): ToolHandler {
+    return async (args) => {
+      const result = await ctx.bindWorkspace({
+        cwd: args["cwd"] as string,
+        worktreeRoot: args["worktreeRoot"] as string | undefined,
+        gitCommonDir: args["gitCommonDir"] as string | undefined,
+        repoId: args["repoId"] as string | undefined,
+      }, "workspace_bind");
+      return ctx.respond("workspace_bind", { ...result });
+    };
+  },
+};

package/src/mcp/tools/workspace-rebind.ts

@@ -0,0 +1,25 @@
+import { z } from "zod";
+import type { ToolDefinition, ToolContext, ToolHandler } from "../context.js";
+
+export const workspaceRebindTool: ToolDefinition = {
+  name: "workspace_rebind",
+  description:
+    "Rebind the current daemon session to a different workspace and start a fresh session-local slice.",
+  schema: {
+    cwd: z.string(),
+    worktreeRoot: z.string().optional(),
+    gitCommonDir: z.string().optional(),
+    repoId: z.string().optional(),
+  },
+  createHandler(ctx: ToolContext): ToolHandler {
+    return async (args) => {
+      const result = await ctx.rebindWorkspace({
+        cwd: args["cwd"] as string,
+        worktreeRoot: args["worktreeRoot"] as string | undefined,
+        gitCommonDir: args["gitCommonDir"] as string | undefined,
+        repoId: args["repoId"] as string | undefined,
+      }, "workspace_rebind");
+      return ctx.respond("workspace_rebind", { ...result });
+    };
+  },
+};

package/src/mcp/tools/workspace-revoke.ts

@@ -0,0 +1,18 @@
+import { z } from "zod";
+import type { ToolDefinition, ToolContext, ToolHandler } from "../context.js";
+
+export const workspaceRevokeTool: ToolDefinition = {
+  name: "workspace_revoke",
+  description:
+    "Revoke daemon authorization for a workspace while leaving any already-open sessions visible to the control plane.",
+  schema: {
+    cwd: z.string(),
+  },
+  createHandler(ctx: ToolContext): ToolHandler {
+    return async (args) => {
+      return ctx.respond("workspace_revoke", { ...await ctx.revokeWorkspace({
+        cwd: args["cwd"] as string,
+      }) });
+    };
+  },
+};

package/src/mcp/tools/workspace-status.ts

@@ -0,0 +1,12 @@
+import type { ToolDefinition, ToolContext, ToolHandler } from "../context.js";
+
+export const workspaceStatusTool: ToolDefinition = {
+  name: "workspace_status",
+  description:
+    "Return the current daemon workspace binding state and resolved capability profile.",
+  createHandler(ctx: ToolContext): ToolHandler {
+    return () => {
+      return ctx.respond("workspace_status", { ...ctx.getWorkspaceStatus() });
+    };
+  },
+};

package/src/mcp/warp-pool.ts

@@ -0,0 +1,36 @@
+import type WarpApp from "@git-stunts/git-warp";
+import { DEFAULT_WARP_WRITER_ID } from "../warp/writer-id.js";
+
+export interface WarpPool {
+  getOrOpen(repoId: string, worktreeRoot: string, writerId?: string): Promise<WarpApp>;
+  size(): number;
+}
+
+export class InMemoryWarpPool implements WarpPool {
+  private readonly opened = new Map<string, Map<string, Promise<WarpApp>>>();
+
+  constructor(private readonly openWarp: (worktreeRoot: string, writerId: string) => Promise<WarpApp>) {}
+
+  getOrOpen(repoId: string, worktreeRoot: string, writerId: string = DEFAULT_WARP_WRITER_ID): Promise<WarpApp> {
+    const repoHandles = this.opened.get(repoId);
+    const cached = repoHandles?.get(writerId);
+    if (cached !== undefined) return cached;
+
+    const nextRepoHandles = repoHandles ?? new Map<string, Promise<WarpApp>>();
+    const opened = this.openWarp(worktreeRoot, writerId).catch((error: unknown) => {
+      const current = this.opened.get(repoId);
+      current?.delete(writerId);
+      if (current?.size === 0) {
+        this.opened.delete(repoId);
+      }
+      throw error;
+    });
+    nextRepoHandles.set(writerId, opened);
+    this.opened.set(repoId, nextRepoHandles);
+    return opened;
+  }
+
+  size(): number {
+    return this.opened.size;
+  }
+}