Colin McDonnell
58 lines
1.9 KiB
TypeScript
58 lines
1.9 KiB
TypeScript
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 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) => {
|
|
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 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,
|
|
});
|
|
|
|
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,
|
|
};
|
|
}),
|
|
});
|