@arii/boomtick-mcp 0.4.0 → 0.4.2

package/src/tools/contract.ts

@@ -1228,7 +1228,7 @@ export const CreateBranchInputJsonSchema = {
   "type": "object"
 } as const;
 
-export const CreateIssueInputSchema = z.object({ "title": z.string().min(1), "body": z.union([z.string().min(1), z.null()]).default(null), "file": z.union([z.string().min(1), z.null()]).default(null), "repo": z.union([z.string(), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
+export const CreateIssueInputSchema = z.object({ "title": z.string().min(1), "body": z.union([z.string().min(1), z.null()]).default(null), "file": z.union([z.string().min(1), z.null()]).default(null), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type CreateIssueInput = z.infer<typeof CreateIssueInputSchema>;
 
@@ -1268,6 +1268,7 @@ export const CreateIssueInputJsonSchema = {
     "repo": {
       "anyOf": [
         {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
           "type": "string"
         },
         {
@@ -1410,7 +1411,7 @@ export const CreateJulesSessionInputJsonSchema = {
   "type": "object"
 } as const;
 
-export const CreatePullRequestInputSchema = z.object({ "title": z.string().describe("PR Title."), "body": z.string().describe("Description of changes."), "head": z.string().describe("The branch containing changes to merge."), "base": z.string().describe("The target branch to merge into.").default("main"), "draft": z.boolean().describe("Whether to create the PR as a draft.").default(false), "repo": z.union([z.string(), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
+export const CreatePullRequestInputSchema = z.object({ "title": z.string().describe("PR Title."), "body": z.string().describe("Description of changes."), "head": z.string().describe("The branch containing changes to merge."), "base": z.string().describe("The target branch to merge into.").default("main"), "draft": z.boolean().describe("Whether to create the PR as a draft.").default(false), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type CreatePullRequestInput = z.infer<typeof CreatePullRequestInputSchema>;
 
@@ -1446,6 +1447,7 @@ export const CreatePullRequestInputJsonSchema = {
     "repo": {
       "anyOf": [
         {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
           "type": "string"
         },
         {
@@ -1595,7 +1597,7 @@ export const GetPackageScriptsInputJsonSchema = {
   "type": "object"
 } as const;
 
-export const GetPrDiffInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the pull request to get the diff for.") });
+export const GetPrDiffInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the pull request to get the diff for."), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type GetPrDiffInput = z.infer<typeof GetPrDiffInputSchema>;
 
@@ -1605,6 +1607,20 @@ export const GetPrDiffInputJsonSchema = {
       "description": "The number of the pull request to get the diff for.",
       "title": "Prnumber",
       "type": "integer"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
     }
   },
   "required": [
@@ -1614,7 +1630,7 @@ export const GetPrDiffInputJsonSchema = {
   "type": "object"
 } as const;
 
-export const GetPrInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the PR to view.") });
+export const GetPrInputSchema = z.object({ "prNumber": z.number().int().describe("The number of the PR to view."), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type GetPrInput = z.infer<typeof GetPrInputSchema>;
 
@@ -1624,6 +1640,20 @@ export const GetPrInputJsonSchema = {
       "description": "The number of the PR to view.",
       "title": "Prnumber",
       "type": "integer"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
     }
   },
   "required": [
@@ -1674,7 +1704,7 @@ export const HealthCheckInputJsonSchema = {
   "type": "object"
 } as const;
 
-export const IssueCommentInputSchema = z.object({ "issueNumber": z.number().int().describe("The number of the issue to comment on."), "body": z.string().describe("The content of the comment.") });
+export const IssueCommentInputSchema = z.object({ "issueNumber": z.number().int().describe("The number of the issue to comment on."), "body": z.string().describe("The content of the comment."), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type IssueCommentInput = z.infer<typeof IssueCommentInputSchema>;
 
@@ -1689,6 +1719,20 @@ export const IssueCommentInputJsonSchema = {
       "description": "The content of the comment.",
       "title": "Body",
       "type": "string"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
     }
   },
   "required": [
@@ -1732,7 +1776,7 @@ export const IssueSummaryJsonSchema = {
   "type": "object"
 } as const;
 
-export const IssueUpdateInputSchema = z.object({ "issueNumber": z.number().int().describe("The number of the issue to update."), "body": z.union([z.string(), z.null()]).describe("The new body content for the issue.").default(null), "file": z.union([z.string(), z.null()]).describe("Path to file containing new issue body.").default(null), "labels": z.union([z.array(z.string()), z.null()]).describe("Comma-separated list of labels to set.").default(null), "addLabels": z.union([z.array(z.string()), z.null()]).describe("Comma-separated list of labels to add.").default(null), "removeLabels": z.union([z.array(z.string()), z.null()]).describe("Comma-separated list of labels to remove.").default(null), "state": z.union([z.enum(["open","closed"]), z.null()]).describe("The state to set the issue to.").default(null) });
+export const IssueUpdateInputSchema = z.object({ "issueNumber": z.number().int().describe("The number of the issue to update."), "body": z.union([z.string(), z.null()]).describe("The new body content for the issue.").default(null), "file": z.union([z.string(), z.null()]).describe("Path to file containing new issue body.").default(null), "labels": z.union([z.array(z.string()), z.null()]).describe("Comma-separated list of labels to set.").default(null), "addLabels": z.union([z.array(z.string()), z.null()]).describe("Comma-separated list of labels to add.").default(null), "removeLabels": z.union([z.array(z.string()), z.null()]).describe("Comma-separated list of labels to remove.").default(null), "state": z.union([z.enum(["open","closed"]), z.null()]).describe("The state to set the issue to.").default(null), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type IssueUpdateInput = z.infer<typeof IssueUpdateInputSchema>;
 
@@ -1833,6 +1877,20 @@ export const IssueUpdateInputJsonSchema = {
       "default": null,
       "description": "The state to set the issue to.",
       "title": "State"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
     }
   },
   "required": [
@@ -1921,7 +1979,7 @@ export const IssueUpdateResponseJsonSchema = {
   "type": "object"
 } as const;
 
-export const IssueViewInputSchema = z.object({ "issueNumber": z.number().int().describe("The number of the issue to view.") });
+export const IssueViewInputSchema = z.object({ "issueNumber": z.number().int().describe("The number of the issue to view."), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type IssueViewInput = z.infer<typeof IssueViewInputSchema>;
 
@@ -1931,6 +1989,20 @@ export const IssueViewInputJsonSchema = {
       "description": "The number of the issue to view.",
       "title": "Issuenumber",
       "type": "integer"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
     }
   },
   "required": [
@@ -2998,7 +3070,7 @@ export const SearchDdgsInputJsonSchema = {
   "type": "object"
 } as const;
 
-export const SearchPRsInputSchema = z.object({ "state": z.enum(["open","closed","all"]).describe("The state of the PRs to search for.").default("open"), "limit": z.number().int().gte(1).lte(100).describe("The maximum number of PRs to return (default: 100, range: 1-100).").default(100), "includeDrafts": z.boolean().describe("Whether to include draft PRs in the results.").default(true), "labels": z.union([z.array(z.string()), z.null()]).describe("Filter PRs by labels.").default(null) });
+export const SearchPRsInputSchema = z.object({ "state": z.enum(["open","closed","all"]).describe("The state of the PRs to search for.").default("open"), "limit": z.number().int().gte(1).lte(100).describe("The maximum number of PRs to return (default: 100, range: 1-100).").default(100), "includeDrafts": z.boolean().describe("Whether to include draft PRs in the results.").default(true), "labels": z.union([z.array(z.string()), z.null()]).describe("Filter PRs by labels.").default(null), "repo": z.union([z.string().regex(new RegExp("^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$")), z.null()]).describe("The target repository override (e.g. org/repo).").default(null) });
 
 export type SearchPRsInput = z.infer<typeof SearchPRsInputSchema>;
 
@@ -3044,6 +3116,20 @@ export const SearchPRsInputJsonSchema = {
       "default": null,
       "description": "Filter PRs by labels.",
       "title": "Labels"
+    },
+    "repo": {
+      "anyOf": [
+        {
+          "pattern": "^[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+$",
+          "type": "string"
+        },
+        {
+          "type": "null"
+        }
+      ],
+      "default": null,
+      "description": "The target repository override (e.g. org/repo).",
+      "title": "Repo"
     }
   },
   "title": "SearchPRsInput",

package/src/tools/github.create_issue.test.ts

@@ -27,6 +27,16 @@ describe("github.create_issue", () => {
     });
 
     const result = await createIssueHandler({ title: "Test Issue", body: "Test Body", file: null, repo: null });
+    expect(shell.runCommand).toHaveBeenCalledWith("td-cli", [
+      "gh",
+      "create-issue",
+      "--title",
+      "Test Issue",
+      "--body",
+      "Test Body",
+      "--repo",
+      expect.any(String)
+    ]);
     expect(result.status).toBe("success");
     expect(result.issue?.number).toBe(123);
     expect(result.issue?.title).toBe("Test Issue");

package/src/tools/github.create_issue.ts

@@ -1,5 +1,6 @@
 import { z } from "zod";
 import { runCommand } from "../lib/shell.js";
+import { resolveRepo } from "../lib/git.js";
 import { sanitizeError } from "../lib/error_utils.js";
 import { CreateIssueInputSchema, CreateIssueResponseSchema } from "./contract.js";
 
@@ -7,11 +8,12 @@ export { CreateIssueInputSchema };
 
 export async function createIssueHandler(args: z.input<typeof CreateIssueInputSchema>) {
   const params = CreateIssueInputSchema.parse(args);
+  const targetRepo = resolveRepo(params.repo);
 
   const cmdArgs = ["gh", "create-issue", "--title", params.title];
   if (params.body) cmdArgs.push("--body", params.body);
   if (params.file) cmdArgs.push("--file", params.file);
-  if (params.repo) cmdArgs.push("--repo", params.repo);
+  cmdArgs.push("--repo", targetRepo);
 
   const result = await runCommand("td-cli", cmdArgs);
 

package/src/tools/github.create_pull_request.ts

@@ -1,11 +1,13 @@
 import { z } from "zod";
 import { runCommand } from "../lib/shell.js";
+import { resolveRepo } from "../lib/git.js";
 import { CreatePullRequestInputSchema } from "./contract.js";
 
 export { CreatePullRequestInputSchema };
 
 export async function createPullRequestHandler(args: z.input<typeof CreatePullRequestInputSchema>) {
   const params = CreatePullRequestInputSchema.parse(args);
+  const targetRepo = resolveRepo(params.repo);
 
   const tdArgs = [
     "gh",
@@ -13,17 +15,14 @@ export async function createPullRequestHandler(args: z.input<typeof CreatePullRe
     "--title", params.title,
     "--body", params.body,
     "--head", params.head,
-    "--base", params.base
+    "--base", params.base,
+    "--repo", targetRepo
   ];
 
   if (params.draft) {
     tdArgs.push("--draft");
   }
 
-  if (params.repo) {
-    tdArgs.push("--repo", params.repo);
-  }
-
   const result = await runCommand("td-cli", tdArgs);
 
   if (result.exitCode !== 0) {

package/src/tools/github.get_pr.ts

@@ -1,15 +1,18 @@
 import { z } from "zod";
 import { runCommand } from "../lib/shell.js";
+import { resolveRepo } from "../lib/git.js";
 import { sanitizeError } from "../lib/error_utils.js";
 
 export const GetPrInputSchema = z.object({
   prNumber: z.number().describe("The number of the PR to view."),
+  repo: z.string().optional().describe("The target repository override (e.g. org/repo)."),
 });
 
 export async function getPrHandler(args: z.infer<typeof GetPrInputSchema>) {
   const params = GetPrInputSchema.parse(args);
+  const targetRepo = resolveRepo(params.repo);
 
-  const result = await runCommand("td-cli", ["gh", "view", params.prNumber.toString()]);
+  const result = await runCommand("td-cli", ["gh", "view", params.prNumber.toString(), "--repo", targetRepo]);
 
   if (result.exitCode !== 0) {
     throw new Error(`Failed to get PR details: ${sanitizeError(result.stderr)}`);

package/src/tools/github.get_pr_diff.ts

@@ -1,13 +1,18 @@
 import { z } from "zod";
 import { runCommand } from "../lib/shell.js";
+import { resolveRepo } from "../lib/git.js";
 
 export const GetPrDiffInputSchema = z.object({
-  prNumber: z.number(),
+  prNumber: z.number().describe("The number of the pull request to get the diff for."),
+  repo: z.string().optional().describe("The target repository override (e.g. org/repo)."),
 });
 
 export async function getPrDiffHandler(args: z.infer<typeof GetPrDiffInputSchema>) {
+  const params = GetPrDiffInputSchema.parse(args);
+  const targetRepo = resolveRepo(params.repo);
+
   const result = await runCommand("td-cli", [
-    "gh", "pr-diff", args.prNumber.toString()
+    "gh", "pr-diff", params.prNumber.toString(), "--repo", targetRepo
   ]);
 
   if (result.exitCode !== 0) {

package/src/tools/github.issue_comment.ts

@@ -1,10 +1,12 @@
 import { z } from "zod";
 import { runCommand } from "../lib/shell.js";
+import { resolveRepo } from "../lib/git.js";
 import { sanitizeError } from "../lib/error_utils.js";
 
 export const IssueCommentInputSchema = z.object({
   issueNumber: z.number().describe("The number of the issue to comment on."),
   body: z.string().min(1, "Comment body cannot be empty").describe("The content of the comment."),
+  repo: z.string().optional().describe("The target repository override (e.g. org/repo)."),
 });
 
 const IssueCommentOutputSchema = z.object({
@@ -15,8 +17,13 @@ const IssueCommentOutputSchema = z.object({
 
 export async function issueCommentHandler(args: z.infer<typeof IssueCommentInputSchema>) {
   const params = IssueCommentInputSchema.parse(args);
+  const targetRepo = resolveRepo(params.repo);
 
-  const result = await runCommand("td-cli", ["gh", "issue-comment", params.issueNumber.toString(), "--body", params.body]);
+  const result = await runCommand("td-cli", [
+    "gh", "issue-comment", params.issueNumber.toString(),
+    "--body", params.body,
+    "--repo", targetRepo
+  ]);
 
   if (result.exitCode !== 0) {
     throw new Error(`Failed to post comment: ${sanitizeError(result.stderr)}`);

package/src/tools/github.issue_update.ts

@@ -1,12 +1,14 @@
 import { z } from "zod";
 import { runCommand } from "../lib/shell.js";
+import { resolveRepo } from "../lib/git.js";
 import { sanitizeError } from "../lib/error_utils.js";
 import { IssueUpdateInputSchema, IssueUpdateResponseSchema } from "./contract.js";
 
 export { IssueUpdateInputSchema };
 
-export async function issueUpdateHandler(args: any) {
+export async function issueUpdateHandler(args: z.infer<typeof IssueUpdateInputSchema>) {
   const params = IssueUpdateInputSchema.parse(args);
+  const targetRepo = resolveRepo(params.repo);
 
   const cmdArgs = ["gh", "issue-update", params.issueNumber.toString()];
   if (params.body) cmdArgs.push("--body", params.body);
@@ -15,6 +17,7 @@ export async function issueUpdateHandler(args: any) {
   if (params.addLabels && params.addLabels.length > 0) cmdArgs.push("--add-labels", params.addLabels.join(","));
   if (params.removeLabels && params.removeLabels.length > 0) cmdArgs.push("--remove-labels", params.removeLabels.join(","));
   if (params.state) cmdArgs.push("--state", params.state);
+  cmdArgs.push("--repo", targetRepo);
 
   const result = await runCommand("td-cli", cmdArgs);
 

package/src/tools/github.issue_view.ts

@@ -1,9 +1,11 @@
 import { z } from "zod";
 import { runCommand } from "../lib/shell.js";
+import { resolveRepo } from "../lib/git.js";
 import { sanitizeError } from "../lib/error_utils.js";
 
 export const IssueViewInputSchema = z.object({
   issueNumber: z.number().describe("The number of the issue to view."),
+  repo: z.string().optional().describe("The target repository override (e.g. org/repo)."),
 });
 
 const IssueViewOutputSchema = z.object({
@@ -19,8 +21,9 @@ const IssueViewOutputSchema = z.object({
 
 export async function issueViewHandler(args: z.infer<typeof IssueViewInputSchema>) {
   const params = IssueViewInputSchema.parse(args);
+  const targetRepo = resolveRepo(params.repo);
 
-  const result = await runCommand("td-cli", ["gh", "issue-view", params.issueNumber.toString()]);
+  const result = await runCommand("td-cli", ["gh", "issue-view", params.issueNumber.toString(), "--repo", targetRepo]);
 
   if (result.exitCode !== 0) {
     throw new Error(`Failed to view issue: ${sanitizeError(result.stderr)}`);

package/src/tools/github.search_open_prs.ts

@@ -1,20 +1,24 @@
 import { z } from "zod";
 import { runCommand } from "../lib/shell.js";
+import { resolveRepo } from "../lib/git.js";
 
 export const SearchOpenPrsInputSchema = z.object({
   state: z.enum(["open", "closed", "all"]).optional().default("open"),
   includeDrafts: z.boolean().optional().default(true),
   limit: z.number().min(1).max(100).optional().default(100),
   labels: z.array(z.string()).optional(),
+  repo: z.string().optional().describe("The target repository override (e.g. org/repo)."),
 });
 
 export async function searchOpenPrsHandler(args: z.infer<typeof SearchOpenPrsInputSchema>) {
   const params = SearchOpenPrsInputSchema.parse(args);
+  const targetRepo = resolveRepo(params.repo);
 
   const tdArgs = [
     "gh", "search-prs",
     "--state", params.state,
     "--limit", params.limit.toString(),
+    "--repo", targetRepo,
   ];
 
   if (!params.includeDrafts) {

package/src/tools/repo.commit_patch.ts

@@ -1,21 +1,45 @@
 import { z } from "zod";
 import { runCommand } from "../lib/shell.js";
+import { config } from "../config.js";
 
 export const CommitPatchInputSchema = z.object({
-  worktreePath: z.string(),
+  worktreePath: z.string().optional(),
   message: z.string(),
   allowedFiles: z.array(z.string()),
-  writeMode: z.boolean().optional().default(false),
+  writeMode: z.boolean().optional().default(true),
 });
 
-export async function commitPatchHandler(args: z.infer<typeof CommitPatchInputSchema>) {
-  if (!args.writeMode) {
+export type CommitPatchInput = z.input<typeof CommitPatchInputSchema>;
+
+export async function commitPatchHandler(args: CommitPatchInput) {
+  const parsed = CommitPatchInputSchema.parse(args || {});
+  const worktreePath = parsed.worktreePath || config.repoPath;
+
+  if (parsed.writeMode === false) {
     throw new Error("writeMode must be true to commit changes.");
   }
-  const statusResult = await runCommand("git", ["status", "--porcelain"], { cwd: args.worktreePath });
-  const changedFiles = statusResult.stdout.trim().split("\n").filter(l => l.length > 0).map(l => l.substring(3));
 
-  const forbiddenFiles = changedFiles.filter(f => !args.allowedFiles.includes(f));
+  const statusResult = await runCommand("git", ["status", "--porcelain"], { cwd: worktreePath });
+
+  const changedFiles = statusResult.stdout
+    .split("\n")
+    .map((l) => (l.endsWith("\r") ? l.slice(0, -1) : l))
+    .filter((l) => l.trim().length > 0)
+    .map((l) => {
+      const match = l.match(/^..\s+(.*)$/);
+      if (!match) return l.slice(3).trim();
+      let file = match[1].trim();
+      if (file.includes(" -> ")) {
+        file = file.split(" -> ").pop()!.trim();
+      }
+      if (file.startsWith('"') && file.endsWith('"')) {
+        file = file.slice(1, -1);
+      }
+      return file;
+    })
+    .filter(Boolean);
+
+  const forbiddenFiles = changedFiles.filter((f) => !parsed.allowedFiles.includes(f));
   if (forbiddenFiles.length > 0) {
     throw new Error(`The following files are not in the allowed list: ${forbiddenFiles.join(", ")}`);
   }
@@ -24,18 +48,18 @@ export async function commitPatchHandler(args: z.infer<typeof CommitPatchInputSc
     throw new Error("No changes to commit.");
   }
 
-  await runCommand("git", ["add", "."], { cwd: args.worktreePath });
-  const commitResult = await runCommand("git", ["commit", "-m", args.message], { cwd: args.worktreePath });
+  await runCommand("git", ["add", "--", ...changedFiles], { cwd: worktreePath });
+  const commitResult = await runCommand("git", ["commit", "-m", parsed.message], { cwd: worktreePath });
 
   if (commitResult.exitCode !== 0) {
     throw new Error(`Failed to commit: ${commitResult.stderr}`);
   }
 
-  const shaResult = await runCommand("git", ["rev-parse", "HEAD"], { cwd: args.worktreePath });
+  const shaResult = await runCommand("git", ["rev-parse", "HEAD"], { cwd: worktreePath });
 
   return {
     success: true,
     commitSha: shaResult.stdout.trim(),
-    changedFiles
+    changedFiles,
   };
 }

package/src/tools/repo.run_playwright.ts

@@ -4,18 +4,25 @@ import { runCommand } from "../lib/shell.js";
 export const RunPlaywrightInputSchema = z.object({
   grep: z.string().optional(),
   worktreePath: z.string().optional(),
+  timeoutSeconds: z.number().optional().default(180),
 });
 
-export async function runPlaywrightHandler(args: z.infer<typeof RunPlaywrightInputSchema>) {
+export type RunPlaywrightInput = z.input<typeof RunPlaywrightInputSchema>;
+
+export async function runPlaywrightHandler(args: RunPlaywrightInput) {
+  const parsed = RunPlaywrightInputSchema.parse(args || {});
   const tdArgs = ["repo", "run-playwright"];
-  if (args.grep) {
-    tdArgs.push("--grep", args.grep);
+  if (parsed.grep) {
+    tdArgs.push("--grep", parsed.grep);
   }
-  if (args.worktreePath) {
-    tdArgs.push("--worktree", args.worktreePath);
+  if (parsed.worktreePath) {
+    tdArgs.push("--worktree", parsed.worktreePath);
   }
 
-  const result = await runCommand("td-cli", tdArgs);
+  const result = await runCommand("td-cli", tdArgs, {
+    cwd: parsed.worktreePath,
+    timeout: parsed.timeoutSeconds * 1000,
+  });
 
   if (result.exitCode !== 0) {
     // Playwright might exit with non-zero if tests fail, but our td-cli should ideally handle it

package/src/tools/repo.run_tests.ts

@@ -7,8 +7,31 @@ export const RunTestsInputSchema = z.object({
   worktreePath: z.string().optional(),
 });
 
-export async function runTestsHandler(args: z.infer<typeof RunTestsInputSchema>) {
-  const commands = args.commands || [
+export type RunTestsInput = z.input<typeof RunTestsInputSchema>;
+
+export function parseCommandArguments(fullCmd: string): [string, string[]] {
+  const tokens: string[] = [];
+  const regex = /"([^"\\]*(?:\\.[^"\\]*)*)"|'([^'\\]*(?:\\.[^'\\]*)*)'|(\S+)/g;
+  let match: RegExpExecArray | null;
+  while ((match = regex.exec(fullCmd)) !== null) {
+    if (match[1] !== undefined) {
+      tokens.push(match[1].replace(/\\"/g, '"'));
+    } else if (match[2] !== undefined) {
+      tokens.push(match[2].replace(/\\'/g, "'"));
+    } else {
+      tokens.push(match[3]);
+    }
+  }
+  if (tokens.length === 0) {
+    return ["", []];
+  }
+  const [cmd, ...args] = tokens;
+  return [cmd, args];
+}
+
+export async function runTestsHandler(args: RunTestsInput) {
+  const parsed = RunTestsInputSchema.parse(args || {});
+  const commands = parsed.commands || [
     "pnpm install --frozen-lockfile",
     "pnpm lint",
     "pnpm test",
@@ -19,11 +42,11 @@ export async function runTestsHandler(args: z.infer<typeof RunTestsInputSchema>)
   let success = true;
 
   for (const fullCmd of commands) {
-    const [cmd, ...cmdArgs] = fullCmd.split(" ");
+    const [cmd, cmdArgs] = parseCommandArguments(fullCmd);
     try {
       const res = await runCommand(cmd, cmdArgs, {
-        cwd: args.worktreePath,
-        timeout: args.timeoutSeconds * 1000
+        cwd: parsed.worktreePath,
+        timeout: parsed.timeoutSeconds * 1000
       });
       results.push(res);
       if (res.exitCode !== 0) {