Tool factories

mcp/pr.ts

@@ -1,8 +1,9 @@
 import { type } from "arktype";
+import type { Context } from "../main.ts";
 import { log } from "../utils/cli.ts";
 import { containsSecrets } from "../utils/secrets.ts";
 import { $ } from "../utils/shell.ts";
-import { contextualize, tool } from "./shared.ts";
+import { execute, tool } from "./shared.ts";
 
 export const PullRequest = type({
   title: type.string.describe("the title of the pull request"),
@@ -10,48 +11,50 @@ export const PullRequest = type({
   base: type.string.describe("the base branch to merge into (e.g., 'main')"),
 });
 
-export const PullRequestTool = tool({
-  name: "create_pull_request",
-  description: "Create a pull request from the current branch",
-  parameters: PullRequest,
-  execute: contextualize(async ({ title, body, base }, ctx) => {
-    const currentBranch = $("git", ["rev-parse", "--abbrev-ref", "HEAD"], { log: false });
-    log.info(`Current branch: ${currentBranch}`);
+export function PullRequestTool(ctx: Context) {
+  return tool({
+    name: "create_pull_request",
+    description: "Create a pull request from the current branch",
+    parameters: PullRequest,
+    execute: execute(ctx, async ({ title, body, base }) => {
+      const currentBranch = $("git", ["rev-parse", "--abbrev-ref", "HEAD"], { log: false });
+      log.info(`Current branch: ${currentBranch}`);
 
-    // validate PR title and body for secrets
-    if (containsSecrets(title) || containsSecrets(body)) {
-      throw new Error(
-        "PR creation blocked: secrets detected in PR title or body. " +
-          "Please remove any sensitive information (API keys, tokens, passwords) before creating a PR."
-      );
-    }
+      // validate PR title and body for secrets
+      if (containsSecrets(title) || containsSecrets(body)) {
+        throw new Error(
+          "PR creation blocked: secrets detected in PR title or body. " +
+            "Please remove any sensitive information (API keys, tokens, passwords) before creating a PR."
+        );
+      }
 
-    // validate all changes that would be in the PR (from base to HEAD)
-    const diff = $("git", ["diff", `origin/${base}...HEAD`], { log: false });
-    if (containsSecrets(diff)) {
-      throw new Error(
-        "PR creation blocked: secrets detected in changes. " +
-          "Please remove any sensitive information (API keys, tokens, passwords) before creating a PR."
-      );
-    }
+      // validate all changes that would be in the PR (from base to HEAD)
+      const diff = $("git", ["diff", `origin/${base}...HEAD`], { log: false });
+      if (containsSecrets(diff)) {
+        throw new Error(
+          "PR creation blocked: secrets detected in changes. " +
+            "Please remove any sensitive information (API keys, tokens, passwords) before creating a PR."
+        );
+      }
 
-    const result = await ctx.octokit.rest.pulls.create({
-      owner: ctx.owner,
-      repo: ctx.name,
-      title: title,
-      body: body,
-      head: currentBranch,
-      base: base,
-    });
+      const result = await ctx.octokit.rest.pulls.create({
+        owner: ctx.owner,
+        repo: ctx.name,
+        title: title,
+        body: body,
+        head: currentBranch,
+        base: base,
+      });
 
-    return {
-      success: true,
-      pullRequestId: result.data.id,
-      number: result.data.number,
-      url: result.data.html_url,
-      title: result.data.title,
-      head: result.data.head.ref,
-      base: result.data.base.ref,
-    };
-  }),
-});
+      return {
+        success: true,
+        pullRequestId: result.data.id,
+        number: result.data.number,
+        url: result.data.html_url,
+        title: result.data.title,
+        head: result.data.head.ref,
+        base: result.data.base.ref,
+      };
+    }),
+  });
+}