kandev/apps/web/components/task-create-dialog-form-body.tsx

"use client";

import { memo } from "react";
import Link from "@/components/routing/app-link";
import type { AgentProfileOption } from "@/lib/state/slices";
import type { WorkflowSnapshotData } from "@/lib/state/slices/kanban/types";
import { WorkflowSelectorRow } from "@/components/workflow-selector-row";
import { AgentLogo } from "@/components/agent-logo";
import type { DialogFormState } from "@/components/task-create-dialog-types";
import type { DialogPromptEnhance } from "@/components/task-create-dialog-types";
import type { useKeyboardShortcutHandler } from "@/hooks/use-keyboard-shortcut";
import { TaskFormInputs } from "@/components/task-create-dialog-selectors";
import { PromptResultRecovery } from "@/components/prompt-result-recovery";
import type { JiraTicket } from "@/lib/types/jira";
import type { LinearIssue } from "@/lib/types/linear";
import { useTranslation } from "react-i18next";

type SelectorOption = {
  value: string;
  label: string;
  renderLabel: () => React.ReactNode;
};

type CreateEditSelectorsProps = {
  isTaskStarted: boolean;
  agentProfiles: AgentProfileOption[];
  agentProfilesLoading: boolean;
  agentProfileOptions: SelectorOption[];
  agentProfileId: string;
  onAgentProfileChange: (value: string) => void;
  isCreatingSession: boolean;
  executorProfileOptions: Array<{
    value: string;
    label: string;
    renderLabel?: () => React.ReactNode;
  }>;
  executorProfileId: string;
  onExecutorProfileChange: (value: string) => void;
  executorsLoading: boolean;
  AgentSelectorComponent: React.ComponentType<{
    options: SelectorOption[];
    value: string;
    onValueChange: (value: string) => void;
    disabled: boolean;
    placeholder: string;
    triggerClassName?: string;
    popoverPortal?: boolean;
  }>;
  ExecutorProfileSelectorComponent: React.ComponentType<{
    options: Array<{ value: string; label: string; renderLabel?: () => React.ReactNode }>;
    value: string;
    onValueChange: (value: string) => void;
    disabled: boolean;
    placeholder: string;
    triggerClassName?: string;
    popoverPortal?: boolean;
  }>;
  workflowAgentLocked: boolean;
  noCompatibleAgent: boolean;
  executorProfileName: string | null;
};

type AgentColumnProps = Pick<
  CreateEditSelectorsProps,
  | "agentProfiles"
  | "agentProfilesLoading"
  | "agentProfileOptions"
  | "agentProfileId"
  | "onAgentProfileChange"
  | "isCreatingSession"
  | "AgentSelectorComponent"
  | "workflowAgentLocked"
  | "noCompatibleAgent"
  | "executorProfileName"
  | "executorProfileId"
>;

function NoCompatibleAgentState({
  executorProfileName,
  executorProfileId,
}: {
  executorProfileName: string | null;
  executorProfileId: string;
}) {
  const { t } = useTranslation();
  const target = executorProfileName ? `“${executorProfileName}”` : "this executor";
  const href = executorProfileId
    ? `/settings/executors/${executorProfileId}`
    : "/settings/executors";
  return (
    <div
      className="flex h-auto min-h-7 items-center justify-between gap-3 rounded-sm border border-input px-3 py-1.5 text-xs text-muted-foreground"
      data-testid="agent-profile-empty-state"
    >
      <span>{t("task:noCompatibleAgentProfilesFor", { target })}</span>
      <Link href={href} className="shrink-0 cursor-pointer text-primary hover:underline">
        {t("task:configureCredentials")}
      </Link>
    </div>
  );
}

