api/controller/placeholder.js

'use strict';

const _ = require('lodash');

const logger = require('pelias-logger').get('api');
const logging = require( '../helper/logging' );
const Document = require('pelias-model').Document;

// composition of toNumber and isFinite, useful for single call to convert a value
//  to a number, then checking to see if it's finite
function isFiniteNumber() {
  return _.flow(_.toNumber, _.isFinite);
}

// returns true if all 4 ,-delimited (max) substrings are parseable as numbers
// '12.12,21.21,13.13,31.31'       returns true
// '12.12,21.21,13.13,blah'        returns false
// '12.12,21.21,13.13,31.31,blah'  returns false
function validBoundingBox(bbox) {
  return bbox.
    split(',').
    filter(isFiniteNumber).length === 4;
}

function synthesizeDocs(result) {
  const doc = new Document('whosonfirst', result.placetype, result.id.toString());
  doc.setName('default', result.name);

  // only assign centroid if both lat and lon are finite numbers
  if (_.conformsTo(result.geom, { 'lat': isFiniteNumber, 'lon': isFiniteNumber } )) {
    doc.setCentroid( { lat: result.geom.lat, lon: result.geom.lon } );
  } else {
    console.error('what\'s up with ' + result.id);
  }

  // lodash conformsTo verifies that an object has a property with a certain format
  if (_.conformsTo(result.geom, { 'bbox': validBoundingBox } )) {
    const parsedBoundingBox = result.geom.bbox.split(',').map(_.toNumber);
    doc.setBoundingBox({
      upperLeft: {
        lat: parsedBoundingBox[3],
        lon: parsedBoundingBox[0]
      },
      lowerRight: {
        lat: parsedBoundingBox[1],
        lon: parsedBoundingBox[2]
      }
    });
  }

  if (_.isEmpty(result.lineage)) {
    // there are no hierarchies so just return what's been assembled so far
    return buildESDoc(doc);

  }

  result.lineage.map((hierarchy) => {
    Object.keys(hierarchy)
      .filter(doc.isSupportedParent)
      .filter((placetype) => { return !_.isEmpty(_.trim(hierarchy[placetype].name)); } )
      .forEach((placetype) => {
        if (hierarchy[placetype].hasOwnProperty('abbr') && placetype === 'country') {
          doc.setAlpha3(hierarchy[placetype].abbr);
        }

        doc.addParent(
          placetype,
          hierarchy[placetype].name,
          hierarchy[placetype].id.toString(),
          hierarchy[placetype].abbr);

    });
  });

  return buildESDoc(doc);

}

function buildESDoc(doc) {
  const esDoc = doc.toESDocument();
  esDoc.data._id = esDoc._id;
  esDoc.data._type = esDoc._type;
  return esDoc.data;
}

function setup(placeholderService, should_execute) {
  function controller( req, res, next ){
    // bail early if req/res don't pass conditions for execution
    if (!should_execute(req, res)) {
      return next();
    }

    placeholderService(req, (err, results) => {
      if (err) {
        // bubble up an error if one occurred
        if (_.isObject(err) && err.message) {
          req.errors.push( err.message );
        } else {
          req.errors.push( err );
        }

      } else {
        // filter that passes only results that match on requested layers
        // passes everything if req.clean.layers is not found
        const matchesLayers = (result) => {
          return req.clean.layers.indexOf(result.placetype) >= 0;
        };
        const layersFilter = _.get(req, 'clean.layers', []).length ?
          matchesLayers : _.constant(true);

        // filter that passes only documents that match on boundary.country
        // passed everything if req.clean['boundary.country'] is not found
        const matchesBoundaryCountry = (doc) => {
          return doc.parent.country_a &&
                 doc.parent.country_a.indexOf(req.clean['boundary.country']) > -1;
        };
        const countryFilter = _.has(req, ['clean', 'boundary.country']) ?
          matchesBoundaryCountry : _.constant(true);

        // convert results to ES docs
        // boundary.country filter must happen after synthesis since multiple
        //  lineages may produce different country docs
        res.meta = {};
        res.data = _.flatten(
          results.filter(layersFilter).map(synthesizeDocs))
        .filter(countryFilter);

        const messageParts = [
          '[controller:placeholder]',
          `[result_count:${_.get(res, 'data', []).length}]`
        ];

        logger.info(messageParts.join(' '));
      }

      return next();
    });

  }

  return controller;
}

module.exports = setup;