import { scopeProjectRef, scopeThreadRef } from "@t3tools/client-runtime/environment"; import { isAtomCommandInterrupted, squashAtomCommandFailure, } from "@t3tools/client-runtime/state/runtime"; import type { EnvironmentId, VcsRef, ThreadId } from "@t3tools/contracts"; import { LegendList, type LegendListRef } from "@legendapp/list/react"; import { ChevronDownIcon, GitBranchIcon, RefreshCwIcon, SearchIcon } from "lucide-react"; import { useCallback, useDeferredValue, useEffect, useId, useLayoutEffect, useMemo, useOptimistic, useRef, useState, useTransition, } from "react"; import { useComposerDraftStore, type DraftId } from "../composerDraftStore"; import { useOpenPrLink } from "../lib/openPullRequestLink"; import { usePaginatedBranches } from "../state/queries"; import { useProject, useThread } from "../state/entities"; import { useEnvironmentQuery } from "../state/query"; import { threadEnvironment } from "../state/threads"; import { useAtomCommand } from "../state/use-atom-command"; import { vcsEnvironment } from "../state/vcs"; import { cn } from "../lib/utils"; import { parsePullRequestReference } from "../pullRequestReference"; import { getSourceControlPresentation } from "../sourceControlPresentation"; import { deriveLocalBranchNameFromRemoteRef, resolveBranchSelectionTarget, resolveBranchToolbarValue, resolveDraftEnvModeAfterBranchChange, resolveEffectiveEnvMode, shouldIncludeBranchPickerItem, } from "./BranchToolbar.logic"; import { ChangeRequestStatusIcon, prStatusIndicator, resolveThreadPr, } from "./ThreadStatusIndicators"; import { Button } from "./ui/button"; import { Switch } from "./ui/switch"; import { Combobox, ComboboxEmpty, ComboboxInput, ComboboxItem, ComboboxListVirtualized, ComboboxPopup, ComboboxStatus, ComboboxTrigger, } from "./ui/combobox"; import { stackedThreadToast, toastManager } from "./ui/toast"; import { Tooltip, TooltipPopup, TooltipTrigger } from "./ui/tooltip"; interface BranchToolbarBranchSelectorProps { className?: string; environmentId: EnvironmentId; threadId: ThreadId; draftId?: DraftId; envLocked: boolean; effectiveEnvModeOverride?: "local" | "worktree"; activeThreadBranchOverride?: string | null; onActiveThreadBranchOverrideChange?: (refName: string | null) => void; startFromOrigin: boolean; onStartFromOriginChange: (startFromOrigin: boolean) => void; onCheckoutPullRequestRequest?: (reference: string) => void; onComposerFocusRequest?: () => void; } function toBranchActionErrorMessage(error: unknown): string { return error instanceof Error ? error.message : "An error occurred."; } function getBranchTriggerLabel(input: { activeWorktreePath: string | null; effectiveEnvMode: "local" | "worktree"; resolvedActiveBranch: string | null; }): string { const { activeWorktreePath, effectiveEnvMode, resolvedActiveBranch } = input; if (!resolvedActiveBranch) { return "Select ref"; } if (effectiveEnvMode === "worktree" && !activeWorktreePath) { return `From ${resolvedActiveBranch}`; } return resolvedActiveBranch; } export function BranchToolbarBranchSelector({ className, environmentId, threadId, draftId, envLocked, effectiveEnvModeOverride, activeThreadBranchOverride, onActiveThreadBranchOverrideChange, startFromOrigin, onStartFromOriginChange, onCheckoutPullRequestRequest, onComposerFocusRequest, }: BranchToolbarBranchSelectorProps) { const startFromOriginSwitchId = useId(); const stopThreadSession = useAtomCommand(threadEnvironment.stopSession, "thread session stop"); const updateThreadMetadata = useAtomCommand( threadEnvironment.updateMetadata, "thread metadata update", ); const switchRef = useAtomCommand(vcsEnvironment.switchRef, { reportFailure: false, }); const createRefMutation = useAtomCommand(vcsEnvironment.createRef, { reportFailure: false, }); // --------------------------------------------------------------------------- // Thread / project state (pushed down from parent to colocate with mutation) // --------------------------------------------------------------------------- const threadRef = useMemo( () => scopeThreadRef(environmentId, threadId), [environmentId, threadId], ); const serverThread = useThread(threadRef); const serverSession = serverThread?.session ?? null; const draftThread = useComposerDraftStore((store) => draftId ? store.getDraftSession(draftId) : store.getDraftThreadByRef(threadRef), ); const setDraftThreadContext = useComposerDraftStore((store) => store.setDraftThreadContext); const activeProjectRef = serverThread ? scopeProjectRef(serverThread.environmentId, serverThread.projectId) : draftThread ? scopeProjectRef(draftThread.environmentId, draftThread.projectId) : null; const activeProject = useProject(activeProjectRef); const activeThreadId = serverThread?.id ?? (draftThread ? threadId : undefined); const activeThreadBranch = activeThreadBranchOverride !== undefined ? activeThreadBranchOverride : (serverThread?.branch ?? draftThread?.branch ?? null); const activeWorktreePath = serverThread?.worktreePath ?? draftThread?.worktreePath ?? null; const activeProjectCwd = activeProject?.workspaceRoot ?? null; const branchCwd = activeWorktreePath ?? activeProjectCwd; const hasServerThread = serverThread !== null; const effectiveEnvMode = effectiveEnvModeOverride ?? resolveEffectiveEnvMode({ activeWorktreePath, hasServerThread, draftThreadEnvMode: draftThread?.envMode, }); // --------------------------------------------------------------------------- // Thread branch mutation (colocated — only this component calls it) // --------------------------------------------------------------------------- const setThreadBranch = useCallback( (branch: string | null, worktreePath: string | null) => { if (!activeThreadId || !activeProject) return; if (serverSession && worktreePath !== activeWorktreePath) { void stopThreadSession({ environmentId, input: { threadId: activeThreadId }, }); } if (hasServerThread) { void updateThreadMetadata({ environmentId, input: { threadId: activeThreadId, branch, worktreePath, }, }); } if (hasServerThread) { onActiveThreadBranchOverrideChange?.(branch); return; } const nextDraftEnvMode = resolveDraftEnvModeAfterBranchChange({ nextWorktreePath: worktreePath, currentWorktreePath: activeWorktreePath, effectiveEnvMode, }); setDraftThreadContext(draftId ?? threadRef, { branch, worktreePath, envMode: nextDraftEnvMode, projectRef: scopeProjectRef(environmentId, activeProject.id), }); }, [ activeThreadId, activeProject, serverSession, activeWorktreePath, hasServerThread, onActiveThreadBranchOverrideChange, setDraftThreadContext, draftId, threadRef, environmentId, effectiveEnvMode, stopThreadSession, updateThreadMetadata, ], ); // --------------------------------------------------------------------------- // Git ref queries // --------------------------------------------------------------------------- const [isBranchMenuOpen, setIsBranchMenuOpen] = useState(false); const [branchQuery, setBranchQuery] = useState(""); const deferredBranchQuery = useDeferredValue(branchQuery); const branchStatusQuery = useEnvironmentQuery( branchCwd === null ? null : vcsEnvironment.status({ environmentId, input: { cwd: branchCwd }, }), ); const trimmedBranchQuery = branchQuery.trim(); const deferredTrimmedBranchQuery = deferredBranchQuery.trim(); const branchRefTarget = useMemo( () => ({ environmentId, cwd: branchCwd, query: deferredTrimmedBranchQuery, }), [branchCwd, deferredTrimmedBranchQuery, environmentId], ); const branchRefState = usePaginatedBranches(branchRefTarget); const refs = branchRefState.refs; const hasNextPage = branchRefState.data?.nextCursor !== null && branchRefState.data?.nextCursor !== undefined; const isFetchingNextPage = branchRefState.isPending && branchRefState.data !== null; const isInitialBranchesLoadPending = branchRefState.isPending && branchRefState.data === null; const currentGitBranch = branchStatusQuery.data?.refName ?? refs.find((refName) => refName.current)?.name ?? null; const sourceControlPresentation = useMemo( () => getSourceControlPresentation(branchStatusQuery.data?.sourceControlProvider), [branchStatusQuery.data?.sourceControlProvider], ); const SourceControlIcon = sourceControlPresentation.Icon; const canonicalActiveBranch = resolveBranchToolbarValue({ envMode: effectiveEnvMode, activeWorktreePath, activeThreadBranch, currentGitBranch, }); const branchNames = useMemo(() => refs.map((refName) => refName.name), [refs]); const branchByName = useMemo( () => new Map(refs.map((refName) => [refName.name, refName] as const)), [refs], ); const normalizedDeferredBranchQuery = deferredTrimmedBranchQuery.toLowerCase(); const prReference = parsePullRequestReference(trimmedBranchQuery); const isSelectingWorktreeBase = effectiveEnvMode === "worktree" && !envLocked && !activeWorktreePath; const checkoutPullRequestItemValue = prReference && onCheckoutPullRequestRequest ? `__checkout_pull_request__:${prReference}` : null; const canCreateBranch = !isSelectingWorktreeBase && trimmedBranchQuery.length > 0; const hasExactBranchMatch = branchByName.has(trimmedBranchQuery); const createBranchItemValue = canCreateBranch ? `__create_new_branch__:${trimmedBranchQuery}` : null; const branchPickerItems = useMemo(() => { const items = [...branchNames]; if (createBranchItemValue && !hasExactBranchMatch) { items.push(createBranchItemValue); } if (checkoutPullRequestItemValue) { items.unshift(checkoutPullRequestItemValue); } return items; }, [branchNames, checkoutPullRequestItemValue, createBranchItemValue, hasExactBranchMatch]); const filteredBranchPickerItems = useMemo( () => normalizedDeferredBranchQuery.length === 0 ? branchPickerItems : branchPickerItems.filter((itemValue) => shouldIncludeBranchPickerItem({ itemValue, normalizedQuery: normalizedDeferredBranchQuery, createBranchItemValue, checkoutPullRequestItemValue, }), ), [ branchPickerItems, checkoutPullRequestItemValue, createBranchItemValue, normalizedDeferredBranchQuery, ], ); const [resolvedActiveBranch, setOptimisticBranch] = useOptimistic( canonicalActiveBranch, (_currentBranch: string | null, optimisticBranch: string | null) => optimisticBranch, ); const [isBranchActionPending, startBranchActionTransition] = useTransition(); const totalBranchCount = branchRefState.data?.totalCount ?? 0; const branchStatusText = isInitialBranchesLoadPending ? "Loading refs..." : isFetchingNextPage ? "Loading more refs..." : hasNextPage ? `Showing ${refs.length} of ${totalBranchCount} refs` : null; // --------------------------------------------------------------------------- // Branch actions // --------------------------------------------------------------------------- const runBranchAction = (action: () => Promise) => { startBranchActionTransition(async () => { await action(); branchRefState.refresh(); branchStatusQuery.refresh(); }); }; const selectBranch = (refName: VcsRef) => { if (!branchCwd || !activeProjectCwd || isBranchActionPending) return; if (isSelectingWorktreeBase) { setThreadBranch(refName.name, null); setIsBranchMenuOpen(false); onComposerFocusRequest?.(); return; } const selectionTarget = resolveBranchSelectionTarget({ activeProjectCwd, activeWorktreePath, refName, }); if (selectionTarget.reuseExistingWorktree) { setThreadBranch(refName.name, selectionTarget.nextWorktreePath); setIsBranchMenuOpen(false); onComposerFocusRequest?.(); return; } const selectedBranchName = refName.isRemote ? deriveLocalBranchNameFromRemoteRef(refName.name) : refName.name; setIsBranchMenuOpen(false); onComposerFocusRequest?.(); runBranchAction(async () => { const previousBranch = resolvedActiveBranch; setOptimisticBranch(selectedBranchName); const checkoutResult = await switchRef({ environmentId, input: { cwd: selectionTarget.checkoutCwd, refName: refName.name, }, }); if (checkoutResult._tag === "Success") { const nextBranchName = refName.isRemote ? (checkoutResult.value.refName ?? selectedBranchName) : selectedBranchName; setOptimisticBranch(nextBranchName); setThreadBranch(nextBranchName, selectionTarget.nextWorktreePath); return; } setOptimisticBranch(previousBranch); if (!isAtomCommandInterrupted(checkoutResult)) { toastManager.add( stackedThreadToast({ type: "error", title: "Failed to switch ref.", description: toBranchActionErrorMessage(squashAtomCommandFailure(checkoutResult)), }), ); } }); }; const createRef = (rawName: string) => { const name = rawName.trim(); if (!branchCwd || !name || isBranchActionPending) return; setIsBranchMenuOpen(false); onComposerFocusRequest?.(); runBranchAction(async () => { const previousBranch = resolvedActiveBranch; setOptimisticBranch(name); const createBranchResult = await createRefMutation({ environmentId, input: { cwd: branchCwd, refName: name, switchRef: true, }, }); if (createBranchResult._tag === "Success") { setOptimisticBranch(createBranchResult.value.refName); setThreadBranch(createBranchResult.value.refName, activeWorktreePath); return; } setOptimisticBranch(previousBranch); if (!isAtomCommandInterrupted(createBranchResult)) { toastManager.add( stackedThreadToast({ type: "error", title: "Failed to create and switch ref.", description: toBranchActionErrorMessage(squashAtomCommandFailure(createBranchResult)), }), ); } }); }; useEffect(() => { if ( effectiveEnvMode !== "worktree" || activeWorktreePath || activeThreadBranch || !currentGitBranch ) { return; } setThreadBranch(currentGitBranch, null); }, [activeThreadBranch, activeWorktreePath, currentGitBranch, effectiveEnvMode, setThreadBranch]); // --------------------------------------------------------------------------- // Combobox / list plumbing // --------------------------------------------------------------------------- const handleOpenChange = useCallback( (open: boolean) => { setIsBranchMenuOpen(open); if (!open) { setBranchQuery(""); return; } branchRefState.refresh(); }, [branchRefState.refresh], ); const branchListScrollElementRef = useRef(null); const [showTopBranchScrollFade, setShowTopBranchScrollFade] = useState(false); const [showBottomBranchScrollFade, setShowBottomBranchScrollFade] = useState(false); const fetchNextBranchPage = useCallback(() => { if (!hasNextPage || isFetchingNextPage) { return; } branchRefState.loadNext(); }, [branchRefState.loadNext, hasNextPage, isFetchingNextPage]); const maybeFetchNextBranchPage = useCallback(() => { if (!isBranchMenuOpen || !hasNextPage || isFetchingNextPage) { return; } const scrollElement = branchListScrollElementRef.current; if (!scrollElement) { return; } const distanceFromBottom = scrollElement.scrollHeight - scrollElement.scrollTop - scrollElement.clientHeight; if (distanceFromBottom > 96) { return; } fetchNextBranchPage(); }, [fetchNextBranchPage, hasNextPage, isBranchMenuOpen, isFetchingNextPage]); const branchListRef = useRef(null); const updateBranchListScrollFades = useCallback(() => { const scrollElement = branchListRef.current?.getScrollableNode?.(); if (!(scrollElement instanceof HTMLElement)) { return; } branchListScrollElementRef.current = scrollElement; const maxScrollOffset = Math.max(0, scrollElement.scrollHeight - scrollElement.clientHeight); setShowTopBranchScrollFade(scrollElement.scrollTop > 1); setShowBottomBranchScrollFade(maxScrollOffset - scrollElement.scrollTop > 1); }, []); useLayoutEffect(() => { if (!isBranchMenuOpen) { return; } setShowTopBranchScrollFade(false); setShowBottomBranchScrollFade(filteredBranchPickerItems.length > 8); let nestedFrame = 0; const frame = requestAnimationFrame(() => { updateBranchListScrollFades(); nestedFrame = requestAnimationFrame(updateBranchListScrollFades); }); return () => { cancelAnimationFrame(frame); cancelAnimationFrame(nestedFrame); }; }, [ deferredTrimmedBranchQuery, filteredBranchPickerItems.length, isBranchMenuOpen, updateBranchListScrollFades, ]); useEffect(() => { if (!isBranchMenuOpen) { return; } branchListRef.current?.scrollToOffset?.({ offset: 0, animated: false }); }, [deferredTrimmedBranchQuery, isBranchMenuOpen]); useEffect(() => { maybeFetchNextBranchPage(); }, [refs.length, maybeFetchNextBranchPage]); const triggerLabel = getBranchTriggerLabel({ activeWorktreePath, effectiveEnvMode, resolvedActiveBranch, }); // PR pill shown next to the branch selector when the active branch has one. const branchPr = resolveThreadPr(resolvedActiveBranch, branchStatusQuery.data ?? null); const branchPrStatus = prStatusIndicator(branchPr, branchStatusQuery.data?.sourceControlProvider); // Action-oriented tooltip (the pill opens the PR), distinct from the sidebar's // state-description tooltip. const branchPrTooltip = branchPr ? `Open ${sourceControlPresentation.terminology.singular} #${branchPr.number} (${branchPr.state}) in browser` : ""; const openPrLink = useOpenPrLink(); function renderPickerItem(itemValue: string, index: number) { if (checkoutPullRequestItemValue && itemValue === checkoutPullRequestItemValue) { return ( { if (!prReference || !onCheckoutPullRequestRequest) { return; } setIsBranchMenuOpen(false); setBranchQuery(""); onComposerFocusRequest?.(); onCheckoutPullRequestRequest(prReference); }} >
Checkout {sourceControlPresentation.terminology.singular} {prReference}
); } if (createBranchItemValue && itemValue === createBranchItemValue) { return ( createRef(trimmedBranchQuery)} > Create new ref "{trimmedBranchQuery}" ); } const refName = branchByName.get(itemValue); if (!refName) return null; const hasSecondaryWorktree = refName.worktreePath && activeProjectCwd && refName.worktreePath !== activeProjectCwd; const badge = refName.current ? "current" : hasSecondaryWorktree ? "worktree" : refName.isRemote ? "remote" : refName.isDefault ? "default" : null; return ( selectBranch(refName)} >
{itemValue} {badge && {badge}}
); } return ( { if (!isBranchMenuOpen || eventDetails.index < 0 || eventDetails.reason !== "keyboard") { return; } branchListRef.current?.scrollIndexIntoView?.({ index: eventDetails.index, animated: false, }); }} onOpenChange={handleOpenChange} open={isBranchMenuOpen} value={resolvedActiveBranch} >
{branchPr && branchPrStatus ? ( openPrLink(event, branchPrStatus.url)} className={cn( "inline-flex shrink-0 items-center gap-0.5 rounded px-1 py-0.5 text-[11px] font-medium tabular-nums transition-colors hover:bg-muted/60", branchPrStatus.colorClass, )} /> } > #{branchPr.number} {branchPrTooltip} ) : null} } className="min-w-0 text-muted-foreground/70 hover:text-foreground/80" disabled={isInitialBranchesLoadPending || isBranchActionPending} > {triggerLabel}
No refs found.
ref={branchListRef} data={filteredBranchPickerItems} keyExtractor={(item) => item} renderItem={({ item, index }) => renderPickerItem(item, index)} estimatedItemSize={28} drawDistance={336} onEndReached={() => { if (hasNextPage && !isFetchingNextPage) { fetchNextBranchPage(); } }} onLayout={() => { updateBranchListScrollFades(); maybeFetchNextBranchPage(); }} onScroll={() => { updateBranchListScrollFades(); maybeFetchNextBranchPage(); }} className={cn( "scrollbar-gutter-stable overflow-x-hidden overscroll-y-contain ps-1 pe-0 pt-2 pb-1 [--fade-size:1.5rem]", showTopBranchScrollFade && "mask-t-from-[calc(100%-var(--fade-size))]", showBottomBranchScrollFade && "mask-b-from-[calc(100%-var(--fade-size))]", )} style={{ maxHeight: "14rem" }} />
{isSelectingWorktreeBase ? ( onStartFromOriginChange(Boolean(checked))} /> } /> Creates the worktree from the latest matching branch on origin instead of your local branch. ) : null} {branchStatusText ? {branchStatusText} : null}
); }