import {
  BezierEdge,
  StraightEdge,
  StepEdge,
  SmoothStepEdge,
  SimpleBezierEdge,
} from "@xyflow/react";
import {
  svgDrawSmoothLinePath,
  svgDrawStraightLinePath,
  svgDrawSmoothStepLinePath,
  svgDrawSimpleBezierLinePath,
  pathfindingAStarDiagonal,
  pathfindingJumpPointNoDiagonal,
} from "./functions";
import type { SmartEdgeOptions } from "./SmartEdge";

export const smartEdgePresets = {
  bezier: {
    drawEdge: svgDrawSmoothLinePath,
    generatePath: pathfindingAStarDiagonal,
    fallback: BezierEdge,
  },
  straight: {
    drawEdge: svgDrawStraightLinePath,
    generatePath: pathfindingAStarDiagonal,
    fallback: StraightEdge,
  },
  step: {
    drawEdge: svgDrawStraightLinePath,
    generatePath: pathfindingJumpPointNoDiagonal,
    fallback: StepEdge,
  },
  smoothstep: {
    drawEdge: svgDrawSmoothStepLinePath(),
    generatePath: pathfindingJumpPointNoDiagonal,
    fallback: SmoothStepEdge,
  },
  simplebezier: {
    drawEdge: svgDrawSimpleBezierLinePath,
    generatePath: pathfindingAStarDiagonal,
    fallback: SimpleBezierEdge,
  },
} as const satisfies Record<string, SmartEdgeOptions>;

export type SmartEdgePreset = keyof typeof smartEdgePresets;