pelias-query/layout/StructuredFallbackQuery.js

// This query is used for component geocodes, where the individual fields have
// been specified by the user and therefore doesn't need to be parsed.  It functions
// much like FallbackQuery but with (currently) one notable exceptions:
//
// - if locality is available but borough isn't, query borough layer with locality value
// - boosts are hardcoded

var _ = require('lodash');
var baseQuery = require('./baseQuery');

function Layout(){
  this._score = [];
  this._filter = [];
}

Layout.prototype.score = function( view ){
  this._score.push( view );
  return this;
};

Layout.prototype.filter = function( view ){
  this._filter.push( view );
  return this;
};

function addPrimary(value, layer, fields) {
  // base primary query should match on layer and one of the admin fields via
  // multi_match
  var o = {
    bool: {
      _name: 'fallback.' + layer,
      must: [
        {
          multi_match: {
            query: value,
            type: 'phrase',
            fields: fields
          }
        }
      ],
      filter: {
        term: {
          layer: layer
        }
      }
    }
  };

  return o;

}

// Secondary matches are for less granular administrative areas that have been
// specified.  For example, in "Socorro, NM", "Socorro" is primary, whereas
// "NM" is secondary.
function addSecondary(value, fields) {
  return {
      multi_match: {
        query: value,
        type: 'phrase',
        fields: fields
      }
  };

}

// add the postal code if supplied
function addSecPostCode(vs, o) {
  // add postcode if specified
  if (vs.isset('input:postcode')) {
    o.bool.should.push({
      match_phrase: {
        'address_parts.zip': {
          query: vs.var('input:postcode').toString()
        }
      }
    });
  }
}

function addSecNeighbourhood(vs, o) {
  // add neighbourhood if specified
  if (vs.isset('input:neighbourhood')) {
    o.bool.must.push(addSecondary(
      vs.var('input:neighbourhood').toString(),
      [
        'parent.neighbourhood',
        'parent.neighbourhood_a'
      ]
    ));
  }
}

function addSecBorough(vs, o) {
  // add borough if specified
  if (vs.isset('input:borough')) {
    o.bool.must.push(addSecondary(
      vs.var('input:borough').toString(),
      [
        'parent.borough',
        'parent.borough_a'
      ]
    ));
  }
}

function addSecLocality(vs, o) {
  // add locality if specified
  if (vs.isset('input:locality')) {
    o.bool.must.push(addSecondary(
      vs.var('input:locality').toString(),
      [
        'parent.locality',
        'parent.locality_a',
        'parent.localadmin',
        'parent.localadmin_a'
      ]
    ));
  }
}

function addSecCounty(vs, o) {
  // add county if specified
  if (vs.isset('input:county')) {
    o.bool.must.push(addSecondary(
      vs.var('input:county').toString(),
      [
        'parent.county',
        'parent.county_a',
        'parent.macrocounty',
        'parent.macrocounty_a'
      ]
    ));
  }
}

function addSecRegion(vs, o) {
  // add region if specified
  if (vs.isset('input:region')) {
    o.bool.must.push(addSecondary(
      vs.var('input:region').toString(),
      [
        'parent.region',
        'parent.region_a',
        'parent.macroregion',
        'parent.macroregion_a'
      ]
    ));
  }
}

function addSecCountry(vs, o) {
  // add country if specified
  if (vs.isset('input:country')) {
    o.bool.must.push(addSecondary(
      vs.var('input:country').toString(),
      [
        'parent.country',
        'parent.country_a',
        'parent.dependency',
        'parent.dependency_a'
      ]
    ));
  }
}


