"use client";
import { useRouter } from "next/navigation";
import { useToast } from "@/components/Toast";
import { type ApiError, api } from "@/lib/client/api";

export function WorkerActions({ id, status, draining }: { id: string; status: string; draining: boolean }) {
  const router = useRouter();
  const toast = useToast();
  const run = async (path: string, body: unknown, msg: string) => {
    try {
      await api(path, { json: body });
      toast.push("success", msg);
      router.refresh();
    } catch (e) {
      toast.push("error", (e as ApiError).message);
    }
  };
  return (
    <div className="flex gap-2">
      <button
        type="button"
        className="btn btn-sm"
        onClick={() =>
          run(
            `/api/admin/workers/${id}/drain`,
            { drain: !draining },
            draining ? "Drain cleared." : "Draining: no new jobs will be leased; running jobs finish.",
          )
        }
      >
        {draining ? "Stop draining" : "Drain"}
      </button>
      {status === "quarantined" ? (
        <button
          type="button"
          className="btn btn-sm"
          onClick={() => run(`/api/admin/workers/${id}/quarantine`, { reason: null }, "Quarantine lifted.")}
        >
          Lift quarantine
        </button>
      ) : (
        <button
          type="button"
          className="btn btn-sm btn-danger"
          onClick={() => {
            const reason = prompt("Quarantine reason (recorded in the audit log):");
            if (reason) run(`/api/admin/workers/${id}/quarantine`, { reason }, "Worker quarantined.");
          }}
        >
          Quarantine
        </button>
      )}
    </div>
  );
}