import { execSync } from "node:child_process";
import { type } from "arktype";
import { contextualize, tool } from "./shared.ts";

export const PullRequest = type({
  title: type.string.describe("the title of the pull request"),
  body: type.string.describe("the body content of the pull request"),
  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) => {
    // Get the current branch name
    const currentBranch = execSync("git rev-parse --abbrev-ref HEAD", {
      encoding: "utf8",
    }).trim();

    console.log(`Current branch: ${currentBranch}`);

    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,
    };
  }),
});