path: root/lib/rfqs/table/rfqs-table-columns.tsx

"use client"

import * as React from "react"
import { type DataTableRowAction } from "@/types/table"
import { type ColumnDef } from "@tanstack/react-table"
import { Ellipsis, Paperclip, Package } from "lucide-react"
import { toast } from "sonner"

import { formatDate } from "@/lib/utils"
import { Badge } from "@/components/ui/badge"
import { Button } from "@/components/ui/button"
import { Checkbox } from "@/components/ui/checkbox"
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuRadioGroup,
  DropdownMenuRadioItem,
  DropdownMenuSeparator,
  DropdownMenuShortcut,
  DropdownMenuSub,
  DropdownMenuSubContent,
  DropdownMenuSubTrigger,
  DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu"

import { getRFQStatusIcon } from "@/lib/tasks/utils"
import { rfqsColumnsConfig } from "@/config/rfqsColumnsConfig"
import { RfqWithItemCount } from "@/db/schema/rfq"
import { DataTableColumnHeaderSimple } from "@/components/data-table/data-table-column-simple-header"
import { useRouter } from "next/navigation"
import { RfqType } from "../validations"

type NextRouter = ReturnType<typeof useRouter>;

interface GetColumnsProps {
  setRowAction: React.Dispatch<
    React.SetStateAction<DataTableRowAction<RfqWithItemCount> | null>
  >
  openItemsModal: (rfqId: number) => void
  openAttachmentsSheet: (rfqId: number) => void
  router: NextRouter
  rfqType?: RfqType
}

/**
 * tanstack table 컬럼 정의 (중첩 헤더 버전)
 */
