#AI工作流节点前端扩展开发指南

// 在 NodeType 接口中添加新节点的参数类型
interface NodeType extends Partial<Node> {
  type: string;
  name: string;
  category: string;        // 节点分类
  icon: string;            // Element Plus 图标名称
  description: string;     // 节点描述
  canBeSource: boolean;
  canBeTarget: boolean;

  // 添加你的新节点参数类型
  customParams?: {
    apiKey: string;
    endpoint: string;
  };
}

// 在 nodeTypes 数组中注册新节点
export const nodeTypes: NodeType[] = [
  // ...现有节点

  {
    type: 'custom',
    name: '自定义节点',
    category: nodeCategories.DATA_PROCESSING.key,
    icon: 'Setting',
    description: '自定义数据处理节点',
    canBeSource: true,
    canBeTarget: true,
    customParams: {
      apiKey: '',
      endpoint: ''
    },
    inputParams: [],
    outputParams: [
      { name: 'result', type: 'String' }
    ]
  }
];

export const getNodeConfig = (type: string): NodeType => {
  const nodeConfig = nodeTypes.find(node => node.type === type) || nodeTypes[0];
  return deepClone(nodeConfig) as NodeType; // 深拷贝避免共享引用
};

// panels/common.ts 返回的结构
[
  {
    id: 'node_abc123',
    name: 'HTTP请求',
    list: [
      { name: 'body', type: 'Object' },
      { name: 'status_code', type: 'Number' }
    ]
  },
  {
    id: 'global',
    name: '全局环境变量',
    list: [
      { name: 'API_KEY', value: 'global' },
      { name: 'DB_URL', value: 'global' }
    ]
  },
  {
    id: 'loop',
    name: '循环变量',
    list: [
      { name: 'index', type: 'Number', description: '当前迭代索引' }
    ]
  }
]

{
  type: 'custom',
  name: '自定义节点',
  icon: 'Setting',  // Element Plus 图标名称
  // ...
}

// panels/common.ts 第86-104行
previousNodes(): Node[] {
  const nodes: Node[] = [];
  const findPreviousNodes = (nodeId: string): void => {
    const connection = this.actualParent.connections
      .find(conn => conn.targetId === nodeId);
    if (!connection) return;

    const node = this.actualParent.nodes
      .find(node => node.id === connection.sourceId);
    if (node) {
      nodes.push(node);
      findPreviousNodes(node.id); // 递归查找
    }
  };

  findPreviousNodes(this.node.id);
  return nodes;
}

// panels/common.ts 第56-66行
if (this.actualParent.env && this.actualParent.env.length) {
  nodeOutputs.push({
    id: 'global',
    name: '全局环境变量',
    list: this.actualParent.env.map(env => ({
      name: env.name,
      value: 'global'
    }))
  });
}

// panels/common.ts 第68-82行
if (this.childNodeParent?.loopNode) {
  const loopVarName = this.childNodeParent.loopVarName || 'index';
  nodeOutputs.push({
    id: 'loop',
    name: '循环变量',
    list: [
      {
        name: loopVarName,
        type: 'Number',
        description: '当前迭代索引（从0开始）'
      }
    ]
  });
}

// LLM 节点的消息配置
{
  messages: [
    {
      role: 'user',
      content: '你好，${nodeA.name}，今天天气是${global.weather}，这是第${loop.index}次迭代'
    }
  ]
}

validate(): ValidationResult {
  return { valid: true, errors: [] };
}

handleClose(done) {
  const panel = this.$refs.panelComponent;
  if (panel?.validate) {
    const result = panel.validate();
    this.node.hasValidationError = !result.valid;
    if (!result.valid) {
      this.$message.warning(result.errors[0]?.message || '请完善必填项');
    }
  }
  if (typeof done === 'function') done();
  this.$emit('close');
}

{
  type: 'mcp',
  name: 'MCP服务',
  category: nodeCategories.AI_CAPABILITY.key,
  icon: 'Puzzle',
  description: '调用模型上下文协议服务',
  canBeSource: true,
  canBeTarget: true,
  mcpParams: {  // MCP 特定参数
    mcpId: '',
    mcpName: '',
    prompt: ''
  },
  inputParams: [],
  outputParams: [
    { name: 'result', type: 'Object' },
    { name: 'status', type: 'String' }
  ]
}

export const nodeCategories = {
  FLOW_CONTROL: {
    key: 'flow-control',
    label: '流程控制',
    order: 1,
    defaultCollapsed: false
  },
  AI_CAPABILITY: {
    key: 'ai-capability',
    label: 'AI能力',
    order: 2,
    defaultCollapsed: false
  },
  MULTIMODAL: {
    key: 'multimodal',
    label: '多模态',
    order: 3,
    defaultCollapsed: false
  },
  DATA_PROCESSING: {
    key: 'data-processing',
    label: '数据处理',
    order: 4,
    defaultCollapsed: false
  },
  EXTERNAL_INTEGRATION: {
    key: 'external-integration',
    label: '外部集成',
    order: 5,
    defaultCollapsed: false
  }
} as const;