'use client';
import { useState } from "react";
import { useGraphQlCompanyList } from "./queries";
import { LivingSpaceCompanyDataTable } from "./data-table";

const PageUsersCompanyTableSection = (
    {
        selectedCompanyIDS,
        appendCompanyID,
        removeCompanyID,
        defaultSelection,
        setDefaultSelection
    }: {
        selectedCompanyIDS: string[],
        appendCompanyID: (id: string) => void,
        removeCompanyID: (id: string) => void,
        defaultSelection: string,
        setDefaultSelection: (id: string) => void
    }
) => {

    const [page, setPage] = useState(1);
    const [limit, setLimit] = useState(10);
    const [sort, setSort] = useState({ createdAt: 'desc' });
    const [filters, setFilters] = useState({});
    const { data, isLoading, error, refetch } = useGraphQlCompanyList({ limit, skip: (page - 1) * limit, sort, filters });

    const handlePageChange = (newPage: number) => { setPage(newPage) };
    const handlePageSizeChange = (newSize: number) => { setLimit(newSize); setPage(1) };

    if (isLoading) { return <div className="flex items-center justify-center p-8">Loading...</div> }
    if (error) { return <div className="flex items-center justify-center p-8 text-red-500">Error loading users</div> }

    return < LivingSpaceCompanyDataTable
        data={data?.data || []} totalCount={data?.totalCount || 0} currentPage={page} pageSize={limit} onPageChange={handlePageChange} onPageSizeChange={handlePageSizeChange}
        refetchTable={refetch} selectedCompanyIDS={selectedCompanyIDS} appendCompanyID={appendCompanyID} removeCompanyID={removeCompanyID} defaultSelection={defaultSelection} setDefaultSelection={setDefaultSelection}
    />

}

export default PageUsersCompanyTableSection;
