"use client"

import * as React from "react"
import { useRouter } from "next/navigation"
import type {
  DataTableAdvancedFilterField,
  DataTableFilterField,
  DataTableRowAction,
} from "@/types/table"
import { toast } from "sonner"
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 { fetchRfqAttachmentsbyCommentId, getTBEforVendor } from "../../rfqs/service"
import { CommentSheet, TbeComment } from "./comments-sheet"
import { TbeVendorFields } from "@/config/vendorTbeColumnsConfig"
import { useTbeFileHandlers } from "./tbeFileHandler"
import { useSession } from "next-auth/react"
import { RfqDeailDialog } from "./rfq-detail-dialog"

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

export function TbeVendorTable({ promises }: VendorsTableProps) {
  const { data: session } = useSession()
  const userVendorId = session?.user?.companyId
  const userId = Number(session?.user?.id)
  // Suspense로 받아온 데이터
  const [{ data, pageCount }] = React.use(promises)
  const [rowAction, setRowAction] = React.useState<DataTableRowAction<TbeVendorFields> | null>(null)


  // router 획득
  const router = useRouter()

  const [initialComments, setInitialComments] = React.useState<TbeComment[]>([])
  const [isLoadingComments, setIsLoadingComments] = React.useState(false)
  
  const [commentSheetOpen, setCommentSheetOpen] = React.useState(false)
  const [selectedRfqIdForComments, setSelectedRfqIdForComments] = React.useState<number | null>(null)
  const [isRfqDetailDialogOpen, setIsRfqDetailDialogOpen] = React.useState(false)

  const [selectedRfqId, setSelectedRfqId] = React.useState<number | null>(null)
  const [selectedRfq, setSelectedRfq] = React.useState<TbeVendorFields | null>(null)

  const openVendorContactsDialog = (rfqId: number, rfq: TbeVendorFields) => {
    setSelectedRfqId(rfqId)
    setSelectedRfq(rfq)
    setIsRfqDetailDialogOpen(true)
  }

  // TBE 파일 핸들러 훅 사용
  const {
    handleDownloadTbeTemplate,
    handleUploadTbeResponse,
    UploadDialog,
  } = useTbeFileHandlers()

  React.useEffect(() => {
    if (rowAction?.type === "comments") {
      // rowAction가 새로 세팅된 뒤 여기서 openCommentSheet 실행
      openCommentSheet(Number(rowAction.row.original.id))
    }
  }, [rowAction])

  async function openCommentSheet(vendorId: 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: userId, // DB나 API 응답에 있다고 가정
            attachments,
          }
        })
      )

      setInitialComments(commentWithAttachments)
    }

    setSelectedRfqIdForComments(vendorId)
    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)
  }
}

  // getColumns() 호출 시, 필요한 모든 핸들러 함수 주입
  const columns = React.useMemo(
    () => getColumns({
      setRowAction,
      router,
      openCommentSheet,
      handleDownloadTbeTemplate,
      handleUploadTbeResponse,
      openVendorContactsDialog
    }),
    [setRowAction, router, openCommentSheet, handleDownloadTbeTemplate, handleUploadTbeResponse, openVendorContactsDialog]
  )

  const filterFields: DataTableFilterField<TbeVendorFields>[] = []

  const advancedFilterFields: DataTableAdvancedFilterField<TbeVendorFields>[] = [
    { id: "rfqCode", label: "RFQ Code", type: "text" },
    { id: "projectCode", label: "Project Code", type: "text" },
    { id: "projectName", label: "Project Name", type: "text" },
    { id: "rfqCode", label: "RFQ Code", type: "text" },
    { id: "tbeResult", label: "TBE Result", type: "text" },
    { id: "tbeNote", label: "TBE Note", type: "text" },
    { id: "rfqCode", label: "RFQ Code", type: "text" },
    { id: "hasResponse", label: "Response?", type: "boolean" },
    { id: "rfqVendorUpdated", label: "Updated at", type: "date" },
    { id: "dueDate", label: "Project Name", type: "date" },

  ]

  const { table } = useDataTable({
    data,
    columns,
    pageCount,
    filterFields,
    enablePinning: true,
    enableAdvancedFilter: true,
    initialState: {
      sorting: [{ id: "rfqVendorUpdated", desc: true }],
      columnPinning: { right: ["comments", "tbeDocuments"] }, // tbeDocuments 컬럼을 우측에 고정
    },
    getRowId: (originalRow) => String(originalRow.id),
    shallow: false,
    clearOnDefault: true,
  })

  return (
    <>
      <DataTable table={table}>
        <DataTableAdvancedToolbar
          table={table}
          filterFields={advancedFilterFields}
          shallow={false}
        />
      </DataTable>

      {/* 코멘트 시트 */}
      {commentSheetOpen && selectedRfqIdForComments && (
        <CommentSheet
          open={commentSheetOpen}
          onOpenChange={setCommentSheetOpen}
          rfqId={selectedRfqIdForComments}
          initialComments={initialComments}
          vendorId={userVendorId || 0}
          currentUserId={userId || 0}
          isLoading={isLoadingComments} // Pass the loading state

        />
      )}

      <RfqDeailDialog
        isOpen={isRfqDetailDialogOpen}
        onOpenChange={setIsRfqDetailDialogOpen}
        rfqId={selectedRfqId}
        rfq={selectedRfq}
      />

      {/* TBE 파일 다이얼로그 */}
      <UploadDialog />
    </>
  )
}