ahriman/frontend/src/hooks/useTableState.ts

import { useState } from "react";
import type { GridFilterModel } from "@mui/x-data-grid";
import { useLocalStorage } from "hooks/useLocalStorage";

export type DialogType = "dashboard" | "add" | "rebuild" | "keyImport";

export interface UseTableStateResult {
    selectionModel: string[];
    setSelectionModel: (model: string[]) => void;

    dialogOpen: DialogType | null;
    setDialogOpen: (dialog: DialogType | null) => void;
    selectedPackage: string | null;
    setSelectedPackage: (base: string | null) => void;

    paginationModel: { pageSize: number; page: number };
    setPaginationModel: (model: { pageSize: number; page: number }) => void;
    columnVisibility: Record<string, boolean>;
    setColumnVisibility: (model: Record<string, boolean>) => void;
    filterModel: GridFilterModel;
    setFilterModel: (model: GridFilterModel) => void;
    searchText: string;
    setSearchText: (text: string) => void;
}

export function useTableState(): UseTableStateResult {
    const [selectionModel, setSelectionModel] = useState<string[]>([]);
    const [dialogOpen, setDialogOpen] = useState<DialogType | null>(null);
    const [selectedPackage, setSelectedPackage] = useState<string | null>(null);
    const [searchText, setSearchText] = useState("");

    const [paginationModel, setPaginationModel] = useLocalStorage("ahriman-packages-pagination", {
        pageSize: 10,
        page: 0,
    });
    const [columnVisibility, setColumnVisibility] = useLocalStorage<Record<string, boolean>>(
        "ahriman-packages-columns",
        { groups: false, licenses: false, packager: false },
    );
    const [filterModel, setFilterModel] = useLocalStorage<GridFilterModel>(
        "ahriman-packages-filters",
        { items: [] },
    );

    return {
        selectionModel,
        setSelectionModel,

        dialogOpen,
        setDialogOpen,
        selectedPackage,
        setSelectedPackage,

        paginationModel,
        setPaginationModel,
        columnVisibility,
        setColumnVisibility,
        filterModel,
        setFilterModel,
        searchText,
        setSearchText,
    };
}