@attivio/suit/lib/util/SolrToQueryResponse.js

Attivio SUIT, the Search UI Toolkit, is a library for creating search clients for searching the Attivio platform.

'use strict';

exports.__esModule = true;
exports.default = undefined;

var _QueryResponse = require('../api/QueryResponse');

var _QueryResponse2 = _interopRequireDefault(_QueryResponse);

var _SearchDocument = require('../api/SearchDocument');

var _SearchDocument2 = _interopRequireDefault(_SearchDocument);

var _SearchFacet = require('../api/SearchFacet');

var _SearchFacet2 = _interopRequireDefault(_SearchFacet);

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

function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }

/**
 * Convert the response from a call to the Solr engine
 * into an Attivio QueryResponse so that it can be
 * processed by the Search UI.
 */
var SolrToQueryResponse = function () {
  function SolrToQueryResponse() {
    _classCallCheck(this, SolrToQueryResponse);
  }

  /**
   * Do the conversion.
   *
   * @param json          the JSON returned from the call
   * @param customOptions the custom options defined in the configuration
   */
  SolrToQueryResponse.convert = function convert(json, customOptions) {
    var result = new _QueryResponse2.default();

    if (json.responseHeader) {
      result.totalTime = json.responseHeader.QTime;
    }

    if (json.response) {
      result.totalHits = json.response.numFound;
    }
    if (json.response.docs.length > 0) {
      result.documents = SolrToQueryResponse.getSolrDocuments(json.response.docs, customOptions);
    }
    if (json.facets) {
      result.facets = SolrToQueryResponse.getSolrFacets(json.facets, customOptions);
    }

    return result;
  };

  /**
   * Turn non-array field values into arrays.
   */


  SolrToQueryResponse.wrapIfNotArray = function wrapIfNotArray(v) {
    return Array.isArray(v) ? v : [v];
  };

  SolrToQueryResponse.getSolrDocuments = function getSolrDocuments(documents, customOptions) {
    return documents.map(function (doc) {
      var mapp = customOptions.mappings;
      var fields = {};

      Object.keys(mapp).forEach(function (k) {
        if (mapp[k] && doc[mapp[k]]) {
          fields[k] = SolrToQueryResponse.wrapIfNotArray(doc[mapp[k]]);
        }
      });

      if (customOptions.customId && customOptions.customId.length > 0) {
        fields['.id'] = [doc[customOptions.customId]];
      } else {
        fields['.id'] = [doc.id];
      }

      fields['.score'] = [''];

      return _SearchDocument2.default.fromJson({ fields: fields });
    });
  };

  SolrToQueryResponse.getSolrFacets = function getSolrFacets(facets, customOptions) {
    var countlessFacets = Object.assign({}, facets);
    if (countlessFacets.count) {
      delete countlessFacets.count;
    }

    return Object.keys(countlessFacets).map(function (field) {
      var facetConfig = customOptions.facets.find(function (f) {
        return f.field === field;
      }) || {};
      return _SearchFacet2.default.fromJson({
        name: facetConfig.field || field,
        field: facetConfig.field || field,
        label: facetConfig.displayName || field,
        buckets: facets[field].buckets.map(function (b) {
          return {
            value: b.val,
            count: b.count,
            filter: facetConfig.field + ':\'' + b.val + '\''
          };
        })
      });
    });
  };

  return SolrToQueryResponse;
}();

exports.default = SolrToQueryResponse;
module.exports = exports['default'];