Function getNodeDimensions

yaml-flow-editor/src/utils/layout.ts:52–121

(node: FlowNode)

// Calculate node dimensions based on type and content
export function getNodeDimensions(node: FlowNode): { width: number; height: number } {
  // Prefer measured size from ReactFlow when available (post-render).
  // This is the most reliable way to avoid overlaps when nodes auto-grow.
  const anyNode = node as unknown as {
    width?: number;
    height?: number;
    measured?: { width?: number; height?: number };
  };
  const mw = anyNode.measured?.width;
  const mh = anyNode.measured?.height;
  if (typeof mw === 'number' && mw > 0 && typeof mh === 'number' && mh > 0) {
    return { width: mw, height: mh };
  }
  if (typeof anyNode.width === 'number' && anyNode.width > 0 && typeof anyNode.height === 'number' && anyNode.height > 0) {
    return { width: anyNode.width, height: anyNode.height };
  }

  const nodeType = node.type;
  const data = node.data;

  if (nodeType === 'if') {
    const { thenSteps, elseSteps } = data as IfNodeData;
    const branchHeight = Math.max(
      calculateNestedStepsHeight(thenSteps) + 60,
      calculateNestedStepsHeight(elseSteps) + 60,
      60,
    );
    return { width: 320, height: Math.max(60 + branchHeight + 48, 150) };
  }

  if (nodeType === 'while' || nodeType === 'for_each') {
    const { loopSteps } = data as WhileNodeData;
    const loopHeight = calculateNestedStepsHeight(loopSteps);
    return { width: 220, height: Math.max(78 + Math.max(loopHeight + 48, 72) + 48, 150) };
  }
  
  if (nodeType === 'switch') {
    const { cases, defaultSteps } = data as SwitchNodeData;
    let totalCasesHeight = 0;
    const caseCount = cases ? Object.keys(cases).length : 0;
    if (cases) {
      for (const caseSteps of Object.values(cases)) {
        totalCasesHeight += calculateNestedStepsHeight(caseSteps) + 50;
      }
    }
    const defaultHeight = calculateNestedStepsHeight(defaultSteps) + 50;
    return { width: 350, height: Math.max(100 + totalCasesHeight + defaultHeight + 40, 200 + caseCount * 60) };
  }

  if (nodeType === 'gather') {
    const gatherData = data as unknown as { calls?: unknown[]; module?: string };
    const callsCount = gatherData.calls?.length || 0;
    // Base height ~80px + (calls * 30px per call); when module is set, submodule viewer adds more
    const baseHeight = 100 + (callsCount * 30);
    const expandedHeight = gatherData.module || callsCount > 0 ? baseHeight + 220 : baseHeight;
    return { width: DEFAULT_NODE_WIDTH, height: Math.max(expandedHeight, DEFAULT_NODE_HEIGHT) };
  }