mirror of https://github.com/pelias/api.git
Diana Shkolnikov
8 years ago
committed by
GitHub
9 changed files with 686 additions and 6 deletions
@ -0,0 +1,172 @@ |
|||||||
|
|
||||||
|
var logger = require( 'pelias-logger' ).get( 'api' ); |
||||||
|
var service = require('../service/language'); |
||||||
|
|
||||||
|
/** |
||||||
|
example response from language web service: |
||||||
|
{ |
||||||
|
"101748479": { |
||||||
|
"wofid": 101748479, |
||||||
|
"placetype": "locality", |
||||||
|
"iso": "DE", |
||||||
|
"area": 0.031614, |
||||||
|
"lineage": { |
||||||
|
"continent_id": 102191581, |
||||||
|
"country_id": 85633111, |
||||||
|
"county_id": 102063261, |
||||||
|
"locality_id": 101748479, |
||||||
|
"macrocounty_id": 404227567, |
||||||
|
"region_id": 85682571 |
||||||
|
}, |
||||||
|
"rowid": 90425, |
||||||
|
"names": { |
||||||
|
"default": "München", |
||||||
|
"eng": "Munich" |
||||||
|
} |
||||||
|
}, |
||||||
|
} |
||||||
|
**/ |
||||||
|
|
||||||
|
function setup() { |
||||||
|
|
||||||
|
var transport = service.findById(); |
||||||
|
var middleware = function(req, res, next) { |
||||||
|
|
||||||
|
// no-op, request did not require a language change
|
||||||
|
if( !isLanguageChangeRequired( req, res ) ){ |
||||||
|
return next(); |
||||||
|
} |
||||||
|
|
||||||
|
// collect a list of parent ids to fetch translations for
|
||||||
|
var ids = extractIds( res ); |
||||||
|
|
||||||
|
// perform language lookup for all relevant ids
|
||||||
|
var timer = (new Date()).getTime(); |
||||||
|
transport.query( ids, function( err, translations ){ |
||||||
|
|
||||||
|
// update documents using a translation map
|
||||||
|
if( err ){ |
||||||
|
logger.error( '[language] [error]', err ); |
||||||
|
} else { |
||||||
|
updateDocs( req, res, translations ); |
||||||
|
} |
||||||
|
|
||||||
|
logger.info( '[language] [took]', (new Date()).getTime() - timer, 'ms' ); |
||||||
|
next(); |
||||||
|
}); |
||||||
|
}; |
||||||
|
|
||||||
|
middleware.transport = transport; |
||||||
|
return middleware; |
||||||
|
} |
||||||
|
|
||||||
|
// collect a list of parent ids to fetch translations for
|
||||||
|
function extractIds( res ){ |
||||||
|
|
||||||
|
// store ids in an object in order to avoid duplicates
|
||||||
|
var ids = {}; |
||||||
|
|
||||||
|
// convenience function for adding a new id to the object
|
||||||
|
function addId(id) { |
||||||
|
ids[id] = true; |
||||||
|
} |
||||||
|
|
||||||
|
// extract all parent ids from documents
|
||||||
|
res.data.forEach( function( doc ){ |
||||||
|
|
||||||
|
// skip invalid records
|
||||||
|
if( !doc || !doc.parent ){ return; } |
||||||
|
|
||||||
|
// iterate over doc.parent.* attributes
|
||||||
|
for( var attr in doc.parent ){ |
||||||
|
|
||||||
|
// match only attributes ending with '_id'
|
||||||
|
var match = attr.match(/_id$/); |
||||||
|
if( !match ){ continue; } |
||||||
|
|
||||||
|
// skip invalid/empty arrays
|
||||||
|
if( !Array.isArray( doc.parent[attr] ) || !doc.parent[attr].length ){ |
||||||
|
continue; |
||||||
|
} |
||||||
|
|
||||||
|
// add each id as a key in the ids object
|
||||||
|
doc.parent[attr].forEach( addId ); |
||||||
|
} |
||||||
|
}); |
||||||
|
|
||||||
|
// return a deduplicated array of ids
|
||||||
|
return Object.keys( ids ); |
||||||
|
} |
||||||
|
|
||||||
|
// update documents using a translation map
|
||||||
|
function updateDocs( req, res, translations ){ |
||||||
|
|
||||||
|
// sanity check arguments
|
||||||
|
if( !req || !res || !res.data || !translations ){ return; } |
||||||
|
|
||||||
|
// this is the target language we will be translating to
|
||||||
|
var requestLanguage = req.language.iso6393; |
||||||
|
|
||||||
|
// iterate over response documents
|
||||||
|
res.data.forEach( function( doc, p ){ |
||||||
|
|
||||||
|
// skip invalid records
|
||||||
|
if( !doc || !doc.parent ){ return; } |
||||||
|
|
||||||
|
// iterate over doc.parent.* attributes
|
||||||
|
for( var attr in doc.parent ){ |
||||||
|
|
||||||
|
// match only attributes ending with '_id'
|
||||||
|
var match = attr.match(/^(.*)_id$/); |
||||||
|
if( !match ){ continue; } |
||||||
|
|
||||||
|
// adminKey is the property name without the '_id'
|
||||||
|
// eg. for 'country_id', adminKey would be 'country'.
|
||||||
|
var adminKey = match[1]; |
||||||
|
var adminValues = doc.parent[adminKey]; |
||||||
|
|
||||||
|
// skip invalid/empty arrays
|
||||||
|
if( !Array.isArray( adminValues ) || !adminValues.length ){ continue; } |
||||||
|
|
||||||
|
// iterate over adminValues (it's an array and can have more than one value)
|
||||||
|
for( var i in adminValues ){ |
||||||
|
|
||||||
|
// find the corresponding key from the '_id' Array
|
||||||
|
var id = doc.parent[attr][i]; |
||||||
|
if( !id ){ continue; } |
||||||
|
|
||||||
|
// id not found in translation service response
|
||||||
|
if( !translations.hasOwnProperty( id ) ){ |
||||||
|
logger.error( '[language] [error]', 'failed to find translations for', id ); |
||||||
|
continue; |
||||||
|
} |
||||||
|
|
||||||
|
// skip invalid records
|
||||||
|
if( !translations[id].hasOwnProperty( 'names' ) ){ continue; } |
||||||
|
|
||||||
|
// requested language is not available
|
||||||
|
if( !translations[id].names.hasOwnProperty( requestLanguage ) ){ |
||||||
|
logger.info( '[language] [info]', 'missing translation', requestLanguage, id ); |
||||||
|
continue; |
||||||
|
} |
||||||
|
|
||||||
|
// translate 'parent.*' property
|
||||||
|
adminValues[i] = translations[id].names[ requestLanguage ]; |
||||||
|
|
||||||
|
// if the record is an admin record we also translate
|
||||||
|
// the 'name.default' property.
|
||||||
|
if( adminKey === doc.layer ){ |
||||||
|
doc.name.default = translations[id].names[ requestLanguage ]; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
// boolean function to check if changing the language is required
|
||||||
|
function isLanguageChangeRequired( req, res ){ |
||||||
|
return req && res && res.data && res.data.length && |
||||||
|
req.hasOwnProperty('language'); |
||||||
|
} |
||||||
|
|
||||||
|
module.exports = setup; |
||||||
@ -0,0 +1,93 @@ |
|||||||
|
|
||||||
|
var logger = require( 'pelias-logger' ).get( 'api' ), |
||||||
|
request = require( 'superagent' ), |
||||||
|
peliasConfig = require( 'pelias-config' ); |
||||||
|
|
||||||
|
/** |
||||||
|
|
||||||
|
language subsitution service client |
||||||
|
|
||||||
|
this file provides a 'transport' which can be used to access the language |
||||||
|
service via a network connnection. |
||||||
|
|
||||||
|
the exported method for this module checks pelias-config for a configuration block such as: |
||||||
|
|
||||||
|
"language": { |
||||||
|
"client": { |
||||||
|
"adapter": "http", |
||||||
|
"host": "http://localhost:6100" |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
for more info on running the service see: https://github.com/pelias/placeholder
|
||||||
|
|
||||||
|
**/ |
||||||
|
|
||||||
|
/** |
||||||
|
NullTransport |
||||||
|
|
||||||
|
disables the service completely |
||||||
|
**/ |
||||||
|
function NullTransport(){} |
||||||
|
NullTransport.prototype.query = function( coord, number, street, cb ){ |
||||||
|
cb(); // no-op
|
||||||
|
}; |
||||||
|
|
||||||
|
/** |
||||||
|
HttpTransport |
||||||
|
|
||||||
|
allows the api to be used via a remote web service |
||||||
|
**/ |
||||||
|
function HttpTransport( host, settings ){ |
||||||
|
this.query = function( ids, cb ){ |
||||||
|
request |
||||||
|
.get( host + '/parser/findbyid' ) |
||||||
|
.set( 'Accept', 'application/json' ) |
||||||
|
.query({ ids: Array.isArray( ids ) ? ids.join(',') : '' }) |
||||||
|
.timeout( settings && settings.timeout || 1000 ) |
||||||
|
.end( function( err, res ){ |
||||||
|
if( err || !res ){ return cb( err ); } |
||||||
|
if( 200 !== res.status ){ return cb( 'non 200 status' ); } |
||||||
|
return cb( null, res.body ); |
||||||
|
}); |
||||||
|
}; |
||||||
|
} |
||||||
|
HttpTransport.prototype.query = function( coord, number, street, cb ){ |
||||||
|
throw new Error( 'language: transport not connected' ); |
||||||
|
}; |
||||||
|
|
||||||
|
/** |
||||||
|
Setup |
||||||
|
|
||||||
|
allows instantiation of transport depending on configuration and preference |
||||||
|
**/ |
||||||
|
module.exports.findById = function setup(){ |
||||||
|
|
||||||
|
// user config
|
||||||
|
var config = peliasConfig.generate(); |
||||||
|
|
||||||
|
// ensure config variables set correctly
|
||||||
|
if( !config.hasOwnProperty('language') || !config.language.hasOwnProperty('client') ){ |
||||||
|
logger.warn( 'language: configuration not found' ); |
||||||
|
} |
||||||
|
|
||||||
|
// valid configuration found
|
||||||
|
else { |
||||||
|
|
||||||
|
// get adapter settings from config
|
||||||
|
var settings = config.language.client; |
||||||
|
|
||||||
|
// http adapter
|
||||||
|
if( 'http' === settings.adapter && settings.hasOwnProperty('host') ){ |
||||||
|
logger.info( 'language: using http transport:', settings.host ); |
||||||
|
if( settings.hasOwnProperty('timeout') ){ |
||||||
|
return new HttpTransport( settings.host, { timeout: parseInt( settings.timeout, 10 ) } ); |
||||||
|
} |
||||||
|
return new HttpTransport( settings.host ); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// default adapter
|
||||||
|
logger.info( 'language: using null transport' ); |
||||||
|
return new NullTransport(); |
||||||
|
}; |
||||||
@ -0,0 +1,107 @@ |
|||||||
|
|
||||||
|
var fs = require('fs'), |
||||||
|
tmp = require('tmp'), |
||||||
|
setup = require('../../../service/language').findById; |
||||||
|
|
||||||
|
module.exports.tests = {}; |
||||||
|
|
||||||
|
module.exports.tests.interface = function(test, common) { |
||||||
|
test('valid interface', function(t) { |
||||||
|
t.equal(typeof setup, 'function', 'setup is a function'); |
||||||
|
t.end(); |
||||||
|
}); |
||||||
|
}; |
||||||
|
|
||||||
|
// adapter factory
|
||||||
|
module.exports.tests.factory = function(test, common) { |
||||||
|
|
||||||
|
test('http adapter', function(t) { |
||||||
|
var config = { language: { client: { |
||||||
|
adapter: 'http', |
||||||
|
host: 'http://example.com' |
||||||
|
}}}; |
||||||
|
|
||||||
|
// adapter is driven by config
|
||||||
|
var tmpfile = tmp.tmpNameSync({ postfix: '.json' }); |
||||||
|
fs.writeFileSync( tmpfile, JSON.stringify( config ), { encoding: 'utf8' } ); |
||||||
|
process.env.PELIAS_CONFIG = tmpfile; |
||||||
|
var adapter = setup(); |
||||||
|
delete process.env.PELIAS_CONFIG; |
||||||
|
|
||||||
|
t.equal(adapter.constructor.name, 'HttpTransport', 'HttpTransport'); |
||||||
|
t.equal(typeof adapter, 'object', 'adapter is an object'); |
||||||
|
t.equal(typeof adapter.query, 'function', 'query is a function'); |
||||||
|
t.equal(adapter.query.length, 2, 'query function signature'); |
||||||
|
t.end(); |
||||||
|
}); |
||||||
|
|
||||||
|
test('null adapter', function(t) { |
||||||
|
var config = { language: { client: { |
||||||
|
adapter: 'null' |
||||||
|
}}}; |
||||||
|
|
||||||
|
// adapter is driven by config
|
||||||
|
var tmpfile = tmp.tmpNameSync({ postfix: '.json' }); |
||||||
|
fs.writeFileSync( tmpfile, JSON.stringify( config ), { encoding: 'utf8' } ); |
||||||
|
process.env.PELIAS_CONFIG = tmpfile; |
||||||
|
var adapter = setup(); |
||||||
|
delete process.env.PELIAS_CONFIG; |
||||||
|
|
||||||
|
t.equal(adapter.constructor.name, 'NullTransport', 'NullTransport'); |
||||||
|
t.equal(typeof adapter, 'object', 'adapter is an object'); |
||||||
|
t.equal(typeof adapter.query, 'function', 'query is a function'); |
||||||
|
t.equal(adapter.query.length, 4, 'query function signature'); |
||||||
|
t.end(); |
||||||
|
}); |
||||||
|
|
||||||
|
test('default adapter', function(t) { |
||||||
|
var config = {}; |
||||||
|
|
||||||
|
// adapter is driven by config
|
||||||
|
var tmpfile = tmp.tmpNameSync({ postfix: '.json' }); |
||||||
|
fs.writeFileSync( tmpfile, JSON.stringify( config ), { encoding: 'utf8' } ); |
||||||
|
process.env.PELIAS_CONFIG = tmpfile; |
||||||
|
var adapter = setup(); |
||||||
|
delete process.env.PELIAS_CONFIG; |
||||||
|
|
||||||
|
t.equal(adapter.constructor.name, 'NullTransport', 'NullTransport'); |
||||||
|
t.equal(typeof adapter, 'object', 'adapter is an object'); |
||||||
|
t.equal(typeof adapter.query, 'function', 'query is a function'); |
||||||
|
t.equal(adapter.query.length, 4, 'query function signature'); |
||||||
|
t.end(); |
||||||
|
}); |
||||||
|
|
||||||
|
}; |
||||||
|
|
||||||
|
// null transport
|
||||||
|
module.exports.tests.NullTransport = function(test, common) { |
||||||
|
|
||||||
|
test('null transport', function(t) { |
||||||
|
|
||||||
|
// adapter is driven by config
|
||||||
|
var tmpfile = tmp.tmpNameSync({ postfix: '.json' }); |
||||||
|
fs.writeFileSync( tmpfile, '{}', { encoding: 'utf8' } ); |
||||||
|
process.env.PELIAS_CONFIG = tmpfile; |
||||||
|
var adapter = setup(); |
||||||
|
delete process.env.PELIAS_CONFIG; |
||||||
|
|
||||||
|
// test null transport performs a no-op
|
||||||
|
adapter.query( null, null, null, function( err, res ){ |
||||||
|
t.equal(err, undefined, 'no-op'); |
||||||
|
t.equal(res, undefined, 'no-op'); |
||||||
|
t.end(); |
||||||
|
}); |
||||||
|
}); |
||||||
|
|
||||||
|
}; |
||||||
|
|
||||||
|
module.exports.all = function (tape, common) { |
||||||
|
|
||||||
|
function test(name, testFunction) { |
||||||
|
return tape('SERVICE language', testFunction); |
||||||
|
} |
||||||
|
|
||||||
|
for( var testCase in module.exports.tests ){ |
||||||
|
module.exports.tests[testCase](test, common); |
||||||
|
} |
||||||
|
}; |
||||||
Loading…
Reference in new issue