var peliasQuery = require('pelias-query'), defaults = require('./autocomplete_defaults'), textParser = require('./text_parser'), check = require('check-types'); // additional views (these may be merged in to pelias/query at a later date) var views = { ngrams_strict: require('./view/ngrams_strict'), focus_selected_layers: require('./view/focus_selected_layers') }; var ngrams_last_only = function( vs ){ // hack to disable ngrams when query parsing enabled if( vs.var('parsed_text').get() ){ return null; } var name = vs.var('input:name').get(); var vs2 = new peliasQuery.Vars( vs.export() ); vs2.var('input:name').set( name.substr( name.lastIndexOf(' ')+1 ) ); var view = views.ngrams_strict( vs2 ); view.match['name.default'].analyzer = 'peliasPhrase'; return view; }; var phrase_first_only = function( vs ){ // hack to disable substr when query parsing enabled if( !vs.var('parsed_text').get() ){ var name = vs.var('input:name').get(); var s = name.split(' '); // single token only, abort if( s.length < 2 ){ return null; } var vs2 = new peliasQuery.Vars( vs.export() ); vs2.var('input:name').set( name.substr(0, name.lastIndexOf(' ') ) ); return peliasQuery.view.phrase( vs2 ); } return peliasQuery.view.phrase( vs ); }; //------------------------------ // autocomplete query //------------------------------ var query = new peliasQuery.layout.FilteredBooleanQuery(); // mandatory matches query.score( phrase_first_only, 'must' ); query.score( ngrams_last_only, 'must' ); // address components query.score( peliasQuery.view.address('housenumber') ); query.score( peliasQuery.view.address('street') ); query.score( peliasQuery.view.address('postcode') ); // admin components query.score( peliasQuery.view.admin('alpha3') ); query.score( peliasQuery.view.admin('admin0') ); query.score( peliasQuery.view.admin('admin1') ); query.score( peliasQuery.view.admin('admin1_abbr') ); query.score( peliasQuery.view.admin('admin2') ); query.score( peliasQuery.view.admin('local_admin') ); query.score( peliasQuery.view.admin('locality') ); query.score( peliasQuery.view.admin('neighborhood') ); // scoring boost query.score( views.focus_selected_layers( views.ngrams_strict ) ); query.score( peliasQuery.view.popularity( views.ngrams_strict ) ); query.score( peliasQuery.view.population( views.ngrams_strict ) ); // -------------------------------- /** map request variables to query variables for all inputs provided by this HTTP request. **/ function generateQuery( clean ){ var vs = new peliasQuery.Vars( defaults ); vs.var( 'parsed_text', false ); // remove single grams at end var text = clean.text.replace(/( .$)/g,'').trim(); if( clean.hasOwnProperty('parsed_text') ){ if( clean.parsed_text.hasOwnProperty('name') ){ vs.var( 'parsed_text', true ); text = clean.parsed_text.name; } } // input text vs.var( 'input:name', text ); // focus point if( check.number(clean['focus.point.lat']) && check.number(clean['focus.point.lon']) ){ vs.set({ 'focus:point:lat': clean['focus.point.lat'], 'focus:point:lon': clean['focus.point.lon'] }); } // run the address parser if( clean.parsed_text ){ textParser( clean.parsed_text, vs ); } return query.render( vs ); } module.exports = generateQuery;