import { ModelNode } from './ModelNode';
import { ModelNodeV2 } from './ModelNodeV2';
import { SolverNode } from './SolverNode';
import { DesignVarNode } from './DesignVarNode';
import { ExtractorNode } from './ExtractorNode';
import { ObjectiveNode } from './ObjectiveNode';
import { ConstraintNode } from './ConstraintNode';
import { AlgorithmNode } from './AlgorithmNode';
import { SurrogateNode } from './SurrogateNode';

export {
  ModelNode,
  ModelNodeV2,
  SolverNode,
  DesignVarNode,
  ExtractorNode,
  ObjectiveNode,
  ConstraintNode,
  AlgorithmNode,
  SurrogateNode,
};

// Use ModelNodeV2 by default for enhanced dependency display
// Set USE_LEGACY_MODEL_NODE=true to use the original
const useEnhancedModelNode = !import.meta.env.VITE_USE_LEGACY_MODEL_NODE;

export const nodeTypes = {
  model: useEnhancedModelNode ? ModelNodeV2 : ModelNode,
  solver: SolverNode,
  designVar: DesignVarNode,
  extractor: ExtractorNode,
  objective: ObjectiveNode,
  constraint: ConstraintNode,
  algorithm: AlgorithmNode,
  surrogate: SurrogateNode,
};