mirror of
https://github.com/FoggedLens/deflock.git
synced 2026-02-12 15:02:45 +00:00
89 lines
2.5 KiB
TypeScript
89 lines
2.5 KiB
TypeScript
|
|
import { createCache, Cache } from 'cache-manager';
|
|
import { Type, Static } from '@sinclair/typebox';
|
|
const { DiskStore } = require('cache-manager-fs-hash');
|
|
|
|
export const NominatimResultSchema = Type.Object({
|
|
addresstype: Type.String(),
|
|
boundingbox: Type.Tuple([Type.String(), Type.String(), Type.String(), Type.String()]),
|
|
class: Type.String(),
|
|
display_name: Type.String(),
|
|
geojson: Type.Object({
|
|
coordinates: Type.Any(),
|
|
type: Type.String(),
|
|
}),
|
|
importance: Type.Number(),
|
|
lat: Type.String(),
|
|
licence: Type.String(),
|
|
lon: Type.String(),
|
|
name: Type.String(),
|
|
osm_id: Type.Number(),
|
|
osm_type: Type.String(),
|
|
place_id: Type.Number(),
|
|
place_rank: Type.Number(),
|
|
type: Type.String(),
|
|
});
|
|
|
|
export type NominatimResult = Static<typeof NominatimResultSchema>;
|
|
|
|
const cache: Cache = createCache({
|
|
stores: [new DiskStore({
|
|
path: '/tmp/nominatim-cache',
|
|
ttl: 3600 * 24, // 24 hours
|
|
maxsize: 1000 * 1000 * 100, // 100MB
|
|
subdirs: true,
|
|
zip: false,
|
|
})]
|
|
});
|
|
|
|
export class NominatimClient {
|
|
baseUrl = 'https://nominatim.openstreetmap.org/search';
|
|
|
|
async geocodePhrase(query: string): Promise<NominatimResult[]> {
|
|
const cacheKey = `geocode:${query}`;
|
|
const cached = await cache.get(cacheKey);
|
|
if (cached) {
|
|
return cached as NominatimResult[];
|
|
}
|
|
const url = `${this.baseUrl}?q=${encodeURIComponent(query)}&polygon_geojson=1&format=json`;
|
|
const response = await fetch(url, {
|
|
headers: {
|
|
'User-Agent': 'DeFlock/1.1',
|
|
},
|
|
});
|
|
if (!response.ok) {
|
|
throw new Error(`Failed to geocode phrase: ${response.status}`);
|
|
}
|
|
const json = await response.json();
|
|
await cache.set(cacheKey, json);
|
|
return json;
|
|
}
|
|
|
|
async geocodeSingleResult(query: string): Promise<NominatimResult | null> {
|
|
const results = await this.geocodePhrase(query);
|
|
|
|
if (!results.length) return null;
|
|
|
|
const cityStatePattern = /(.+),\s*(\w{2})/;
|
|
const postalCodePattern = /\d{5}/;
|
|
|
|
if (cityStatePattern.test(query)) {
|
|
const cityStateResults = results.filter((result: NominatimResult) =>
|
|
["city", "town", "village", "hamlet", "suburb", "quarter", "neighbourhood", "borough"].includes(result.addresstype)
|
|
);
|
|
if (cityStateResults.length) {
|
|
return cityStateResults[0];
|
|
}
|
|
}
|
|
|
|
if (postalCodePattern.test(query)) {
|
|
const postalCodeResults = results.filter((result: NominatimResult) => result.addresstype === "postcode");
|
|
if (postalCodeResults.length) {
|
|
return postalCodeResults[0];
|
|
}
|
|
}
|
|
|
|
return results[0];
|
|
}
|
|
}
|