ERP-node/frontend/components/dataflow/DataFlowDesigner.tsx

"use client";

import React, { useState, useCallback, useEffect, useRef } from "react";
import toast from "react-hot-toast";
import {
  ReactFlow,
  Node,
  Edge,
  Controls,
  Background,
  useNodesState,
  useEdgesState,
  BackgroundVariant,
  SelectionMode,
} from "@xyflow/react";
import "@xyflow/react/dist/style.css";
import { TableNode } from "./TableNode";
import { TableSelector } from "./TableSelector";
import { ConnectionSetupModal } from "./ConnectionSetupModal";
import {
  TableDefinition,
  TableRelationship,
  DataFlowAPI,
  DataFlowDiagram,
  JsonRelationship,
  CreateDiagramRequest,
} from "@/lib/api/dataflow";
import SaveDiagramModal from "./SaveDiagramModal";
import { useAuth } from "@/hooks/useAuth";

// 고유 ID 생성 함수
const generateUniqueId = (prefix: string, diagramId?: number): string => {
  const timestamp = Date.now();
  const random = Math.random().toString(36).substr(2, 9);
  return `${prefix}-${diagramId || timestamp}-${random}`;
};

// 테이블 노드 데이터 타입 정의
interface TableNodeData extends Record<string, unknown> {
  table: {
    tableName: string;
    displayName: string;
    description: string;
    columns: Array<{
      name: string;
      type: string;
      description: string;
    }>;
  };
  onColumnClick: (tableName: string, columnName: string) => void;
  selectedColumns: string[];
}

// 노드 및 엣지 타입 정의
const nodeTypes = {
  tableNode: TableNode,
};

const edgeTypes = {};

interface DataFlowDesignerProps {
  companyCode?: string;
  onSave?: (relationships: TableRelationship[]) => void;
  selectedDiagram?: DataFlowDiagram | string | null;
  diagramId?: number;
  relationshipId?: string; // 하위 호환성 유지
  onBackToList?: () => void;
}

// TableRelationship 타입은 dataflow.ts에서 import

