Added location resolver.

This commit is contained in:
DebaucheryLibrarian 2024-10-31 02:50:13 +01:00
parent 3bf8ce6e23
commit b2d040dc3c
1 changed files with 72 additions and 0 deletions

72
geo.mjs Normal file
View File

@ -0,0 +1,72 @@
export async function resolvePlace(query, context) {
if (!query) {
return null;
}
const cacheKey = `place-${context.slugify(query)}`;
const cachedPlace = await context.redis.hGetAll(cacheKey);
if (context.argv.placeCache !== false && await context.redis.exists(cacheKey)) {
await context.redis.expire(cacheKey, 3600 * 24 * 30);
context.logger.debug(`Using cached place '${cacheKey}' for query '${query}': ${JSON.stringify(cachedPlace)}`);
return cachedPlace;
}
// query is a nationality, lookup would get weird results (British resolves to British, Northern Ireland)
const country = await context.knex('countries')
.where('nationality', 'ilike', `%${query}%`)
.orWhere('alpha3', 'ilike', `%${query}%`)
.orWhere('alpha2', 'ilike', `%${query}%`)
.orderBy('priority', 'desc')
.first();
if (country) {
return {
country: country.alpha2,
};
}
try {
// https://operations.osmfoundation.org/policies/nominatim/
const res = await context.unprint.get(`https://nominatim.openstreetmap.org/search?q=${encodeURI(query)}&format=json&accept-language=en&addressdetails=1`, {
headers: {
'User-Agent': context.config.location.userAgent,
},
interval: 1000,
concurrency: 1,
});
const [item] = res.body;
if (item && item.address) {
const rawPlace = item.address;
const place = {};
if (item.class === 'place' || item.class === 'boundary') {
const location = rawPlace[item.type] || rawPlace.city || rawPlace.place || rawPlace.town;
if (location) {
place.place = location;
place.city = rawPlace.city || location;
}
}
if (rawPlace.state) place.state = rawPlace.state;
if (rawPlace.country_code) place.country = rawPlace.country_code.toUpperCase();
if (rawPlace.continent) place.continent = rawPlace.continent;
context.logger.debug(`Resolved place '${query}' to ${JSON.stringify(place)}`);
await context.redis.hSet(cacheKey, place);
await context.redis.expire(cacheKey, 3600 * 24 * 30);
return place;
}
} catch (error) {
context.logger.error(`Failed to resolve place '${query}': ${error.message}`);
}
return null;
}