timestring/index.js

var _ = require('lodash');

/**
 * Exports
 */

module.exports = parseTimestring;

/**
 * Default options to use when parsing a timestring
 *
 * @type {Object}
 */

var defaultOpts = {
  hoursPerDay: 24,
  daysPerWeek: 7,
  weeksPerMonth: 4,
  monthsPerYear: 12,
};

/**
 * Map of accepted strings to unit
 *
 * @type {Object}
 */

var unitMap = {
  s: ['s', 'sec', 'secs', 'second', 'seconds'],
  m: ['m', 'min', 'mins', 'minute', 'minutes'],
  h: ['h', 'hr', 'hrs', 'hour', 'hours'],
  d: ['d', 'day', 'days'],
  w: ['w', 'week', 'weeks'],
  mth: ['mth', 'mths','month', 'months'],
  y: ['y', 'yr', 'yrs', 'year', 'years'],
};

/**
 * Parse a timestring
 *
 * @param {string} string
 * @param {string} [returnUnit]
 * @param {Object} [opts]
 * @return {number}
 */

function parseTimestring(string, returnUnit, opts) {
  opts = _.extend(_.clone(defaultOpts), opts || {});

  var totalSeconds = 0;
  var unitValues = getUnitValues(opts);
  var groups = string
    .toLowerCase()
    .replace(/[^\.\w+-]+/g, '')
    .match(/[-+]?[0-9]+[a-z]+/g);

  if (groups !== null) {
    _.each(groups, function(group) {
      var value = group.match(/[0-9]+/g)[0];
      var unit = group.match(/[a-z]+/g)[0];

      totalSeconds += getSeconds(value, unit, unitValues);
    });
  }

  if (returnUnit) {
    return convert(totalSeconds, returnUnit, unitValues);
  }

  return totalSeconds;
}

/**
 * Get unit values based on the passed options
 *
 * @param {Object} opts
 * @returns {Object}
 */

function getUnitValues(opts) {
  var unitValues = {
    s: 1,
    m: 60,
    h: 3600,
  };

  unitValues.d = opts.hoursPerDay * unitValues.h;
  unitValues.w = opts.daysPerWeek * unitValues.d;
  unitValues.mth = opts.weeksPerMonth * unitValues.w;
  unitValues.y = opts.monthsPerYear * unitValues.mth;

  return unitValues;
}

/**
 * Get the key for a unit
 *
 * @param {string} unit
 * @returns {string}
 */

function getUnitKey(unit) {
  for (var k in unitMap) {
    for (var u in unitMap[k]) {
      if (unit === unitMap[k][u]) {
        return k;
      }
    }
  }

  throw new Error('The unit [' + unit + '] is not supported by timestring');
}

/**
 *  Get the number of seconds for a value, based on the unit
 *
 * @param {number} value
 * @param {string} unit
 * @param {Object} unitValues
 * @returns {number}
 */

function getSeconds(value, unit, unitValues) {
  var baseValue = unitValues[getUnitKey(unit)];

  return value * baseValue;
}

/**
 * Convert a value from its existing unit to a new unit
 *
 * @param {number} value
 * @param {string} unit
 * @param {Object} unitValues
 * @returns {number}
 */

function convert(value, unit, unitValues) {
  var baseValue = unitValues[getUnitKey(unit)];

  return value / baseValue;
}