DataLineageVisualizer/script.js
2024-05-12 17:24:49 +02:00

99 lines
3 KiB
JavaScript

import graphology from 'https://cdn.jsdelivr.net/npm/graphology/+esm';
import Sigma from 'https://cdn.jsdelivr.net/npm/sigma/+esm';
import forceAtlas2 from 'https://cdn.jsdelivr.net/npm/graphology-layout-forceatlas2/+esm';
document.addEventListener('DOMContentLoaded', function() {
main();
});
async function main(){
try {
const network_graph_json = await getNetworkGraphJson();
let graph = createGraph(network_graph_json);
graph = addLayout(graph);
console.log("Graph data:", network_graph_json);
console.log("Graph object:", graph);
const sigmaInstance = new Sigma(graph, document.getElementById('container'));
add_drawer(sigmaInstance, graph);
} catch (error) {
console.error('Failed to load or create the graph:', error);
}
}
function createGraph(network_graph_json){
let graph = new graphology.Graph();
for (const node of network_graph_json.nodes) {
graph.addNode(
node.key,
{
label: node.label,
tag: node.tag,
URL: node.URL,
cluster: node.cluster,
size: network_graph_json.tags.find((el) => el.key === node.tag).size,
color: network_graph_json.tags.find((el) => el.key === node.tag).color,
}
);
}
for (const edge of network_graph_json.edges) {
graph.addEdge(edge.source, edge.target);
}
return graph;
}
function addLayout(graph){
graph.forEachNode((node) => {
if (!graph.getNodeAttribute(node, 'x') && !graph.getNodeAttribute(node, 'y')) {
graph.setNodeAttribute(node, 'x', Math.random());
graph.setNodeAttribute(node, 'y', Math.random());
}
});
forceAtlas2.assign(graph, {
iterations: 100,
settings: {
gravity: 0.5,
scalingRatio: 2.0
}
});
return graph;
}
async function getNetworkGraphJson(){
const response = await fetch('./assets/network_graph_json.json');
if (!response.ok) {
throw new Error('Network response was not ok');
}
return await response.json();
}
function add_drawer(sigmaInstance, graph){
sigmaInstance.on('clickNode', ({node}) => {
const attributes = graph.getNodeAttributes(node);
const drawer = document.getElementById('nodeDrawer');
const drawerText = document.getElementById('drawerText');
drawerText.innerHTML = `Node Label: ${attributes.label}<br>Node Type: ${attributes.tag}<br>More info: <a href="${attributes.URL}">Click here</a>`;
drawer.style.width = "30%";
drawer.style.display = "block";
});
const close = document.getElementsByClassName("close")[0];
close.onclick = function() {
const drawer = document.getElementById('nodeDrawer');
drawer.style.display = "none";
}
window.onclick = function(event) {
const drawer = document.getElementById('nodeDrawer');
if (event.target === drawer) {
drawer.style.display = "none";
}
}
}