opencode working

2025-12-10 01:02:39 -05:00
parent bbda005ee9
commit ccb28d8cf5
13 changed files with 567 additions and 89 deletions
@@ -0,0 +1,154 @@
+import { type } from "arktype";
+import { log } from "../utils/cli.ts";
+import { containsSecrets } from "../utils/secrets.ts";
+import { $ } from "../utils/shell.ts";
+import { contextualize, tool } from "./shared.ts";
+
+export const CreateBranch = type({
+  branchName: type.string.describe(
+    "The name of the branch to create (e.g., 'pullfrog/123-fix-bug')"
+  ),
+  baseBranch: type.string.describe("The base branch to create from (e.g., 'main')").default("main"),
+});
+
+export const CreateBranchTool = tool({
+  name: "create_branch",
+  description:
+    "Create a new git branch from the specified base branch. The branch will be created locally and pushed to the remote repository.",
+  parameters: CreateBranch,
+  execute: contextualize(async ({ branchName, baseBranch }) => {
+    // validate branch name for secrets
+    if (containsSecrets(branchName)) {
+      throw new Error(
+        "Branch creation blocked: secrets detected in branch name. " +
+          "Please remove any sensitive information (API keys, tokens, passwords) before creating a branch."
+      );
+    }
+
+    log.info(`Creating branch ${branchName} from ${baseBranch}`);
+
+    // fetch base branch to ensure we're up to date
+    $("git", ["fetch", "origin", baseBranch, "--depth=1"]);
+
+    // checkout base branch
+    $("git", ["checkout", baseBranch]);
+
+    // create and checkout new branch
+    $("git", ["checkout", "-b", branchName]);
+
+    // push branch to remote (set upstream)
+    $("git", ["push", "-u", "origin", branchName]);
+
+    log.info(`Successfully created and pushed branch ${branchName}`);
+
+    return {
+      success: true,
+      branchName,
+      baseBranch,
+      message: `Branch ${branchName} created from ${baseBranch} and pushed to remote`,
+    };
+  }),
+});
+
+export const CommitFiles = type({
+  message: type.string.describe("The commit message"),
+  files: type.string
+    .array()
+    .describe(
+      "Array of file paths to commit (relative to repo root). If empty, commits all staged changes."
+    ),
+});
+
+export const CommitFilesTool = tool({
+  name: "commit_files",
+  description:
+    "Stage and commit files with a commit message. If files array is empty, commits all staged changes. The commit will be attributed to the correct bot account.",
+  parameters: CommitFiles,
+  execute: contextualize(async ({ message, files }) => {
+    // validate commit message for secrets
+    if (containsSecrets(message)) {
+      throw new Error(
+        "Commit blocked: secrets detected in commit message. " +
+          "Please remove any sensitive information (API keys, tokens, passwords) before committing."
+      );
+    }
+
+    // validate files for secrets if provided
+    if (files.length > 0) {
+      for (const file of files) {
+        try {
+          // try to read file content - if it exists, check for secrets
+          const content = $("cat", [file], { log: false });
+          if (containsSecrets(content)) {
+            throw new Error(
+              `Commit blocked: secrets detected in file ${file}. ` +
+                "Please remove any sensitive information (API keys, tokens, passwords) before committing."
+            );
+          }
+        } catch (error) {
+          // if error is about secrets, re-throw it
+          if (error instanceof Error && error.message.includes("Commit blocked")) {
+            throw error;
+          }
+          // if file doesn't exist (cat fails), that's ok - it will be created by git add
+          // other errors are also ok - git add will handle them
+        }
+      }
+    }
+
+    const currentBranch = $("git", ["rev-parse", "--abbrev-ref", "HEAD"], { log: false });
+    log.info(`Committing files on branch ${currentBranch}`);
+
+    // stage files if provided, otherwise stage all changes
+    if (files.length > 0) {
+      $("git", ["add", ...files]);
+    } else {
+      $("git", ["add", "."]);
+    }
+
+    // commit with message
+    $("git", ["commit", "-m", message]);
+
+    const commitSha = $("git", ["rev-parse", "HEAD"], { log: false });
+    log.info(`Successfully committed: ${commitSha.substring(0, 7)}`);
+
+    return {
+      success: true,
+      commitSha,
+      branch: currentBranch,
+      message: `Committed ${files.length > 0 ? files.length + " file(s)" : "all changes"} with message: ${message}`,
+    };
+  }),
+});
+
+export const PushBranch = type({
+  branchName: type.string
+    .describe("The branch name to push (defaults to current branch)")
+    .optional(),
+  force: type.boolean.describe("Force push (use with caution)").default(false),
+});
+
+export const PushBranchTool = tool({
+  name: "push_branch",
+  description:
+    "Push the current branch (or specified branch) to the remote repository. Never force push unless explicitly requested.",
+  parameters: PushBranch,
+  execute: contextualize(async ({ branchName, force }) => {
+    const branch = branchName || $("git", ["rev-parse", "--abbrev-ref", "HEAD"], { log: false });
+
+    if (force) {
+      log.warning(`Force pushing branch ${branch} - this will overwrite remote history`);
+      $("git", ["push", "--force", "origin", branch]);
+    } else {
+      log.info(`Pushing branch ${branch} to remote`);
+      $("git", ["push", "origin", branch]);
+    }
+
+    return {
+      success: true,
+      branch,
+      force,
+      message: `Successfully pushed branch ${branch} to remote`,
+    };
+  }),
+});