import { useEffect, useMemo, useState } from "react"; import { useTranslation } from "react-i18next"; import { toast } from "sonner"; import type { CreatedWorktree, GitState } from "@/shared/types/git"; import { Button } from "@/shared/ui/button"; import { Checkbox } from "@/shared/ui/checkbox"; import { Dialog, DialogContent, DialogDescription, DialogFooter, DialogHeader, DialogTitle, } from "@/shared/ui/dialog"; import { Input } from "@/shared/ui/input"; import { Label } from "@/shared/ui/label"; import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue, } from "@/shared/ui/select"; import type { ActiveWorkspace } from "../../stores/chatSessionStore"; import { formatErrorMessage } from "./formatError"; import { shortenPath } from "./WorkingContextPicker"; const UNSET_SELECT_VALUE = "__unset__"; export type WorkspaceCreateMode = "branch" | "worktree"; interface WorkspaceCreateDialogProps { mode: WorkspaceCreateMode | null; gitState: GitState; currentPath: string; activeBranch: string | null; onClose: () => void; onContextChange: (context: ActiveWorkspace) => void; onCreateBranch: ( path: string, name: string, baseBranch: string, ) => Promise; onCreateWorktree: ( path: string, name: string, branch: string, createBranch: boolean, baseBranch?: string, ) => Promise; } function worktreePreviewPath(rootPath: string, name: string) { const normalizedPath = rootPath.replace(/\/+$/, ""); const pathSegments = normalizedPath.split("/"); const repoName = pathSegments[pathSegments.length - 1]; const parentPath = pathSegments.slice(0, -1).join("/"); return `${parentPath}/${repoName}-worktrees/${name}`; } export function WorkspaceCreateDialog({ mode, gitState, currentPath, activeBranch, onClose, onContextChange, onCreateBranch, onCreateWorktree, }: WorkspaceCreateDialogProps) { const { t } = useTranslation("chat"); const [branchName, setBranchName] = useState(""); const [baseBranch, setBaseBranch] = useState(""); const [worktreeName, setWorktreeName] = useState(""); const [useNewBranch, setUseNewBranch] = useState(true); const [existingBranch, setExistingBranch] = useState(""); const [branchNameManuallyEdited, setBranchNameManuallyEdited] = useState(false); const [error, setError] = useState(null); const [saving, setSaving] = useState(false); const occupiedBranches = useMemo( () => new Set( gitState.worktrees .map((worktree) => worktree.branch) .filter((branch): branch is string => Boolean(branch)), ), [gitState.worktrees], ); const availableExistingBranches = useMemo( () => gitState.localBranches.filter((branch) => !occupiedBranches.has(branch)), [gitState.localBranches, occupiedBranches], ); const defaultBaseBranch = activeBranch ?? gitState.currentBranch ?? gitState.localBranches[0] ?? ""; const previewRootPath = gitState.mainWorktreePath ?? currentPath; const trimmedBranchName = branchName.trim(); const trimmedWorktreeName = worktreeName.trim(); const previewPath = mode === "worktree" && trimmedWorktreeName ? shortenPath(worktreePreviewPath(previewRootPath, trimmedWorktreeName)) : null; const branchDialogValid = trimmedBranchName.length > 0 && baseBranch.length > 0 && !saving; const worktreeDialogValid = trimmedWorktreeName.length > 0 && (useNewBranch ? trimmedBranchName.length > 0 && baseBranch.length > 0 : existingBranch.length > 0) && !saving; const canSubmit = mode === "branch" ? branchDialogValid : mode === "worktree" ? worktreeDialogValid : false; useEffect(() => { if (!mode) { return; } const defaultUseNewBranch = true; setBranchName(""); setBaseBranch(defaultBaseBranch); setWorktreeName(""); setUseNewBranch(defaultUseNewBranch); setExistingBranch( defaultUseNewBranch ? "" : (availableExistingBranches[0] ?? ""), ); setBranchNameManuallyEdited(false); setError(null); setSaving(false); }, [availableExistingBranches, defaultBaseBranch, mode]); const handleSubmit = async (event: React.FormEvent) => { event.preventDefault(); if (!mode) { return; } setSaving(true); setError(null); try { if (mode === "branch") { await onCreateBranch(currentPath, trimmedBranchName, baseBranch); onContextChange({ path: currentPath, branch: trimmedBranchName }); toast.success( t("contextPanel.createDialog.branchSuccess", { branch: trimmedBranchName, }), ); onClose(); return; } const targetBranch = useNewBranch ? trimmedBranchName : existingBranch; const createdWorktree = await onCreateWorktree( currentPath, trimmedWorktreeName, targetBranch, useNewBranch, useNewBranch ? baseBranch : undefined, ); onContextChange({ path: createdWorktree.path, branch: createdWorktree.branch, }); toast.success( t("contextPanel.createDialog.worktreeSuccess", { worktree: trimmedWorktreeName, }), ); onClose(); } catch (submitError) { const fallback = mode === "branch" ? t("contextPanel.createDialog.branchError", { branch: trimmedBranchName, }) : t("contextPanel.createDialog.worktreeError", { worktree: trimmedWorktreeName, }); const message = formatErrorMessage(submitError, fallback); setError(message); toast.error(message); } finally { setSaving(false); } }; return ( ); }