admin / Synapse-Cortex

import type { Edge } from '@xyflow/react'
import type { PlaybookGraph, PlaybookGraphNode, PlaybookNodeType } from '../api/types'
import { makeStartNode, type PlaybookNode } from './types'

/** Strips React Flow's runtime-only fields (selected/dragging/measured/
 * width/height/positionAbsolute/...) down to exactly what the backend's
 * app/ai/guardrails.py reads: id, type, data, position. */
export function toGraphJson(nodes: PlaybookNode[], edges: Edge[]): PlaybookGraph {
  return {
    nodes: nodes.map((n) => ({
      id: n.id,
      type: (n.type ?? 'command') as PlaybookNodeType,
      data: n.data,
      position: n.position,
    })),
    edges: edges.map((e) => ({
      id: e.id,
      source: e.source,
      target: e.target,
      sourceHandle: e.sourceHandle ?? null,
    })),
  }
}

/** The inverse conversion. A brand-new playbook has no graph yet, so an
 * empty/missing graph_json falls back to a single Start node rather than
 * an empty canvas. */
export function fromGraphJson(graph: PlaybookGraph | null | undefined): { nodes: PlaybookNode[]; edges: Edge[] } {
  if (!graph || !graph.nodes || graph.nodes.length === 0) {
    return { nodes: [makeStartNode()], edges: [] }
  }
  const nodes: PlaybookNode[] = graph.nodes.map((n: PlaybookGraphNode) => ({
    id: n.id,
    type: n.type,
    position: n.position ?? { x: 0, y: 0 },
    data: n.data ?? {},
    deletable: n.type !== 'start',
  })) as PlaybookNode[]

  const edges: Edge[] = graph.edges.map((e) => ({
    id: e.id,
    source: e.source,
    target: e.target,
    sourceHandle: e.sourceHandle ?? undefined,
    label: e.sourceHandle === 'true' ? 'True' : e.sourceHandle === 'false' ? 'False' : undefined,
  }))

  return { nodes, edges }
}