Layerserver: improve docs, add stub of script that generates configuration file

2024-01-17 02:30:13 +01:00 · 2024-01-17 02:30:13 +01:00 · fb088059a5
commit fb088059a5
parent c98891d4bd
3 changed files with 194 additions and 5 deletions
--- a/scripts/osm2pgsql/generateLayerFile.ts
+++ b/scripts/osm2pgsql/generateLayerFile.ts
@ -0,0 +1,136 @@
+import LayerConfig from "../../src/Models/ThemeConfig/LayerConfig"
+import { TagsFilter } from "../../src/Logic/Tags/TagsFilter"
+import { Tag } from "../../src/Logic/Tags/Tag"
+import { And } from "../../src/Logic/Tags/And"
+import Script from "../Script"
+import { AllSharedLayers } from "../../src/Customizations/AllSharedLayers"
+import fs from "fs"
+import { Or } from "../../src/Logic/Tags/Or"
+import { RegexTag } from "../../src/Logic/Tags/RegexTag"
+
+class LuaSnippets{
+    /**
+     * The main piece of code that calls `process_poi`
+     */
+    static tail = [
+        "function osm2pgsql.process_node(object)",
+        "  process_poi(object, object:as_point())",
+        "end",
+        "",
+        "function osm2pgsql.process_way(object)",
+        "  if object.is_closed then",
+        "    process_poi(object, object:as_polygon():centroid())",
+        "  end",
+        "end",
+        ""].join("\n")
+
+    public static combine(calls: string[]): string{
+        return [
+            `function process_poi(object, geom)`,
+            ...calls.map(c => "  "+c+"(object, geom)"),
+            `end`,
+        ].join("\n")
+    }
+}
+class GenerateLayerLua {
+    private readonly _layer: LayerConfig
+
+    constructor(layer: LayerConfig) {
+        this._layer = layer
+    }
+    public functionName(){
+        const l = this._layer
+        return `process_poi_${l.id}`
+    }
+
+    public generateFunction(): string {
+        const l = this._layer
+        return [
+            `local pois_${l.id} = osm2pgsql.define_table({`,
+            `  name = '${l.id}',`,
+            "  ids = { type = 'any', type_column = 'osm_type', id_column = 'osm_id' },",
+            "  columns = {",
+            "    { column = 'tags', type = 'jsonb' },",
+            "    { column = 'geom', type = 'point', not_null = true },",
+            "  }" +
+            "})",
+            "",
+            "",
+            `function ${this.functionName()}(object, geom)`,
+            "  local matches_filter = " + this.toLuaFilter(l.source.osmTags),
+            "  if( not matches_filter) then",
+            "    return",
+            "  end",
+            "  local a = {",
+            "    geom = geom,",
+            "    tags = object.tags",
+            "  }",
+            "  ",
+            `  pois_${l.id}:insert(a)`,
+            "end",
+            ""
+        ].join("\n")
+    }
+
+
+    private toLuaFilter(tag: TagsFilter, useParens: boolean = false): string {
+        if (tag instanceof Tag) {
+            return `object.tags["${tag.key}"] == "${tag.value}"`
+        }
+        if (tag instanceof And) {
+            const expr = tag.and.map(t => this.toLuaFilter(t, true)).join(" and ")
+            if (useParens) {
+                return "(" + expr + ")"
+            }
+            return expr
+        }
+        if (tag instanceof Or) {
+            const expr = tag.or.map(t => this.toLuaFilter(t, true)).join(" or ")
+            if (useParens) {
+                return "(" + expr + ")"
+            }
+            return expr
+        }
+        if (tag instanceof RegexTag) {
+            if(typeof tag.value === "string" && tag.invert){
+                return `object.tags["${tag.key}"] ~= "${tag.value}"`
+            }
+
+            let expr = `not string.find(object.tags["${tag.key}"], "${tag.value}")`
+            if (!tag.invert) {
+                expr = "not " + expr
+            }
+            if (useParens) {
+                expr = "(" + expr + ")"
+            }
+            return expr
+        }
+        let msg = "Could not handle" + tag.asHumanString(false, false, {})
+        console.error(msg)
+        throw msg
+    }
+}
+
+class GenerateLayerFile extends Script {
+    constructor() {
+        super("Generates a .lua-file to use with osm2pgsql")
+    }
+
+    async main(args: string[]) {
+        let dw = AllSharedLayers.sharedLayers.get("drinking_water")
+        let t = AllSharedLayers.sharedLayers.get("toilet")
+
+        const generators = [dw, t].map(l => new GenerateLayerLua(l))
+
+        const script = [
+            ...generators.map(g => g.generateFunction()),
+            LuaSnippets.combine(generators.map(g => g.functionName())),
+            LuaSnippets.tail
+        ].join("\n\n\n")
+        const path = "build_db.lua"
+        fs.writeFileSync(path,script, "utf-8")
+        console.log("Written", path)
+    }
+}
+
+new GenerateLayerFile().run()