76 lines
3.1 KiB
TypeScript
76 lines
3.1 KiB
TypeScript
import { type } from "arktype";
|
|
import { buildShockbotFooter, stripExistingFooter } from "../utils/buildShockbotFooter.ts";
|
|
import { log } from "../utils/cli.ts";
|
|
import { fixDoubleEscapedString } from "../utils/fixDoubleEscapedString.ts";
|
|
import { $ } from "../utils/shell.ts";
|
|
import type { ToolContext } from "./server.ts";
|
|
import { execute, tool } from "./shared.ts";
|
|
|
|
interface GiteaPull { number: number; html_url?: string; title?: string; head?: { ref?: string }; base?: { ref?: string } }
|
|
|
|
function buildPrBodyWithFooter(ctx: ToolContext, body: string): string {
|
|
return `${stripExistingFooter(fixDoubleEscapedString(body))}${buildShockbotFooter({ model: ctx.toolState.model })}`;
|
|
}
|
|
|
|
export const UpdatePullRequestBody = type({
|
|
pull_number: type.number.describe("the pull request number to update"),
|
|
body: type.string.describe("the new body content for the pull request"),
|
|
});
|
|
|
|
export function UpdatePullRequestBodyTool(ctx: ToolContext) {
|
|
return tool({
|
|
name: "update_pull_request_body",
|
|
description: "Update the body/description of an existing pull request",
|
|
parameters: UpdatePullRequestBody,
|
|
execute: execute(async (params) => {
|
|
const r = await ctx.gitea.request(
|
|
"PATCH /repos/{owner}/{repo}/pulls/{index}",
|
|
{ owner: ctx.repo.owner, repo: ctx.repo.name, index: params.pull_number, body: buildPrBodyWithFooter(ctx, params.body) }
|
|
);
|
|
const data = r.data as GiteaPull;
|
|
log.info(`» updated pull request #${data.number}`);
|
|
ctx.toolState.wasUpdated = true;
|
|
return { success: true, number: data.number, url: data.html_url };
|
|
}),
|
|
});
|
|
}
|
|
|
|
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')"),
|
|
"draft?": type.boolean,
|
|
});
|
|
|
|
export function CreatePullRequestTool(ctx: ToolContext) {
|
|
return tool({
|
|
name: "create_pull_request",
|
|
description: "Create a pull request from the current branch",
|
|
parameters: PullRequest,
|
|
execute: execute(async (params) => {
|
|
const currentBranch = $("git", ["rev-parse", "--abbrev-ref", "HEAD"], { log: false }).trim();
|
|
const r = await ctx.gitea.request(
|
|
"POST /repos/{owner}/{repo}/pulls",
|
|
{
|
|
owner: ctx.repo.owner, repo: ctx.repo.name,
|
|
title: params.title, body: buildPrBodyWithFooter(ctx, params.body),
|
|
head: currentBranch, base: params.base,
|
|
}
|
|
);
|
|
const data = r.data as GiteaPull;
|
|
log.info(`» created pull request #${data.number}`);
|
|
|
|
const reviewer = ctx.payload.triggerer;
|
|
if (reviewer && data.number) {
|
|
try {
|
|
await ctx.gitea.request(
|
|
"POST /repos/{owner}/{repo}/pulls/{index}/requested_reviewers",
|
|
{ owner: ctx.repo.owner, repo: ctx.repo.name, index: data.number, reviewers: [reviewer] }
|
|
);
|
|
} catch { log.debug(`failed to request review from ${reviewer}`); }
|
|
}
|
|
return { success: true, number: data.number, url: data.html_url, title: data.title, head: data.head?.ref, base: data.base?.ref };
|
|
}),
|
|
});
|
|
}
|