function AgentColumn({
  agentProfiles,
  agentProfilesLoading,
  agentProfileOptions,
  agentProfileId,
  onAgentProfileChange,
  isCreatingSession,
  AgentSelectorComponent,
  workflowAgentLocked,
  noCompatibleAgent,
  executorProfileName,
  executorProfileId,
}: AgentColumnProps) {
  const { t } = useTranslation();
  if (agentProfiles.length === 0 && !agentProfilesLoading) {
    return (
      <div
        className="flex h-7 items-center justify-center gap-2 rounded-sm border border-input px-3 text-xs text-muted-foreground"
        data-testid="agent-profile-empty-state"
      >
        <span>{t("task:noAgentsFound")}</span>
        <Link href="/settings/agents" className="cursor-pointer text-primary hover:underline">
          {t("task:addAgent")}
        </Link>
      </div>
    );
  }
  if (noCompatibleAgent && !agentProfilesLoading) {
    return (
      <NoCompatibleAgentState
        executorProfileName={executorProfileName}
        executorProfileId={executorProfileId}
      />
    );
  }
  const placeholder = agentProfilesLoading ? t("task:loadingAgents") : t("task:selectAgent");
  return (
    <>
      <AgentSelectorComponent
        options={agentProfileOptions}
        value={agentProfileId}
        onValueChange={onAgentProfileChange}
        placeholder={placeholder}
        disabled={agentProfilesLoading || isCreatingSession || workflowAgentLocked}
        popoverPortal
      />
      {workflowAgentLocked && (
        <p className="text-[11px] text-muted-foreground mt-1">{t("task:agentSetByWorkflow")}</p>
      )}
    </>
  );
}

export const CreateEditSelectors = memo(function CreateEditSelectors(
  props: CreateEditSelectorsProps,
) {
  const { t } = useTranslation();
  if (props.isTaskStarted) return null;
  const {
    executorProfileOptions,
    executorProfileId,
    onExecutorProfileChange,
    executorsLoading,
    ExecutorProfileSelectorComponent,
  } = props;

  // Branch + repo selection (and the FreshBranchToggle, which is per-task
  // branch strategy) live in the chip row above the description; this row
  // carries only agent and executor profile selectors.
  return (
    <div className="grid min-w-0 grid-cols-1 gap-4 sm:grid-cols-2">
      <div className="min-w-0">
        <AgentColumn {...props} />
      </div>
      <div className="min-w-0">
        <ExecutorProfileSelectorComponent
          options={executorProfileOptions}
          value={executorProfileId}
          onValueChange={onExecutorProfileChange}
          placeholder={executorsLoading ? t("task:loadingProfiles") : t("task:selectProfile")}
          disabled={executorsLoading}
          popoverPortal
        />
      </div>
    </div>
  );
});

type SessionSelectorsProps = {
  agentProfileOptions: SelectorOption[];
  agentProfileId: string;
  onAgentProfileChange: (value: string) => void;
  agentProfilesLoading: boolean;
  isCreatingSession: boolean;
  executorProfileOptions: Array<{
    value: string;
    label: string;
    renderLabel?: () => React.ReactNode;
  }>;
  executorProfileId: string;
  onExecutorProfileChange: (value: string) => void;
  executorsLoading: boolean;
  AgentSelectorComponent: React.ComponentType<{
    options: SelectorOption[];
    value: string;
    onValueChange: (value: string) => void;
    disabled: boolean;
    placeholder: string;
    triggerClassName?: string;
    popoverPortal?: boolean;
  }>;
  ExecutorProfileSelectorComponent: React.ComponentType<{
    options: Array<{ value: string; label: string; renderLabel?: () => React.ReactNode }>;
    value: string;
    onValueChange: (value: string) => void;
    disabled: boolean;
    placeholder: string;
    triggerClassName?: string;
    popoverPortal?: boolean;
  }>;
};

