|
|
|
const peliasQuery = require('pelias-query'),
|
|
|
|
defaults = require('./search_defaults'),
|
|
|
|
textParser = require('./text_parser'),
|
|
|
|
check = require('check-types');
|
|
|
|
|
|
|
|
//------------------------------
|
|
|
|
// general-purpose search query
|
|
|
|
//------------------------------
|
|
|
|
const structuredQuery = new peliasQuery.layout.StructuredFallbackQuery();
|
|
|
|
|
|
|
|
// scoring boost
|
|
|
|
structuredQuery.score( peliasQuery.view.focus_only_function( peliasQuery.view.phrase ) );
|
|
|
|
structuredQuery.score( peliasQuery.view.popularity_only_function );
|
|
|
|
structuredQuery.score( peliasQuery.view.population_only_function );
|
|
|
|
// --------------------------------
|
|
|
|
|
|
|
|
// non-scoring hard filters
|
|
|
|
structuredQuery.filter( peliasQuery.view.boundary_country );
|
|
|
|
structuredQuery.filter( peliasQuery.view.boundary_circle );
|
|
|
|
structuredQuery.filter( peliasQuery.view.boundary_rect );
|
|
|
|
structuredQuery.filter( peliasQuery.view.sources );
|
|
|
|
structuredQuery.filter( peliasQuery.view.layers );
|
|
|
|
structuredQuery.filter( peliasQuery.view.categories );
|
|
|
|
// --------------------------------
|
|
|
|
|
|
|
|
/**
|
|
|
|
map request variables to query variables for all inputs
|
|
|
|
provided by this HTTP request.
|
|
|
|
**/
|
|
|
|
function generateQuery( clean ){
|
|
|
|
|
|
|
|
const 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);
|
|
|
|
|
|
|
|
// 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 );
|
|
|
|
}
|
|
|
|
|
|
|
|
const q = getQuery(vs);
|
|
|
|
|
|
|
|
// console.log(JSON.stringify(q.body, null, 2));
|
|
|
|
|
|
|
|
return q;
|
|
|
|
}
|
|
|
|
|
|
|
|
function getQuery(vs) {
|
|
|
|
return {
|
|
|
|
type: 'structured',
|
|
|
|
body: structuredQuery.render(vs)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
module.exports = generateQuery;
|