signoz/frontend/src/container/NewWidget/RightContainer/ContextLinks/useContextLinkModal.ts

import { Dispatch, SetStateAction, useState } from 'react';
import { ContextLinkProps, ContextLinksData } from 'types/api/dashboard/getAll';

interface ContextLinkModalProps {
	isModalOpen: boolean;
	selectedContextLink: ContextLinkProps | null;
	handleEditContextLink: (contextLink: ContextLinkProps) => void;
	handleAddContextLink: () => void;
	handleCancelModal: () => void;
	handleSaveContextLink: (newContextLink: ContextLinkProps) => void;
}

const useContextLinkModal = ({
	setContextLinks,
}: {
	setContextLinks: Dispatch<SetStateAction<ContextLinksData>>;
}): ContextLinkModalProps => {
	const [isModalOpen, setIsModalOpen] = useState(false);
	const [
		selectedContextLink,
		setSelectedContextLink,
	] = useState<ContextLinkProps | null>(null);

	const handleEditContextLink = (contextLink: ContextLinkProps): void => {
		setSelectedContextLink(contextLink);
		setIsModalOpen(true);
	};

	const handleAddContextLink = (): void => {
		setSelectedContextLink(null);
		setIsModalOpen(true);
	};

	const handleCancelModal = (): void => {
		setIsModalOpen(false);
		setSelectedContextLink(null);
	};

	const handleSaveContextLink = (newContextLink: ContextLinkProps): void => {
		setContextLinks((prev) => {
			const links = [...prev.linksData];
			const existing = links.filter((link) => link.id === newContextLink.id)[0];
			if (existing) {
				const idx = links.findIndex((link) => link.id === newContextLink.id);
				links[idx] = { ...existing, ...newContextLink };
				return { ...prev, linksData: links };
			}
			links.push(newContextLink);
			return { ...prev, linksData: links };
		});
		setIsModalOpen(false);
		setSelectedContextLink(null);
	};

	return {
		isModalOpen,
		selectedContextLink,
		handleEditContextLink,
		handleAddContextLink,
		handleCancelModal,
		handleSaveContextLink,
	};
};

export default useContextLinkModal;