koa-ip-geo/lib/tools.js

IP and GeoLocation filter middleware for koa, support allow lists and block lists.

const fs = require('fs');

// fix for IPv6 Dotted-quad notation
exports.corrIP = function (ip) {
  if (ip.indexOf(':') !== -1 && ip.indexOf('.') !== -1) {
    ip = ip.split(':');
    ip = ip[ip.length - 1];
  }
  return (ip);
};

exports.corrLocalhost = function (list) {
  if (list.indexOf('127.0.0.1') !== -1 || list.indexOf(':.1') !== -1 || list.indexOf('localhost') !== -1) {
    if (list.indexOf('127.0.0.1') === -1) list.push('127.0.0.1');
    if (list.indexOf('::1') === -1) list.push('::1');
    if (list.indexOf('localhost') === -1) list.push('localhost');
  }
  return (list);
};

exports.getJsonFile = function (fileName) {
  fs.accessSync(fileName, fs.constants.R_OK);
  let result = {};
  try {
    result = JSON.parse(fs.readFileSync(fileName, 'utf8'));
  } catch (e) {
  }
  return result;
}