fusion/packages/dashboard/app/components/AgentImportModal.tsx

import "./AgentImportModal.css";
import { useState, useRef, useCallback, useEffect } from "react";
import { Upload, FileText, CheckCircle, AlertTriangle, X, Loader2, FolderOpen, Globe, Search, RefreshCw } from "lucide-react";
import { fetchCompanies, type CompanyEntry } from "../api";
import { useMobileScrollLock } from "../hooks/useMobileScrollLock";

export interface AgentImportModalProps {
  isOpen: boolean;
  onClose: () => void;
  onImported: () => void;
  projectId?: string;
}

/** Parsed agent preview item for display before import */
interface AgentPreview {
  name: string;
  role: string;
  title?: string;
  icon?: string;
  reportsTo?: string;
  instructionsText?: string;
  skills?: string[];
}

interface SkillPreview {
  name: string;
  description?: string;
}

/** Skill import result from the API */
interface SkillImportResult {
  imported: Array<{ name: string; path: string }>;
  skipped: string[];
  errors: Array<{ name: string; error: string }>;
}

/** Import result from the API */
interface ImportResult {
  companyName?: string;
  companySlug?: string;
  created: Array<{ id: string; name: string }>;
  skipped: string[];
  errors: Array<{ name: string; error: string }>;
  skills?: SkillImportResult;
}

interface DirectoryAgentInput {
  name: string;
  title?: string;
  icon?: string;
  role?: string;
  reportsTo?: string;
  skills?: string[];
  instructionBody?: string;
}

/** API error response shape */
interface ApiErrorResponse {
  error: string;
}

type ModalStep = "input" | "preview" | "result";
type InputMethod = "paste" | "file" | "directory" | "browse";