export const SessionSelectors = memo(function SessionSelectors({
  agentProfileOptions,
  agentProfileId,
  onAgentProfileChange,
  agentProfilesLoading,
  isCreatingSession,
  executorProfileOptions,
  executorProfileId,
  onExecutorProfileChange,
  executorsLoading,
  AgentSelectorComponent,
  ExecutorProfileSelectorComponent,
}: SessionSelectorsProps) {
  const { t } = useTranslation();
  return (
    <div className="grid min-w-0 grid-cols-1 gap-3 sm:grid-cols-2">
      <AgentSelectorComponent
        options={agentProfileOptions}
        value={agentProfileId}
        onValueChange={onAgentProfileChange}
        placeholder={
          agentProfilesLoading ? t("task:loadingAgentProfiles") : t("task:selectAgentProfile")
        }
        disabled={agentProfilesLoading || isCreatingSession}
        popoverPortal
      />
      <ExecutorProfileSelectorComponent
        options={executorProfileOptions}
        value={executorProfileId}
        onValueChange={onExecutorProfileChange}
        placeholder={executorsLoading ? t("task:loadingProfiles") : t("task:selectProfile")}
        disabled={executorsLoading || isCreatingSession}
        popoverPortal
      />
    </div>
  );
});

type WorkflowSectionProps = {
  isCreateMode: boolean;
  isTaskStarted: boolean;
  workflows: Array<{
    id: string;
    name: string;
    agent_profile_id?: string;
    hidden?: boolean;
    [key: string]: unknown;
  }>;
  snapshots: Record<string, WorkflowSnapshotData>;
  effectiveWorkflowId: string | null;
  onWorkflowChange: (value: string) => void;
  agentProfiles: AgentProfileOption[];
  /**
   * When true the picker is hidden entirely. Used by feature wrappers
   * (Improve Kandev) where the workflow is enforced and the user must not be
   * able to switch to a different one. The wrapper is responsible for
   * surfacing the workflow elsewhere (e.g. a steps preview).
   */
  workflowLocked?: boolean;
};

function renderWorkflowSection({
  isCreateMode,
  isTaskStarted,
  workflows: allWorkflows,
  snapshots,
  effectiveWorkflowId,
  onWorkflowChange,
  agentProfiles,
  workflowLocked,
}: WorkflowSectionProps) {
  // Hidden workflows (e.g. improve-kandev) are excluded from the picker; they
  // remain reachable via their dedicated entry point.
  const workflows = allWorkflows.filter((w) => !w.hidden);

  if (!isCreateMode || isTaskStarted) return null;
  if (workflowLocked) return null;

  if (!effectiveWorkflowId || workflows.length > 1) {
    return (
      <WorkflowSelectorRow
        workflows={workflows}
        snapshots={snapshots}
        selectedWorkflowId={effectiveWorkflowId ?? null}
        onWorkflowChange={onWorkflowChange}
        agentProfiles={agentProfiles}
      />
    );
  }

  // Single selected workflow — show agent override info if any overrides exist
  if (workflows.length === 1) {
    const singleWorkflow = workflows[0];
    if (!singleWorkflow) return null;
    const snapshot = snapshots[singleWorkflow.id];
    const workflowProfile = singleWorkflow.agent_profile_id
      ? agentProfiles.find((p) => p.id === singleWorkflow.agent_profile_id)
      : null;
    const stepsWithOverrides = (snapshot?.steps ?? [])
      .filter((s) => s.agent_profile_id)
      .map((s) => ({
        name: s.title,
        profile: agentProfiles.find((p) => p.id === s.agent_profile_id),
      }))
      .filter((s) => s.profile);
    if (!workflowProfile && stepsWithOverrides.length === 0) return null;
    return (
      <div
        className="flex flex-wrap items-center gap-2 text-xs text-muted-foreground"
        data-testid="workflow-override-info"
      >
        {workflowProfile && (
          <span className="flex items-center gap-1">
            <AgentLogo agentName={workflowProfile.agent_name} size={14} className="shrink-0" />
            <span>{workflowProfile.label}</span>
          </span>
        )}
        {stepsWithOverrides.map((s) => (
          <span key={s.name} className="flex items-center gap-1">
            <span className="text-muted-foreground/50">{s.name}:</span>
            <AgentLogo agentName={s.profile!.agent_name} size={14} className="shrink-0" />
            <span>{s.profile!.label}</span>
          </span>
        ))}
      </div>
    );
  }

  return null;
}