function addQuery(vs) {
  var o = addPrimary(
    vs.var('input:query').toString(),
    'venue',
    [
      'phrase.default',
      'category'
    ],
    false
  );

  addSecNeighbourhood(vs, o);
  addSecBorough(vs, o);
  addSecLocality(vs, o);
  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addUnitAndHouseNumberAndStreet(vs) {
  var o = {
    bool: {
      _name: 'fallback.address',
      must: [
        {
          match_phrase: {
            'address_parts.unit': {
              query: vs.var('input:unit').toString()
            }
          }
        },
        {
          match_phrase: {
            'address_parts.number': {
              query: vs.var('input:housenumber').toString()
            }
          }
        },
        {
          match_phrase: {
            'address_parts.street': {
              query: vs.var('input:street').toString()
            }
          }
        }
      ],
      should: [],
      filter: {
        term: {
          layer: 'address'
        }
      }
    }
  };

  if (vs.isset('boost:address')) {
    o.bool.boost = vs.var('boost:address');
  }

  addSecPostCode(vs, o);
  addSecNeighbourhood(vs, o);
  addSecBorough(vs, o);
  addSecLocality(vs, o);
  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addHouseNumber(vs) {
  var o = {
    bool: {
      _name: 'fallback.housenumber',
      must: [
        {
          match_phrase: {
            'address_parts.number': {
              query: vs.var('input:housenumber').toString()
            }
          }
        }
      ],
      should: [],
      filter: {
        term: {
          layer: 'address'
        }
      }
    }
  };

  if (vs.isset('boost:address')) {
    o.bool.boost = vs.var('boost:address');
  }

  addSecPostCode(vs, o);
  addSecNeighbourhood(vs, o);
  addSecBorough(vs, o);
  addSecLocality(vs, o);
  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;
}

function addHouseNumberAndStreet(vs) {
  var o = {
    bool: {
      _name: 'fallback.address',
      must: [
        {
          match_phrase: {
            'address_parts.number': {
              query: vs.var('input:housenumber').toString()
            }
          }
        },
        {
          match_phrase: {
            'address_parts.street': {
              query: vs.var('input:street').toString()
            }
          }
        }
      ],
      should: [],
      filter: {
        term: {
          layer: 'address'
        }
      }
    }
  };

  if (vs.isset('boost:address')) {
    o.bool.boost = vs.var('boost:address');
  }

  addSecPostCode(vs, o);
  addSecNeighbourhood(vs, o);
  addSecBorough(vs, o);
  addSecLocality(vs, o);
  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addStreet(vs) {
  var o = {
    bool: {
      _name: 'fallback.street',
      must: [
        {
          match_phrase: {
            'address_parts.street': {
              query: vs.var('input:street').toString()
            }
          }
        }
      ],
      should: [],
      filter: {
        term: {
          layer: 'street'
        }
      }
    }
  };

  if (vs.isset('boost:street')) {
    o.bool.boost = vs.var('boost:street');
  }

  addSecPostCode(vs, o);
  addSecNeighbourhood(vs, o);
  addSecBorough(vs, o);
  addSecLocality(vs, o);
  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addNeighbourhood(vs) {
  var o = addPrimary(
    vs.var('input:neighbourhood').toString(),
    'neighbourhood',
    [
      'parent.neighbourhood',
      'parent.neighbourhood_a'
    ],
    false
  );

  addSecBorough(vs, o);
  addSecLocality(vs, o);
  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addBorough(vs) {
  var o = addPrimary(
    vs.var('input:borough').toString(),
    'borough',
    [
      'parent.borough',
      'parent.borough_a'
    ],
    false
  );

  addSecLocality(vs, o);
  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addLocalityAsBorough(vs) {
  var o = addPrimary(vs.var('input:locality').toString(),
            'borough', ['parent.borough', 'parent.borough_a'], false);

  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addLocality(vs) {
  var o = addPrimary(vs.var('input:locality').toString(),
            'locality', ['parent.locality', 'parent.locality_a'], false);

  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addLocalAdmin(vs) {
  var o = addPrimary(vs.var('input:locality').toString(),
            'localadmin', ['parent.localadmin', 'parent.localadmin_a'], false);

  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addCounty(vs) {
  var o = addPrimary(
    vs.var('input:county').toString(),
    'county',
    [
      'parent.county',
      'parent.county_a'
    ],
    false
  );

  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addMacroCounty(vs) {
  var o = addPrimary(
    vs.var('input:county').toString(),
    'macrocounty',
    [
      'parent.macrocounty',
      'parent.macrocounty_a'
    ],
    false
  );

  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}

function addRegion(vs) {
  var o = addPrimary(
    vs.var('input:region').toString(),
    'region',
    [
      'parent.region',
      'parent.region_a'
    ],
    true
  );

  addSecCountry(vs, o);

  return o;

}

function addMacroRegion(vs) {
  var o = addPrimary(
    vs.var('input:region').toString(),
    'macroregion',
    [
      'parent.macroregion',
      'parent.macroregion_a'
    ],
    true
  );

  addSecCountry(vs, o);

  return o;

}

function addDependency(vs) {
  var o = addPrimary(
    vs.var('input:country').toString(),
    'dependency',
    [
      'parent.dependency',
      'parent.dependency_a'
    ],
    true
  );

  return o;

}

function addCountry(vs) {
  var o = addPrimary(
    vs.var('input:country').toString(),
    'country',
    [
      'parent.country',
      'parent.country_a'
    ],
    true
  );

  return o;

}

function addPostCode(vs) {
  var o = addPrimary(
    vs.var('input:postcode').toString(),
    'postalcode',
    [
      'parent.postalcode'
    ],
    false
  );

  // same position in hierarchy as borough according to WOF
  // https://github.com/whosonfirst/whosonfirst-placetypes#here-is-a-pretty-picture
  addSecLocality(vs, o);
  addSecCounty(vs, o);
  addSecRegion(vs, o);
  addSecCountry(vs, o);

  return o;

}


Layout.prototype.render = function( vs ){
  var q = Layout.base( vs );

  var funcScoreShould = q.query.function_score.query.bool.should;

  if (vs.isset('input:query')) {
    funcScoreShould.push(addQuery(vs));
  }
  if (vs.isset('input:street')) {
    if (vs.isset('input:unit') && vs.isset('input:housenumber')) {
      funcScoreShould.push(addUnitAndHouseNumberAndStreet(vs));
    } else if (vs.isset('input:housenumber')) {
      funcScoreShould.push(addHouseNumberAndStreet(vs));
    }
    funcScoreShould.push(addStreet(vs));
  } else if (vs.isset('input:housenumber')) {
      funcScoreShould.push(addHouseNumber(vs));
  }

  if (vs.isset('input:postcode')) {
    funcScoreShould.push(addPostCode(vs));
  }
  if (vs.isset('input:neighbourhood')) {
    funcScoreShould.push(addNeighbourhood(vs));
  }
  if (vs.isset('input:borough')) {
    funcScoreShould.push(addBorough(vs));
  }
  if (vs.isset('input:locality')) {
    if (!vs.isset('input:borough')) {
      funcScoreShould.push(addLocalityAsBorough(vs));
    }
    funcScoreShould.push(addLocality(vs));
    funcScoreShould.push(addLocalAdmin(vs));
  }
  if (vs.isset('input:county')) {
    funcScoreShould.push(addCounty(vs));
    funcScoreShould.push(addMacroCounty(vs));
  }
  if (vs.isset('input:region')) {
    funcScoreShould.push(addRegion(vs));
    funcScoreShould.push(addMacroRegion(vs));
  }
  if (vs.isset('input:country')) {
    funcScoreShould.push(addDependency(vs));
    funcScoreShould.push(addCountry(vs));
  }

  // handle scoring views under 'query' section (both 'must' & 'should')
  if( this._score.length ){
    this._score.forEach( function( view ){
      var rendered = view( vs );
      if( rendered ){
        q.query.function_score.functions.push( rendered );
      }
    });
  }

  // handle filter views under 'filter' section (only 'must' is allowed here)
  if( this._filter.length ){
    this._filter.forEach( function( view ){
      var rendered = view( vs );
      if( rendered ){
        if( !q.query.function_score.query.bool.hasOwnProperty( 'filter' ) ){
          q.query.function_score.query.bool.filter = {
            bool: {
              must: []
            }
          };
        }
        q.query.function_score.query.bool.filter.bool.must.push( rendered );
      }
    });
  }

  return q;
};

Layout.base = function( vs ){
  var baseQueryCopy = _.cloneDeep(baseQuery);

  baseQueryCopy.size = vs.var('size');
  baseQueryCopy.track_scores = vs.var('track_scores');

  return baseQueryCopy;

};

module.exports = Layout;