src/store/useStore.ts

import { create } from 'zustand';
import { applyNodeChanges, applyEdgeChanges, NodeChange, EdgeChange, Node, Edge } from 'reactflow';
import { AppState, Message } from '../types';

export const useStore = create<AppState>((set, get) => ({
  nodes: [
    { id: '1', type: 'input', position: { x: 250, y: 0 }, data: { label: '开始' } },
    { id: '2', type: 'default', position: { x: 250, y: 150 }, data: { label: '处理步骤' } },
    { id: '3', type: 'output', position: { x: 250, y: 300 }, data: { label: '结束' } },
  ],
  edges: [
    { id: 'e1-2', source: '1', target: '2', animated: true },
    { id: 'e2-3', source: '2', target: '3', animated: true },
  ],
  messages: [
    {
      id: '1',
      role: 'assistant',
      content: '你好！我是你的工作流助手。请告诉我你想创建什么样的流程，或者直接描述你的需求。',
      createdAt: Date.now(),
    },
  ],
  isLoading: false,

  setNodes: (nodes: Node[]) => set({ nodes }),
  setEdges: (edges: Edge[]) => set({ edges }),

  onNodesChange: (changes: NodeChange[]) => {
    set((state) => ({
      nodes: applyNodeChanges(changes, state.nodes),
    }));
  },

  onEdgesChange: (changes: EdgeChange[]) => {
    set((state) => ({
      edges: applyEdgeChanges(changes, state.edges),
    }));
  },

  addMessage: (message: Message) => {
    set((state) => ({
      messages: [...state.messages, message],
    }));
  },

  setLoading: (loading: boolean) => set({ isLoading: loading }),

  updateFlow: (nodes: Node[], edges: Edge[]) => {
    set({ nodes, edges });
  },
}));