46 lines
1.6 KiB
TypeScript
46 lines
1.6 KiB
TypeScript
import { type } from "arktype";
|
|
import { log } from "../utils/cli.ts";
|
|
import type { ToolContext } from "./server.ts";
|
|
import { execute, tool } from "./shared.ts";
|
|
|
|
export const AddLabelsParams = type({
|
|
issue_number: type.number.describe("the issue or PR number to add labels to"),
|
|
labels: type.string.array().atLeastLength(1).describe("array of label names to add"),
|
|
});
|
|
|
|
export function AddLabelsTool(ctx: ToolContext) {
|
|
return tool({
|
|
name: "add_labels",
|
|
description:
|
|
"Add labels to a Gitea issue or pull request. Only use labels that already exist in the repository.",
|
|
parameters: AddLabelsParams,
|
|
execute: execute(async ({ issue_number, labels }) => {
|
|
// Resolve label names to IDs (Gitea uses IDs, not names)
|
|
const allLabels = await ctx.gitea.paginate(ctx.gitea.rest.issue.issueListLabels, {
|
|
owner: ctx.repo.owner,
|
|
repo: ctx.repo.name,
|
|
});
|
|
const labelIds = labels
|
|
.map((name) => allLabels.find((l) => l.name === name)?.id)
|
|
.filter((id): id is number => typeof id === "number");
|
|
|
|
if (labelIds.length === 0) {
|
|
return { success: true, labels: [], message: "No matching labels found in repository" };
|
|
}
|
|
|
|
const result = await ctx.gitea.rest.issue.issueAddLabel({
|
|
owner: ctx.repo.owner,
|
|
repo: ctx.repo.name,
|
|
index: issue_number,
|
|
body: { labels: labelIds },
|
|
});
|
|
log.info(`» added labels [${labels.join(", ")}] to issue #${issue_number}`);
|
|
|
|
return {
|
|
success: true,
|
|
labels: result.data.map((label) => label.name).filter(Boolean),
|
|
};
|
|
}),
|
|
});
|
|
}
|