aboutsummaryrefslogtreecommitdiff
path: root/react-frontend/src/components/Visualization.js
blob: 1975e864fb611037aeb3580b3b04f00311f1b1fd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
// JS module imports
import { useEffect, useRef, useState } from "react";
import uuid from 'react-uuid';
import Graph from 'vis-react';

// CSS imports
import '../css/Canvas.css';

/**
 * This function renders and mantains thhe canvas.
 * @param {Object} props The props for the canvas.
 * @returns {import("react").HtmlHTMLAttributes} The canvas to be retured.
 */
function Visualization(props) {
  const options = {
    edges: {
      color: "#ffffff"
    }
  };
  const events = {
    select: () => event => props.setSelected(event.nodes[0])
  };

  const [graphState, setGraphState] = useState({
    nodes: [],
    edges: []
  });
  const getNodes = () => {
    let nodes = [];
    props.data.forEach(hub => {
      if (hub.followers) {
        nodes.push({
          id: hub.id,
          label: hub.name,
          size: hub.suspicionScore
        });
      }
    });
    return nodes;
  }
  const getEdges = () => {
    let edges = []
    props.data.forEach(hub => {
      hub.followers.forEach(follower => {
        edges.push({
          from: hub.id,
          to: follower.id
        });
      });
    });
    return edges;
  }

  // Hooks to update graph state
  useEffect(() => setGraphState({nodes: getNodes(), edges: getEdges()}), [JSON.stringify(props.data)]);

  return (
    <div className="Map-canvas">
      <Graph
        key={uuid()} 
        graph={graphState}
        options={options}
        events={events}>
      </Graph>
    </div>
  );
}

export default Visualization;