import { create } from "zustand";
import type { GraphData, GraphNode, GraphLink } from "../types";

interface GraphState {
  data: GraphData;
  highlightNodes: Set<string>;
  highlightLinks: Set<GraphLink>;
  isLinkMode: boolean;
  selectedNode: GraphNode | null;

  // Действия с данными
  setData: (data: GraphData) => void;
  addNode: (node: GraphNode) => void;
  removeNode: (nodeId: string) => void;
  addLink: (link: GraphLink) => void;
  removeLink: (link: GraphLink) => void;

  // Подсветка
  setHighlight: (nodeIds: Set<string>, links: Set<GraphLink>) => void;

  // Режим связи
  toggleLinkMode: () => void;
  setSelectedNode: (node: GraphNode | null) => void;
  createLink: (sourceId: string, targetId: string) => void;

  // Экспорт
  exportData: () => void;
}

export const useGraphStore = create<GraphState>((set, get) => ({
  data: { nodes: [], links: [] },
  highlightNodes: new Set(),
  highlightLinks: new Set(),
  isLinkMode: false,
  selectedNode: null,

  setData: (data) => set({ data }),

  addNode: (node) => {
    set((state) => ({
      data: {
        ...state.data,
        nodes: [...state.data.nodes, node],
      },
    }));
  },

  removeNode: (nodeId) => {
    set((state) => ({
      data: {
        nodes: state.data.nodes.filter((n) => n.id !== nodeId),
        links: state.data.links.filter(
          (l) => l.source !== nodeId && l.target !== nodeId,
        ),
      },
    }));
  },

  addLink: (link) => {
    set((state) => ({
      data: {
        ...state.data,
        links: [...state.data.links, link],
      },
    }));
  },

  removeLink: (linkToRemove) => {
    set((state) => ({
      data: {
        ...state.data,
        links: state.data.links.filter((l) => l !== linkToRemove),
      },
    }));
  },

  setHighlight: (nodeIds, links) =>
    set({ highlightNodes: nodeIds, highlightLinks: links }),

  toggleLinkMode: () =>
    set((state) => ({
      isLinkMode: !state.isLinkMode,
      selectedNode: null,
    })),

  setSelectedNode: (node) => set({ selectedNode: node }),

  createLink: (sourceId, targetId) => {
    const { data, addLink } = get();

    const linkExists = data.links.some(
      (link) =>
        (link.source === sourceId && link.target === targetId) ||
        (link.source === targetId && link.target === sourceId),
    );

    if (!linkExists) {
      addLink({ source: sourceId, target: targetId, type: "custom" });
    }
  },

  exportData: () => {
    const { data } = get();
    const dataStr = JSON.stringify(data, null, 2);
    const blob = new Blob([dataStr], { type: "application/json" });
    const url = URL.createObjectURL(blob);
    const link = document.createElement("a");
    link.href = url;
    link.download = "graph-data.json";
    link.click();
    URL.revokeObjectURL(url);
  },
}));