path: root/lib/tbe/table/tbe-table.tsx

"use client"

import * as React from "react"
import { useRouter } from "next/navigation"
import type {
  DataTableAdvancedFilterField,
  DataTableFilterField,
  DataTableRowAction,
} from "@/types/table"

import { toSentenceCase } from "@/lib/utils"
import { useDataTable } from "@/hooks/use-data-table"
import { DataTable } from "@/components/data-table/data-table"
import { DataTableAdvancedToolbar } from "@/components/data-table/data-table-advanced-toolbar"
import { useFeatureFlags } from "./feature-flags-provider"
import { getColumns } from "./tbe-table-columns"
import { Vendor, vendors } from "@/db/schema/vendors"
import { CommentSheet, TbeComment } from "./comments-sheet"
import { VendorWithTbeFields } from "@/config/vendorTbeColumnsConfig"
import { TBEFileDialog } from "./file-dialog"
import { fetchRfqAttachmentsbyCommentId, getAllTBE } from "@/lib/rfqs/service"
import { VendorsTableToolbarActions } from "./tbe-table-toolbar-actions"
import { TbeResultDialog } from "./tbe-result-dialog"
import { toast } from "sonner"
import { VendorContactsDialog } from "./vendor-contact-dialog"

interface VendorsTableProps {
  promises: Promise<[
    Awaited<ReturnType<typeof getAllTBE>>,
  ]>
}

