import { Client } from "pg" import http from "node:http" import { Tiles } from "../../src/Models/TileRange" /** * Just the OSM2PGSL default database */ interface PoiDatabaseMeta { attributes current_timestamp db_format flat_node_file import_timestamp output prefix replication_base_url replication_sequence_number replication_timestamp style updatable version } /** * Connects with a Postgis database, gives back how much items there are within the given BBOX */ class OsmPoiDatabase { private static readonly prefixes: ReadonlyArray = ["pois", "lines", "polygons"] private readonly _client: Client private isConnected = false private supportedLayers: Set = undefined private supportedLayersDate: Date = undefined private metaCache: PoiDatabaseMeta = undefined private metaCacheDate: Date = undefined constructor(connectionString: string) { this._client = new Client(connectionString) } async getCount( layer: string, bbox: [[number, number], [number, number]] = undefined ): Promise { if (!this.isConnected) { await this._client.connect() this.isConnected = true } let total = 0 for (const prefix of OsmPoiDatabase.prefixes) { let query = "SELECT COUNT(*) FROM " + prefix + "_" + layer if (bbox) { query += ` WHERE ST_MakeEnvelope (${bbox[0][0]}, ${bbox[0][1]}, ${bbox[1][0]}, ${bbox[1][1]}, 4326) ~ geom` } const result = await this._client.query(query) total += Number(result.rows[0].count) } return total } disconnect() { this._client.end() } async getLayers(): Promise> { if ( this.supportedLayers !== undefined && new Date().getTime() - this.supportedLayersDate.getTime() < 1000 * 60 * 60 * 24 ) { return this.supportedLayers } const q = "SELECT table_name \n" + "FROM information_schema.tables \n" + "WHERE table_schema = 'public' AND table_name LIKE 'lines_%';" const result = await this._client.query(q) const layers = result.rows.map((r) => r.table_name.substring("lines_".length)) this.supportedLayers = new Set(layers) this.supportedLayersDate = new Date() return this.supportedLayers } async getMeta(): Promise { const now = new Date() if (this.metaCache !== undefined) { const diffSec = (this.metaCacheDate.getTime() - now.getTime()) / 1000 if (diffSec < 120) { return this.metaCache } } const result = await this._client.query("SELECT * FROM public.osm2pgsql_properties") const meta = {} for (const { property, value } of result.rows) { meta[property] = value } this.metaCacheDate = now this.metaCache = meta return this.metaCache } } class Server { constructor( port: number, options: { ignorePathPrefix?: string[] }, handle: { mustMatch: string | RegExp mimetype: string handle: (path: string) => Promise }[] ) { handle.push({ mustMatch: "", mimetype: "text/html", handle: async () => { return `Supported endpoints are

" + l + "

` }, }) http.createServer(async (req: http.IncomingMessage, res) => { try { console.log( req.method + " " + req.url, "from:", req.headers.origin, new Date().toISOString() ) const url = new URL(`http://127.0.0.1/` + req.url) let path = url.pathname while (path.startsWith("/")) { path = path.substring(1) } if (options?.ignorePathPrefix) { for (const toIgnore of options.ignorePathPrefix) { if (path.startsWith(toIgnore)) { path = path.substring(toIgnore.length + 1) break } } } console.log("Handling ", path) const handler = handle.find((h) => { if (typeof h.mustMatch === "string") { return h.mustMatch === path } if (path.match(h.mustMatch)) { return true } }) if (handler === undefined || handler === null) { res.writeHead(404, { "Content-Type": "text/html" }) res.write("

Not found...

") res.end() return } res.setHeader( "Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept" ) res.setHeader("Access-Control-Allow-Origin", req.headers.origin ?? "*") if (req.method === "OPTIONS") { res.setHeader( "Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE, UPDATE" ) res.writeHead(204, { "Content-Type": handler.mimetype }) res.end() return } if (req.method === "POST" || req.method === "UPDATE") { return } if (req.method === "DELETE") { return } try { const result = await handler.handle(path) res.writeHead(200, { "Content-Type": handler.mimetype }) res.write(result) res.end() } catch (e) { console.error("Could not handle request:", e) res.writeHead(500) res.write(e) res.end() } } catch (e) { console.error("FATAL:", e) res.end() } }).listen(port) console.log( "Server is running on port " + port, ". Supported endpoints are: " + handle.map((h) => h.mustMatch).join(", ") ) } } const connectionString = "postgresql://user:password@localhost:5444/osm-poi" const tcs = new OsmPoiDatabase(connectionString) const server = new Server(2345, { ignorePathPrefix: ["summary"] }, [ { mustMatch: "status.json", mimetype: "application/json", handle: async (path: string) => { const layers = await tcs.getLayers() const meta = await tcs.getMeta() return JSON.stringify({ meta, layers: Array.from(layers) }) }, }, { mustMatch: /[a-zA-Z0-9+_-]+\/[0-9]+\/[0-9]+\/[0-9]+\.json/, mimetype: "application/json", // "application/vnd.geo+json", async handle(path) { console.log("Path is:", path, path.split(".")[0]) const [layers, z, x, y] = path.split(".")[0].split("/") let sum = 0 let properties: Record = {} const availableLayers = await tcs.getLayers() for (const layer of layers.split("+")) { console.log("Getting layer", layer) if (!availableLayers.has(layer)) { continue } const count = await tcs.getCount( layer, Tiles.tile_bounds_lon_lat(Number(z), Number(x), Number(y)) ) properties[layer] = count sum += count } return JSON.stringify({ ...properties, total: sum }) }, }, ]) console.log( ">>>", await tcs.getCount("drinking_water", [ [3.194358020772171, 51.228073636083394], [3.2839964396059145, 51.172701162680994], ]) )