"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 { 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"

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


export function TbeTable({ promises, rfqId }: VendorsTableProps) {
  const { featureFlags } = useFeatureFlags()

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

  console.log(data)

  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 [selectedRfqIdForComments, setSelectedRfqIdForComments] = React.useState<number | null>(null)
  
  const [isFileDialogOpen, setIsFileDialogOpen] = React.useState(false)
  const [selectedVendorId, setSelectedVendorId] = React.useState<number | null>(null)
  const [selectedTbeId, setSelectedTbeId] = React.useState<number | null>(null)

  console.log(selectedVendorId,"selectedVendorId")
  console.log(rfqId,"rfqId")

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

  React.useEffect(() => {
    if (rowAction?.type === "comments") {
      // rowAction가 새로 세팅된 뒤 여기서 openCommentSheet 실행
      openCommentSheet(Number(rowAction.row.original.id))
    } 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(vendorId: number) {
    setInitialComments([])

    const comments = rowAction?.row.original.comments

    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,
          }
        })
      )
      // 3) state에 저장 -> CommentSheet에서 initialComments로 사용
      setInitialComments(commentWithAttachments)
    }

    setSelectedRfqIdForComments(vendorId)
    setCommentSheetOpen(true)
  }

  const openFilesDialog = (tbeId: number, vendorId: number) => {
    setSelectedTbeId(tbeId)
    setSelectedVendorId(vendorId)
    setIsFileDialogOpen(true)
  }


  // getColumns() 호출 시, router를 주입
  const columns = React.useMemo(
    () => getColumns({ setRowAction, router, openCommentSheet, openFilesDialog }),
    [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: ["actions"] },
    },
    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={1}
        open={commentSheetOpen}
        onOpenChange={setCommentSheetOpen}
        rfqId={rfqId}
        vendorId={selectedRfqIdForComments ?? 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}
      />
    </div>
  )
}