Checkmate/client/src/Hooks/monitorHooks.js

import { useEffect, useState } from "react";
import { networkService } from "../main";
import { createToast } from "../Utils/toastUtils";
import { useTheme } from "@emotion/react";
import { useMonitorUtils } from "./useMonitorUtils";
import { useNavigate } from "react-router-dom";
import { useTranslation } from "react-i18next";

const useFetchMonitorsWithSummary = ({ types, monitorUpdateTrigger }) => {
	const [isLoading, setIsLoading] = useState(false);
	const [monitors, setMonitors] = useState(undefined);
	const [monitorsSummary, setMonitorsSummary] = useState(undefined);
	const [networkError, setNetworkError] = useState(false);

	useEffect(() => {
		const fetchMonitors = async () => {
			try {
				setIsLoading(true);
				const res = await networkService.getMonitorsWithSummaryByTeamId({
					types,
				});
				const { monitors, summary } = res?.data?.data ?? {};
				setMonitors(monitors);
				setMonitorsSummary(summary);
			} catch (error) {
				console.error(error);
				setNetworkError(true);
				createToast({
					body: error.message,
				});
			} finally {
				setIsLoading(false);
			}
		};
		fetchMonitors();
	}, [types, monitorUpdateTrigger]);
	return [monitors, monitorsSummary, isLoading, networkError];
};

const useFetchMonitorsWithChecks = ({
	types,
	limit,
	page,
	rowsPerPage,
	filter,
	field,
	order,
	monitorUpdateTrigger,
}) => {
	const [isLoading, setIsLoading] = useState(false);
	const [count, setCount] = useState(undefined);
	const [monitors, setMonitors] = useState(undefined);
	const [networkError, setNetworkError] = useState(false);

	const theme = useTheme();
	const { getMonitorWithPercentage } = useMonitorUtils();
	useEffect(() => {
		const fetchMonitors = async () => {
			try {
				setIsLoading(true);
				const res = await networkService.getMonitorsWithChecksByTeamId({
					limit,
					types,
					page,
					rowsPerPage,
					filter,
					field,
					order,
				});
				const { count, monitors } = res?.data?.data ?? {};
				const mappedMonitors = monitors.map((monitor) =>
					getMonitorWithPercentage(monitor, theme)
				);
				setMonitors(mappedMonitors);
				setCount(count?.monitorsCount ?? 0);
			} catch (error) {
				console.error(error);
				setNetworkError(true);
				createToast({
					body: error.message,
				});
			} finally {
				setIsLoading(false);
			}
		};
		fetchMonitors();
	}, [
		field,
		filter,
		getMonitorWithPercentage,
		limit,
		order,
		page,
		rowsPerPage,
		theme,
		types,
		monitorUpdateTrigger,
	]);
	return [monitors, count, isLoading, networkError];
};

const useFetchMonitorsByTeamId = ({
	types,
	limit,
	page,
	rowsPerPage,
	filter,
	field,
	order,
	checkOrder,
	normalize,
	status,
	updateTrigger,
}) => {
	const [isLoading, setIsLoading] = useState(false);
	const [monitors, setMonitors] = useState(undefined);
	const [summary, setSummary] = useState(undefined);
	const [networkError, setNetworkError] = useState(false);

	useEffect(() => {
		const fetchMonitors = async () => {
			try {
				setIsLoading(true);
				const res = await networkService.getMonitorsByTeamId({
					limit,
					types,
					page,
					rowsPerPage,
					filter,
					field,
					order,
					checkOrder,
					status,
					normalize,
				});
				if (res?.data?.data?.filteredMonitors) {
					setMonitors(res.data.data.filteredMonitors);
					setSummary(res.data.data.summary);
				}
			} catch (error) {
				setNetworkError(true);
				createToast({
					body: error.message,
				});
			} finally {
				setIsLoading(false);
			}
		};
		fetchMonitors();
	}, [
		types,
		limit,
		page,
		rowsPerPage,
		filter,
		field,
		order,
		updateTrigger,
		checkOrder,
		normalize,
		status,
	]);
	return [monitors, summary, isLoading, networkError];
};

