api/query/search.js

var queries = require('geopipes-elasticsearch-backend').queries,
    sort = require('../query/sort');

function generate( params ){
  var centroid = null;

  if ( params.lat && params.lon ){
    centroid = {
      lat: params.lat,
      lon: params.lon
    };
  } 
  
  var query = queries.distance( centroid, { size: params.size } );
  var input = params.input;

  if (params.bbox) {
    query = queries.bbox ( centroid, { size: params.size, bbox: params.bbox } );
  }

  query.query.filtered.query = {
    'bool': {
      'must': [],
      'should': []
    }
  };

  if (params.parsed_input) {

    query.query.filtered.query.bool.should = [];

    var admin_fields = [];
    var qb = function(admin_fields, value) {
      admin_fields.forEach(function(admin_field) {
        var match = {};
        match[admin_field] = value;
        query.query.filtered.query.bool.should.push({
           'match': match
        });
      });  
    };

    // update input
    if (params.parsed_input.number && params.parsed_input.street) {
      input = params.parsed_input.number + ' ' + params.parsed_input.street;
    } else if (params.parsed_input.admin_parts) {
      input = params.parsed_input.name;
    }

    // address
    // number, street, zip
    if (params.parsed_input.number) {
      qb(['address.number'], params.parsed_input.number);
    } 
    if (params.parsed_input.street) {
      qb(['address.street'], params.parsed_input.street);
    } 
    if (params.parsed_input.zip) {
      qb(['address.zip'], params.parsed_input.zip);
    } 

    // city
    // admin2, locality, local_admin, neighborhood
    if (params.parsed_input.admin2) {
      qb(['admin2'], params.parsed_input.admin2);
    } else {
      admin_fields.push('admin2');
    }

    // state
    // admin1, admin1_abbr
    if (params.parsed_input.admin1) {
      qb(['admin1', 'admin1_abbr'], params.parsed_input.admin1);
    } else {
      admin_fields.push('admin1', 'admin1_abbr');
    }

    // country
    // admin0, alpha3
    if (params.parsed_input.admin0) {
      qb(['admin0', 'alpha3'], params.parsed_input.admin0);
    } else {
      admin_fields.push('admin0', 'alpha3');
    }

    var input_regions = params.parsed_input.regions.join(' ');
    if (admin_fields.length === 5 &&  input_regions !== params.input) {
      if (params.parsed_input.admin_parts) {
        qb(admin_fields, params.parsed_input.admin_parts);
      } else {
        qb(admin_fields, input_regions);
      }
    }
  }

  // add search condition to distance query
  query.query.filtered.query.bool.must.push({ 
    'match': {
      'name.default': input
    }
  });

  query.sort = query.sort.concat( sort( params ) );

  return query;
}

module.exports = generate;