miolo 0.9.21 → 0.9.23

package/dist/server/miolo.server.mjs

@@ -1,5 +1,5 @@
 /**
- * miolo v0.9.21
+ * miolo v0.9.23
  *
  * Copyright (c) Donato Lorenzo <donato@afialapis.com>
  *
@@ -28,8 +28,9 @@ import koa_mount from 'koa-mount';
 import koa_serve from 'koa-static';
 import koa_favicon from 'koa-favicon';
 import { performance } from 'node:perf_hooks';
+import fs, { readFileSync } from 'node:fs';
+import { Reader } from '@maxmind/geoip2-node';
 import Router from '@koa/router';
-import { readFileSync } from 'node:fs';
 import jwt from 'jwt-simple';
 import passport from 'koa-passport';
 import LocalStrategy from 'passport-local';
@@ -156,11 +157,15 @@ var base_config = {
       // seconds to consider slow a request
       onStart: undefined,
       // (ctx, times) => { return begin_result}
-      onDone: undefined
-      // (ctx, begin_result, times) => {}
+      onDone: undefined,
+      // (ctx, begin_result, times) => {},
+      geoip: {
+        enabled: false,
+        db: '/var/lib/GeoIP/GeoLite2-City.mmdb',
+        local_ips: ['127.0.0.1']
+      }
     }
   },
-
   session: {
     salt: 'SUPER_SALTY_YES?',
     secret: 'SUPER_SECRET_KEY_KERE',
@@ -800,6 +805,53 @@ var init_static_middleware = (app, config) => {
   }
 };
 
+var _geoip_reader = undefined;
+var _geoip_local_ips = ['127.0.0.1'];
+function _geoip_init() {
+  var db = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : '/var/lib/GeoIP/GeoLite2-City.mmdb';
+  var local_ips = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : ['127.0.0.1'];
+  var logger = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : console;
+  try {
+    if (_geoip_reader != undefined) {
+      return _geoip_reader;
+    }
+    _geoip_local_ips = [..._geoip_local_ips, ...(local_ips || [])];
+    var dbBuffer = fs.readFileSync(db);
+    _geoip_reader = Reader.openBuffer(dbBuffer);
+    return _geoip_reader;
+  } catch (error) {
+    logger.error("[geoip] Error initing:");
+    logger.error(error);
+    return undefined;
+  }
+}
+var geoip_localize_ip = function geoip_localize_ip(ip, config) {
+  var logger = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : console;
+  if (_geoip_local_ips.indexOf(ip) >= 0) {
+    return {
+      local: true,
+      country: '',
+      city: ''
+    };
+  }
+  try {
+    var _resp$city;
+    var reader = _geoip_init(config === null || config === void 0 ? void 0 : config.db, config === null || config === void 0 ? void 0 : config.local_ipds, logger);
+    var resp = reader.city(ip);
+    return {
+      country: resp.country.isoCode,
+      city: (_resp$city = resp.city) === null || _resp$city === void 0 || (_resp$city = _resp$city.names) === null || _resp$city === void 0 ? void 0 : _resp$city.en
+    };
+  } catch (error) {
+    logger.error("[geoip] Error localizing IP ".concat(ip, ":"));
+    logger.error(error);
+  }
+  return {
+    country: '',
+    city: ''
+  };
+};
+
 var REQUEST_COUNTER = {
   total: 0
 };
@@ -827,14 +879,17 @@ function init_request_middleware(app, config) {
     lazy: (config === null || config === void 0 ? void 0 : config.lazy) || 1,
     slow: (config === null || config === void 0 ? void 0 : config.slow) || 2,
     onStart: (config === null || config === void 0 ? void 0 : config.onStart) || _def_on_begin,
-    onDone: (config === null || config === void 0 ? void 0 : config.onDone) || _def_on_done
+    onDone: (config === null || config === void 0 ? void 0 : config.onDone) || _def_on_done,
+    geoip: (config === null || config === void 0 ? void 0 : config.geoip) || {
+      enabled: false
+    }
   };
   function request_middleware(_x6, _x7) {
     return _request_middleware.apply(this, arguments);
   }
   function _request_middleware() {
     _request_middleware = _asyncToGenerator(function* (ctx, next) {
-      var _ctx$session;
+      var _reqConfig$geoip, _geo_info, _geo_info2, _geo_info3, _geo_info4, _ctx$session;
       var logger = ctx.miolo.logger;
       var ip = ctx.headers["x-real-ip"] || '127.0.0.1';
       var started = performance.now();
@@ -850,9 +905,18 @@ function init_request_middleware(app, config) {
       ctx.requestId = REQUEST_COUNTER.total;
       ctx.request.ip = ip;
 
+      // If wanted, geo localize ip
+      var geo_enabled = (reqConfig === null || reqConfig === void 0 || (_reqConfig$geoip = reqConfig.geoip) === null || _reqConfig$geoip === void 0 ? void 0 : _reqConfig$geoip.enabled) === true;
+      var geo_info = {};
+      if (geo_enabled) {
+        geo_info = geoip_localize_ip(ip, reqConfig.geoip, logger);
+      }
+      ctx.request.geoip = geo_info;
+
       // Log something
       var clurl = ctx.request.url.indexOf('?') >= 0 ? ctx.request.url.substr(0, ctx.request.url.indexOf('?')) : ctx.request.url;
-      var sreq = "".concat(magenta(ip), " ").concat(cyan(ctx.request.method), " ").concat(cyan(clurl), " [").concat(cyan_light(REQUEST_COUNTER[ip]), "/").concat(cyan_light(ctx.requestId), "]");
+      var geo_string = geo_enabled ? ((_geo_info = geo_info) === null || _geo_info === void 0 ? void 0 : _geo_info.local) === true ? '' : (_geo_info2 = geo_info) !== null && _geo_info2 !== void 0 && _geo_info2.country ? (_geo_info3 = geo_info) !== null && _geo_info3 !== void 0 && _geo_info3.city ? " (".concat((_geo_info4 = geo_info) === null || _geo_info4 === void 0 ? void 0 : _geo_info4.city, ", ").concat(geo_info.country, ")") : " (".concat(geo_info.country, ")") : '' : '';
+      var sreq = "".concat(magenta(ip)).concat(geo_string, " ").concat(cyan(ctx.request.method), " ").concat(cyan(clurl), " [").concat(cyan_light(REQUEST_COUNTER[ip]), "/").concat(cyan_light(ctx.requestId), "]");
       var sbody = ctx.request.body != undefined ? JSON.stringify(ctx.request.body) : '';
       logger.info("".concat(sreq, " - START"));
       logger.debug("".concat(sreq, " - Body: ").concat(sbody));