"use client"

import type { ColumnDef } from "@tanstack/react-table"
import type { ProcessedPub } from "contracts"
import type { PubsId } from "db/public"
import type * as React from "react"
import type { DataTableRowAction } from "ui/data-table-paged"

import { Checkbox } from "ui/checkbox"
import { DataTableColumnHeader } from "ui/data-table"
import { cn } from "utils"

import { dateFormatOptions } from "~/lib/dates"
import { getPubTitle } from "~/lib/pubs"

interface GetColumnsProps {
	setRowAction?: React.Dispatch<
		React.SetStateAction<DataTableRowAction<ProcessedPub<{ withPubType: true }>> | null>
	>
	disabledRows?: PubsId[]
}

export function getColumns({
	setRowAction,
	disabledRows = [],
}: GetColumnsProps): ColumnDef<ProcessedPub<{ withPubType: true }>>[] {
	return [
		{
			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"
					disabled={disabledRows.includes(row.original.id)}
				/>
			),
			enableSorting: false,
			enableHiding: false,
			maxSize: 1,
		},
		{
			header: ({ column }) => <DataTableColumnHeader column={column} title="Name" />,
			accessorKey: "name",
			cell: ({ row }) => {
				return (
					<div
						className={cn("flex items-center gap-2", {
							"text-muted-foreground": disabledRows.includes(row.original.id),
						})}
					>
						<span>{getPubTitle(row.original)}</span>
					</div>
				)
			},
			enableSorting: false,
		},
		{
			header: ({ column }) => <DataTableColumnHeader column={column} title="Updated At" />,
			accessorKey: "updatedAt",
			cell: ({ getValue, row }) => (
				<time
					dateTime={new Date().toString()}
					suppressHydrationWarning
					className={cn({
						"text-muted-foreground": disabledRows.includes(row.original.id),
					})}
				>
					{new Date(getValue<string>()).toLocaleString(undefined, dateFormatOptions)}
				</time>
			),
			enableSorting: true,
		},
	] as const satisfies ColumnDef<ProcessedPub<{ withPubType: true }>, unknown>[]
}