mirror of https://github.com/pelias/api.git
Stephen Hess
8 years ago
1 changed files with 131 additions and 0 deletions
@ -0,0 +1,131 @@
|
||||
var peliasQuery = require('pelias-query'), |
||||
defaults = require('./search_defaults'), |
||||
textParser = require('./text_parser'), |
||||
check = require('check-types'); |
||||
|
||||
var placeTypes = require('../helper/placeTypes'); |
||||
|
||||
// region_a is also an admin field. addressit tries to detect
|
||||
// region_a, in which case we use a match query specifically for it.
|
||||
// but address it doesn't know about all of them so it helps to search
|
||||
// against this with the other admin parts as a fallback
|
||||
var adminFields = placeTypes.concat(['region_a']); |
||||
|
||||
//------------------------------
|
||||
// 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( peliasQuery.view.phrase ) ); |
||||
query.score( peliasQuery.view.popularity( peliasQuery.view.phrase ) ); |
||||
query.score( peliasQuery.view.population( peliasQuery.view.phrase ) ); |
||||
|
||||
// address components
|
||||
query.score( peliasQuery.view.address('housenumber') ); |
||||
query.score( peliasQuery.view.address('street') ); |
||||
query.score( peliasQuery.view.address('postcode') ); |
||||
|
||||
// admin components
|
||||
// country_a and region_a are left as matches here because the text-analyzer
|
||||
// can sometimes detect them, in which case a query more specific than a
|
||||
// multi_match is appropriate.
|
||||
query.score( peliasQuery.view.admin('country_a') ); |
||||
query.score( peliasQuery.view.admin('region_a') ); |
||||
query.score( peliasQuery.view.admin_multi_match(adminFields, 'peliasAdmin') ); |
||||
|
||||
// non-scoring hard filters
|
||||
query.filter( peliasQuery.view.boundary_circle ); |
||||
query.filter( peliasQuery.view.boundary_rect ); |
||||
query.filter( peliasQuery.view.sources ); |
||||
query.filter( peliasQuery.view.layers ); |
||||
query.filter( peliasQuery.view.categories ); |
||||
|
||||
// --------------------------------
|
||||
|
||||
/** |
||||
map request variables to query variables for all inputs |
||||
provided by this HTTP request. |
||||
**/ |
||||
function generateQuery( clean ){ |
||||
|
||||
var vs = new peliasQuery.Vars( defaults ); |
||||
|
||||
// input text
|
||||
vs.var( 'input:name', clean.text ); |
||||
|
||||
// sources
|
||||
vs.var( 'sources', clean.sources); |
||||
|
||||
// layers
|
||||
vs.var( 'layers', clean.layers); |
||||
|
||||
// categories
|
||||
if (clean.categories) { |
||||
vs.var('input:categories', clean.categories); |
||||
} |
||||
|
||||
// size
|
||||
if( clean.querySize ) { |
||||
vs.var( 'size', clean.querySize ); |
||||
} |
||||
|
||||
// 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'] |
||||
}); |
||||
} |
||||
|
||||
// boundary rect
|
||||
if( check.number(clean['boundary.rect.min_lat']) && |
||||
check.number(clean['boundary.rect.max_lat']) && |
||||
check.number(clean['boundary.rect.min_lon']) && |
||||
check.number(clean['boundary.rect.max_lon']) ){ |
||||
vs.set({ |
||||
'boundary:rect:top': clean['boundary.rect.max_lat'], |
||||
'boundary:rect:right': clean['boundary.rect.max_lon'], |
||||
'boundary:rect:bottom': clean['boundary.rect.min_lat'], |
||||
'boundary:rect:left': clean['boundary.rect.min_lon'] |
||||
}); |
||||
} |
||||
|
||||
// boundary circle
|
||||
// @todo: change these to the correct request variable names
|
||||
if( check.number(clean['boundary.circle.lat']) && |
||||
check.number(clean['boundary.circle.lon']) ){ |
||||
vs.set({ |
||||
'boundary:circle:lat': clean['boundary.circle.lat'], |
||||
'boundary:circle:lon': clean['boundary.circle.lon'] |
||||
}); |
||||
|
||||
if( check.number(clean['boundary.circle.radius']) ){ |
||||
vs.set({ |
||||
'boundary:circle:radius': Math.round( clean['boundary.circle.radius'] ) + 'km' |
||||
}); |
||||
} |
||||
} |
||||
|
||||
// boundary country
|
||||
if( check.string(clean['boundary.country']) ){ |
||||
vs.set({ |
||||
'boundary:country': clean['boundary.country'] |
||||
}); |
||||
} |
||||
|
||||
// run the address parser
|
||||
if( clean.parsed_text ){ |
||||
textParser( clean.parsed_text, vs ); |
||||
} |
||||
|
||||
return query.render( vs ); |
||||
} |
||||
|
||||
module.exports = generateQuery; |
Loading…
Reference in new issue