import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
import { isToolCallEventType } from "@mariozechner/pi-coding-agent";

export default function mutationGuardRepro(pi: ExtensionAPI) {
        pi.on("tool_call", async (event, ctx) => {
                if (!isToolCallEventType("edit", event)) return undefined;

                const path = event.input.path;

                if (!ctx.hasUI) {
                        return {
                                block: true,
                                reason: `Edit ${path} blocked: confirmation requires UI.`,
                        };
                }

                const choice = await ctx.ui.select(`Confirm edit ${path}`, [
                        "Allow",
                        "Block with note",
                ]);

                if (choice === "Allow") {
                        return undefined;
                }

                const note = (await ctx.ui.input("Reject note", "Type note")) ?? "";

                return {
                        block: true,
                        reason: `Mutation rejected by user for edit ${path}. User note: ${note.trim() || "none"}`,
                };
        });
}