export function getColumns({
  setRowAction,
  openItemsModal,
  openAttachmentsSheet,
  router,
  rfqType,
}: GetColumnsProps): ColumnDef<RfqWithItemCount>[] {
  // ----------------------------------------------------------------
  // 1) select 컬럼 (체크박스)
  // ----------------------------------------------------------------
  const selectColumn: ColumnDef<RfqWithItemCount> = {
    id: "select",
    header: ({ table }) => (
      <Checkbox
        checked={
          table.getIsAllPageRowsSelected() ||
          (table.getIsSomePageRowsSelected() && "indeterminate")
        }
        onCheckedChange={(value) => table.toggleAllPageRowsSelected(!!value)}
        aria-label="Select all"
        className="translate-y-0.5"
      />
    ),
    cell: ({ row }) => (
      <Checkbox
        checked={row.getIsSelected()}
        onCheckedChange={(value) => row.toggleSelected(!!value)}
        aria-label="Select row"
        className="translate-y-0.5"
      />
    ),
    size: 40,
    enableSorting: false,
    enableHiding: false,
  }

  // ----------------------------------------------------------------
  // 2) actions 컬럼 (Dropdown 메뉴)
  // ----------------------------------------------------------------
  const actionsColumn: ColumnDef<RfqWithItemCount> = {
    id: "actions",
    enableHiding: false,
    cell: function Cell({ row }) {
      const [isUpdatePending, startUpdateTransition] = React.useTransition()

      // Proceed 버튼 클릭 시 호출되는 함수
      const handleProceed = () => {
        const rfq = row.original
        const itemCount = Number(rfq.itemCount || 0)
        const attachCount = Number(rfq.attachCount || 0)

        // 아이템과 첨부파일이 모두 0보다 커야 진행 가능
        if (itemCount > 0 && attachCount > 0) {
          router.push(
            rfqType === RfqType.PURCHASE
              ? `/evcp/rfq/${rfq.rfqId}`
              : `/evcp/budgetary/${rfq.rfqId}`
          )
        } else {
          // 조건을 충족하지 않는 경우 토스트 알림 표시
          if (itemCount === 0 && attachCount === 0) {
            toast.error("아이템과 첨부파일을 먼저 추가해주세요.")
          } else if (itemCount === 0) {
            toast.error("아이템을 먼저 추가해주세요.")
          } else {
            toast.error("첨부파일을 먼저 추가해주세요.")
          }
        }
      }

      return (
        <DropdownMenu>
          <DropdownMenuTrigger asChild>
            <Button
              aria-label="Open menu"
              variant="ghost"
              className="flex size-8 p-0 data-[state=open]:bg-muted"
            >
              <Ellipsis className="size-4" aria-hidden="true" />
            </Button>
          </DropdownMenuTrigger>
          <DropdownMenuContent align="end" className="w-40">
            <DropdownMenuItem
              onSelect={() => setRowAction({ row, type: "update" })}
            >
              Edit
            </DropdownMenuItem>
            <DropdownMenuSeparator />
            <DropdownMenuItem onSelect={handleProceed}>
              {row.original.status ==="DRAFT"?"Proceed":"View Detail"}
              <DropdownMenuShortcut>↵</DropdownMenuShortcut>
            </DropdownMenuItem>
            <DropdownMenuSeparator />
            <DropdownMenuItem
              onSelect={() => setRowAction({ row, type: "delete" })}
            >
              Delete
              <DropdownMenuShortcut>⌘⌫</DropdownMenuShortcut>
            </DropdownMenuItem>
          </DropdownMenuContent>
        </DropdownMenu>
      )
    },
    size: 40,
  }

  // ----------------------------------------------------------------
  // 3) itemsColumn (아이템 개수 표시: 아이콘 + Badge)
  // ----------------------------------------------------------------
  const itemsColumn: ColumnDef<RfqWithItemCount> = {
    id: "items",
    header: "Items",
    cell: ({ row }) => {
      const rfq = row.original
      const itemCount = rfq.itemCount || 0

      const handleClick = () => {
        openItemsModal(rfq.rfqId)
      }

      return (
        <Button
          variant="ghost"
          size="sm"
          className="relative h-8 w-8 p-0 group"
          onClick={handleClick}
          aria-label={
            itemCount > 0 ? `View ${itemCount} items` : "Add items"
          }
        >
          <Package className="h-4 w-4 text-muted-foreground group-hover:text-primary transition-colors" />
          {itemCount > 0 && (
            <Badge
              variant="secondary"
              className="pointer-events-none absolute -top-1 -right-1 h-4 min-w-[1rem] p-0 text-[0.625rem] leading-none flex items-center justify-center"
            >
              {itemCount}
            </Badge>
          )}
          <span className="sr-only">
            {itemCount > 0 ? `${itemCount} Items` : "Add Items"}
          </span>
        </Button>
      )
    },
    enableSorting: false,
    size: 60,
  }

  // ----------------------------------------------------------------
  // 4) attachmentsColumn (첨부파일 개수 표시: 아이콘 + Badge)
  // ----------------------------------------------------------------
  const attachmentsColumn: ColumnDef<RfqWithItemCount> = {
    id: "attachments",
    header: "Attachments",
    cell: ({ row }) => {
      const fileCount = row.original.attachCount ?? 0

      const handleClick = () => {
        openAttachmentsSheet(row.original.rfqId)
      }

      return (
        <Button
          variant="ghost"
          size="sm"
          className="relative h-8 w-8 p-0 group"
          onClick={handleClick}
          aria-label={
            fileCount > 0 ? `View ${fileCount} files` : "Add files"
          }
        >
          <Paperclip className="h-4 w-4 text-muted-foreground group-hover:text-primary transition-colors" />
          {fileCount > 0 && (
            <Badge
              variant="secondary"
              className="pointer-events-none absolute -top-1 -right-1 h-4 min-w-[1rem] p-0 text-[0.625rem] leading-none flex items-center justify-center"
            >
              {fileCount}
            </Badge>
          )}
          <span className="sr-only">
            {fileCount > 0 ? `${fileCount} Files` : "Add Files"}
          </span>
        </Button>
      )
    },
    enableSorting: false,
    size: 60,
  }

  // ----------------------------------------------------------------
  // 5) 일반 컬럼들을 "그룹"별로 묶어 중첩 columns 생성
  // ----------------------------------------------------------------
  const groupMap: Record<string, ColumnDef<RfqWithItemCount>[]> = {}

  rfqsColumnsConfig.forEach((cfg) => {
    const groupName = cfg.group || "_noGroup"

    if (!groupMap[groupName]) {
      groupMap[groupName] = []
    }

    // child column 정의
    const childCol: ColumnDef<RfqWithItemCount> = {
      accessorKey: cfg.id,
      enableResizing: true,
      header: ({ column }) => (
        <DataTableColumnHeaderSimple column={column} title={cfg.label} />
      ),
      meta: {
        excelHeader: cfg.excelHeader,
        group: cfg.group,
        type: cfg.type,
      },
      cell: ({ row, cell }) => {
        if (cfg.id === "status") {
          const statusVal = row.original.status
          if (!statusVal) return null
          const Icon = getRFQStatusIcon(
            statusVal as "DRAFT" | "PUBLISHED" | "EVALUATION" | "AWARDED"
          )
          return (
            <div className="flex w-[6.25rem] items-center">
              <Icon className="mr-2 size-4 text-muted-foreground" aria-hidden="true" />
              <span className="capitalize">{statusVal}</span>
            </div>
          )
        }

        if (cfg.id === "createdAt" || cfg.id === "updatedAt") {
          const dateVal = cell.getValue() as Date
          return formatDate(dateVal, "KR")
        }

        return row.getValue(cfg.id) ?? ""
      },
    }

    groupMap[groupName].push(childCol)
  })

  // groupMap -> nestedColumns
  const nestedColumns: ColumnDef<RfqWithItemCount>[] = []
  Object.entries(groupMap).forEach(([groupName, colDefs]) => {
    if (groupName === "_noGroup") {
      nestedColumns.push(...colDefs)
    } else {
      nestedColumns.push({
        id: groupName,
        header: groupName,
        columns: colDefs,
      })
    }
  })

  // ----------------------------------------------------------------
  // 6) 최종 컬럼 배열
  // ----------------------------------------------------------------
  return [
    selectColumn,
    ...nestedColumns,
    attachmentsColumn, // 첨부파일
    actionsColumn,
    itemsColumn,       // 아이템
  ]
}