export const DataFlowDesigner: React.FC<DataFlowDesignerProps> = ({
  companyCode: propCompanyCode = "*",
  diagramId,
  relationshipId, // 하위 호환성 유지
  onSave, // eslint-disable-line @typescript-eslint/no-unused-vars
  selectedDiagram,
  onBackToList, // eslint-disable-line @typescript-eslint/no-unused-vars
}) => {
  const { user } = useAuth();

  // 실제 사용자 회사 코드 사용 (prop보다 사용자 정보 우선)
  const companyCode = user?.company_code || user?.companyCode || propCompanyCode;

  const [nodes, setNodes, onNodesChange] = useNodesState<Node<TableNodeData>>([]);
  const [edges, setEdges, onEdgesChange] = useEdgesState<Edge>([]);
  const [selectedColumns, setSelectedColumns] = useState<{
    [tableName: string]: string[];
  }>({});
  const [selectionOrder, setSelectionOrder] = useState<string[]>([]);
  const [selectedNodes, setSelectedNodes] = useState<string[]>([]);
  const [pendingConnection, setPendingConnection] = useState<{
    fromNode: { id: string; tableName: string; displayName: string };
    toNode: { id: string; tableName: string; displayName: string };
    fromColumn?: string;
    toColumn?: string;
    selectedColumnsData?: {
      [tableName: string]: {
        displayName: string;
        columns: string[];
      };
    };
  } | null>(null);
  const [relationships, setRelationships] = useState<TableRelationship[]>([]); // eslint-disable-line @typescript-eslint/no-unused-vars
  const [currentDiagramId, setCurrentDiagramId] = useState<number | null>(null); // 현재 화면의 diagram_id
  const [selectedEdgeInfo, setSelectedEdgeInfo] = useState<{
    relationshipId: string;
    relationshipName: string;
    fromTable: string;
    toTable: string;
    fromColumns: string[];
    toColumns: string[];
    relationshipType: string;
    connectionType: string;
    connectionInfo: string;
  } | null>(null); // 선택된 엣지 정보

  // 새로운 메모리 기반 상태들
  const [tempRelationships, setTempRelationships] = useState<JsonRelationship[]>([]); // 메모리에 저장된 관계들
  const [hasUnsavedChanges, setHasUnsavedChanges] = useState(false); // 저장되지 않은 변경사항
  const [showSaveModal, setShowSaveModal] = useState(false); // 저장 모달 표시 상태
  const [isSaving, setIsSaving] = useState(false); // 저장 중 상태
  const [currentDiagramName, setCurrentDiagramName] = useState<string>(""); // 현재 편집 중인 관계도 이름
  const toastShownRef = useRef(false); // eslint-disable-line @typescript-eslint/no-unused-vars

  // 편집 모드일 때 관계도 이름 로드
  useEffect(() => {
    const loadDiagramName = async () => {
      if (diagramId && diagramId > 0) {
        try {
          const jsonDiagram = await DataFlowAPI.getJsonDataFlowDiagramById(diagramId, companyCode);
          if (jsonDiagram && jsonDiagram.diagram_name) {
            setCurrentDiagramName(jsonDiagram.diagram_name);
          }
        } catch (error) {
          console.error("관계도 이름 로드 실패:", error);
        }
      } else {
        setCurrentDiagramName(""); // 신규 생성 모드
      }
    };

    loadDiagramName();
  }, [diagramId, companyCode]);

  // 키보드 이벤트 핸들러 (Del 키로 선택된 노드 삭제)
  useEffect(() => {
    const handleKeyDown = (event: KeyboardEvent) => {
      if (event.key === "Delete" && selectedNodes.length > 0) {
        // 선택된 노드들 삭제
        setNodes((prevNodes) => prevNodes.filter((node) => !selectedNodes.includes(node.id)));

        // 삭제된 노드들과 관련된 선택된 컬럼들도 정리
        const deletedTableNames = selectedNodes
          .filter((nodeId) => nodeId.startsWith("table-"))
          .map((nodeId) => nodeId.replace("table-", ""));

        setSelectedColumns((prev) => {
          const newColumns = { ...prev };
          deletedTableNames.forEach((tableName) => {
            delete newColumns[tableName];
          });
          return newColumns;
        });

        // 선택 순서도 정리
        setSelectionOrder((prev) => prev.filter((tableName) => !deletedTableNames.includes(tableName)));

        // 선택된 노드 초기화
        setSelectedNodes([]);

        toast.success(`${selectedNodes.length}개 테이블이 삭제되었습니다.`);
      }
    };

    window.addEventListener("keydown", handleKeyDown);
    return () => window.removeEventListener("keydown", handleKeyDown);
  }, [selectedNodes, setNodes]);

  // 컬럼 클릭 처리 (토글 방식, 최대 2개 테이블만 허용)
  const handleColumnClick = useCallback((tableName: string, columnName: string) => {
    setSelectedColumns((prev) => {
      const currentColumns = prev[tableName] || [];
      const isSelected = currentColumns.includes(columnName);
      const selectedTables = Object.keys(prev).filter((name) => prev[name] && prev[name].length > 0);

      if (isSelected) {
        // 선택 해제
        const newColumns = currentColumns.filter((column) => column !== columnName);
        if (newColumns.length === 0) {
          // eslint-disable-next-line @typescript-eslint/no-unused-vars
          const { [tableName]: removed, ...rest } = prev;
          // 선택 순서에서도 제거 (다음 렌더링에서)
          setTimeout(() => {
            setSelectionOrder((order) => order.filter((name) => name !== tableName));
          }, 0);
          return rest;
        }
        return { ...prev, [tableName]: newColumns };
      } else {
        // 새 선택
        if (selectedTables.length >= 2 && !selectedTables.includes(tableName)) {
          toast.error("최대 2개 테이블까지만 선택할 수 있습니다.");
          return prev;
        }

        const newColumns = [...currentColumns, columnName];
        const newSelection = { ...prev, [tableName]: newColumns };

        // 선택 순서 업데이트 (다음 렌더링에서)
        setTimeout(() => {
          setSelectionOrder((order) => {
            if (!order.includes(tableName)) {
              return [...order, tableName];
            }
            return order;
          });
        }, 0);

        return newSelection;
      }
    });
  }, []);

  // 선택된 관계도의 관계 로드
  const loadSelectedDiagramRelationships = useCallback(async () => {
    const currentDiagramId = diagramId || (relationshipId ? parseInt(relationshipId) : null);
    if (!currentDiagramId || isNaN(currentDiagramId)) return;

    try {
      console.log("🔍 JSON 관계도 로드 시작 (diagramId):", currentDiagramId);
      toast.loading("관계도를 불러오는 중...", { id: "load-diagram" });

      // 새로운 JSON API로 관계도 조회
      const jsonDiagram = await DataFlowAPI.getJsonDataFlowDiagramById(currentDiagramId);
      console.log("📋 JSON 관계도 데이터:", jsonDiagram);

      if (!jsonDiagram || !jsonDiagram.relationships) {
        throw new Error("관계도 데이터를 찾을 수 없습니다.");
      }

      const relationships = jsonDiagram.relationships.relationships || [];
      const tableNames = jsonDiagram.relationships.tables || [];

      console.log("📋 관계 목록:", relationships);
      console.log("📊 테이블 목록:", tableNames);

      // 메모리에 관계 저장 (기존 관계도 편집 시)
      setTempRelationships(relationships);
      setCurrentDiagramId(currentDiagramId);

      // 테이블 노드 생성을 위한 테이블 정보 로드

      // 테이블 정보 로드
      const allTables = await DataFlowAPI.getTables();
      console.log("🏢 전체 테이블 수:", allTables.length);
      const tableDefinitions: TableDefinition[] = [];

      for (const tableName of tableNames) {
        const foundTable = allTables.find((t) => t.tableName === tableName);
        console.log(`🔍 테이블 ${tableName} 검색 결과:`, foundTable);
        if (foundTable) {
          // 각 테이블의 컬럼 정보를 별도로 가져옴
          const columns = await DataFlowAPI.getTableColumns(tableName);
          const safeColumns = Array.isArray(columns) ? columns : [];
          console.log(`📋 테이블 ${tableName}의 컬럼 수:`, safeColumns.length);
          tableDefinitions.push({
            tableName: foundTable.tableName,
            displayName: foundTable.displayName,
            description: foundTable.description,
            columns: safeColumns,
          });
        } else {
          console.warn(`⚠️ 테이블 ${tableName}을 찾을 수 없습니다`);
        }
      }

      // 연결된 컬럼 정보 계산
      const connectedColumnsInfo: {
        [tableName: string]: { [columnName: string]: { direction: "source" | "target" | "both" } };
      } = {};

      relationships.forEach((rel: JsonRelationship) => {
        const fromTable = rel.fromTable;
        const toTable = rel.toTable;
        const fromColumns = rel.fromColumns || [];
        const toColumns = rel.toColumns || [];

        // 소스 테이블의 컬럼들을 source로 표시
        if (!connectedColumnsInfo[fromTable]) connectedColumnsInfo[fromTable] = {};
        fromColumns.forEach((col: string) => {
          if (connectedColumnsInfo[fromTable][col]) {
            connectedColumnsInfo[fromTable][col].direction = "both";
          } else {
            connectedColumnsInfo[fromTable][col] = { direction: "source" };
          }
        });

        // 타겟 테이블의 컬럼들을 target으로 표시
        if (!connectedColumnsInfo[toTable]) connectedColumnsInfo[toTable] = {};
        toColumns.forEach((col: string) => {
          if (connectedColumnsInfo[toTable][col]) {
            connectedColumnsInfo[toTable][col].direction = "both";
          } else {
            connectedColumnsInfo[toTable][col] = { direction: "target" };
          }
        });
      });

      console.log("🔌 연결된 컬럼 정보:", connectedColumnsInfo);

      // 테이블을 노드로 변환 (자동 레이아웃)
      const tableNodes = tableDefinitions.map((table, index) => {
        const x = (index % 3) * 400 + 100; // 3열 배치
        const y = Math.floor(index / 3) * 300 + 100;

        return {
          id: `table-${table.tableName}`,
          type: "tableNode",
          position: { x, y },
          data: {
            table: {
              tableName: table.tableName,
              displayName: table.displayName,
              description: "", // 기존 로드된 노드도 description 없이 통일
              columns: Array.isArray(table.columns)
                ? table.columns.map((col) => ({
                    name: col.columnName,
                    type: col.dataType || "varchar",
                    description: col.description || "",
                  }))
                : [],
            },
            onColumnClick: handleColumnClick,
            selectedColumns: [], // 관계도 로드 시에는 빈 상태로 시작
            connectedColumns: connectedColumnsInfo[table.tableName] || {},
          } as TableNodeData,
        };
      });

      console.log("🎨 생성된 테이블 노드 수:", tableNodes.length);
      console.log("📍 테이블 노드 상세:", tableNodes);
      setNodes(tableNodes);

      // JSON 관계를 엣지로 변환하여 표시 (테이블 간 번들 연결)
      const relationshipEdges: Edge[] = [];

      relationships.forEach((rel: JsonRelationship) => {
        const fromTable = rel.fromTable;
        const toTable = rel.toTable;
        const fromColumns = rel.fromColumns || [];
        const toColumns = rel.toColumns || [];

        if (fromColumns.length === 0 || toColumns.length === 0) {
          console.warn("⚠️ 컬럼 정보가 없습니다:", { fromColumns, toColumns });
          return;
        }

        // 테이블 간 하나의 번들 엣지 생성 (컬럼별 개별 엣지 대신)
        relationshipEdges.push({
          id: generateUniqueId("edge", currentDiagramId),
          source: `table-${fromTable}`,
          target: `table-${toTable}`,
          type: "smoothstep",
          animated: false,
          style: {
            stroke: "#3b82f6",
            strokeWidth: 2,
            strokeDasharray: "none",
          },
          data: {
            relationshipId: rel.id,
            relationshipName: "기존 관계",
            relationshipType: rel.relationshipType,
            connectionType: rel.connectionType,
            fromTable: fromTable,
            toTable: toTable,
            fromColumns: fromColumns,
            toColumns: toColumns,
            // 클릭 시 표시할 상세 정보
            details: {
              connectionInfo: `${fromTable}(${fromColumns.join(", ")}) → ${toTable}(${toColumns.join(", ")})`,
              relationshipType: rel.relationshipType,
              connectionType: rel.connectionType,
            },
          },
        });
      });

      console.log("🔗 생성된 관계 에지 수:", relationshipEdges.length);
      console.log("📍 관계 에지 상세:", relationshipEdges);
      setEdges(relationshipEdges);
      toast.success("관계도를 불러왔습니다.", { id: "load-diagram" });
    } catch (error) {
      console.error("선택된 관계도 로드 실패:", error);
      toast.error("관계도를 불러오는데 실패했습니다.", { id: "load-diagram" });
    }
  }, [diagramId, relationshipId, setNodes, setEdges, handleColumnClick]);

  // 기존 관계 로드 (새 관계도 생성 시)
  const loadExistingRelationships = useCallback(async () => {
    if (selectedDiagram) return; // 선택된 관계도가 있으면 실행하지 않음

    try {
      // 새로운 JSON 기반 시스템에서는 기존 관계를 미리 로드하지 않음
      console.log("새 관계도 생성 모드: 빈 캔버스로 시작");
      setRelationships([]);

      // 빈 캔버스로 시작
      setEdges([]);
    } catch (error) {
      console.error("기존 관계 로드 실패:", error);
      toast.error("기존 관계를 불러오는데 실패했습니다.");
    }
  }, [setEdges, selectedDiagram]);

  // 컴포넌트 마운트 시 관계 로드
  useEffect(() => {
    if (companyCode) {
      if (diagramId || relationshipId) {
        loadSelectedDiagramRelationships();
      } else {
        loadExistingRelationships();
      }
    }
  }, [companyCode, diagramId, relationshipId, loadExistingRelationships, loadSelectedDiagramRelationships]);

  // 노드 선택 변경 핸들러
  const onSelectionChange = useCallback(({ nodes }: { nodes: Node<TableNodeData>[] }) => {
    const selectedNodeIds = nodes.map((node) => node.id);
    setSelectedNodes(selectedNodeIds);
  }, []);

  // 캔버스 클릭 시 엣지 정보 섹션 닫기
  const onPaneClick = useCallback(() => {
    if (selectedEdgeInfo) {
      setSelectedEdgeInfo(null);
    }
  }, [selectedEdgeInfo]);

  // 빈 onConnect 함수 (드래그 연결 비활성화)
  const onConnect = useCallback(() => {
    // 드래그로 연결하는 것을 방지
    return;
  }, []);

  // 엣지 클릭 시 연결 정보 표시
  const onEdgeClick = useCallback((event: React.MouseEvent, edge: Edge) => {
    event.stopPropagation();
    const edgeData = edge.data as {
      relationshipId: string;
      relationshipName: string;
      fromTable: string;
      toTable: string;
      fromColumns: string[];
      toColumns: string[];
      relationshipType: string;
      connectionType: string;
      details?: {
        connectionInfo: string;
        relationshipType: string;
        connectionType: string;
      };
    };
    if (edgeData) {
      setSelectedEdgeInfo({
        relationshipId: edgeData.relationshipId,
        relationshipName: edgeData.relationshipName || "관계",
        fromTable: edgeData.fromTable,
        toTable: edgeData.toTable,
        fromColumns: edgeData.fromColumns || [],
        toColumns: edgeData.toColumns || [],
        relationshipType: edgeData.relationshipType,
        connectionType: edgeData.connectionType,
        connectionInfo: edgeData.details?.connectionInfo || `${edgeData.fromTable} → ${edgeData.toTable}`,
      });
    }
  }, []);

  // 엣지 마우스 엔터 시 색상 변경
  const onEdgeMouseEnter = useCallback(
    (event: React.MouseEvent, edge: Edge) => {
      setEdges((eds) =>
        eds.map((e) =>
          e.id === edge.id
            ? {
                ...e,
                style: {
                  ...e.style,
                  stroke: "#1d4ed8", // hover 색상
                  strokeWidth: 3,
                },
              }
            : e,
        ),
      );
    },
    [setEdges],
  );

  // 엣지 마우스 리브 시 원래 색상으로 복원
  const onEdgeMouseLeave = useCallback(
    (event: React.MouseEvent, edge: Edge) => {
      setEdges((eds) =>
        eds.map((e) =>
          e.id === edge.id
            ? {
                ...e,
                style: {
                  ...e.style,
                  stroke: "#3b82f6", // 기본 색상
                  strokeWidth: 2,
                },
              }
            : e,
        ),
      );
    },
    [setEdges],
  );

  // 선택된 컬럼이 변경될 때마다 기존 노드들 업데이트 및 selectionOrder 정리
  useEffect(() => {
    setNodes((prevNodes) =>
      prevNodes.map((node) => ({
        ...node,
        data: {
          ...node.data,
          selectedColumns: selectedColumns[node.data.table.tableName] || [],
        },
      })),
    );

    // selectionOrder에서 선택되지 않은 테이블들 제거
    const activeTables = Object.keys(selectedColumns).filter(
      (tableName) => selectedColumns[tableName] && selectedColumns[tableName].length > 0,
    );
    setSelectionOrder((prev) => prev.filter((tableName) => activeTables.includes(tableName)));
  }, [selectedColumns, setNodes]);

  // 연결 가능한 상태인지 확인
  const canCreateConnection = () => {
    const selectedTables = Object.keys(selectedColumns).filter(
      (tableName) => selectedColumns[tableName] && selectedColumns[tableName].length > 0,
    );

    // 최소 2개의 서로 다른 테이블에서 컬럼이 선택되어야 함
    return selectedTables.length >= 2;
  };

  // 컬럼 연결 설정 모달 열기
  const openConnectionModal = () => {
    const selectedTables = Object.keys(selectedColumns).filter(
      (tableName) => selectedColumns[tableName] && selectedColumns[tableName].length > 0,
    );

    if (selectedTables.length < 2) return;

    // 선택 순서에 따라 첫 번째와 두 번째 테이블 설정
    const orderedTables = selectionOrder.filter((name) => selectedTables.includes(name));
    const firstTableName = orderedTables[0];
    const secondTableName = orderedTables[1];
    const firstNode = nodes.find((node) => node.data.table.tableName === firstTableName);
    const secondNode = nodes.find((node) => node.data.table.tableName === secondTableName);

    if (!firstNode || !secondNode) return;

    // 첫 번째로 선택된 컬럼들 가져오기
    const firstTableColumns = selectedColumns[firstTableName] || [];
    const secondTableColumns = selectedColumns[secondTableName] || [];

    setPendingConnection({
      fromNode: {
        id: firstNode.id,
        tableName: firstNode.data.table.tableName,
        displayName: firstNode.data.table.displayName,
      },
      toNode: {
        id: secondNode.id,
        tableName: secondNode.data.table.tableName,
        displayName: secondNode.data.table.displayName,
      },
      // 선택된 첫 번째 컬럼을 연결 컬럼으로 설정
      fromColumn: firstTableColumns[0] || "",
      toColumn: secondTableColumns[0] || "",
      // 선택된 모든 컬럼 정보를 선택 순서대로 전달
      selectedColumnsData: (() => {
        const orderedData: { [key: string]: { displayName: string; columns: string[] } } = {};
        // selectionOrder 순서대로 데이터 구성 (첫 번째 선택이 먼저)
        orderedTables.forEach((tableName) => {
          const node = nodes.find((n) => n.data.table.tableName === tableName);
          if (node && selectedColumns[tableName]) {
            orderedData[tableName] = {
              displayName: node.data.table.displayName,
              columns: selectedColumns[tableName],
            };
          }
        });
        return orderedData;
      })(),
    });
  };

  // 실제 테이블 노드 추가
  const addTableNode = useCallback(
    async (table: TableDefinition) => {
      try {
        const newNode: Node<TableNodeData> = {
          id: `table-${table.tableName}`,
          type: "tableNode",
          position: { x: Math.random() * 300, y: Math.random() * 200 },
          data: {
            table: {
              tableName: table.tableName,
              displayName: table.displayName || table.tableName,
              description: "", // 새로 추가된 노드는 description 없이 통일
              columns: Array.isArray(table.columns)
                ? table.columns.map((col) => ({
                    name: col.columnName || "unknown",
                    type: col.dataType || "varchar", // 기존과 동일한 기본값 사용
                    description: col.description || "",
                  }))
                : [],
            },
            onColumnClick: handleColumnClick,
            selectedColumns: selectedColumns[table.tableName] || [],
            connectedColumns: {}, // 새로 추가된 노드는 연결 정보 없음
          },
        };

        setNodes((nds) => nds.concat(newNode));
      } catch (error) {
        console.error("테이블 노드 추가 실패:", error);
        toast.error("테이블 정보를 불러오는데 실패했습니다.");
      }
    },
    [handleColumnClick, selectedColumns, setNodes],
  );

  // 노드 전체 삭제
  const clearNodes = useCallback(() => {
    setNodes([]);
    setEdges([]);
    setSelectedColumns({});
    setSelectionOrder([]);
    setSelectedNodes([]);
    setCurrentDiagramId(null); // 현재 diagram_id도 초기화
  }, [setNodes, setEdges]);

  // 현재 추가된 테이블명 목록 가져오기
  const getSelectedTableNames = useCallback(() => {
    return nodes.filter((node) => node.id.startsWith("table-")).map((node) => node.data.table.tableName);
  }, [nodes]);

  // 연결 설정 확인
  const handleConfirmConnection = useCallback(
    (relationship: TableRelationship) => {
      if (!pendingConnection) return;

      // 메모리 기반 관계 생성 (DB 저장 없이)
      const fromTable = relationship.from_table_name;
      const toTable = relationship.to_table_name;
      const fromColumns = relationship.from_column_name
        .split(",")
        .map((col) => col.trim())
        .filter((col) => col);
      const toColumns = relationship.to_column_name
        .split(",")
        .map((col) => col.trim())
        .filter((col) => col);

      // JSON 형태의 관계 객체 생성
      const newRelationship: JsonRelationship = {
        id: generateUniqueId("rel", Date.now()),
        fromTable,
        toTable,
        fromColumns,
        toColumns,
        relationshipType: relationship.relationship_type,
        connectionType: relationship.connection_type,
        settings: relationship.settings || {},
      };

      // 메모리에 관계 추가
      setTempRelationships((prev) => [...prev, newRelationship]);
      setHasUnsavedChanges(true);

      // 캔버스에 엣지 즉시 표시
      const newEdge: Edge = {
        id: generateUniqueId("edge", Date.now()),
        source: pendingConnection.fromNode.id,
        target: pendingConnection.toNode.id,
        type: "smoothstep",
        animated: false,
        style: {
          stroke: "#3b82f6",
          strokeWidth: 2,
          strokeDasharray: "none",
        },
        data: {
          relationshipId: newRelationship.id,
          relationshipName: "임시 관계",
          relationshipType: relationship.relationship_type,
          connectionType: relationship.connection_type,
          fromTable,
          toTable,
          fromColumns,
          toColumns,
          details: {
            connectionInfo: `${fromTable}(${fromColumns.join(", ")}) → ${toTable}(${toColumns.join(", ")})`,
            relationshipType: relationship.relationship_type,
            connectionType: relationship.connection_type,
          },
        },
      };

      setEdges((eds) => [...eds, newEdge]);
      setPendingConnection(null);

      // 관계 생성 후 선택된 컬럼들 초기화
      setSelectedColumns({});
      setSelectionOrder([]);

      console.log("메모리에 관계 생성 완료:", newRelationship);
      toast.success("관계가 생성되었습니다. 저장 버튼을 눌러 관계도를 저장하세요.");
    },
    [pendingConnection, setEdges],
  );

  // 연결 설정 취소
  const handleCancelConnection = useCallback(() => {
    setPendingConnection(null);
  }, []);

  // 관계도 저장 함수
  const handleSaveDiagram = useCallback(
    async (diagramName: string) => {
      if (tempRelationships.length === 0) {
        toast.error("저장할 관계가 없습니다.");
        return;
      }

      setIsSaving(true);
      try {
        // 연결된 테이블 목록 추출
        const connectedTables = Array.from(
          new Set([...tempRelationships.map((rel) => rel.fromTable), ...tempRelationships.map((rel) => rel.toTable)]),
        ).sort();

        // 저장 요청 데이터 생성
        const createRequest: CreateDiagramRequest = {
          diagram_name: diagramName,
          relationships: {
            relationships: tempRelationships,
            tables: connectedTables,
          },
        };

        let savedDiagram;

        // 편집 모드 vs 신규 생성 모드 구분
        if (diagramId && diagramId > 0) {
          // 편집 모드: 기존 관계도 업데이트
          savedDiagram = await DataFlowAPI.updateJsonDataFlowDiagram(
            diagramId,
            createRequest,
            companyCode,
            user?.userId || "SYSTEM",
          );
          toast.success(`관계도 "${diagramName}"가 성공적으로 수정되었습니다.`);
        } else {
          // 신규 생성 모드: 새로운 관계도 생성
          savedDiagram = await DataFlowAPI.createJsonDataFlowDiagram(
            createRequest,
            companyCode,
            user?.userId || "SYSTEM",
          );
          toast.success(`관계도 "${diagramName}"가 성공적으로 생성되었습니다.`);
        }

        // 성공 처리
        setHasUnsavedChanges(false);
        setShowSaveModal(false);
        setCurrentDiagramId(savedDiagram.diagram_id);

        console.log("관계도 저장 완료:", savedDiagram);
      } catch (error) {
        console.error("관계도 저장 실패:", error);
        toast.error("관계도 저장 중 오류가 발생했습니다.");
      } finally {
        setIsSaving(false);
      }
    },
    [tempRelationships, diagramId, companyCode, user?.userId],
  );

  // 저장 모달 열기
  const handleOpenSaveModal = useCallback(() => {
    if (tempRelationships.length === 0) {
      toast.error("저장할 관계가 없습니다. 먼저 테이블을 연결해주세요.");
      return;
    }
    setShowSaveModal(true);
  }, [tempRelationships.length]);

  // 저장 모달 닫기
  const handleCloseSaveModal = useCallback(() => {
    if (!isSaving) {
      setShowSaveModal(false);
    }
  }, [isSaving]);

  return (
    <div className="data-flow-designer h-screen bg-gray-100">
      <div className="flex h-full">
        {/* 사이드바 */}
        <div className="w-80 border-r border-gray-200 bg-white shadow-lg">
          <div className="p-6">
            <h2 className="mb-6 text-xl font-bold text-gray-800">테이블 간 데이터 관계 설정</h2>

            {/* 테이블 선택기 */}
            <TableSelector
              companyCode={companyCode}
              onTableAdd={addTableNode}
              selectedTables={getSelectedTableNames()}
            />

            {/* 컨트롤 버튼들 */}
            <div className="space-y-3">
              <button
                onClick={clearNodes}
                className="w-full rounded-lg bg-red-500 p-3 font-medium text-white transition-colors hover:bg-red-600"
              >
                전체 삭제
              </button>

              <button
                onClick={handleOpenSaveModal}
                disabled={tempRelationships.length === 0}
                className={`w-full rounded-lg p-3 font-medium text-white transition-colors ${
                  tempRelationships.length > 0 ? "bg-green-500 hover:bg-green-600" : "cursor-not-allowed bg-gray-400"
                } ${hasUnsavedChanges ? "animate-pulse" : ""}`}
              >
                💾 관계도 저장 {tempRelationships.length > 0 && `(${tempRelationships.length})`}
              </button>
            </div>

            {/* 통계 정보 */}
            <div className="mt-6 rounded-lg bg-gray-50 p-4">
              <div className="mb-2 text-sm font-semibold text-gray-700">통계</div>
              <div className="space-y-1 text-sm text-gray-600">
                <div className="flex justify-between">
                  <span>테이블 노드:</span>
                  <span className="font-medium">{nodes.length}개</span>
                </div>
                <div className="flex justify-between">
                  <span>연결:</span>
                  <span className="font-medium">{edges.length}개</span>
                </div>
                <div className="flex justify-between">
                  <span>메모리 관계:</span>
                  <span className="font-medium text-orange-600">{tempRelationships.length}개</span>
                </div>
                <div className="flex justify-between">
                  <span>관계도 ID:</span>
                  <span className="font-medium">{currentDiagramId || "미설정"}</span>
                </div>
                {hasUnsavedChanges && (
                  <div className="mt-2 text-xs font-medium text-orange-600">⚠️ 저장되지 않은 변경사항이 있습니다</div>
                )}
              </div>
            </div>

            {/* 선택된 컬럼 정보 */}
            {Object.keys(selectedColumns).length > 0 && (
              <div className="mt-6 space-y-4">
                <div className="rounded-lg border border-blue-200 bg-blue-50 p-4">
                  <div className="mb-3 text-sm font-semibold text-blue-800">선택된 컬럼</div>
                  <div className="space-y-3">
                    {[...new Set(selectionOrder)]
                      .filter((tableName) => selectedColumns[tableName] && selectedColumns[tableName].length > 0)
                      .map((tableName, index, filteredOrder) => {
                        const columns = selectedColumns[tableName];
                        const node = nodes.find((n) => n.data.table.tableName === tableName);
                        const displayName = node?.data.table.displayName || tableName;

                        return (
                          <div key={`selected-${tableName}-${index}`}>
                            <div className="w-full min-w-0 rounded-lg border border-blue-300 bg-white p-3">
                              <div className="mb-2 flex flex-wrap items-center gap-2">
                                <div className="flex-shrink-0 rounded px-2 py-1 text-xs font-medium text-blue-600">
                                  {displayName}
                                </div>
                              </div>
                              <div className="flex w-full min-w-0 flex-wrap gap-1">
                                {columns.map((column, columnIndex) => (
                                  <div
                                    key={`${tableName}-${column}-${columnIndex}`}
                                    className="max-w-full truncate rounded-full border border-blue-200 bg-blue-100 px-2 py-1 text-xs text-blue-800"
                                    title={column}
                                  >
                                    {column}
                                  </div>
                                ))}
                              </div>
                            </div>
                            {/* 첫 번째 테이블 다음에 화살표 표시 */}
                            {index === 0 && filteredOrder.length > 1 && (
                              <div className="flex justify-center py-2">
                                <div className="text-gray-400">↓</div>
                              </div>
                            )}
                          </div>
                        );
                      })}
                  </div>
                  <div className="mt-3 flex gap-2">
                    <button
                      onClick={openConnectionModal}
                      disabled={!canCreateConnection()}
                      className={`w-full rounded px-3 py-1 text-xs font-medium transition-colors ${
                        canCreateConnection()
                          ? "cursor-pointer bg-blue-600 text-white hover:bg-blue-700"
                          : "cursor-not-allowed bg-gray-300 text-gray-500"
                      }`}
                    >
                      컬럼 연결 설정
                    </button>
                    <button
                      onClick={() => {
                        setSelectedColumns({});
                        setSelectionOrder([]);
                      }}
                      className="w-full cursor-pointer rounded bg-gray-200 px-3 py-1 text-xs font-medium text-gray-600 hover:bg-gray-300"
                    >
                      선택 초기화
                    </button>
                  </div>
                </div>
              </div>
            )}

            {/* 선택된 엣지 정보 */}
            {selectedEdgeInfo && (
              <div className="mt-6 space-y-4">
                <div className="rounded-lg border border-green-200 bg-green-50 p-4">
                  <div className="mb-3 flex items-center justify-between">
                    <div className="text-sm font-semibold text-green-800">🔗 연결 정보</div>
                    <button onClick={() => setSelectedEdgeInfo(null)} className="text-green-600 hover:text-green-800">
                      ✕
                    </button>
                  </div>
                  <div className="space-y-2 text-xs">
                    <div className="rounded bg-white p-2">
                      <div className="font-medium text-gray-700">연결</div>
                      <div className="text-gray-600">
                        {selectedEdgeInfo.fromTable}({selectedEdgeInfo.fromColumns.join(", ")}) →{" "}
                        {selectedEdgeInfo.toTable}({selectedEdgeInfo.toColumns.join(", ")})
                      </div>
                    </div>
                    <div className="flex gap-2">
                      <div className="flex-1 rounded bg-white p-2">
                        <div className="font-medium text-gray-700">관계 유형</div>
                        <div className="text-gray-600">{selectedEdgeInfo.relationshipType}</div>
                      </div>
                      <div className="flex-1 rounded bg-white p-2">
                        <div className="font-medium text-gray-700">연결 유형</div>
                        <div className="text-gray-600">{selectedEdgeInfo.connectionType}</div>
                      </div>
                    </div>
                  </div>
                </div>
              </div>
            )}
          </div>
        </div>

        {/* React Flow 캔버스 */}
        <div className="relative flex-1">
          <ReactFlow
            className="[&_.react-flow\_\_pane]:cursor-default [&_.react-flow\_\_pane.dragging]:cursor-grabbing"
            nodes={nodes}
            edges={edges}
            onNodesChange={onNodesChange}
            onEdgesChange={onEdgesChange}
            onConnect={onConnect}
            onSelectionChange={onSelectionChange}
            onPaneClick={onPaneClick}
            onEdgeClick={onEdgeClick}
            onEdgeMouseEnter={onEdgeMouseEnter}
            onEdgeMouseLeave={onEdgeMouseLeave}
            nodeTypes={nodeTypes}
            edgeTypes={edgeTypes}
            fitView
            attributionPosition="bottom-left"
            panOnScroll={false}
            zoomOnScroll={true}
            zoomOnPinch={true}
            panOnDrag={[1, 2]}
            selectionOnDrag={true}
            multiSelectionKeyCode={null}
            selectNodesOnDrag={false}
            selectionMode={SelectionMode.Partial}
          >
            <Controls />

            <Background variant={BackgroundVariant.Dots} gap={20} size={1} color="#E5E7EB" />
          </ReactFlow>

          {/* 안내 메시지 */}
          {nodes.length === 0 && (
            <div className="pointer-events-none absolute inset-0 flex items-center justify-center">
              <div className="text-center text-gray-500">
                <div className="mb-2 text-2xl">📊</div>
                <div className="mb-1 text-lg font-medium">테이블 간 데이터 관계 설정을 시작하세요</div>
                <div className="text-sm">
                  <div>왼쪽 사이드바에서 테이블을 더블클릭하여 추가하세요</div>
                  <div className="mt-1 text-xs text-gray-400">테이블 선택 후 Del 키로 삭제 가능</div>
                </div>
              </div>
            </div>
          )}
        </div>
      </div>

      {/* 연결 설정 모달 */}
      <ConnectionSetupModal
        isOpen={!!pendingConnection}
        connection={pendingConnection}
        companyCode={companyCode}
        diagramId={currentDiagramId || diagramId || (relationshipId ? parseInt(relationshipId) : undefined)}
        onConfirm={handleConfirmConnection}
        onCancel={handleCancelConnection}
      />

      {/* 관계도 저장 모달 */}
      <SaveDiagramModal
        isOpen={showSaveModal}
        onClose={handleCloseSaveModal}
        onSave={handleSaveDiagram}
        relationships={tempRelationships}
        defaultName={
          diagramId && diagramId > 0 && currentDiagramName
            ? currentDiagramName // 편집 모드: 기존 관계도 이름
            : `관계도 ${new Date().toLocaleDateString()}` // 신규 생성 모드: 새로운 이름
        }
        isLoading={isSaving}
      />
    </div>
  );
};