mirror of https://github.com/pelias/api.git
Diana Shkolnikov
8 years ago
committed by
GitHub
6 changed files with 599 additions and 2 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