'use client';

import {
    ColumnDef,
    ColumnFiltersState,
    flexRender,
    getCoreRowModel,
    useReactTable,
} from '@tanstack/react-table';

import { DataTablePagination } from '@/Components/DataTablePagination';
import ShowIf from '@/Components/ShowIf';
import {
    Table,
    TableBody,
    TableCell,
    TableHead,
    TableHeader,
    TableRow,
} from '@/Components/ui/table';
import DrugCreateForm from '@/Pages/Drug/DrugCreateForm';
import DrugEditForm from '@/Pages/Drug/DrugEditForm';
import DrugSearchInput from '@/Pages/Drug/DrugSearchInput';
import { Drug } from '@/types';
import { router, usePage } from '@inertiajs/react';
import React, { useState } from 'react';

interface DataTableProps<TData, TValue> {
    columns: ColumnDef<TData, TValue>[];
    data: TData[];
}

interface DrugsTableProps<TData, TValue> extends DataTableProps<TData, TValue> {
    rowCount: number;
    pageSize: number;
    pageCount: number;
    canEdit: boolean;
    canCreate: boolean;
}

export function DrugsTable<TData, TValue>({
    columns,
    data,
    rowCount,
    pageSize,
    pageCount,
    canEdit,
    canCreate,
}: DrugsTableProps<TData, TValue>) {
    const { filters } = usePage().props;
    const [search, setSearch] = useState(filters?.search || '');
    const [page, setPage] = useState(1);
    const [columnFilters, setColumnFilters] =
        React.useState<ColumnFiltersState>([]);

    const table = useReactTable({
        data,
        columns,
        getCoreRowModel: getCoreRowModel(),
        onColumnFiltersChange: setColumnFilters,
        manualPagination: true,
        manualFiltering: true,
        rowCount,
        initialState: {
            pagination: { pageSize, pageIndex: 0 },
        },
        state: {
            columnFilters,
        },
    });

    const handlePageChange = (pageIndex: number) => {
        setPage(pageIndex);

        // Immediately send search request
        router.get(
            route('admin-drugs'),
            { search, page: pageIndex },
            {
                preserveState: true,
                preserveScroll: true,
                replace: true, // Prevent duplicate history entries
            },
        );
    };

    return (
        <div>
            <div className="flex w-full items-baseline justify-between gap-4 py-4">
                <DrugSearchInput search={search} setSearch={setSearch} />
                <ShowIf condition={canCreate}>
                    <DrugCreateForm />
                </ShowIf>
            </div>
            <div className="rounded-md border bg-white">
                <Table>
                    <TableHeader>
                        {table.getHeaderGroups().map((headerGroup) => (
                            <TableRow key={headerGroup.id}>
                                {headerGroup.headers.map((header) => {
                                    return (
                                        <TableHead key={header.id}>
                                            {header.isPlaceholder
                                                ? null
                                                : flexRender(
                                                      header.column.columnDef
                                                          .header,
                                                      header.getContext(),
                                                  )}
                                        </TableHead>
                                    );
                                })}
                            </TableRow>
                        ))}
                    </TableHeader>
                    <TableBody>
                        {table.getRowModel().rows?.length ? (
                            table.getRowModel().rows.map((row) => (
                                <DrugEditForm
                                    canEdit={canEdit}
                                    key={row.id}
                                    drug={row.original as Drug}
                                >
                                    <TableRow
                                        className="hover:cursor-pointer"
                                        data-state={
                                            row.getIsSelected() && 'selected'
                                        }
                                    >
                                        {row.getVisibleCells().map((cell) => (
                                            <TableCell key={cell.id}>
                                                {flexRender(
                                                    cell.column.columnDef.cell,
                                                    cell.getContext(),
                                                )}
                                            </TableCell>
                                        ))}
                                    </TableRow>
                                </DrugEditForm>
                            ))
                        ) : (
                            <TableRow>
                                <TableCell
                                    colSpan={columns.length}
                                    className="h-24 text-center"
                                >
                                    Nessun risultato.
                                </TableCell>
                            </TableRow>
                        )}
                    </TableBody>
                </Table>
            </div>
            <div className="py-4">
                <DataTablePagination
                    table={table}
                    changePage={handlePageChange}
                    currentPage={page}
                    pageCount={pageCount}
                />
            </div>
        </div>
    );
}