// @ts-check

import { api } from "../../api.js";
import { ComfyButton } from "../components/button.js";

export function getInteruptButton(visibility) {
	const btn = new ComfyButton({
		icon: "close",
		tooltip: "Cancel current generation",
		enabled: false,
		action: () => {
			api.interrupt();
		},
		classList: ["comfyui-button", "comfyui-interrupt-button", visibility],
	});

	api.addEventListener("status", ({ detail }) => {
		const sz = detail?.exec_info?.queue_remaining;
		btn.enabled = sz > 0;
	});

	return btn;
}