export const WorkflowSection = memo(function WorkflowSection(workflowProps: WorkflowSectionProps) {
  if (!workflowProps.isCreateMode || workflowProps.isTaskStarted) return null;
  return renderWorkflowSection(workflowProps);
});

export type DialogPromptSectionProps = {
  isSessionMode: boolean;
  isTaskStarted: boolean;
  initialDescription: string;
  fs: DialogFormState;
  onPendingAttachmentUploadsChange?: (pending: boolean) => void;
  handleKeyDown: ReturnType<typeof useKeyboardShortcutHandler>;
  enhance?: DialogPromptEnhance;
  workspaceId?: string | null;
  onJiraImport?: (ticket: JiraTicket) => void;
  onLinearImport?: (issue: LinearIssue) => void;
  /** Extension slot rendered below the description textarea (e.g. log-capture toggle). */
  extraFormSlot?: React.ReactNode;
  /** Optional override for the description textarea placeholder. */
  descriptionPlaceholder?: string;
  /** Optional slot rendered above the description textarea (e.g. a tab toggle). */
  aboveDescriptionSlot?: React.ReactNode;
  /**
   * Whether the description textarea should grab focus on mount. Defaults to
   * `!isTaskStarted`. Callers that render a task-name input above the
   * description should pass `false` so the name field wins focus.
   */
  autoFocusDescription?: boolean;
  /** Submits the form for a plugin composer action that finished producing text. */
  onComposerSubmit?: () => boolean | Promise<boolean>;
};

// importBindings collapses the optional Jira/Linear import callbacks into the
// shape TaskFormInputs expects, dropping integrations that aren't applicable
// (session mode, started tasks, or no callback wired). Keeps DialogPromptSection
// below the cyclomatic-complexity bar.
function importBindings<T>(
  enabled: boolean,
  workspaceId: string | null,
  onImport: ((value: T) => void) | undefined,
) {
  if (!enabled || !onImport) return undefined;
  return { workspaceId, disabled: false, onImport };
}

export function DialogPromptSection({
  isSessionMode,
  isTaskStarted,
  initialDescription,
  fs,
  onPendingAttachmentUploadsChange,
  handleKeyDown,
  enhance,
  workspaceId,
  onJiraImport,
  onLinearImport,
  extraFormSlot,
  descriptionPlaceholder,
  aboveDescriptionSlot,
  autoFocusDescription,
  onComposerSubmit,
}: DialogPromptSectionProps) {
  const importsEnabled = !isSessionMode && !isTaskStarted;
  const ws = workspaceId ?? null;
  const shouldAutoFocus = autoFocusDescription ?? !isTaskStarted;
  return (
    <>
      {aboveDescriptionSlot}
      <TaskFormInputs
        key={fs.openCycle}
        isSessionMode={isSessionMode}
        workspaceId={workspaceId}
        autoFocus={shouldAutoFocus}
        initialDescription={initialDescription}
        onDescriptionChange={fs.setHasDescription}
        onPendingAttachmentUploadsChange={onPendingAttachmentUploadsChange}
        onKeyDown={handleKeyDown}
        descriptionValueRef={fs.descriptionInputRef}
        disabled={isTaskStarted}
        placeholder={descriptionPlaceholder}
        onEnhancePrompt={enhance?.onEnhance}
        isEnhancingPrompt={enhance?.isLoading}
        isUtilityConfigured={enhance?.isConfigured}
        jiraImport={importBindings(importsEnabled, ws, onJiraImport)}
        linearImport={importBindings(importsEnabled, ws, onLinearImport)}
        onComposerSubmit={onComposerSubmit}
      />
      <PromptResultRecovery
        pendingResult={enhance?.pendingResult ?? null}
        onApply={enhance?.onApplyPending ?? (() => undefined)}
        onCopy={enhance?.onCopyPending ?? (() => undefined)}
      />
      {extraFormSlot}
    </>
  );
}