export function AllTbeTable({ promises }: VendorsTableProps) {
  const { featureFlags } = useFeatureFlags()
  const router = useRouter()

  // Suspense로 받아온 데이터
  const [{ data, pageCount }] = React.use(promises)

  const [rowAction, setRowAction] = React.useState<DataTableRowAction<VendorWithTbeFields> | null>(null)

  // 댓글 시트 관련 state
  const [initialComments, setInitialComments] = React.useState<TbeComment[]>([])
  const [isLoadingComments, setIsLoadingComments] = React.useState(false)

  const [commentSheetOpen, setCommentSheetOpen] = React.useState(false)
  const [selectedVendorIdForComments, setSelectedVendorIdForComments] = React.useState<number | null>(null)
  const [selectedRfqIdForComments, setSelectedRfqIdForComments] = React.useState<number | null>(null)

  // 파일 다이얼로그 관련 state
  const [isFileDialogOpen, setIsFileDialogOpen] = React.useState(false)
  const [selectedVendorIdForFiles, setSelectedVendorIdForFiles] = React.useState<number | null>(null)
  const [selectedTbeIdForFiles, setSelectedTbeIdForFiles] = React.useState<number | null>(null)
  const [selectedRfqIdForFiles, setSelectedRfqIdForFiles] = React.useState<number | null>(null)

  const [isContactDialogOpen, setIsContactDialogOpen] = React.useState(false)
  const [selectedVendor, setSelectedVendor] = React.useState<VendorWithTbeFields | null>(null)
  const [selectedVendorId, setSelectedVendorId] = React.useState<number | null>(null)

  // 테이블 리프레시용
  const handleRefresh = React.useCallback(() => {
    router.refresh();
  }, [router]);

  // -----------------------------------------------------------
  // 특정 action이 설정될 때마다 실행되는 effect
  // -----------------------------------------------------------
  React.useEffect(() => {
    if (!rowAction) return

    if (rowAction.type === "comments") {
      // rowAction가 새로 세팅되면 openCommentSheet 실행
      // row.original에 rfqId가 있다고 가정
      openCommentSheet(
        rowAction.row.original.vendorId ?? 0,
        rowAction.row.original.rfqId ?? 0,
      )
    } else if (rowAction.type === "files") {
      openFilesDialog(
        rowAction.row.original.tbeId ?? 0,
        rowAction.row.original.vendorId ?? 0,
        rowAction.row.original.rfqId ?? 0,
      )
    }
  }, [rowAction])

  // -----------------------------------------------------------
  // 댓글 시트 열기
  // -----------------------------------------------------------
  async function openCommentSheet(vendorId: number, rfqId: number) {
    setInitialComments([])
    setIsLoadingComments(true)
    const comments = rowAction?.row.original.comments
    try {
      if (comments && comments.length > 0) {
        const commentWithAttachments: TbeComment[] = await Promise.all(
          comments.map(async (c) => {
            const attachments = await fetchRfqAttachmentsbyCommentId(c.id)
            return {
              ...c,
              commentedBy: 1, // DB나 API 응답에 있다고 가정
              attachments,
            }
          })
        )
        setInitialComments(commentWithAttachments)
      }

      setSelectedVendorIdForComments(vendorId)
      setSelectedRfqIdForComments(rfqId)
      setCommentSheetOpen(true)
    } catch (error) {
      console.error("Error loading comments:", error)
      toast.error("Failed to load comments")
    } finally {
      // End loading regardless of success/failure
      setIsLoadingComments(false)
    }
  }

  // -----------------------------------------------------------
  // 파일 다이얼로그 열기
  // -----------------------------------------------------------
  const openFilesDialog = (tbeId: number, vendorId: number, rfqId: number) => {
    setSelectedTbeIdForFiles(tbeId)
    setSelectedVendorIdForFiles(vendorId)
    setSelectedRfqIdForFiles(rfqId)
    setIsFileDialogOpen(true)
  }

  const openVendorContactsDialog = (vendorId: number, vendor: VendorWithTbeFields) => {
    setSelectedVendorId(vendorId)
    setSelectedVendor(vendor)
    setIsContactDialogOpen(true)
  }


  // -----------------------------------------------------------
  // 테이블 컬럼
  // -----------------------------------------------------------
  const columns = React.useMemo(
    () =>
      getColumns({
        setRowAction,
        router,
        openCommentSheet,  // 필요하면 직접 호출 가능
        openFilesDialog,
        openVendorContactsDialog,
      }),
    [setRowAction, router]
  )

  // -----------------------------------------------------------
  // 필터 필드
  // -----------------------------------------------------------
  const filterFields: DataTableFilterField<VendorWithTbeFields>[] = [
    // 예: 표준 필터
  ]
  const advancedFilterFields: DataTableAdvancedFilterField<VendorWithTbeFields>[] = [
    { id: "vendorName", label: "Vendor Name", type: "text" },
    { id: "vendorCode", label: "Vendor Code", type: "text" },
    { id: "email", label: "Email", type: "text" },
    { id: "country", label: "Country", type: "text" },
    {
      id: "vendorStatus",
      label: "Vendor Status",
      type: "multi-select",
      options: vendors.status.enumValues.map((status) => ({
        label: toSentenceCase(status),
        value: status,
      })),
    },
    { id: "rfqVendorUpdated", label: "Updated at", type: "date" },
  ]

  // -----------------------------------------------------------
  // 테이블 생성 훅
  // -----------------------------------------------------------
  const { table } = useDataTable({
    data,
    columns,
    pageCount,
    filterFields,
    enablePinning: true,
    enableAdvancedFilter: true,
    initialState: {
      sorting: [{ id: "rfqVendorUpdated", desc: true }],
      columnPinning: { right: ["files", "comments"] },
    },
    getRowId: (originalRow) => (`${originalRow.id}${originalRow.rfqId}`),
    shallow: false,
    clearOnDefault: true,
  })

  return (
    <>
      <DataTable table={table}>
        <DataTableAdvancedToolbar
          table={table}
          filterFields={advancedFilterFields}
          shallow={false}
        >
          <VendorsTableToolbarActions table={table} rfqId={selectedRfqIdForFiles ?? 0} />
        </DataTableAdvancedToolbar>
      </DataTable>

      {/* 댓글 시트 */}
      <CommentSheet
        currentUserId={1}
        open={commentSheetOpen}
        onOpenChange={setCommentSheetOpen}
        vendorId={selectedVendorIdForComments ?? 0}
        rfqId={selectedRfqIdForComments ?? 0}
        isLoading={isLoadingComments}
        initialComments={initialComments}
      />

      {/* 파일 업로드/다운로드 다이얼로그 */}
      <TBEFileDialog
        isOpen={isFileDialogOpen}
        onOpenChange={setIsFileDialogOpen}
        tbeId={selectedTbeIdForFiles ?? 0}
        vendorId={selectedVendorIdForFiles ?? 0}
        rfqId={selectedRfqIdForFiles ?? 0}     // ← 여기!
        onRefresh={handleRefresh}
      />

      <TbeResultDialog
        open={rowAction?.type === "tbeResult"}
        onOpenChange={() => setRowAction(null)}
        tbe={rowAction?.row.original ?? null}
      />

      <VendorContactsDialog
        isOpen={isContactDialogOpen}
        onOpenChange={setIsContactDialogOpen}
        vendorId={selectedVendorId}
        vendor={selectedVendor}
      />

    </>
  )
}