import { useEndpoint } from '@rocket.chat/ui-contexts';
import { useQuery } from '@tanstack/react-query';

import type { Period } from '../../../../components/dashboards/periods';
import { getPeriodRange } from '../../../../components/dashboards/periods';

type UseMessageOriginsOptions = { period: Period['key'] };

export const useMessageOrigins = ({ period }: UseMessageOriginsOptions) => {
	const getMessageOrigins = useEndpoint('GET', '/v1/engagement-dashboard/messages/origin');

	return useQuery({
		queryKey: ['admin/engagement-dashboard/messages/origins', { period }],

		queryFn: async () => {
			const { start, end } = getPeriodRange(period);

			const response = await getMessageOrigins({
				start: start.toISOString(),
				end: end.toISOString(),
			});

			return response
				? {
						...response,
						start,
						end,
					}
				: undefined;
		},

		refetchInterval: 5 * 60 * 1000,
		throwOnError: true,
	});
};