import { SquarePen } from "lucide-react-native";
import invariant from "tiny-invariant";
import { WorkspaceDraftAgentTab } from "@/screens/workspace/workspace-draft-agent-tab";
import { usePaneContext } from "@/panels/pane-context";
import type { PanelRegistration } from "@/panels/panel-registry";
import { useSessionStore } from "@/stores/session-store";
import { normalizeAgentSnapshot } from "@/utils/agent-snapshots";

function useDraftPanelDescriptor() {
  return {
    label: "New Agent",
    subtitle: "New Agent",
    titleState: "ready" as const,
    icon: SquarePen,
    statusBucket: null,
  };
}

function DraftPanel() {
  const {
    serverId,
    workspaceId,
    tabId,
    target,
    openFileInWorkspace,
    retargetCurrentTab,
  } = usePaneContext();
  invariant(target.kind === "draft", "DraftPanel requires draft target");

  return (
    <WorkspaceDraftAgentTab
      serverId={serverId}
      workspaceId={workspaceId}
      tabId={tabId}
      draftId={target.draftId}
      onOpenWorkspaceFile={({ filePath }) => {
        openFileInWorkspace(filePath);
      }}
      onCreated={(agentSnapshot) => {
        const normalized = normalizeAgentSnapshot(agentSnapshot, serverId);
        retargetCurrentTab({ kind: "agent", agentId: agentSnapshot.id });
        useSessionStore.getState().setAgents(serverId, (prev) => {
          const next = new Map(prev);
          next.set(agentSnapshot.id, normalized);
          return next;
        });
      }}
    />
  );
}

export const draftPanelRegistration: PanelRegistration<"draft"> = {
  kind: "draft",
  component: DraftPanel,
  useDescriptor: useDraftPanelDescriptor,
};