craft-ai/lib/formatter.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _PROPERTY_FORMATTER, _OPERATORS$IN, _FORMATTER_FROM_DECIS;

exports.formatProperty = formatProperty;
exports.formatDecisionRules = formatDecisionRules;

var _lodash = require('lodash');

var _lodash2 = _interopRequireDefault(_lodash);

var _time2 = require('./time');

var _time3 = _interopRequireDefault(_time2);

var _constants = require('./constants');

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }

var DAYS = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun'];

var MONTH = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'];

var PROPERTY_FORMATTER = (_PROPERTY_FORMATTER = {}, _defineProperty(_PROPERTY_FORMATTER, _constants.TYPE_ANY, function (value) {
  return value;
}), _defineProperty(_PROPERTY_FORMATTER, _constants.TYPES.continuous, function (number) {
  return '' + Math.round(number * 100) / 100;
}), _defineProperty(_PROPERTY_FORMATTER, _constants.TYPES.time_of_day, function (time) {
  var _time = time instanceof _time3.default ? time.time_of_day : time;
  var hours = Math.floor(_time);
  var hoursStr = _lodash2.default.padStart(hours, 2, '0');
  var decMinutes = Math.round((_time - hours) * 60 * 100) / 100;
  var minutes = Math.floor(decMinutes);
  var minutesStr = _lodash2.default.padStart(minutes, 2, '0');
  var seconds = Math.round((decMinutes - minutes) * 60);
  var secondsStr = _lodash2.default.padStart(seconds, 2, '0');

  if (seconds > 0) {
    return hoursStr + ':' + minutesStr + ':' + secondsStr;
  } else {
    return hoursStr + ':' + minutesStr;
  }
}), _defineProperty(_PROPERTY_FORMATTER, _constants.TYPES.day_of_week, function (day) {
  var _day = day instanceof _time3.default ? day.day_of_week : day;
  return DAYS[_day];
}), _defineProperty(_PROPERTY_FORMATTER, _constants.TYPES.day_of_month, function (day) {
  var _day = day instanceof _time3.default ? day.day_of_month : day;
  return _lodash2.default.padStart(_day, 2, '0');
}), _defineProperty(_PROPERTY_FORMATTER, _constants.TYPES.month_of_year, function (month) {
  var _month = month instanceof _time3.default ? month.month_of_year : month;
  return MONTH[_month - 1];
}), _PROPERTY_FORMATTER);

function formatProperty(type) {
  var value = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : undefined;

  var formatter = PROPERTY_FORMATTER[type] || PROPERTY_FORMATTER[_constants.TYPE_ANY];
  var extendedFormatter = function extendedFormatter(value) {
    // A `null` value corresponds to a null/MVs branch
    if (_lodash2.default.isNull(value)) {
      return 'null';
    }
    // The empty object `{}` corresponds to an optional branch
    else if (_lodash2.default.isPlainObject(value) && _lodash2.default.isEmpty(value)) {
        return 'N/A';
      }
    return formatter(value);
  };
  if (!_lodash2.default.isUndefined(value)) {
    return extendedFormatter(value);
  }
  return extendedFormatter;
}