function parseDirectoryAgentManifest(content: string): DirectoryAgentInput {
  const match = content.match(/^---\s*\r?\n([\s\S]*?)\r?\n---(?:\r?\n([\s\S]*))?$/);
  if (!match) {
    throw new Error("Missing YAML frontmatter delimiters (---)");
  }

  const frontmatterLines = match[1].split(/\r?\n/);
  const body = match[2] ?? "";
  const result: DirectoryAgentInput = { name: "" };
  const skills: string[] = [];
  let inSkills = false;

  for (const rawLine of frontmatterLines) {
    const line = rawLine.trimEnd();
    const trimmed = line.trim();
    if (!trimmed) continue;

    if (trimmed.startsWith("skills:")) {
      inSkills = true;
      continue;
    }

    if (inSkills && trimmed.startsWith("- ")) {
      skills.push(trimmed.slice(2).trim());
      continue;
    }

    inSkills = false;

    const [key, ...valueParts] = trimmed.split(":");
    const value = valueParts.join(":").trim();
    const normalizedValue = value.replace(/^['"]|['"]$/g, "");

    if (key === "name") result.name = normalizedValue;
    if (key === "title") result.title = normalizedValue;
    if (key === "icon") result.icon = normalizedValue;
    if (key === "role") result.role = normalizedValue;
    if (key === "reportsTo") result.reportsTo = normalizedValue;
  }

  if (!result.name) {
    throw new Error("Missing required field: name");
  }

  if (skills.length > 0) {
    result.skills = skills;
  }
  if (body.trim().length > 0) {
    result.instructionBody = body;
  }

  return result;
}

/**
 * Modal for importing agents from Agent Companies manifests.
 *
 * Supports three input methods:
 * - File upload (.md/.txt files)
 * - Directory upload (webkitdirectory)
 * - Paste raw manifest content
 *
 * Flow: Input → Preview parsed agents → Import → Show results
 */
export function AgentImportModal({ isOpen, onClose, onImported, projectId }: AgentImportModalProps) {
  useMobileScrollLock(isOpen);
  const [step, setStep] = useState<ModalStep>("input");
  const [inputMethod, setInputMethod] = useState<InputMethod>("paste");
  const [manifestContent, setManifestContent] = useState("");
  const [directoryAgents, setDirectoryAgents] = useState<DirectoryAgentInput[]>([]);
  const [companyName, setCompanyName] = useState("Unknown");
  const [agents, setAgents] = useState<AgentPreview[]>([]);
  const [skills, setSkills] = useState<SkillPreview[]>([]);
  const [selectedAgentNames, setSelectedAgentNames] = useState<string[]>([]);
  const [selectedSkillNames, setSelectedSkillNames] = useState<string[]>([]);
  const [isParsing, setIsParsing] = useState(false);
  const [isImporting, setIsImporting] = useState(false);
  const [parseError, setParseError] = useState<string | null>(null);
  const [importResult, setImportResult] = useState<ImportResult | null>(null);
  const [importError, setImportError] = useState<string | null>(null);
  const fileInputRef = useRef<HTMLInputElement>(null);
  const directoryInputRef = useRef<HTMLInputElement>(null);

  // Browse mode state
  const [companies, setCompanies] = useState<CompanyEntry[]>([]);
  const [searchQuery, setSearchQuery] = useState("");
  const [selectedCompany, setSelectedCompany] = useState<CompanyEntry | null>(null);
  const [isLoadingCompanies, setIsLoadingCompanies] = useState(false);
  const [companiesError, setCompaniesError] = useState<string | null>(null);

  // Track whether we've attempted to fetch to prevent infinite retry loops
  const fetchAttemptedRef = useRef(false);

  // Load companies when browse mode is selected
  useEffect(() => {
    if (inputMethod === "browse" && !fetchAttemptedRef.current && !isLoadingCompanies) {
      fetchAttemptedRef.current = true;
      setIsLoadingCompanies(true);
      setCompaniesError(null);
      fetchCompanies()
        .then((data) => {
          if (data.error) {
            setCompaniesError(data.error);
          } else if (data.companies.length > 0) {
            setCompanies(data.companies);
          } else {
            setCompaniesError("No companies available");
          }
        })
        .catch((err) => {
          setCompaniesError(err instanceof Error ? err.message : "Failed to load companies");
        })
        .finally(() => {
          setIsLoadingCompanies(false);
        });
    }
  }, [inputMethod, isLoadingCompanies]);

  /** Retry fetching companies after an error - calls fetch directly to bypass useEffect */
  const handleRetryFetchCompanies = useCallback(() => {
    fetchAttemptedRef.current = true; // Prevent useEffect from also firing
    setCompaniesError(null);
    setCompanies([]);
    setSelectedCompany(null);
    setIsLoadingCompanies(true);
    fetchCompanies()
      .then((data) => {
        if (data.error) {
          setCompaniesError(data.error);
        } else if (data.companies.length > 0) {
          setCompanies(data.companies);
        } else {
          setCompaniesError("No companies available");
        }
      })
      .catch((err) => {
        setCompaniesError(err instanceof Error ? err.message : "Failed to load companies");
      })
      .finally(() => {
        setIsLoadingCompanies(false);
      });
  }, []);

  const reset = useCallback(() => {
    setStep("input");
    setInputMethod("paste");
    setManifestContent("");
    setDirectoryAgents([]);
    setCompanyName("Unknown");
    setAgents([]);
    setSkills([]);
    setSelectedAgentNames([]);
    setSelectedSkillNames([]);
    setIsParsing(false);
    setIsImporting(false);
    setParseError(null);
    setImportResult(null);
    setImportError(null);
    setCompanies([]);
    setSearchQuery("");
    setSelectedCompany(null);
    setIsLoadingCompanies(false);
    setCompaniesError(null);
    fetchAttemptedRef.current = false;
  }, []);

  const handleClose = useCallback(() => {
    reset();
    onClose();
  }, [reset, onClose]);

  const handleFileChange = useCallback((e: React.ChangeEvent<HTMLInputElement>) => {
    const file = e.target.files?.[0];
    if (!file) return;

    const reader = new FileReader();
    reader.onload = (ev) => {
      const content = ev.target?.result as string;
      setInputMethod("file");
      setDirectoryAgents([]);
      setManifestContent(content);
      setParseError(null);
    };
    reader.onerror = () => {
      setParseError("Failed to read file");
    };
    reader.readAsText(file);

    // Reset file input so the same file can be re-selected
    e.target.value = "";
  }, []);

  const handleDirectoryChange = useCallback(async (e: React.ChangeEvent<HTMLInputElement>) => {
    const files = Array.from(e.target.files ?? []);
    if (files.length === 0) return;

    try {
      const agentFiles = files
        .filter((file) => (file.webkitRelativePath || file.name).toLowerCase().endsWith("agents.md"))
        .sort((a, b) => {
          const aPath = a.webkitRelativePath || a.name;
          const bPath = b.webkitRelativePath || b.name;
          return aPath.localeCompare(bPath);
        });

      if (agentFiles.length === 0) {
        setParseError("Selected directory has no AGENTS.md files");
        return;
      }

      const parsedAgents: DirectoryAgentInput[] = [];
      for (const file of agentFiles) {
        const content = await file.text();
        parsedAgents.push(parseDirectoryAgentManifest(content));
      }

      setInputMethod("directory");
      setDirectoryAgents(parsedAgents);
      setManifestContent("");
      setParseError(null);
    } catch {
      setParseError("Failed to parse AGENTS.md files from selected directory");
    } finally {
      e.target.value = "";
    }
  }, []);

  /** Build the API URL with optional projectId */
  function buildUrl(path: string): string {
    if (!projectId) return `/api${path}`;
    const separator = path.includes("?") ? "&" : "?";
    return `/api${path}${separator}projectId=${encodeURIComponent(projectId)}`;
  }

  /** Parse the manifest content by calling the API with dryRun=true */
  const handleParse = useCallback(async () => {
    if (inputMethod === "directory" && directoryAgents.length === 0) {
      setParseError("Please select a directory containing AGENTS.md files");
      return;
    }
    if (inputMethod === "browse" && !selectedCompany) {
      setParseError("Please select a company from the catalog");
      return;
    }
    if (inputMethod !== "directory" && inputMethod !== "browse" && !manifestContent.trim()) {
      setParseError("Please provide manifest content");
      return;
    }

    setIsParsing(true);
    setParseError(null);

    try {
      let body: Record<string, unknown>;

      if (inputMethod === "directory") {
        body = { agents: directoryAgents, dryRun: true };
      } else if (inputMethod === "browse" && selectedCompany) {
        body = { importSource: "companies.sh", companySlug: selectedCompany.slug, dryRun: true };
      } else {
        body = { manifest: manifestContent, dryRun: true };
      }

      const res = await fetch(buildUrl("/agents/import"), {
        method: "POST",
        headers: { "Content-Type": "application/json" },
        body: JSON.stringify(body),
      });

      if (!res.ok) {
        const data = await res.json() as ApiErrorResponse;
        throw new Error(data.error ?? `Parse failed (${res.status})`);
      }

      const data = await res.json() as {
        companyName?: string;
        agents?: AgentPreview[];
        skills?: SkillPreview[];
        created: string[];
        skipped: string[];
        errors: Array<{ name: string; error: string }>;
      };

      const previewAgents = (data.agents && data.agents.length > 0)
        ? data.agents
        : data.created.map((name) => ({ name, role: "custom" }));
      const previewSkills = Array.isArray(data.skills) ? data.skills : [];

      setCompanyName(data.companyName ?? "Unknown");
      setAgents(previewAgents);
      setSkills(previewSkills);
      setSelectedAgentNames(previewAgents.map((agent) => agent.name));
      setSelectedSkillNames(previewSkills.map((skill) => skill.name));
      setStep("preview");
    } catch (err) {
      setParseError(err instanceof Error ? err.message : "Failed to parse manifest");
    } finally {
      setIsParsing(false);
    }
  }, [inputMethod, directoryAgents, manifestContent, selectedCompany, projectId]);

  /** Execute the actual import */
  const handleImport = useCallback(async () => {
    setIsImporting(true);
    setImportError(null);

    try {
      let body: Record<string, unknown>;

      if (inputMethod === "directory") {
        body = {
          agents: directoryAgents,
          skipExisting: true,
          selectedAgents: selectedAgentNames,
          selectedSkills: selectedSkillNames,
        };
      } else if (inputMethod === "browse" && selectedCompany) {
        body = {
          importSource: "companies.sh",
          companySlug: selectedCompany.slug,
          skipExisting: true,
          selectedAgents: selectedAgentNames,
          selectedSkills: selectedSkillNames,
        };
      } else {
        body = {
          manifest: manifestContent,
          skipExisting: true,
          selectedAgents: selectedAgentNames,
          selectedSkills: selectedSkillNames,
        };
      }

      const res = await fetch(buildUrl("/agents/import"), {
        method: "POST",
        headers: { "Content-Type": "application/json" },
        body: JSON.stringify(body),
      });

      if (!res.ok) {
        const data = await res.json() as ApiErrorResponse;
        throw new Error(data.error ?? `Import failed (${res.status})`);
      }

      const data = await res.json() as ImportResult;
      setImportResult(data);
      setStep("result");
      onImported();
    } catch (err) {
      setImportError(err instanceof Error ? err.message : "Failed to import agents");
    } finally {
      setIsImporting(false);
    }
  }, [
    inputMethod,
    directoryAgents,
    manifestContent,
    selectedCompany,
    selectedAgentNames,
    selectedSkillNames,
    projectId,
    onImported,
  ]);

  const selectedAgentCount = selectedAgentNames.length;
  const selectedSkillCount = selectedSkillNames.length;
  const selectedAgentLabel = `${selectedAgentCount} Agent${selectedAgentCount !== 1 ? "s" : ""}`;
  const selectedSkillLabel = `${selectedSkillCount} Skill${selectedSkillCount !== 1 ? "s" : ""}`;
  const importActionLabel = selectedAgentCount > 0 && selectedSkillCount > 0
    ? `${selectedAgentLabel} + ${selectedSkillLabel}`
    : selectedSkillCount > 0
      ? selectedSkillLabel
      : selectedAgentLabel;
  const importLoadingLabel = selectedAgentCount > 0 && selectedSkillCount > 0
    ? `Importing ${selectedAgentCount} agent${selectedAgentCount !== 1 ? "s" : ""} and ${selectedSkillCount} skill${selectedSkillCount !== 1 ? "s" : ""}...`
    : selectedSkillCount > 0
      ? `Importing ${selectedSkillCount} skill${selectedSkillCount !== 1 ? "s" : ""}...`
      : `Importing ${selectedAgentCount} agent${selectedAgentCount !== 1 ? "s" : ""}...`;

  const toggleAgentSelection = (name: string) => {
    setSelectedAgentNames((current) => (
      current.includes(name)
        ? current.filter((selectedName) => selectedName !== name)
        : [...current, name]
    ));
  };

  const toggleSkillSelection = (name: string) => {
    setSelectedSkillNames((current) => (
      current.includes(name)
        ? current.filter((selectedName) => selectedName !== name)
        : [...current, name]
    ));
  };

  if (!isOpen) return null;

  return (
    <div className="agent-dialog-overlay" onClick={(e) => { if (e.target === e.currentTarget) handleClose(); }}>
      <div className="agent-dialog agent-import-dialog" role="dialog" aria-modal="true" aria-label="Import agents">
        {/* Header */}
        <div className="agent-dialog-header">
          <span className="agent-dialog-header-title">Import Agents</span>
          <button className="modal-close" onClick={handleClose} aria-label="Close">
            &times;
          </button>
        </div>

        {/* Body */}
        <div className="agent-dialog-body">
          {/* Step 1: Input */}
          {step === "input" && (
            <div className="agent-import-input">
              <p className="agent-import-description">
                Import agents from an Agent Companies package. Browse the companies.sh catalog to discover published agents, upload an AGENTS.md file, select a directory, or paste manifest content.
              </p>

              {/* File upload */}
              <div className="agent-import-file-upload">
                <input
                  ref={fileInputRef}
                  type="file"
                  accept=".md,.txt"
                  onChange={handleFileChange}
                  className="agent-import-file-input"
                  aria-label="Upload agent manifest file"
                />
                <input
                  ref={directoryInputRef}
                  type="file"
                  // @ts-expect-error webkitdirectory is non-standard but supported by Chromium browsers
                  webkitdirectory=""
                  multiple
                  onChange={handleDirectoryChange}
                  className="agent-import-file-input"
                  aria-label="Select directory"
                />
                <button
                  type="button"
                  className="btn agent-import-upload-btn"
                  onClick={() => fileInputRef.current?.click()}
                >
                  <Upload size={16} />
                  Choose File
                </button>
                <button
                  type="button"
                  className="btn agent-import-upload-btn"
                  onClick={() => directoryInputRef.current?.click()}
                >
                  <FolderOpen size={16} />
                  Select Directory
                </button>
                <button
                  type="button"
                  className="btn agent-import-upload-btn"
                  onClick={() => {
                    setInputMethod("browse");
                    setDirectoryAgents([]);
                    setManifestContent("");
                    setSelectedCompany(null);
                    setParseError(null);
                  }}
                >
                  <Globe size={16} />
                  Browse Catalog
                </button>
                <span className="agent-import-file-hint">.md and .txt files supported</span>
              </div>

              {/* Browse Catalog Mode */}
              {inputMethod === "browse" && (
                <div className="agent-import-browse">
                  <div className="agent-import-browse-header">
                    <div className="agent-import-browse-search">
                      <Search size={16} className="agent-import-browse-search-icon" />
                      <input
                        type="text"
                        className="agent-import-browse-search-input"
                        placeholder="Search companies..."
                        value={searchQuery}
                        onChange={(e) => setSearchQuery(e.target.value)}
                        aria-label="Search companies"
                      />
                    </div>
                    {selectedCompany && (
                      <div className="agent-import-browse-selected">
                        <span className="agent-import-browse-selected-label">Selected:</span>
                        <span className="agent-import-browse-selected-name">{selectedCompany.name}</span>
                        <button
                          type="button"
                          className="btn btn-sm"
                          onClick={() => setSelectedCompany(null)}
                        >
                          Change
                        </button>
                      </div>
                    )}
                  </div>

                  {isLoadingCompanies && (
                    <div className="agent-import-browse-loading">
                      <Loader2 size={20} className="spin" />
                      <span>Loading companies...</span>
                    </div>
                  )}

                  {companiesError && (
                    <div className="agent-import-browse-error">
                      <AlertTriangle size={16} />
                      <span>{companiesError}</span>
                      <button
                        type="button"
                        className="btn btn-sm"
                        onClick={handleRetryFetchCompanies}
                      >
                        <RefreshCw size={14} />
                        Retry
                      </button>
                    </div>
                  )}

                  {!isLoadingCompanies && !companiesError && (
                    <div className="agent-import-browse-list">
                      {companies
                        .filter((company) =>
                          searchQuery === "" ||
                          company.name.toLowerCase().includes(searchQuery.toLowerCase()) ||
                          (company.tagline?.toLowerCase().includes(searchQuery.toLowerCase()) ?? false)
                        )
                        .map((company) => (
                          <div
                            key={company.slug}
                            className={`agent-import-browse-item ${selectedCompany?.slug === company.slug ? "agent-import-browse-item--selected" : ""}`}
                            onClick={() => setSelectedCompany(company)}
                            role="button"
                            tabIndex={0}
                            onKeyDown={(e) => {
                              if (e.key === "Enter" || e.key === " ") {
                                setSelectedCompany(company);
                              }
                            }}
                          >
                            <div className="agent-import-browse-item-header">
                              <span className="agent-import-browse-item-name">{company.name}</span>
                              {company.installs !== undefined && (
                                <span className="agent-import-browse-item-installs">{company.installs.toLocaleString()} installs</span>
                              )}
                            </div>
                            {company.tagline && (
                              <span className="agent-import-browse-item-tagline">{company.tagline}</span>
                            )}
                            {company.repo && (
                              <span className="agent-import-browse-item-repo">{company.repo}</span>
                            )}
                          </div>
                        ))}
                      {companies.filter((company) =>
                        searchQuery === "" ||
                        company.name.toLowerCase().includes(searchQuery.toLowerCase()) ||
                        (company.tagline?.toLowerCase().includes(searchQuery.toLowerCase()) ?? false)
                      ).length === 0 && (
                        <p className="agent-import-browse-empty">
                          {searchQuery ? "No companies match your search" : "No companies available"}
                        </p>
                      )}
                    </div>
                  )}
                </div>
              )}

              {/* Or divider - only show when not in browse mode */}
              {inputMethod !== "browse" && (
                <>
                  <div className="agent-import-divider">
                    <span>or paste manifest content</span>
                  </div>

                  {/* Text area for paste */}
                  <textarea
                    className="agent-import-textarea"
                    placeholder={"---\nname: CEO\ntitle: Chief Executive Officer\nreportsTo: null\nskills:\n  - review\n---\nAgent instructions go here..."}
                    value={manifestContent}
                    onChange={(e) => {
                      setInputMethod("paste");
                      setDirectoryAgents([]);
                      setManifestContent(e.target.value);
                      setParseError(null);
                    }}
                    rows={8}
                    aria-label="Manifest content"
                  />
                </>
              )}

              <p className="agent-import-file-hint">Current input: {inputMethod}</p>

              {parseError && (
                <p className="agent-dialog-error">
                  <AlertTriangle size={14} />
                  {parseError}
                </p>
              )}
            </div>
          )}

          {/* Step 2: Preview */}
          {step === "preview" && (
            <div className="agent-import-preview">
              <div className="agent-import-company">
                <span className="agent-import-company-label">Company</span>
                <span className="agent-import-company-name">{companyName}</span>
              </div>

              <div className="agent-import-count">
                <FileText size={14} />
                <span>{agents.length} agent{agents.length !== 1 ? "s" : ""} found</span>
              </div>

              {agents.length > 0 && (
                <div className="agent-import-selection-controls">
                  <button
                    type="button"
                    className="btn btn-sm"
                    onClick={() => setSelectedAgentNames(agents.map((agent) => agent.name))}
                  >
                    Select all agents
                  </button>
                  <button
                    type="button"
                    className="btn btn-sm"
                    onClick={() => setSelectedAgentNames([])}
                  >
                    Clear agents
                  </button>
                </div>
              )}

              {agents.length > 0 ? (
                <div className="agent-import-agent-list">
                  {agents.map((agent, idx) => (
                    <div key={idx} className="agent-import-agent-item">
                      <label className="checkbox-label">
                        <input
                          type="checkbox"
                          aria-label={`Select agent ${agent.name}`}
                          checked={selectedAgentNames.includes(agent.name)}
                          onChange={() => toggleAgentSelection(agent.name)}
                        />
                      </label>
                      <span className="agent-import-agent-icon">{agent.icon || "🤖"}</span>
                      <div className="agent-import-agent-details">
                        <span className="agent-import-agent-name">{agent.name}</span>
                        <span className="agent-import-agent-meta">
                          {agent.title && <span className="agent-import-agent-title">{agent.title} · </span>}
                          <span className="agent-import-agent-role">{agent.role}</span>
                          {agent.reportsTo && (
                            <span className="agent-import-agent-reports"> · reports to {agent.reportsTo}</span>
                          )}
                          {agent.skills && agent.skills.length > 0 && (
                            <span className="agent-import-agent-model"> · skills: {agent.skills.join(", ")}</span>
                          )}
                        </span>
                        {agent.instructionsText && (
                          <span className="agent-import-agent-instructions">
                            {agent.instructionsText.slice(0, 100)}{agent.instructionsText.length > 100 ? "..." : ""}
                          </span>
                        )}
                      </div>
                    </div>
                  ))}
                </div>
              ) : (
                <p className="agent-import-empty">No agents found in the manifest.</p>
              )}

              {skills.length > 0 && (
                <div className="agent-import-skills-section">
                  <div className="agent-import-count">
                    <FileText size={14} />
                    <span>{skills.length} skill{skills.length !== 1 ? "s" : ""} found</span>
                  </div>
                  <div className="agent-import-selection-controls">
                    <button
                      type="button"
                      className="btn btn-sm"
                      onClick={() => setSelectedSkillNames(skills.map((skill) => skill.name))}
                    >
                      Select all skills
                    </button>
                    <button
                      type="button"
                      className="btn btn-sm"
                      onClick={() => setSelectedSkillNames([])}
                    >
                      Clear skills
                    </button>
                  </div>
                  <div className="agent-import-skill-list">
                    {skills.map((skill, idx) => (
                      <div key={`${skill.name}-${idx}`} className="agent-import-skill-item">
                        <label className="checkbox-label">
                          <input
                            type="checkbox"
                            aria-label={`Select skill ${skill.name}`}
                            checked={selectedSkillNames.includes(skill.name)}
                            onChange={() => toggleSkillSelection(skill.name)}
                          />
                        </label>
                        <span className="agent-import-skill-icon">⚡</span>
                        <div className="agent-import-skill-details">
                          <span className="agent-import-skill-name">{skill.name}</span>
                          {skill.description && (
                            <span className="agent-import-skill-description">{skill.description}</span>
                          )}
                        </div>
                      </div>
                    ))}
                  </div>
                </div>
              )}

              {importError && (
                <p className="agent-dialog-error">
                  <AlertTriangle size={14} />
                  {importError}
                </p>
              )}
            </div>
          )}

          {/* Step 3: Result */}
          {step === "result" && importResult && (
            <div className="agent-import-result">
              <div className="agent-import-result-icon">
                <CheckCircle size={32} />
              </div>
              <h3 className="agent-import-result-title">Import Complete</h3>
              <p className="agent-import-result-company">
                From <strong>{importResult.companyName ?? "Unknown"}</strong>
              </p>

              <div className="agent-import-result-stats">
                {importResult.created.length > 0 && (
                  <div className="agent-import-result-stat agent-import-result-stat--success">
                    <span>{importResult.created.length} created</span>
                  </div>
                )}
                {importResult.skipped.length > 0 && (
                  <div className="agent-import-result-stat agent-import-result-stat--skipped">
                    <span>{importResult.skipped.length} skipped (already exist)</span>
                  </div>
                )}
                {importResult.errors.length > 0 && (
                  <div className="agent-import-result-stat agent-import-result-stat--error">
                    <span>{importResult.errors.length} error{importResult.errors.length !== 1 ? "s" : ""}</span>
                  </div>
                )}
              </div>

              {importResult.created.length > 0 && (
                <div className="agent-import-result-agents">
                  {importResult.created.map((a, idx) => (
                    <div key={idx} className="agent-import-result-agent">
                      <CheckCircle size={12} />
                      <span>{a.name}</span>
                    </div>
                  ))}
                </div>
              )}

              {importResult.errors.length > 0 && (
                <div className="agent-import-result-errors">
                  {importResult.errors.map((err, idx) => (
                    <div key={idx} className="agent-import-result-error">
                      <X size={12} />
                      <span>{err.name}: {err.error}</span>
                    </div>
                  ))}
                </div>
              )}

              {importResult.skills && (
                <>
                  <div className="agent-import-result-divider" />
                  <h4 className="agent-import-result-section-title">Skills</h4>
                  <div className="agent-import-result-stats">
                    {importResult.skills.imported.length > 0 && (
                      <div className="agent-import-result-stat agent-import-result-stat--success">
                        <span>{importResult.skills.imported.length} skill{importResult.skills.imported.length !== 1 ? "s" : ""} imported</span>
                      </div>
                    )}
                    {importResult.skills.skipped.length > 0 && (
                      <div className="agent-import-result-stat agent-import-result-stat--skipped">
                        <span>{importResult.skills.skipped.length} skill{importResult.skills.skipped.length !== 1 ? "s" : ""} skipped (already exist)</span>
                      </div>
                    )}
                    {importResult.skills.errors.length > 0 && (
                      <div className="agent-import-result-stat agent-import-result-stat--error">
                        <span>{importResult.skills.errors.length} skill{importResult.skills.errors.length !== 1 ? "s" : ""} error{importResult.skills.errors.length !== 1 ? "s" : ""}</span>
                      </div>
                    )}
                    {importResult.skills.imported.length === 0 && importResult.skills.skipped.length === 0 && importResult.skills.errors.length === 0 && (
                      <div className="agent-import-result-stat agent-import-result-stat--skipped">
                        <span>No skills in package</span>
                      </div>
                    )}
                  </div>

                  {importResult.skills.imported.length > 0 && (
                    <div className="agent-import-result-agents">
                      {importResult.skills.imported.map((skill, idx) => (
                        <div key={idx} className="agent-import-result-agent">
                          <CheckCircle size={12} />
                          <span>{skill.name}</span>
                        </div>
                      ))}
                    </div>
                  )}

                  {importResult.skills.errors.length > 0 && (
                    <div className="agent-import-result-errors">
                      {importResult.skills.errors.map((err, idx) => (
                        <div key={idx} className="agent-import-result-error">
                          <X size={12} />
                          <span>{err.name}: {err.error}</span>
                        </div>
                      ))}
                    </div>
                  )}
                </>
              )}
            </div>
          )}
        </div>

        {/* Footer */}
        <div className="agent-dialog-footer">
          {step === "preview" && (
            <button className="btn" onClick={() => setStep("input")} disabled={isImporting}>
              Back
            </button>
          )}
          <button className="btn" onClick={handleClose} disabled={isImporting}>
            {step === "result" ? "Close" : "Cancel"}
          </button>
          {step === "input" && (
            <button
              className="btn btn-task-create"
              onClick={() => void handleParse()}
              disabled={
                isParsing || (
                  inputMethod === "directory"
                    ? directoryAgents.length === 0
                    : inputMethod === "browse"
                      ? !selectedCompany
                      : !manifestContent.trim()
                )
              }
            >
              {isParsing ? (
                <>
                  <Loader2 size={14} className="spin" />
                  Parsing...
                </>
              ) : (
                "Preview"
              )}
            </button>
          )}
          {step === "preview" && (
            <button
              className="btn btn-task-create"
              onClick={() => void handleImport()}
              disabled={isImporting || (selectedAgentCount === 0 && selectedSkillCount === 0)}
            >
              {isImporting ? (
                <>
                  <Loader2 size={14} className="spin" />
                  {importLoadingLabel}
                </>
              ) : (
                `Import ${importActionLabel}`
              )}
            </button>
          )}
        </div>
      </div>
    </div>
  );
}