var peliasQuery = require('pelias-query'), sort = require('../query/sort'), adminFields = require('../helper/adminFields')(); //------------------------------ // general-purpose search query //------------------------------ var query = new peliasQuery.layout.FilteredBooleanQuery(); // mandatory matches query.score( peliasQuery.view.boundary_country, 'must' ); query.score( peliasQuery.view.ngrams, 'must' ); // scoring boost query.score( peliasQuery.view.phrase ); query.score( peliasQuery.view.focus ); // 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') ); // non-scoring hard filters query.filter( peliasQuery.view.boundary_circle ); query.filter( peliasQuery.view.boundary_rect ); // -------------------------------- function generateQuery( clean ){ var vs = new peliasQuery.Vars( peliasQuery.defaults ); // set input text vs.var( 'input:name', clean.input ); // set size if( clean.size ){ vs.var( 'size', clean.size ); } // focus point if( clean.lat && clean.lon ){ vs.set({ 'focus:point:lat': clean.lat, 'focus:point:lon': clean.lon }); } // bbox if( clean.bbox ){ vs.set({ 'boundary:rect:top': clean.bbox.top, 'boundary:rect:right': clean.bbox.right, 'boundary:rect:bottom': clean.bbox.bottom, 'boundary:rect:left': clean.bbox.left }); } // address parsing if( clean.parsed_input ){ // is it a street address? var isStreetAddress = clean.parsed_input.hasOwnProperty('number') && clean.parsed_input.hasOwnProperty('street'); if( isStreetAddress ){ vs.var( 'input:name', clean.parsed_input.number + ' ' + clean.parsed_input.street ); } // I don't understand this else if( clean.parsed_input.admin_parts ) { vs.var( 'input:name', clean.parsed_input.name ); } // or this.. else { console.warn( 'chaos monkey asks: what happens now?' ); console.log( clean ); try{ throw new Error(); } catch(e){ console.error( e.stack ); } // print a stack trace } // ==== add parsed matches [address components] ==== // house number if( clean.parsed_input.hasOwnProperty('number') ){ vs.var( 'input:housenumber', clean.parsed_input.number ); } // street name if( clean.parsed_input.hasOwnProperty('street') ){ vs.var( 'input:street', clean.parsed_input.street ); } // postal code if( clean.parsed_input.hasOwnProperty('postalcode') ){ vs.var( 'input:postcode', clean.parsed_input.postalcode ); } // ==== add parsed matches [admin components] ==== // city if( clean.parsed_input.hasOwnProperty('city') ){ vs.var( 'input:admin2', clean.parsed_input.city ); } // state if( clean.parsed_input.hasOwnProperty('state') ){ vs.var( 'input:admin1_abbr', clean.parsed_input.state ); } // country if( clean.parsed_input.hasOwnProperty('country') ){ vs.var( 'input:alpha3', clean.parsed_input.country ); } // ==== deal with the 'leftover' components ==== // @todo: clean up this code // a concept called 'leftovers' which is just 'admin_parts' /or 'regions'. var leftoversString = ''; if( clean.parsed_input.hasOwnProperty('admin_parts') ){ leftoversString = clean.parsed_input.admin_parts; } else if( clean.parsed_input.hasOwnProperty('regions') ){ leftoversString = clean.parsed_input.regions.join(' '); } // if we have 'leftovers' then assign them to any fields which // currently don't have a value assigned. if( leftoversString.length ){ var unmatchedAdminFields = adminFields.slice(); // cycle through fields and set fields which // are still currently unset unmatchedAdminFields.forEach( function( key ){ if( !vs.isset( 'input:' + key ) ){ vs.var( 'input:' + key, leftoversString ); } }); } } var result = query.render( vs ); // @todo: remove unnessesary sort conditions result.sort = result.sort.concat( sort( clean ) ); // @todo: remove this hack return JSON.parse( JSON.stringify( result ) ); } module.exports = generateQuery;