"use client";

import * as React from "react";

import {
  getCoreRowModel,
  useReactTable,
  type Header,
} from "@tanstack/react-table";
import { useRouter } from "next/navigation";
import { toast } from "sonner";

import {
  DataTablePagination,
  DataTableShell,
  useServerPagination,
  type ServerPaginationMeta,
} from "@/app/dashboard/_components/data-table";
import { useMasterListFilters } from "@/hooks/use-master-list-filters";
import { requestDashboardApi } from "@/lib/dashboard-api-client";
import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input";
import {
  Select,
  SelectContent,
  SelectItem,
  SelectTrigger,
  SelectValue,
} from "@/components/ui/select";
import { TableCell } from "@/components/ui/table";

import { createCommodityTicketColumns } from "./columns";
import {
  COMMODITY_TICKET_STATUS_OPTIONS,
  type CommodityTicketFilters,
  type CommodityTicketRow,
} from "./schema";
import { exportCommodityTicketCsv } from "./export-csv";
import { toastApiError } from "@/lib/toast-api-error";

const PRIORITY_SAVE_HREF = "/dashboard/master/commodity-ticket/priority";

const FILTER_KEYS = ["masterName", "parentName", "status"] as const;
const TEXT_FILTER_KEYS = ["masterName", "parentName"] as const;

type Props = {
  data: CommodityTicketRow[];
  initialFilters: CommodityTicketFilters;
  pagination: ServerPaginationMeta;
};

function buildPriorityMap(rows: CommodityTicketRow[]) {
  return Object.fromEntries(rows.map((row) => [row.id, String(row.priority)]));
}

export function CommodityTicketsTable({
  data,
  initialFilters,
  pagination: serverPagination,
}: Props) {
  const router = useRouter();
  const { filters, setFilter, drafts, setDraft } = useMasterListFilters(
    initialFilters,
    FILTER_KEYS,
    TEXT_FILTER_KEYS,
  );
  const prioritiesRef = React.useRef<Record<number, string>>(buildPriorityMap(data));
  const { pagination, onPaginationChange, totalCount } = useServerPagination(serverPagination);

  React.useEffect(() => {
    prioritiesRef.current = buildPriorityMap(data);
  }, [data]);

  const onPriorityChange = React.useCallback((id: number, priority: string) => {
    prioritiesRef.current = { ...prioritiesRef.current, [id]: priority };
  }, []);

  const columns = React.useMemo(
    () => createCommodityTicketColumns({ onPriorityChange }),
    [onPriorityChange],
  );

  const table = useReactTable({
    data,
    columns,
    state: { pagination },
    onPaginationChange,
    manualPagination: true,
    pageCount: serverPagination.pageCount,
    rowCount: totalCount,
    getCoreRowModel: getCoreRowModel(),
    getRowId: (row) => String(row.id),
  });

  const updateFilter = <K extends keyof CommodityTicketFilters>(
    key: K,
    value: CommodityTicketFilters[K],
  ) => {
    setFilter(key, value);
  };

  const handlePrioritySave = async () => {
    try {
      const response = await requestDashboardApi<{ message?: string }>({
        url: PRIORITY_SAVE_HREF,
        method: "POST",
        body: {
          priority: Object.fromEntries(
            Object.entries(prioritiesRef.current).map(([id, value]) => [
              id,
              Number.parseInt(value || "0", 10) || 0,
            ]),
          ),
        },
        fallbackError: "Commodity ticket priority could not be updated.",
      });
      toast.success(response.message ?? "Priority successfully updated!");
      router.refresh();
    } catch (error) {
      toastApiError(error, "Commodity ticket priority could not be updated.");
    }
  };

  return (
    <div className="space-y-4">
      <div className="flex justify-end">
        <Button
          type="button"
          variant="secondary"
          size="sm"
          onClick={() => exportCommodityTicketCsv(data)}
        >
          Export filtered
        </Button>
      </div>

      <DataTableShell
        table={table}
        columnCount={columns.length}
        emptyMessage="No commodity tickets match your filters."
        renderFilterCell={(header) =>
          renderFilterCell(header, filters, drafts, updateFilter, setDraft)
        }
      />

      <DataTablePagination
        table={table}
        totalRows={totalCount}
        itemNoun="commodity ticket"
        idPrefix="commodity-tickets"
      />

      <div className="flex justify-end border-t pt-4">
        <Button type="button" onClick={handlePrioritySave}>
          Update priority
        </Button>
      </div>
    </div>
  );
}

function renderFilterCell(
  header: Header<CommodityTicketRow, unknown>,
  filters: CommodityTicketFilters,
  drafts: Record<string, string>,
  updateFilter: <K extends keyof CommodityTicketFilters>(
    key: K,
    value: CommodityTicketFilters[K],
  ) => void,
  setDraft: (key: keyof CommodityTicketFilters & string, value: string) => void,
) {
  const columnId = header.column.id;

  if (columnId === "masterName") {
    return (
      <TableCell key={header.id} className="p-2">
        <Input
          value={drafts.masterName}
          onChange={(event) => setDraft("masterName", event.target.value.toUpperCase())}
          placeholder="Ticker"
          className="h-8 bg-background font-mono text-xs uppercase"
        />
      </TableCell>
    );
  }

  if (columnId === "parentName") {
    return (
      <TableCell key={header.id} className="p-2">
        <Input
          value={drafts.parentName}
          onChange={(event) => setDraft("parentName", event.target.value)}
          placeholder="Commodity name"
          className="h-8 bg-background text-xs"
        />
      </TableCell>
    );
  }

  if (columnId === "status") {
    return (
      <TableCell key={header.id} className="p-2">
        <Select value={filters.status} onValueChange={(value) => updateFilter("status", value)}>
          <SelectTrigger className="h-8 bg-background text-xs">
            <SelectValue placeholder="Status" />
          </SelectTrigger>
          <SelectContent>
            {COMMODITY_TICKET_STATUS_OPTIONS.map((option) => (
              <SelectItem key={option.value} value={option.value}>
                {option.label}
              </SelectItem>
            ))}
          </SelectContent>
        </Select>
      </TableCell>
    );
  }

  return null;
}
