import { useQuery } from "@tanstack/react-query";
import { fetch } from "@marzneshin/utils";
import { UserType } from "../types";
import {
    FetchEntityReturn,
    UseEntityQueryProps,
    SortableEntitySortQueryProps,
    EntityQueryKeyType
} from "@marzneshin/components";

export type SortUserBy = "username" | "used_traffic" | "data_limit" | "expire" | "created_at"

interface UserResponse extends UserType {
    service_ids: number[]
    services: any[]
}

export async function fetchUsers({ queryKey }: EntityQueryKeyType): FetchEntityReturn<UserType> {
    return fetch(`/users`, {
        query: {
            page: queryKey[1],
            size: queryKey[2],
            username: queryKey[3],
            descending: queryKey[5],
            order_by: queryKey[4]
        }
    }).then((result) => {
        const users: UserType[] = result.items.map((fetchedUser: UserResponse) => {
            const user: UserType = fetchedUser;
            user.services = fetchedUser.service_ids
            return user
        })
        return {
            entity: users,
            pageCount: result.pages
        };
    });
}

export const UsersQueryFetchKey = "users";

export const useUsersQuery = ({
    page, size, search = "", sortBy = "created_at", desc = false
}: UseEntityQueryProps & SortableEntitySortQueryProps) => {
    return useQuery({
        queryKey: [UsersQueryFetchKey, page, size, search, sortBy, desc],
        queryFn: fetchUsers,
        initialData: { entity: [], pageCount: 0 }
    })
}