var geojsonify = require('../helper/geojsonify').suggest; var async = require('async'); function setup( backend, query ){ // allow overriding of dependencies backend = backend || require('../src/backend'); query = query || require('../query/suggest'); function controller( req, res, next ){ var cmd = { index: 'pelias', body: query( req.clean ) }; var SIZE = req.clean.size || 10; var query_backend = function(cmd, callback) { // query backend backend().client.suggest( cmd, function( err, data ){ var docs = []; // handle backend errors if( err ){ return next( err ); } // map response to a valid FeatureCollection if( data && Array.isArray( data.pelias ) && data.pelias.length ){ docs = data['pelias'][0].options || []; } callback(null, docs); }); }; var dedup = function(combined) { var unique_ids = []; return combined.filter(function(item, pos) { if (unique_ids.indexOf(item.payload.id) == -1) { unique_ids.push(item.payload.id); return true; } return false; }); }; var respond = function(data) { // convert docs to geojson var geojson = geojsonify( data ); // response envelope geojson.date = new Date().getTime(); // respond return res.status(200).json( geojson ); }; if (req.clean.input) { var async_query; // admin only req.admin = {}; for (k in req.clean) { req.admin[k] = req.clean[k] } req.admin.layers = ['admin0','admin1','admin2']; if (req.clean.input.length < 4 && isNaN(parseInt(req.clean.input, 10))) { async_query = { a: function(callback){ cmd.body = query( req.admin, 3 ); query_backend(cmd, callback); }, b: function(callback){ cmd.body = query( req.admin, 1 ); query_backend(cmd, callback); }, c: function(callback) { cmd.body = query( req.clean, 3 ); query_backend(cmd, callback); } } } else { async_query = { a: function(callback){ cmd.body = query( req.clean, 5); query_backend(cmd, callback); }, b: function(callback){ cmd.body = query( req.clean, 3); query_backend(cmd, callback); }, c: function(callback){ cmd.body = query( req.clean, 1 ); query_backend(cmd, callback); }, d: function(callback){ cmd.body = query( req.admin ); query_backend(cmd, callback); } } } async.parallel(async_query, function(err, results) { var splice_length = parseInt((SIZE / Object.keys(results).length), 10); // results is equal to: {a: docs, b: docs, c: docs} var combined = []; for (keys in results) { combined = combined.concat(results[keys].splice(0,splice_length)); } combined = dedup(combined); respond(combined); }); } else { query_backend(cmd, function(err, results) { respond(results); }); } } return controller; } module.exports = setup;