path: root/lib/rfqs-tech/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 { getColumns } from "./tbe-table-columns"
import { vendors } from "@/db/schema/vendors"
import { VendorsTableToolbarActions } from "./tbe-table-toolbar-actions"
import { fetchRfqAttachmentsbyCommentId, getTBE } from "../service"
import { InviteVendorsDialog } from "./invite-vendors-dialog"
import { CommentSheet, TbeComment } from "./comments-sheet"
import { VendorWithTbeFields } from "@/config/vendorTbeColumnsConfig"
import { TBEFileDialog } from "./file-dialog"
import { TbeResultDialog } from "./tbe-result-dialog"
import { VendorContactsDialog } from "./vendor-contact-dialog"
import { useSession } from "next-auth/react" // Next-auth session hook 추가

interface VendorsTableProps {
  promises: Promise<
    [
      Awaited<ReturnType<typeof getTBE>>,
    ]
  >
  rfqId: number
}


export function TbeTable({ promises, rfqId }: VendorsTableProps) {
  // Suspense로 받아온 데이터
  const [{ data: rawData, pageCount }] = React.use(promises)

  // 벤더별로 데이터 그룹화
  const data = React.useMemo(() => {
    const vendorMap = new Map<number, VendorWithTbeFields>()
    
    rawData.forEach((item) => {
      const vendorId = item.vendorId
      
      if (vendorMap.has(vendorId)) {
        // 기존 벤더 데이터가 있으면 파일과 댓글을 합침
        const existing = vendorMap.get(vendorId)!
        
        // 파일 합치기 (중복 제거)
        const existingFileIds = new Set(existing.files.map(f => f.id))
        const newFiles = item.files.filter(f => !existingFileIds.has(f.id))
        existing.files = [...existing.files, ...newFiles]
        
        // 댓글 합치기 (중복 제거)
        const existingCommentIds = new Set(existing.comments.map(c => c.id))
        const newComments = item.comments.filter(c => !existingCommentIds.has(c.id))
        existing.comments = [...existing.comments, ...newComments]
        
      } else {
        // 새로운 벤더 데이터 추가
        vendorMap.set(vendorId, { ...item, vendorResponseId: item.id })
      }
    })
    
    return Array.from(vendorMap.values())
  }, [rawData])
  
  const { data: session } = useSession() // 세션 정보 가져오기

  const currentUserId = session?.user?.id ? parseInt(session.user.id, 10) : 0


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

  // **router** 획득
  const router = useRouter()

  const [initialComments, setInitialComments] = React.useState<TbeComment[]>([])
  const [commentSheetOpen, setCommentSheetOpen] = React.useState(false)

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

  // Add handleRefresh function
  const handleRefresh = React.useCallback(() => {
    router.refresh();
  }, [router]);

  React.useEffect(() => {
    if (rowAction?.type === "comments") {
      // rowAction가 새로 세팅된 뒤 여기서 openCommentSheet 실행
      openCommentSheet()
    } else if (rowAction?.type === "files") {
      // Handle files action
      const vendorId = rowAction.row.original.vendorId;
      const tbeId = rowAction.row.original.tbeId ?? 0;
      openFilesDialog(tbeId, vendorId);
    }
  }, [rowAction])

  async function openCommentSheet() {
    setInitialComments([])
    
    const comments = rowAction?.row.original.comments
    const vendorId = rowAction?.row.original.vendorId
    const tbeId = rowAction?.row.original.tbeId 
    if (comments && comments.length > 0) {
      const commentWithAttachments: TbeComment[] = await Promise.all(
        comments.map(async (c) => {
          const attachments = await fetchRfqAttachmentsbyCommentId(c.id)

          return {
            ...c,
            commentedBy: currentUserId, // DB나 API 응답에 있다고 가정
            attachments,
          }
        })
      )
      // 3) state에 저장 -> CommentSheet에서 initialComments로 사용
      setInitialComments(commentWithAttachments)
    }
    setSelectedTbeId(tbeId ?? 0)
    setSelectedVendorId(vendorId ?? 0)
    setCommentSheetOpen(true)
  }

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

  // getColumns() 호출 시, router를 주입
  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: ["comments"] },
    },
    getRowId: (originalRow) => String(originalRow.id),
    shallow: false,
    clearOnDefault: true,
  })



  return (
    <div style={{ maxWidth: '80vw' }}>
      <DataTable
        table={table}
      >
        <DataTableAdvancedToolbar
          table={table}
          filterFields={advancedFilterFields}
          shallow={false}
        >
          <VendorsTableToolbarActions table={table} rfqId={rfqId} />
        </DataTableAdvancedToolbar>
      </DataTable>
      <InviteVendorsDialog
        vendors={rowAction?.row.original ? [rowAction?.row.original] : []}
        onOpenChange={() => setRowAction(null)}
        rfqId={rfqId}
        open={rowAction?.type === "invite"}
        showTrigger={false}
      />
      <CommentSheet
        currentUserId={currentUserId}
        open={commentSheetOpen}
        onOpenChange={setCommentSheetOpen}
        rfqId={rfqId}
        tbeId={selectedTbeId ?? 0}
        vendorId={selectedVendorId ?? 0}
        initialComments={initialComments}
      />

      <TBEFileDialog
        isOpen={isFileDialogOpen}
        onOpenChange={setIsFileDialogOpen}
        tbeId={selectedTbeId ?? 0}
        vendorId={selectedVendorId ?? 0}
        rfqId={rfqId} // Use the prop directly instead of data[0]?.rfqId
        onRefresh={handleRefresh}
      />

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

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

    </div>
  )
}