import { tool } from "@opencode-ai/plugin";
import type { OpenCodeContext } from "../../../types/plugin";
import { getBackgroundAgent } from "../../../runtime";
import { definePlugin } from "../../../plugin-api/define-plugin";
import { handleBackgroundCancel } from "./handler";
import type { BackgroundCancelArgs } from "./types";

const backgroundCancelTool = tool({
  description: "Cancel running background task(s). Use all=true to cancel ALL before final answer.",
  args: {
    task_id: tool.schema.string().optional().describe("Task ID to cancel (required if all=false)"),
    all: tool.schema
      .boolean()
      .optional()
      .describe("Cancel all running background tasks (default: false)"),
  },
  async execute(args: BackgroundCancelArgs, toolContext) {
    const { manager } = getBackgroundAgent();
    return handleBackgroundCancel(manager, toolContext as unknown as OpenCodeContext, args);
  },
});

export default definePlugin({
  name: "background-cancel",
  version: "1.0.0",
  dependencies: ["delegate-task"],
  tools: {
    background_cancel: backgroundCancelTool,
  },
});