"use client"

import type { ColumnDef, FilterFn } from "@tanstack/react-table"
import { Pencil } from "lucide-react"
import Link from "next/link"

import { Badge } from "@/components/ui/badge"
import { Button } from "@/components/ui/button"
import {
  Tooltip,
  TooltipContent,
  TooltipTrigger,
} from "@/components/ui/tooltip"
import { LIST_HREF } from "./constants"
import type { IsinAssetRow } from "./schema"

const includesStringFilter: FilterFn<IsinAssetRow> = (row, columnId, value) => {
  if (!value || value === "all") return true
  return String(row.getValue(columnId) ?? "")
    .toLowerCase()
    .includes(String(value).toLowerCase())
}

const inPortfolioFilter: FilterFn<IsinAssetRow> = (row, _columnId, value) => {
  if (!value || value === "all") return true
  return row.original.inPortfolio === value
}

export function createIsinAssetColumns(): ColumnDef<IsinAssetRow>[] {
  return [
    {
      accessorKey: "isin",
      header: "ISIN",
      filterFn: includesStringFilter,
      cell: ({ row }) => (
        <span className="font-mono text-sm">{row.original.isin}</span>
      ),
    },
    {
      accessorKey: "assetClassCode",
      header: "Asset class code",
      filterFn: includesStringFilter,
    },
    {
      accessorKey: "inPortfolio",
      header: "In portfolio",
      filterFn: inPortfolioFilter,
      cell: ({ row }) => (
        <Badge variant="outline" className="font-normal">
          {row.original.inPortfolioLabel}
        </Badge>
      ),
    },
    {
      id: "options",
      header: () => <span className="block w-full text-right">Options</span>,
      enableColumnFilter: false,
      cell: ({ row }) => (
        <div className="flex justify-end">
          <Tooltip>
            <TooltipTrigger asChild>
              <Button variant="secondary" size="icon-sm" className="size-7" asChild>
                <Link href={`${LIST_HREF}/${encodeURIComponent(row.original.id)}`}>
                  <Pencil className="size-3.5" />
                </Link>
              </Button>
            </TooltipTrigger>
            <TooltipContent>Update</TooltipContent>
          </Tooltip>
        </div>
      ),
      enableSorting: false,
    },
  ]
}