"use client"
import * as React from "react"
import { type DataTableRowAction } from "@/types/table"
import { type ColumnDef } from "@tanstack/react-table"
import { formatDate } from "@/lib/utils"
import { DataTableColumnHeaderSimple } from "@/components/data-table/data-table-column-simple-header"
import { BiddingProjects } from "@/db/schema"
import { bidProjectsColumnsConfig } from "@/config/bidProjectsColumnsConfig"
import { Button } from "@/components/ui/button"
import { ListFilter, Edit } from "lucide-react" // Import an icon for the button

interface GetColumnsProps {
  setRowAction: React.Dispatch<React.SetStateAction<DataTableRowAction<BiddingProjects> | null>>
}

/**
 * tanstack table 컬럼 정의 (중첩 헤더 버전)
 */
export function getColumns({ setRowAction }: GetColumnsProps): ColumnDef<BiddingProjects>[] {
  // ----------------------------------------------------------------
  // 3) 일반 컬럼들을 "그룹"별로 묶어 중첩 columns 생성
  // ----------------------------------------------------------------
  // 3-1) groupMap: { [groupName]: ColumnDef<BiddingProjects>[] }
  const groupMap: Record<string, ColumnDef<BiddingProjects>[]> = {}
  bidProjectsColumnsConfig.forEach((cfg) => {
    // 만약 group가 없으면 "_noGroup" 처리
    const groupName = cfg.group || "_noGroup"
    if (!groupMap[groupName]) {
      groupMap[groupName] = []
    }
    // child column 정의
    const childCol: ColumnDef<BiddingProjects> = {
      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 === "createdAt" || cfg.id === "updatedAt") {
          const dateVal = cell.getValue() as Date
          return formatDate(dateVal)
        }
        return row.getValue(cfg.id) ?? ""
      },
    }
    groupMap[groupName].push(childCol)
  })
  
  // ----------------------------------------------------------------
  // 3-2) groupMap에서 실제 상위 컬럼(그룹)을 만들기
  // ----------------------------------------------------------------
  const nestedColumns: ColumnDef<BiddingProjects>[] = []
  // 순서를 고정하고 싶다면 group 순서를 미리 정의하거나 sort해야 함
  // 여기서는 그냥 Object.entries 순서
  Object.entries(groupMap).forEach(([groupName, colDefs]) => {
    if (groupName === "_noGroup") {
      // 그룹 없음 → 그냥 최상위 레벨 컬럼
      nestedColumns.push(...colDefs)
    } else {
      // 상위 컬럼
      nestedColumns.push({
        id: groupName,
        header: groupName, // "Basic Info", "Metadata" 등
        columns: colDefs,
      })
    }
  })
  
  // Add action column
  const actionColumn: ColumnDef<BiddingProjects> = {
    id: "actions",
    header: "Actions",
    cell: ({ row }) => {
      const project = row.original
      const isTopType = project.pjtType === "TOP"
      
      return (
        <div className="flex items-center gap-1">
          <Button
            variant="ghost"
            size="sm"
            className="flex items-center gap-1"
            onClick={() => {
              setRowAction({ row, type: "view-series" })
            }}
          >
            <ListFilter className="h-4 w-4" />
            시리즈 보기
          </Button>
          
          {isTopType && (
            <Button
              variant="ghost"
              size="sm"
              className="flex items-center gap-1"
              onClick={() => {
                setRowAction({ row, type: "update" })
              }}
            >
              <Edit className="h-4 w-4" />
              수정
            </Button>
          )}
        </div>
      )
    },
  }
  
  // ----------------------------------------------------------------
  // 4) 최종 컬럼 배열: nestedColumns + actions
  // ----------------------------------------------------------------
  return [
    ...nestedColumns,
    actionColumn, // Add the action column
  ]
}