invyone/frontend/contexts/TableOptionsContext.tsx

import React, { createContext, useContext, useState, useCallback, useMemo, ReactNode } from "react";
import { TableRegistration, TableOptionsContextValue } from "@/types/table-options";
import { useActiveTab } from "./ActiveTabContext";

/**
 * TableOptionsContext — TableSettingsModal / TableOptionsToolbar / ColumnVisibilityPanel /
 * GroupingPanel / FilterPanel 등 사용자 옵션 UI 의 공유 등록처.
 *
 * 등록 contract 는 컴포넌트 type 과 독립적이다. `table_id` 는 단지 unique key 일 뿐이고
 * canonical `lib/registry/components/table/TableComponent` (`table-${id}` prefix) 를
 * 비롯한 다양한 테이블 ID 형식이 같은 Provider 아래 공존해도 충돌하지 않는다.
 */
const TableOptionsContext = createContext<TableOptionsContextValue | undefined>(undefined);

export const TableOptionsProvider: React.FC<{ children: ReactNode }> = ({ children }) => {
  const [registeredTables, setRegisteredTables] = useState<Map<string, TableRegistration>>(new Map());
  const [selectedTableId, setSelectedTableId] = useState<string | null>(null);

  /**
   * 테이블 등록
   */
  const registerTable = useCallback((registration: TableRegistration) => {
    setRegisteredTables((prev) => {
      const newMap = new Map(prev);
      newMap.set(registration.table_id, registration);

      // 첫 번째 테이블이면 자동 선택
      if (newMap.size === 1) {
        setSelectedTableId(registration.table_id);
      }

      return newMap;
    });
  }, []);

  /**
   * 테이블 등록 해제
   * 주의:
   * 1. selectedTableId를 의존성으로 사용하면 무한 루프 발생 가능
   * 2. 재등록 시에도 unregister가 호출되므로 selectedTableId를 변경하면 안됨
   */
  const unregisterTable = useCallback(
    (tableId: string) => {
      setRegisteredTables((prev) => {
        const newMap = new Map(prev);
        newMap.delete(tableId);
        return newMap;
      });

      // 🚫 selectedTableId를 변경하지 않음
      // 이유: useEffect 재실행 시 cleanup → register 순서로 호출되는데,
      // cleanup에서 selectedTableId를 null로 만들면 필터 설정이 초기화됨
      // 다른 테이블이 선택되어야 하면 TableSearchWidget에서 자동 선택함
    },
    [], // 의존성 없음 - 무한 루프 방지
  );

  /**
   * 특정 테이블 조회
   */
  const getTable = useCallback(
    (tableId: string) => {
      return registeredTables.get(tableId);
    },
    [registeredTables],
  );

  /**
   * 테이블 데이터 건수 업데이트
   */
  const updateTableDataCount = useCallback((tableId: string, count: number) => {
    setRegisteredTables((prev) => {
      const table = prev.get(tableId);
      if (table) {
        // 기존 테이블 정보에 data_count만 업데이트 (TableRegistration contract)
        const updatedTable = { ...table, data_count: count };
        const newMap = new Map(prev);
        newMap.set(tableId, updatedTable);
        return newMap;
      }
      return prev;
    });
  }, []);

  // ActiveTab context 사용 (optional - 에러 방지)
  const activeTabContext = useActiveTab();

  /**
   * 현재 활성 탭의 테이블만 반환
   */
  const getActiveTabTables = useCallback(() => {
    const allTables = Array.from(registeredTables.values());
    const activeTabIds = activeTabContext.getAllActiveTabIds();

    // 활성 탭이 없으면 탭에 속하지 않은 테이블만 반환
    if (activeTabIds.length === 0) {
      return allTables.filter((table) => !table.parent_tab_id);
    }

    // 활성 탭에 속한 테이블 + 탭에 속하지 않은 테이블
    return allTables.filter((table) => !table.parent_tab_id || activeTabIds.includes(table.parent_tab_id));
  }, [registeredTables, activeTabContext]);

  /**
   * 특정 탭의 테이블만 반환
   */
  const getTablesForTab = useCallback(
    (tabId: string) => {
      const allTables = Array.from(registeredTables.values());
      return allTables.filter((table) => table.parent_tab_id === tabId);
    },
    [registeredTables],
  );

  return (
    <TableOptionsContext.Provider
      value={{
        registered_tables: registeredTables,
        registerTable,
        unregisterTable,
        getTable,
        updateTableDataCount,
        selected_table_id: selectedTableId,
        setSelectedTableId,
        getActiveTabTables,
        getTablesForTab,
      }}
    >
      {children}
    </TableOptionsContext.Provider>
  );
};

/**
 * Context Hook — Provider 없으면 throw. canonical TableComponent 등 Provider 가
 * 항상 wrap 된 화면에서만 호출.
 */
export const useTableOptions = () => {
  const context = useContext(TableOptionsContext);
  if (!context) {
    throw new Error("useTableOptions must be used within TableOptionsProvider");
  }
  return context;
};

/**
 * Optional variant — Provider 가 없는 위치 (대시보드 카드, 단독 미리보기, embed iframe 등)
 * 에서도 mount 가능해야 하는 canonical TableComponent 용. Provider 가 있으면 context 객체,
 * 없으면 `undefined` 를 돌려준다. 호출자는 `if (!ctx) return;` 로 가드.
 *
 * Phase B.1 (2026-05-20) — canonical `table` 컴포넌트가 옛 본체와 동일 contract 로
 * register 하되, dashboard 등 Provider 가 없는 곳에서도 안전하게 mount 되게 신설.
 */
export const useTableOptionsOptional = () => {
  return useContext(TableOptionsContext);
};