const useFetchStatsByMonitorId = ({
	monitorId,
	sortOrder,
	limit,
	dateRange,
	numToDisplay,
	normalize,
}) => {
	const [monitor, setMonitor] = useState(undefined);
	const [audits, setAudits] = useState(undefined);
	const [isLoading, setIsLoading] = useState(true);
	const [networkError, setNetworkError] = useState(false);
	useEffect(() => {
		const fetchMonitor = async () => {
			try {
				setIsLoading(true);
				const res = await networkService.getStatsByMonitorId({
					monitorId: monitorId,
					sortOrder,
					limit,
					dateRange,
					numToDisplay,
					normalize,
				});
				setMonitor(res?.data?.data ?? undefined);
				setAudits(res?.data?.data?.checks?.[0]?.audits ?? undefined);
			} catch (error) {
				setNetworkError(true);
				createToast({ body: error.message });
			} finally {
				setIsLoading(false);
			}
		};
		fetchMonitor();
	}, [monitorId, dateRange, numToDisplay, normalize, sortOrder, limit]);
	return [monitor, audits, isLoading, networkError];
};

const useFetchMonitorById = ({ monitorId, setMonitor, updateTrigger }) => {
	const [isLoading, setIsLoading] = useState(true);
	useEffect(() => {
		const fetchMonitor = async () => {
			try {
				setIsLoading(true);
				const res = await networkService.getMonitorById({ monitorId: monitorId });
				setMonitor(res.data.data);
			} catch (error) {
				createToast({ body: error.message });
			} finally {
				setIsLoading(false);
			}
		};
		fetchMonitor();
	}, [monitorId, setMonitor, updateTrigger]);
	return [isLoading];
};

const useFetchHardwareMonitorById = ({ monitorId, dateRange }) => {
	const [isLoading, setIsLoading] = useState(true);
	const [networkError, setNetworkError] = useState(false);
	const [monitor, setMonitor] = useState(undefined);

	useEffect(() => {
		const fetchMonitor = async () => {
			try {
				if (!monitorId) {
					return { monitor: undefined, isLoading: false, networkError: undefined };
				}
				const response = await networkService.getHardwareDetailsByMonitorId({
					monitorId: monitorId,
					dateRange: dateRange,
				});
				setMonitor(response.data.data);
			} catch (error) {
				setNetworkError(true);
			} finally {
				setIsLoading(false);
			}
		};
		fetchMonitor();
	}, [monitorId, dateRange]);
	return [monitor, isLoading, networkError];
};

const useFetchUptimeMonitorById = ({ monitorId, dateRange, trigger }) => {
	const [networkError, setNetworkError] = useState(false);
	const [isLoading, setIsLoading] = useState(true);
	const [monitor, setMonitor] = useState(undefined);
	const [monitorStats, setMonitorStats] = useState(undefined);
	useEffect(() => {
		const fetchMonitors = async () => {
			try {
				const res = await networkService.getUptimeDetailsById({
					monitorId: monitorId,
					dateRange: dateRange,
					normalize: true,
				});
				const { monitorData, monitorStats } = res?.data?.data ?? {};
				setMonitor(monitorData);
				setMonitorStats(monitorStats);
			} catch (error) {
				setNetworkError(true);
				createToast({ body: error.message });
			} finally {
				setIsLoading(false);
			}
		};
		fetchMonitors();
	}, [dateRange, monitorId, trigger]);
	return [monitor, monitorStats, isLoading, networkError];
};

const useCreateMonitor = () => {
	const [isLoading, setIsLoading] = useState(false);
	const navigate = useNavigate();
	const createMonitor = async ({ monitor, redirect }) => {
		try {
			setIsLoading(true);
			await networkService.createMonitor({ monitor });
			createToast({ body: "Monitor created successfully!" });
			if (redirect) {
				navigate(redirect);
			}
		} catch (error) {
			createToast({ body: "Failed to create monitor." });
		} finally {
			setIsLoading(false);
		}
	};
	return [createMonitor, isLoading];
};

const useDeleteMonitor = () => {
	const [isLoading, setIsLoading] = useState(false);
	const navigate = useNavigate();
	const deleteMonitor = async ({ monitor, redirect }) => {
		try {
			setIsLoading(true);
			await networkService.deleteMonitorById({ monitorId: monitor._id });
			createToast({ body: "Monitor deleted successfully!" });
			if (redirect) {
				navigate(redirect);
			}
		} catch (error) {
			createToast({ body: "Failed to delete monitor." });
		} finally {
			setIsLoading(false);
		}
	};
	return [deleteMonitor, isLoading];
};

