import { type } from "arktype";
import type { Context } from "../main.ts";
import { execute, tool } from "./shared.ts";

export const Issue = type({
  title: type.string.describe("the title of the issue"),
  body: type.string.describe("the body content of the issue"),
  labels: type.string
    .array()
    .describe("optional array of label names to apply to the issue")
    .optional(),
  assignees: type.string
    .array()
    .describe("optional array of usernames to assign to the issue")
    .optional(),
});

export function IssueTool(ctx: Context) {
  return tool({
    name: "create_issue",
    description: "Create a new GitHub issue",
    parameters: Issue,
    execute: execute(ctx, async ({ title, body, labels, assignees }) => {
      const result = await ctx.octokit.rest.issues.create({
        owner: ctx.owner,
        repo: ctx.name,
        title: title,
        body: body,
        labels: labels ?? [],
        assignees: assignees ?? [],
      });

      return {
        success: true,
        issueId: result.data.id,
        number: result.data.number,
        url: result.data.html_url,
        title: result.data.title,
        state: result.data.state,
        labels: result.data.labels?.map((label) => (typeof label === "string" ? label : label.name)),
        assignees: result.data.assignees?.map((assignee) => assignee.login),
      };
    }),
  });
}