import { getNeighbors } from '../../services/GraphQuery';
import { NeoNode, NeoRelationship } from '../../types';
import { showNormalToast } from '../../utils/Toasts';
export const handleGraphNodeClick = async (
elementId: string,
viewMode: string,
setNeoNodes: React.Dispatch<React.SetStateAction<NeoNode[]>>,
setNeoRels: React.Dispatch<React.SetStateAction<NeoRelationship[]>>,
setOpenGraphView: React.Dispatch<React.SetStateAction<boolean>>,
setViewPoint: React.Dispatch<React.SetStateAction<string>>,
setLoadingGraphView?: React.Dispatch<React.SetStateAction<boolean>>
) => {
if (setLoadingGraphView) {
setLoadingGraphView(true);
}
try {
const result = await getNeighbors(elementId);
if (result && result.data.data.nodes.length > 0) {
let { nodes } = result.data.data;
if (viewMode === 'Chunk') {
nodes = nodes.filter((node: NeoNode) => node.labels.length === 1 && node.properties.id !== null);
}
const nodeIds = new Set(nodes.map((node: NeoNode) => node.element_id));
const relationships = result.data.data.relationships.filter(
(rel: NeoRelationship) => nodeIds.has(rel.end_node_element_id) && nodeIds.has(rel.start_node_element_id)
);
setNeoNodes(nodes);
setNeoRels(relationships);
setOpenGraphView(true);
setViewPoint('chatInfoView');
} else {
showNormalToast('No nodes or relationships found for the selected node.');
}
} catch (error: any) {
console.error('Error fetching neighbors:', error);
} finally {
if (setLoadingGraphView) {
setLoadingGraphView(false);
}
}
};
