"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 type { FormSelectOption } from "@/app/dashboard/_components/form";

import { createUserWidgetColumns } from "./columns";
import {
  USER_WIDGET_STATUS_FILTER_OPTIONS,
  type UserWidgetFilters,
  type UserWidgetRow,
} from "./schema";
import { exportUserWidgetsCsv } from "./export-csv";
import { toastApiError } from "@/lib/toast-api-error";

const PRIORITY_SAVE_HREF = "/dashboard/master/user-widgets/priority";

const FILTER_KEYS = ["name", "categoryId", "status"] as const;
const TEXT_FILTER_KEYS = ["name"] as const;

type Props = {
  data: UserWidgetRow[];
  initialFilters: UserWidgetFilters;
  categoryOptions?: FormSelectOption[];
  pagination: ServerPaginationMeta;
};

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

export function UserWidgetsTable({
  data,
  initialFilters,
  categoryOptions = [],
  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 categoryFilterOptions = React.useMemo<FormSelectOption[]>(
    () => [{ value: "all", label: "All categories" }, ...categoryOptions],
    [categoryOptions],
  );

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

  const columns = React.useMemo(
    () => createUserWidgetColumns({ 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 UserWidgetFilters>(
    key: K,
    value: UserWidgetFilters[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: "User widget priority could not be updated.",
      });
      toast.success(response.message ?? "Priority successfully updated!");
      router.refresh();
    } catch (error) {
      toastApiError(error, "User widget priority could not be updated.");
    }
  };

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

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

      <DataTablePagination
        table={table}
        totalRows={totalCount}
        itemNoun="widget"
        idPrefix="user-widgets"
      />

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

function renderFilterCell(
  header: Header<UserWidgetRow, unknown>,
  filters: UserWidgetFilters,
  nameDraft: string,
  updateFilter: <K extends keyof UserWidgetFilters>(
    key: K,
    value: UserWidgetFilters[K],
  ) => void,
  setDraft: (key: keyof UserWidgetFilters & string, value: string) => void,
  categoryFilterOptions: FormSelectOption[],
) {
  const columnId = header.column.id;

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

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

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

  return null;
}