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 }; }

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

function getFields(info, fragments = info.fragments) {
  if (info.kind !== 'Field') {
    info = info.fieldNodes[0];
  }
  const selections = info.selectionSet.selections;
  const reducer = (fields, selection) => {
    if (selection.kind === 'FragmentSpread') {
      const name = selection.name.value;
      const 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, { depth = 5, colors = true, breakLength = 120 } = {}) {
  console.log(_util2.default.inspect(obj, { depth, colors, breakLength }));
}

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

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