import { Fragment, useCallback, useEffect, useMemo, useRef, useState, type Dispatch, type ReactNode, type SetStateAction, } from "react"; import { DndContext, DragOverlay, KeyboardSensor, PointerSensor, closestCenter, pointerWithin, useSensor, useSensors, type CollisionDetection, type DragEndEvent, type DragMoveEvent, type DragOverEvent, type DragStartEvent, } from "@dnd-kit/core"; import { arrayMove, sortableKeyboardCoordinates } from "@dnd-kit/sortable"; import { Platform, View, Text } from "react-native"; import { StyleSheet, useUnistyles } from "react-native-unistyles"; import { ResizeHandle } from "@/components/resize-handle"; import { shouldFocusPaneFromEventTarget } from "@/components/split-container-pane-focus"; import { computeTabDropPreview, type TabDropPreview, } from "@/components/split-container-tab-drop-preview"; import { SplitDropZone, resolveSplitDropPosition, type SplitDropZoneHover, } from "@/components/split-drop-zone"; import { deriveWorkspacePaneState, getWorkspacePaneDescriptors, } from "@/screens/workspace/workspace-pane-state"; import { WorkspacePaneContent, type WorkspacePaneContentModel, } from "@/screens/workspace/workspace-pane-content"; import { WorkspaceDesktopTabsRow, type WorkspaceDesktopTabRowItem, } from "@/screens/workspace/workspace-desktop-tabs-row"; import { WorkspaceTabPresentationResolver, WorkspaceTabIcon, } from "@/screens/workspace/workspace-tab-presentation"; import type { WorkspaceTabDescriptor } from "@/screens/workspace/workspace-tabs-types"; import { useWorkspaceLayoutStore, type SplitNode, type SplitPane, type WorkspaceLayout, } from "@/stores/workspace-layout-store"; import type { WorkspaceTab } from "@/stores/workspace-tabs-store"; interface SplitContainerProps { layout: WorkspaceLayout; workspaceKey: string; normalizedServerId: string; normalizedWorkspaceId: string; uiTabs: WorkspaceTab[]; hoveredCloseTabKey: string | null; setHoveredTabKey: Dispatch>; setHoveredCloseTabKey: Dispatch>; closingTabIds: Set; onNavigateTab: (tabId: string) => void; onCloseTab: (tabId: string) => Promise | void; onCopyResumeCommand: (agentId: string) => Promise | void; onCopyAgentId: (agentId: string) => Promise | void; onCloseTabsToLeft: (tabId: string, paneTabs: WorkspaceTabDescriptor[]) => Promise | void; onCloseTabsToRight: (tabId: string, paneTabs: WorkspaceTabDescriptor[]) => Promise | void; onCloseOtherTabs: (tabId: string, paneTabs: WorkspaceTabDescriptor[]) => Promise | void; onSelectNewTabOption: (selection: { optionId: "__new_tab_agent__" | "__new_tab_terminal__"; paneId?: string; }) => void; onNewTerminalTab: (input: { paneId?: string }) => void; newTabAgentOptionId?: "__new_tab_agent__" | "__new_tab_terminal__"; buildPaneContentModel: (input: { paneId: string; isPaneFocused: boolean; tab: WorkspaceTabDescriptor; }) => WorkspacePaneContentModel; onFocusPane: (paneId: string) => void; onSplitPane: (input: { tabId: string; targetPaneId: string; position: "left" | "right" | "top" | "bottom"; }) => void; onSplitPaneEmpty: (input: { targetPaneId: string; position: "left" | "right" | "top" | "bottom"; }) => void; onMoveTabToPane: (tabId: string, toPaneId: string) => void; onResizeSplit: (groupId: string, sizes: number[]) => void; onReorderTabsInPane: (paneId: string, tabIds: string[]) => void; renderPaneEmptyState?: () => ReactNode; } interface WorkspaceTabDragData { kind: "workspace-tab"; paneId: string; tabId: string; } interface SplitPaneDropData { kind: "split-pane-drop"; paneId: string; } interface SplitNodeViewProps extends Omit { node: SplitNode; uiTabs: WorkspaceTab[]; focusedPaneId: string; activeDragTabId: string | null; showDropZones: boolean; dropPreview: SplitDropZoneHover | null; tabDropPreview: TabDropPreview | null; } interface SplitPaneViewProps extends Omit< SplitNodeViewProps, | "node" | "workspaceKey" | "focusedPaneId" | "activeDragTabId" | "showDropZones" | "dropPreview" | "onMoveTabToPane" | "onResizeSplit" > { pane: SplitPane; uiTabs: WorkspaceTab[]; isFocused: boolean; activeDragTabId: string | null; showDropZones: boolean; dropPreview: SplitDropZoneHover | null; tabDropPreview: TabDropPreview | null; } const dropCollisionDetection: CollisionDetection = (args) => { const pointerHits = pointerWithin(args); const tabHits = pointerHits.filter( (entry) => entry.data?.droppableContainer.data.current?.kind === "workspace-tab", ); if (tabHits.length > 0) { return tabHits; } const paneHits = pointerHits.filter( (entry) => entry.data?.droppableContainer.data.current?.kind === "split-pane-drop", ); if (paneHits.length > 0) { return paneHits; } return closestCenter(args); }; export function SplitContainer({ layout, workspaceKey, normalizedServerId, normalizedWorkspaceId, uiTabs, hoveredCloseTabKey, setHoveredTabKey, setHoveredCloseTabKey, closingTabIds, onNavigateTab, onCloseTab, onCopyResumeCommand, onCopyAgentId, onCloseTabsToLeft, onCloseTabsToRight, onCloseOtherTabs, onSelectNewTabOption, onNewTerminalTab, newTabAgentOptionId = "__new_tab_agent__", buildPaneContentModel, onFocusPane, onSplitPane, onSplitPaneEmpty, onMoveTabToPane, onResizeSplit, onReorderTabsInPane, renderPaneEmptyState = () => null, }: SplitContainerProps) { const [activeDragTabId, setActiveDragTabId] = useState(null); const [dropPreview, setDropPreview] = useState(null); const [tabDropPreview, setTabDropPreview] = useState(null); const sensors = useSensors( useSensor(PointerSensor, { activationConstraint: { distance: 8, }, }), useSensor(KeyboardSensor, { coordinateGetter: sortableKeyboardCoordinates, }), ); const panesById = useMemo(() => collectPanesById(layout.root), [layout.root]); const handleDragStart = useCallback((event: DragStartEvent) => { const data = event.active.data.current as WorkspaceTabDragData | undefined; if (data?.kind !== "workspace-tab") { setActiveDragTabId(null); setDropPreview(null); setTabDropPreview(null); return; } setActiveDragTabId(data.tabId); }, []); const handleDragCancel = useCallback(() => { setActiveDragTabId(null); setDropPreview(null); setTabDropPreview(null); }, []); const updateDropPreview = useCallback( (event: Pick | Pick) => { const activeData = event.active.data.current as WorkspaceTabDragData | undefined; const overData = event.over?.data.current as | WorkspaceTabDragData | SplitPaneDropData | undefined; if (activeData?.kind !== "workspace-tab") { setDropPreview(null); setTabDropPreview(null); return; } const translatedRect = event.active.rect.current.translated; const overRect = event.over?.rect; if (!translatedRect || !overRect || overRect.width <= 0 || overRect.height <= 0) { setDropPreview(null); setTabDropPreview(null); return; } if (overData?.kind === "workspace-tab") { const targetPane = panesById.get(overData.paneId) ?? null; if (!targetPane) { setDropPreview(null); setTabDropPreview(null); return; } const targetTabs = getWorkspacePaneDescriptors({ pane: targetPane, tabs: uiTabs, }); setDropPreview(null); setTabDropPreview( computeTabDropPreview({ activePaneId: activeData.paneId, activeTabId: activeData.tabId, overPaneId: overData.paneId, overTabId: overData.tabId, targetTabs, activeRect: { left: translatedRect.left, width: translatedRect.width, }, overRect: { left: overRect.left, width: overRect.width, }, }), ); return; } setTabDropPreview(null); if (overData?.kind !== "split-pane-drop") { setDropPreview(null); return; } const centerX = translatedRect.left + translatedRect.width / 2; const centerY = translatedRect.top + translatedRect.height / 2; const relativeX = centerX - overRect.left; const relativeY = centerY - overRect.top; if ( Number.isNaN(relativeX) || Number.isNaN(relativeY) || relativeX < 0 || relativeX > overRect.width || relativeY < 0 || relativeY > overRect.height ) { setDropPreview(null); return; } setDropPreview({ paneId: overData.paneId, position: resolveSplitDropPosition({ width: overRect.width, height: overRect.height, x: relativeX, y: relativeY, }), }); }, [panesById, uiTabs], ); const handleDragEnd = useCallback( (event: DragEndEvent) => { const activeData = event.active.data.current as WorkspaceTabDragData | undefined; const overData = event.over?.data.current as | WorkspaceTabDragData | SplitPaneDropData | undefined; setActiveDragTabId(null); if (activeData?.kind !== "workspace-tab" || !event.over) { setDropPreview(null); setTabDropPreview(null); return; } if (overData?.kind === "workspace-tab") { const sourcePane = panesById.get(activeData.paneId) ?? null; const targetPane = panesById.get(overData.paneId) ?? null; if (!sourcePane || !targetPane) { setDropPreview(null); setTabDropPreview(null); return; } const sourceTabs = getWorkspacePaneDescriptors({ pane: sourcePane, tabs: uiTabs }); const targetTabs = getWorkspacePaneDescriptors({ pane: targetPane, tabs: uiTabs }); const sourceIndex = sourceTabs.findIndex((tab) => tab.tabId === activeData.tabId); const resolvedTabDropPreview = tabDropPreview?.paneId === overData.paneId ? tabDropPreview : null; if (sourceIndex < 0 || !resolvedTabDropPreview) { setDropPreview(null); setTabDropPreview(null); return; } if (activeData.paneId === overData.paneId) { if (sourceIndex !== resolvedTabDropPreview.insertionIndex) { const nextTabs = arrayMove( sourceTabs, sourceIndex, resolvedTabDropPreview.insertionIndex, ); onReorderTabsInPane( activeData.paneId, nextTabs.map((tab) => tab.tabId), ); } setDropPreview(null); setTabDropPreview(null); return; } const nextTargetTabIds = targetTabs.map((tab) => tab.tabId); nextTargetTabIds.splice(resolvedTabDropPreview.insertionIndex, 0, activeData.tabId); onMoveTabToPane(activeData.tabId, overData.paneId); onReorderTabsInPane(overData.paneId, nextTargetTabIds); setDropPreview(null); setTabDropPreview(null); return; } if (overData?.kind === "split-pane-drop" && dropPreview?.paneId === overData.paneId) { if (dropPreview.position === "center") { if (activeData.paneId !== overData.paneId) { onMoveTabToPane(activeData.tabId, overData.paneId); } setDropPreview(null); setTabDropPreview(null); return; } onSplitPane({ tabId: activeData.tabId, targetPaneId: overData.paneId, position: dropPreview.position, }); } setDropPreview(null); setTabDropPreview(null); }, [ dropPreview, onMoveTabToPane, onReorderTabsInPane, onSplitPane, panesById, tabDropPreview, uiTabs, ], ); return ( {activeDragTabId ? ( ) : null} ); } function DragOverlayTabChip({ tabId, uiTabs, normalizedServerId, normalizedWorkspaceId, }: { tabId: string; uiTabs: WorkspaceTab[]; normalizedServerId: string; normalizedWorkspaceId: string; }) { const tab = uiTabs.find((t) => t.tabId === tabId); if (!tab) { return null; } const descriptor: WorkspaceTabDescriptor = { key: tab.tabId, tabId: tab.tabId, kind: tab.target.kind, target: tab.target, }; return ( ); } function DragOverlayTabChipInner({ tab, normalizedServerId, normalizedWorkspaceId, }: { tab: WorkspaceTabDescriptor; normalizedServerId: string; normalizedWorkspaceId: string; }) { const { theme } = useUnistyles(); return ( {(presentation) => { const label = presentation.titleState === "loading" ? "Loading..." : presentation.label; return ( {label} ); }} ); } function SplitNodeView({ node, workspaceKey, uiTabs, focusedPaneId, normalizedServerId, normalizedWorkspaceId, hoveredCloseTabKey, setHoveredTabKey, setHoveredCloseTabKey, closingTabIds, onNavigateTab, onCloseTab, onCopyResumeCommand, onCopyAgentId, onCloseTabsToLeft, onCloseTabsToRight, onCloseOtherTabs, onSelectNewTabOption, onNewTerminalTab, newTabAgentOptionId, buildPaneContentModel, onFocusPane, onSplitPane, onSplitPaneEmpty, onMoveTabToPane, onResizeSplit, onReorderTabsInPane, renderPaneEmptyState, activeDragTabId, showDropZones, dropPreview, tabDropPreview, }: SplitNodeViewProps) { if (node.kind === "pane") { return ( ); } const groupSizes = useWorkspaceLayoutStore( (state) => state.splitSizesByWorkspace[workspaceKey]?.[node.group.id], ) ?? node.group.sizes; return ( {node.group.children.map((child, index) => ( {index < node.group.children.length - 1 ? ( ) : null} ))} ); } function SplitPaneView({ pane, uiTabs, isFocused, normalizedServerId, normalizedWorkspaceId, hoveredCloseTabKey, setHoveredTabKey, setHoveredCloseTabKey, closingTabIds, onNavigateTab, onCloseTab, onCopyResumeCommand, onCopyAgentId, onCloseTabsToLeft, onCloseTabsToRight, onCloseOtherTabs, onSelectNewTabOption, onNewTerminalTab, newTabAgentOptionId, buildPaneContentModel, onFocusPane, onSplitPane, onSplitPaneEmpty, onReorderTabsInPane, renderPaneEmptyState, activeDragTabId, showDropZones, dropPreview, tabDropPreview, }: SplitPaneViewProps) { const { theme } = useUnistyles(); const paneRef = useRef(null); const paneState = useMemo( () => deriveWorkspacePaneState({ pane, tabs: uiTabs, }), [pane, uiTabs], ); const paneTabs = useMemo(() => paneState.tabs.map((tab) => tab.descriptor), [paneState.tabs]); const activeTabDescriptor = paneState.activeTab?.descriptor ?? null; const desktopTabRowItems = useMemo( () => paneTabs.map((tab) => ({ tab, isActive: tab.key === activeTabDescriptor?.key, isCloseHovered: hoveredCloseTabKey === tab.key, isClosingTab: closingTabIds.has(tab.tabId), })), [activeTabDescriptor?.key, closingTabIds, hoveredCloseTabKey, paneTabs], ); const paneContent = useMemo( () => activeTabDescriptor ? buildPaneContentModel({ paneId: pane.id, isPaneFocused: isFocused, tab: activeTabDescriptor, }) : null, [activeTabDescriptor, buildPaneContentModel, pane.id], ); useEffect(() => { if (Platform.OS !== "web") { return; } const paneElement = paneRef.current as unknown as HTMLElement | null; if ( !paneElement || typeof paneElement.addEventListener !== "function" || typeof paneElement.removeEventListener !== "function" ) { return; } const handlePanePointerDown = (event: PointerEvent) => { if (!shouldFocusPaneFromEventTarget(event.target)) { return; } onFocusPane(pane.id); }; const handlePaneFocusIn = (event: FocusEvent) => { if (!shouldFocusPaneFromEventTarget(event.target)) { return; } onFocusPane(pane.id); }; paneElement.addEventListener("pointerdown", handlePanePointerDown, true); paneElement.addEventListener("focusin", handlePaneFocusIn, true); return () => { paneElement.removeEventListener("pointerdown", handlePanePointerDown, true); paneElement.removeEventListener("focusin", handlePaneFocusIn, true); }; }, [onFocusPane, pane.id]); return ( onCloseTabsToLeft(tabId, paneTabs)} onCloseTabsToRight={(tabId) => onCloseTabsToRight(tabId, paneTabs)} onCloseOtherTabs={(tabId) => onCloseOtherTabs(tabId, paneTabs)} onSelectNewTabOption={onSelectNewTabOption} onNewTerminalTab={onNewTerminalTab} newTabAgentOptionId={newTabAgentOptionId ?? "__new_tab_agent__"} onReorderTabs={(nextTabs) => { onReorderTabsInPane( pane.id, nextTabs.map((tab) => tab.tabId), ); }} onSplitRight={() => onSplitPaneEmpty({ targetPaneId: pane.id, position: "right" })} onSplitDown={() => onSplitPaneEmpty({ targetPaneId: pane.id, position: "bottom" })} externalDndContext activeDragTabId={activeDragTabId} tabDropPreviewIndex={ tabDropPreview?.paneId === pane.id ? tabDropPreview.indicatorIndex : null } /> {paneContent ? ( ) : ( (renderPaneEmptyState?.() ?? null) )} ); } function collectPanesById(node: SplitNode): Map { const next = new Map(); function visit(current: SplitNode) { if (current.kind === "pane") { next.set(current.pane.id, current.pane); return; } for (const child of current.group.children) { visit(child); } } visit(node); return next; } function getNodeKey(node: SplitNode): string { if (node.kind === "pane") { return node.pane.id; } return node.group.id; } const styles = StyleSheet.create((theme) => ({ group: { flex: 1, minWidth: 0, minHeight: 0, }, groupHorizontal: { flexDirection: "row", }, groupVertical: { flexDirection: "column", }, groupChild: { flexBasis: 0, minWidth: 0, minHeight: 0, }, pane: { position: "relative", flex: 1, minWidth: 0, minHeight: 0, backgroundColor: theme.colors.surface0, overflow: "hidden", }, paneTabs: { minWidth: 0, }, paneContent: { position: "relative", flex: 1, minWidth: 0, minHeight: 0, }, dragOverlayChip: { flexDirection: "row", alignItems: "center", gap: theme.spacing[2], paddingHorizontal: theme.spacing[3], paddingVertical: theme.spacing[1], borderRadius: theme.borderRadius.md, borderWidth: 1, maxWidth: 200, }, dragOverlayLabel: { fontSize: theme.fontSize.sm, flexShrink: 1, }, }));