oui/update.js

#!/usr/bin/env node
"use strict";

const fs = require("fs");
const path = require("path");
const url = require("url");

const countries = require("country-data").countries;
const fetch = require("make-fetch-happen");
const stringify = require("json-stable-stringify");

const stringifyOpts = {
  space: 1,
  cmp: (a, b) => {
    return parseInt(a.key, 16) > parseInt(b.key, 16) ? 1 : -1;
  },
};

module.exports = function update(opts) {
  return new Promise((resolve, reject) => {
    opts = Object.assign({
      url: "https://linuxnet.ca/ieee/oui.txt",
      file: path.join(__dirname, "oui.json"),
    }, opts);

    const uri = url.parse(opts.url);
    if (!uri.protocol || !uri.hostname) {
      return reject(new Error(`Invalid source URL '${opts.url}'`));
    }

    fetch(opts.url).then(res => res.text()).then(body => {
      if (!body || !body.length || !/^(OUI|[#]|[A-Fa-f0-9])/.exec(body)) {
        throw new Error("Downloaded file does not look like a oui.txt file");
      } else {
        return parse(body.split("\n"));
      }
    }).then(result => {
      if (opts.test) return resolve(result);

      // save oui.js
      fs.writeFile(opts.file, stringify(result, stringifyOpts), err => {
        if (err) return reject(err);
        if (!opts.web) return resolve(result);

        // update oui.web.js
        const resultShort = {};
        Object.keys(result).map(key => {
          resultShort[key] = result[key].match(/^.*$/m)[0];
        });

        const web = path.join(__dirname, "oui.web.js");
        fs.readFile(web, "utf8", (err, js) => {
          if (err) return reject(err);
          js = js.replace(/var db =.+/, `var db = ${stringify(resultShort)};`);
          fs.writeFile(web, js, err => {
            if (err) return reject(err);
            resolve(result);
          });
        });
      });
    }).catch(reject);
  });
};

function isStart(firstLine, secondLine) {
  if (firstLine === undefined || secondLine === undefined) return false;
  return firstLine.trim().length === 0 && /([0-9A-F]{2}[-]){2}([0-9A-F]{2})/.test(secondLine);
}

function parse(lines) {
  const result = {};
  let i = 3;
  while (i !== lines.length) {
    if (isStart(lines[i], lines[i + 1])) {
      let oui   = lines[i + 2].substring(0, 6).trim();
      let owner = lines[i + 1].replace(/\((hex|base 16)\)/, "").substring(10).trim();

      i += 3;
      while (!isStart(lines[i], lines[i + 1]) && i < lines.length) {
        if (lines[i] && lines[i].trim()) owner += `\n${lines[i].trim()}`;
        i++;
      }

      // ensure upper case on hex digits
      oui = oui.toUpperCase();

      // remove excessive whitespace
      owner = owner.replace(/[ \t]+/gm, " ");

      // replace country shortcodes
      const shortCode = (/\n([A-Z]{2})$/.exec(owner) || [])[1];
      if (shortCode && countries[shortCode]) {
        owner = owner.replace(/\n.+$/, `\n${countries[shortCode].name}`);
      }

      result[oui] = owner;
    }
  }
  return result;
}