hopglass/lib/main.js

define(["moment", "router", "leaflet", "gui", "numeral"],
  function (moment, Router, L, GUI, numeral) {
    return function (config) {
      function handleData(data) {
        var dataNodes = {};
        dataNodes.nodes = [];
        dataNodes.nodeIds = [];
        var dataGraph = {};
        dataGraph.batadv = {};
        dataGraph.batadv.nodes = [];
        dataGraph.batadv.links = [];


        function rearrangeLinks(d) {
          d.source += dataGraph.batadv.nodes.length;
          d.target += dataGraph.batadv.nodes.length;
        }

        for (var i = 0; i < data.length; ++i) {
          var vererr;
          if (i % 2) {
            if (data[i].version !== 1) {
              vererr = "Unsupported graph version: " + data[i].version;
              console.log(vererr); //silent fail
            } else {
              data[i].batadv.links.forEach(rearrangeLinks);
              dataGraph.batadv.nodes = dataGraph.batadv.nodes.concat(data[i].batadv.nodes);
              dataGraph.batadv.links = dataGraph.batadv.links.concat(data[i].batadv.links);
              dataGraph.timestamp = data[i].timestamp;
            }
          } else if (data[i].version !== 2) {
            vererr = "Unsupported nodes version: " + data[i].version;
            console.log(vererr); //silent fail
          } else {
            data[i].nodes.forEach(fillData);
            dataNodes.timestamp = data[i].timestamp;
          }
        }

        function fillData(node) {
          var position = dataNodes.nodeIds.indexOf(node.nodeinfo.node_id);
          if (position === -1) {
            dataNodes.nodes.push(node);
            dataNodes.nodeIds.push(node.nodeinfo.node_id);
          }
          else if (node.flags.online === true) {
            dataNodes.nodes[position] = node;
          }
        }

        var nodes = dataNodes.nodes.filter(function (d) {
          return "firstseen" in d && "lastseen" in d;
        });

        nodes.forEach(function (node) {
          node.firstseen = moment.utc(node.firstseen).local();
          node.lastseen = moment.utc(node.lastseen).local();
        });

        var now = moment();
        var age = moment(now).subtract(config.maxAge, "days");

        var newnodes = limit("firstseen", age, sortByKey("firstseen", nodes).filter(online));
        var lostnodes = limit("lastseen", age, sortByKey("lastseen", nodes).filter(offline));

        var graphnodes = {};

        dataNodes.nodes.forEach(function (d) {
          graphnodes[d.nodeinfo.node_id] = d;
        });

        var graph = dataGraph.batadv;

        graph.nodes.forEach(function (d) {
          if (d.node_id in graphnodes) {
            d.node = graphnodes[d.node_id];
            if (d.unseen) {
              d.node.flags.online = true;
              d.node.flags.unseen = true;
            }
          }
        });

        graph.links.forEach(function (d) {
          d.source = graph.nodes[d.source];

          if (graph.nodes[d.target].node) {
            d.target = graph.nodes[d.target];
          } else {
            d.target = undefined;
          }
        });

        var links = graph.links.filter(function (d) {
          return d.target !== undefined;
        });

        links.forEach(function (d) {
          var unknown = (d.source.node === undefined);
          var ids;
          if (unknown) {
            ids = [d.source.id.replace(/:/g, ""), d.target.node.nodeinfo.node_id];
          } else {
            ids = [d.source.node.nodeinfo.node_id, d.target.node.nodeinfo.node_id];
          }
          d.id = ids.join("-");

          if (unknown || !d.source.node.nodeinfo.location || !d.target.node.nodeinfo.location ||
            isNaN(d.source.node.nodeinfo.location.latitude) ||
            isNaN(d.source.node.nodeinfo.location.longitude) ||
            isNaN(d.target.node.nodeinfo.location.latitude) ||
            isNaN(d.target.node.nodeinfo.location.longitude)) {
            return;
          }

          d.latlngs = [];
          d.latlngs.push(L.latLng(d.source.node.nodeinfo.location.latitude, d.source.node.nodeinfo.location.longitude));
          d.latlngs.push(L.latLng(d.target.node.nodeinfo.location.latitude, d.target.node.nodeinfo.location.longitude));

          d.distance = d.latlngs[0].distanceTo(d.latlngs[1]);
        });

        nodes.forEach(function (d) {
          d.neighbours = [];
          if (d.statistics) {
            /*eslint camelcase:0*/
            if ("gateway" in d.statistics && d.statistics.gateway in graphnodes) {
              d.statistics.gateway = {"node": graphnodes[d.statistics.gateway], "id": d.statistics.gateway};
            }
            if ("nexthop" in d.statistics && d.statistics.nexthop in graphnodes) {
              d.statistics.nexthop = {"node": graphnodes[d.statistics.nexthop], "id": d.statistics.nexthop};
            }
            if ("gateway_nexthop" in d.statistics && d.statistics.gateway_nexthop in graphnodes) {
              d.statistics.gateway_nexthop = {
                "node": graphnodes[d.statistics.gateway_nexthop],
                "id": d.statistics.gateway_nexthop
              };
            }
          }
        });

        links.forEach(function (d) {
          if (d.type === "tunnel" || d.vpn) {
            d.type = "VPN";
            d.isVPN = true;
          } else if (d.type === "fastd") {
            d.type = "fastd";
            d.isVPN = true;
          } else if (d.type === "l2tp") {
            d.type = "L2TP";
            d.isVPN = true;
          } else if (d.type === "gre") {
            d.type = "GRE";
            d.isVPN = true;
          } else if (d.type === "wireless") {
            d.type = "Wifi";
            d.isVPN = false;
          } else if (d.type === "other") {
            d.type = "Kabel";
            d.isVPN = false;
          } else {
            d.type = "N/A";
            d.isVPN = false;
          }

          if (d.isVPN && d.target.node) {
            d.target.node.flags.uplink = true;
          }

          var unknown = (d.source.node === undefined);
          if (unknown) {
            d.target.node.neighbours.push({id: d.source.id, link: d, incoming: true});
            return;
          }
          d.source.node.neighbours.push({node: d.target.node, link: d, incoming: false});
          d.target.node.neighbours.push({node: d.source.node, link: d, incoming: true});
          if (!d.isVPN) {
            d.source.node.meshlinks = d.source.node.meshlinks ? d.source.node.meshlinks + 1 : 1;
          }
        });

        links.sort(function (a, b) {
          return b.tq - a.tq;
        });

        return {
          now: now,
          timestamp: moment.utc(dataNodes.timestamp).local(),
          nodes: {
            all: nodes,
            new: newnodes,
            lost: lostnodes
          },
          graph: {
            links: links,
            nodes: graph.nodes
          }
        };
      }

      numeral.language("de");
      moment.locale("de");

      var router = new Router();

      var urls = [];

      if (typeof config.dataPath === "string" || config.dataPath instanceof String) {
        config.dataPath = [config.dataPath];
      }

      for (var i in config.dataPath) {
        urls.push(config.dataPath[i] + "nodes.json");
        urls.push(config.dataPath[i] + "graph.json");
      }

      function update() {
        return Promise.all(urls.map(getJSON))
          .then(handleData);
      }

      update()
        .then(function (d) {
          var gui = new GUI(config, router);
          gui.setData(d);
          router.setData(d);
          router.start();

          window.setInterval(function () {
            update().then(function (d) {
              gui.setData(d);
              router.setData(d);
            });
          }, 60000);
        })
        .catch(function (e) {
          document.body.textContent = e;
          console.log(e);
        });
    };
  });