graphbrainz/lib/util.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.ONE_DAY = undefined;
exports.getFields = getFields;
exports.prettyPrint = prettyPrint;
exports.toFilteredArray = toFilteredArray;
exports.extendIncludes = extendIncludes;

var _util = require('util');

var _util2 = _interopRequireDefault(_util);

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

var ONE_DAY = exports.ONE_DAY = 24 * 60 * 60 * 1000;

function getFields(info) {
  var fragments = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : info.fragments;

  if (info.kind !== 'Field') {
    info = info.fieldNodes[0];
  }
  var selections = info.selectionSet.selections;
  var reducer = function reducer(fields, selection) {
    if (selection.kind === 'FragmentSpread') {
      var name = selection.name.value;
      var fragment = fragments[name];
      if (!fragment) {
        throw new Error('Fragment \'' + name + '\' was not passed to getFields()');
      }
      fragment.selectionSet.selections.reduce(reducer, fields);
    } else if (selection.kind === 'InlineFragment') {
      selection.selectionSet.selections.reduce(reducer, fields);
    } else {
      fields[selection.name.value] = selection;
    }
    return fields;
  };
  return selections.reduce(reducer, {});
}

function prettyPrint(obj) {
  var _ref = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {},
      _ref$depth = _ref.depth,
      depth = _ref$depth === undefined ? 5 : _ref$depth,
      _ref$colors = _ref.colors,
      colors = _ref$colors === undefined ? true : _ref$colors,
      _ref$breakLength = _ref.breakLength,
      breakLength = _ref$breakLength === undefined ? 120 : _ref$breakLength;

  console.log(_util2.default.inspect(obj, { depth: depth, colors: colors, breakLength: breakLength }));
}

function toFilteredArray(obj) {
  return (Array.isArray(obj) ? obj : [obj]).filter(function (x) {
    return x;
  });
}

function extendIncludes(includes, moreIncludes) {
  includes = toFilteredArray(includes);
  moreIncludes = toFilteredArray(moreIncludes);
  var seen = {};
  return includes.concat(moreIncludes).filter(function (x) {
    if (seen[x]) {
      return false;
    }
    seen[x] = true;
    return true;
  });
}