import { type } from "arktype";
import { log } from "../utils/cli.ts";
import { $ } from "../utils/shell.ts";
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 = $("git", ["rev-parse", "--abbrev-ref", "HEAD"], { log: false });

    log.info(`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,
    };
  }),
});