[squeep-api-dingus] / lib / common.js

/* eslint-disable security/detect-object-injection */
'use strict';

/**
 * Utility and miscellaneous functions.
 */

const path = require('path');
const crypto = require('crypto');
const uuid = require('uuid');
const Enum = require('./enum');

/**
 * Return a function which combines a part of the filename with a scope, for use in logging.
 * @param {string} filename
 */
const fileScope = (filename) => {
  let fScope = path.basename(filename, '.js');
  if (fScope === 'index') {
    fScope = path.basename(path.dirname(filename));
  }
  return (scope) => `${fScope}:${scope}`;
}

/**
 * Simple ETag from data.
 * @param {string} filePath 
 * @param {object} fileStat 
 * @param {*} fileData 
 */
const generateETag = (_filePath, fileStat, fileData) => {
  const hash = crypto.createHash('sha256');
  if (fileStat && fileStat.mtimeMs) {
    hash.update(fileStat.mtimeMs.toString());
  }
  hash.update(fileData);
  const digest = hash.digest('base64').replace('=', '');
  return `"${digest}"`;
};

/**
 * @param {object} obj 
 * @param {string} prop 
 * @param {*} def 
 */
const get = (obj, prop, def) => obj && prop && (prop in obj) ? obj[prop] : def;

/**
 * @param {http.ClientRequest} req 
 * @param {http.ServerResponse} res 
 * @param {object} ctx 
 */
const handlerLogData = (req, res, ctx) => ({
  req: requestLogData(req),
  res: responseLogData(res),
  ctx,
});

/**
 * 
 * @param {http.ClientRequest} req 
 * @param {Number} modifiedTimeMs 
 * @param {string} eTag 
 */
const isClientCached = (req, modifiedTimeMs, eTag) => {
  let clientCached = false;

  const ifModifiedSince = req.getHeader(Enum.Header.IfModifiedSince);
  if (ifModifiedSince) {
    const ifModifiedSinceMs = Date.parse(ifModifiedSince);
    if (modifiedTimeMs < ifModifiedSinceMs) {
      clientCached = true;
    }
  }

  const ifNoneMatch = req.getHeader(Enum.Header.IfNoneMatch);
  if (ifNoneMatch) {
    const matches = ifNoneMatch.split(',').map((m) => m.trim());
    if (matches.includes(eTag)
    ||  (ifNoneMatch === '*' && eTag)) {
      clientCached = true;
    } else {
      // If if-none-matched header is present, it takes precedence over modified-since.
      clientCached = false;
    }
  }

  return clientCached;
};

/**
 * Shallow merge for enums, to be called by derived constructor.
 * Expects only one-level deep, is not recursive!
 * @param {Object} origEnum
 * @param {Object} additionalEnum
 */
const mergeEnum = (origEnum, additionalEnum) => {
  for (const e of Object.keys(additionalEnum)) {
    if (typeof additionalEnum[e] === 'object') {
      if (! (e in origEnum)) {
        origEnum[e] = {};
      }
      Object.assign(origEnum[e], additionalEnum[e]);
    } else {
      origEnum[e] = additionalEnum[e];
    }
  }
  return origEnum;
};

/**
 * Isolate the general category of an http status code.
 * @param {Number} statusCode
 * @returns {Number}
 */
const httpStatusCodeClass = (statusCode) => Math.floor(statusCode / 100);

const _isObject = (obj) => obj && typeof obj === 'object';
const _isArray = (obj) => Array.isArray(obj);
/**
 * Return a new object with all objects combined.
 * @param  {...any} objects
 * @returns
 */
const mergeDeep = (...objects) => {
  return objects.reduce((acc, obj) => {
    const objectProperties = [...Object.getOwnPropertyNames(obj), ...Object.getOwnPropertySymbols(obj)];
    objectProperties.forEach((k) => {
      const aVal = acc[k];
      const oVal = obj[k];
      if (_isArray(oVal)) {
        acc[k] = (_isArray(aVal) ? aVal : []).concat(oVal);
      } else if (_isObject(oVal)) {
        acc[k] = mergeDeep(_isObject(aVal) ? aVal : {}, oVal);
      } else {
        acc[k] = oVal;
      }
    });
    return acc;
  }, {});
};


/**
 * Return a new object with selected props.
 * @param {Object} obj
 * @param {string[]} props
 */
const pick = (obj, props) => {
  const picked = {};
  props.forEach((prop) => {
    if (prop in obj) {
      picked[prop] = obj[prop];
    }
  });
  return picked;
};

/**
 * Return a subset of a request object, suitable for logging.
 * @param {http.ClientRequest} req 
 */
const requestLogData = (req) => {
  return pick(req, [
    'method',
    'url',
    'httpVersion',
    'headers',
    'trailers',
  ]);
};


/**
 * Return a subset of a response object, suitable for logging.
 * @param {http.ServerResponse} res 
 */
const responseLogData = (res) => {
  const response = pick(res, [
    'statusCode',
    'statusMessage',
  ]);
  response.headers = res.getHeaders();
  return response;
};


/**
 * Store updates to defaultOptions, but no new properties.
 * @param {Object} target 
 * @param {Object} defaultOptions 
 * @param {Object} options 
 */
const setOptions = (target, defaultOptions, options) => {
  Object.assign(target, defaultOptions, pick(options, Object.keys(defaultOptions)));
};

/**
 * Return a list of source split at first delimiter.
 * @param {string} src 
 * @param {string} delimiter 
 * @param {string} fill trailing stand-in if no delimiter in src
 */
const splitFirst = (src, delimiter, fill) => {
  const idx = src.indexOf(delimiter);
  if (idx >= 0) {
    return [ src.slice(0, idx), src.slice(idx + 1) ];
  } else {
    return [ src, fill ];
  }
};

/**
 * Generate a new request identifier.
 * @returns {String}
 */
const requestId = () => {
  return uuid.v1();
};

const nop = () => { /**/ };
const nullLogger = {
  error: nop,
  warn: nop,
  info: nop,
  log: nop,
  debug: nop,
};

const ensureLoggerLevels = (logger = {}) => {
  for (const level in nullLogger) {
    if (! (level in logger)) {
      logger[level] = nullLogger[level];
    }
  }
  return logger;
};

module.exports = {
  fileScope,
  generateETag,
  get,
  handlerLogData,
  isClientCached,
  httpStatusCodeClass,
  mergeDeep,
  mergeEnum,
  nop,
  nullLogger,
  ensureLoggerLevels,
  pick,
  requestId,
  requestLogData,
  responseLogData,
  setOptions,
  splitFirst,
};