var FORMATTER_FROM_DECISION_RULE = (_FORMATTER_FROM_DECIS = {}, _defineProperty(_FORMATTER_FROM_DECIS, _constants.OPERATORS.IS, _defineProperty({}, _constants.TYPE_ANY, function (_ref) {
  var property = _ref.property,
      operand = _ref.operand,
      operandFormatter = _ref.operandFormatter;

  if (property) {
    return '\'' + property + '\' is ' + operandFormatter(operand);
  }
  return 'is ' + operandFormatter(operand);
})), _defineProperty(_FORMATTER_FROM_DECIS, _constants.OPERATORS.IN, (_OPERATORS$IN = {}, _defineProperty(_OPERATORS$IN, _constants.TYPE_ANY, function (_ref2) {
  var property = _ref2.property,
      operand = _ref2.operand,
      operandFormatter = _ref2.operandFormatter;

  if (property) {
    return '\'' + property + '\' in [' + operandFormatter(operand[0]) + ', ' + operandFormatter(operand[1]) + '[';
  }
  return '[' + operandFormatter(operand[0]) + ', ' + operandFormatter(operand[1]) + '[';
}), _defineProperty(_OPERATORS$IN, _constants.TYPES.day_of_week, function (_ref3) {
  var property = _ref3.property,
      operand = _ref3.operand,
      operandFormatter = _ref3.operandFormatter;

  var day_from = Math.floor(operand[0]);
  var day_to = Math.floor(operand[1]);
  // If there is only one day in the interval
  if (day_to - day_from == 1 || day_from == 6 && day_to == 0) {
    if (property) {
      return '\'' + property + '\' is ' + operandFormatter(day_from);
    }
    return operandFormatter(day_from);
  } else {
    if (property) {
      return '\'' + property + '\' from ' + operandFormatter(day_from) + ' to ' + operandFormatter((7 + day_to - 1) % 7);
    }
    return operandFormatter(day_from) + ' to ' + operandFormatter((7 + day_to - 1) % 7);
  }
}), _defineProperty(_OPERATORS$IN, _constants.TYPES.month_of_year, function (_ref4) {
  var property = _ref4.property,
      operand = _ref4.operand,
      operandFormatter = _ref4.operandFormatter;

  var month_from = Math.floor(operand[0]);
  var month_to = Math.floor(operand[1]);
  if (month_to - month_from == 1 || month_from == 12 && month_to == 1) {
    // One month in the interval
    if (property) {
      return '\'' + property + '\' is ' + operandFormatter(month_from);
    }
    return operandFormatter(month_from);
  } else if (month_to == 1) {
    // (Excluded) upper bound is january
    if (property) {
      return '\'' + property + '\' from ' + operandFormatter(month_from) + ' to ' + operandFormatter(12);
    }
    return operandFormatter(month_from) + ' to ' + operandFormatter(12);
  } else {
    if (property) {
      return '\'' + property + '\' from ' + operandFormatter(month_from) + ' to ' + operandFormatter(month_to - 1);
    }
    return operandFormatter(month_from) + ' to ' + operandFormatter(month_to - 1);
  }
}), _OPERATORS$IN)), _defineProperty(_FORMATTER_FROM_DECIS, _constants.OPERATORS.GTE, _defineProperty({}, _constants.TYPE_ANY, function (_ref5) {
  var property = _ref5.property,
      operand = _ref5.operand,
      operandFormatter = _ref5.operandFormatter;

  if (property) {
    return '\'' + property + '\' >= ' + operandFormatter(operand);
  }
  return '>= ' + operandFormatter(operand);
})), _defineProperty(_FORMATTER_FROM_DECIS, _constants.OPERATORS.LT, _defineProperty({}, _constants.TYPE_ANY, function (_ref6) {
  var property = _ref6.property,
      operand = _ref6.operand,
      operandFormatter = _ref6.operandFormatter;

  if (property) {
    return '\'' + property + '\' < ' + operandFormatter(operand);
  }
  return '< ' + operandFormatter(operand);
})), _FORMATTER_FROM_DECIS);

function formatDecisionRules(decisionRules) {
  return decisionRules.map(function (_ref7) {
    var property = _ref7.property,
        type = _ref7.type,
        operand = _ref7.operand,
        operator = _ref7.operator;

    var operatorFormatters = FORMATTER_FROM_DECISION_RULE[operator];
    if (!operatorFormatters) {
      throw new Error('Unable to format the given decision rule: unknown operator \'' + operator + '\'.');
    }
    var formatter = operatorFormatters[type] || operatorFormatters[_constants.TYPE_ANY];
    var operandFormatter = formatProperty(type || _constants.TYPE_ANY);
    return formatter({ property: property, type: type, operator: operator, operandFormatter: operandFormatter, operand: operand });
  }).join(' and ');
}