craft-ai/lib/interpreter.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.distribution = exports.reduceDecisionRules = exports.formatProperty = exports.formatDecisionRules = undefined;
exports.decide = decide;
exports.getDecisionRulesProperties = getDecisionRulesProperties;
exports.decideFromContextsArray = decideFromContextsArray;

var _lodash = require('lodash');

var _lodash2 = _interopRequireDefault(_lodash);

var _context = require('./context');

var _context2 = _interopRequireDefault(_context);

var _interpreter_v = require('./interpreter_v1');

var _parse4 = require('./parse');

var _parse5 = _interopRequireDefault(_parse4);

var _reducer = require('./reducer');

var _semver = require('semver');

var _semver2 = _interopRequireDefault(_semver);

var _errors = require('./errors');

var _interpreter_v2 = require('./interpreter_v2');

var _formatter = require('./formatter');

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

function _toConsumableArray(arr) { if (Array.isArray(arr)) { for (var i = 0, arr2 = Array(arr.length); i < arr.length; i++) { arr2[i] = arr[i]; } return arr2; } else { return Array.from(arr); } }

var DECISION_FORMAT_VERSION = '1.1.0';

function decide(tree) {
  var _parse = (0, _parse5.default)(tree),
      _version = _parse._version,
      configuration = _parse.configuration,
      trees = _parse.trees;

  for (var _len = arguments.length, args = Array(_len > 1 ? _len - 1 : 0), _key = 1; _key < _len; _key++) {
    args[_key - 1] = arguments[_key];
  }

  var ctx = configuration ? _context2.default.apply(undefined, [configuration].concat(args)) : _lodash2.default.extend.apply(_lodash2.default, [{}].concat(args));
  return _decide(configuration, trees, ctx, _version);
}

function _decide(configuration, trees, ctx, _version) {
  if (_semver2.default.satisfies(_version, '>=1.0.0 <2.0.0')) {
    return (0, _interpreter_v.decide)(configuration, trees, ctx);
  }
  if (_semver2.default.satisfies(_version, '>=2.0.0 <3.0.0')) {
    return (0, _interpreter_v2.decide)(configuration, trees, ctx);
  }
  throw new _errors.CraftAiDecisionError('Invalid decision tree format, "' + _version + '" is not a valid version.');
}

function reduceNodes(tree, fn, initialAccValue) {
  var nodes = [];
  nodes.push(tree);
  var recursiveNext = function recursiveNext(acc) {
    if (nodes.length == 0) {
      // No more nodes
      return acc;
    }

    var node = nodes.pop();
    if (node.children) {
      nodes = node.children.concat(nodes);
    }

    var updatedAcc = fn(acc, node);
    return recursiveNext(updatedAcc);
  };
  return recursiveNext(initialAccValue);
}

function getDecisionRulesProperties(tree) {
  var _parse2 = (0, _parse5.default)(tree),
      configuration = _parse2.configuration,
      trees = _parse2.trees;

  return (0, _lodash2.default)(trees).values().reduce(function (properties, tree) {
    return reduceNodes(tree, function (properties, node) {
      if (node.children) {
        // Skip leaves
        return properties.concat(node.children[0].decision_rule.property);
      }
      return properties;
    }, properties);
  }, (0, _lodash2.default)([])).uniq().map(function (property) {
    return _lodash2.default.extend(configuration.context[property], {
      property: property
    });
  }).value();
}

function decideFromContextsArray(tree, contexts) {
  var _parse3 = (0, _parse5.default)(tree),
      _version = _parse3._version,
      configuration = _parse3.configuration,
      trees = _parse3.trees;

  return _lodash2.default.map(contexts, function (contextsItem) {
    var ctx = void 0;
    if (_lodash2.default.isArray(contextsItem)) {
      ctx = _context2.default.apply(undefined, [configuration].concat(_toConsumableArray(contextsItem)));
    } else {
      ctx = (0, _context2.default)(configuration, contextsItem);
    }
    try {
      return _decide(configuration, trees, ctx, _version);
    } catch (error) {
      if (error instanceof _errors.CraftAiNullDecisionError) {
        var message = error.message,
            metadata = error.metadata;

        return {
          _version: DECISION_FORMAT_VERSION,
          context: ctx,
          error: { message: message, metadata: metadata }
        };
      } else {
        throw error;
      }
    }
  });
}

exports.formatDecisionRules = _formatter.formatDecisionRules;
exports.formatProperty = _formatter.formatProperty;
exports.reduceDecisionRules = _reducer.reduceDecisionRules;
exports.distribution = _interpreter_v2.distribution;