const useUpdateMonitor = () => {
	const [isLoading, setIsLoading] = useState(false);
	const navigate = useNavigate();
	const updateMonitor = async ({ monitor, redirect }) => {
		try {
			setIsLoading(true);
			const updatedFields = {
				name: monitor.name,
				description: monitor.description,
				interval: monitor.interval,
				notifications: monitor.notifications,
				matchMethod: monitor.matchMethod,
				expectedValue: monitor.expectedValue,
				ignoreTlsErrors: monitor.ignoreTlsErrors,
				jsonPath: monitor.jsonPath,
				...(monitor.type === "port" && { port: monitor.port }),
				...(monitor.type === "hardware" && {
					thresholds: monitor.thresholds,
					secret: monitor.secret,
				}),
			};
			await networkService.updateMonitor({
				monitorId: monitor._id,
				updatedFields,
			});

			createToast({ body: "Monitor updated successfully!" });
			if (redirect) {
				navigate(redirect);
			}
		} catch (error) {
			createToast({ body: "Failed to update monitor." });
		} finally {
			setIsLoading(false);
		}
	};
	return [updateMonitor, isLoading];
};

const usePauseMonitor = () => {
	const [isLoading, setIsLoading] = useState(false);
	const [error, setError] = useState(undefined);
	const pauseMonitor = async ({ monitorId, triggerUpdate }) => {
		try {
			setIsLoading(true);
			const res = await networkService.pauseMonitorById({ monitorId });
			createToast({
				body: res.data.data.isActive
					? "Monitor resumed successfully"
					: "Monitor paused successfully",
			});
			triggerUpdate();
		} catch (error) {
			setError(error);
		} finally {
			setIsLoading(false);
		}
	};

	return [pauseMonitor, isLoading, error];
};

const useAddDemoMonitors = () => {
	const [isLoading, setIsLoading] = useState(false);
	const { t } = useTranslation();
	const addDemoMonitors = async () => {
		try {
			setIsLoading(true);
			await networkService.addDemoMonitors();
			createToast({ body: t("settingsDemoMonitorsAdded") });
		} catch (error) {
			createToast({ body: t("settingsFailedToAddDemoMonitors") });
		} finally {
			setIsLoading(false);
		}
	};
	return [addDemoMonitors, isLoading];
};

const useDeleteAllMonitors = () => {
	const [isLoading, setIsLoading] = useState(false);
	const { t } = useTranslation();
	const deleteAllMonitors = async () => {
		try {
			setIsLoading(true);
			await networkService.deleteAllMonitors();
			createToast({ body: t("settingsMonitorsDeleted") });
		} catch (error) {
			createToast({ body: t("settingsFailedToDeleteMonitors") });
		} finally {
			setIsLoading(false);
		}
	};
	return [deleteAllMonitors, isLoading];
};

const useDeleteMonitorStats = () => {
	const { t } = useTranslation();
	const [isLoading, setIsLoading] = useState(false);
	const deleteMonitorStats = async () => {
		setIsLoading(true);
		try {
			await networkService.deleteChecksByTeamId();
			createToast({ body: t("settingsStatsCleared") });
		} catch (error) {
			createToast({ body: t("settingsFailedToClearStats") });
		} finally {
			setIsLoading(false);
		}
	};

	return [deleteMonitorStats, isLoading];
};

const useCreateBulkMonitors = () => {
	const [isLoading, setIsLoading] = useState(false);

	const createBulkMonitors = async (file, user) => {
		setIsLoading(true);

		const formData = new FormData();
		formData.append("csvFile", file);

		try {
			const response = await networkService.createBulkMonitors(formData);
			return [true, response.data, null]; // [success, data, error]
		} catch (err) {
			const errorMessage = err?.response?.data?.msg || err.message;
			return [false, null, errorMessage];
		} finally {
			setIsLoading(false);
		}
	};

	return [createBulkMonitors, isLoading];
};

export {
	useFetchMonitorsWithSummary,
	useFetchMonitorsWithChecks,
	useFetchMonitorsByTeamId,
	useFetchStatsByMonitorId,
	useFetchMonitorById,
	useFetchUptimeMonitorById,
	useFetchHardwareMonitorById,
	useCreateMonitor,
	useDeleteMonitor,
	useUpdateMonitor,
	usePauseMonitor,
	useAddDemoMonitors,
	useDeleteAllMonitors,
	useDeleteMonitorStats,
	useCreateBulkMonitors,
};