import { useQuery } from '@tanstack/react-query';
import { getFileViolations } from '@/app/actions/violationActions';

export interface ViolationFileDetail {
	id: number;
	url: string;
	page_url: string;
	page_title: string;
	host: string;
	status: string;
	created_date: string;
	file_id: string;
}

interface ViolationFile {
	violations: ViolationFileDetail[],
	total_elements: number,
	total_pages: number,
	current_page: number,
	page_size: number,
	has_next: boolean,
	has_previous: boolean
}

export const useFileViolations = (id: string, page: number, status?: string) => {
	return useQuery({
		queryKey: ['fileViolations', id, page, status],
		queryFn: () => {
			return getFileViolations(id, page, status)
		},
		select: (data: ViolationFile | null) => {
			if (!data) {
				return null
			}
			return data;
		},
		retry: false
	});
};