medusa-store/www/packages/docs-ui/src/utils/workflow-diagram-utils.ts

import { WorkflowSteps, WorkflowStepUi } from "types"

export const createNodeClusters = (steps: WorkflowSteps) => {
  const clusters: Record<number, WorkflowStepUi[]> = {}

  steps.forEach((step) => {
    if (!clusters[step.depth]) {
      clusters[step.depth] = []
    }

    if (step.type === "when") {
      const whenSteps = step.steps.map((whenStep) => ({
        ...whenStep,
        depth: step.depth,
        when: step,
      }))
      clusters[step.depth].push(...whenSteps)
    } else {
      clusters[step.depth].push(step)
    }
  })

  return clusters
}

export const getNextCluster = (
  clusters: Record<number, WorkflowStepUi[]>,
  depth: number
) => {
  const nextDepth = depth + 1
  return clusters[nextDepth]
}