feat: generate a map configuration based on a wms

pyproject.toml

@@ -41,7 +41,8 @@ dependencies = [
   "httpx>=0.28.1",
   "setuptools>=65.5.0",
   "backoff>=2.1.0",
-  "pyyaml>=6.0.3"
+  "pyyaml>=6.0.3",
+  "owslib>=0.35.0"
 ]
 description = ""
 license = "GPL-3.0+"

src/datasync/main.py

@@ -8,6 +8,7 @@
     dms,
     gbif_backbone,
     grass,
+    maps,
     ninagen,
     nva,
     pit_registering_salmon,
@@ -25,6 +26,7 @@
 app.add_typer(pit_registering_salmon.app, name="pit-registering-salmon")
 app.add_typer(grass.app, name="grass-gis")
 app.add_typer(services.app, name="services")
+app.add_typer(maps.app, name="maps")
 app.add_typer(gbif_backbone.app, name="gbif-backbone")
 
 if __name__ == "__main__":

src/datasync/maps.py

@@ -0,0 +1,132 @@
+import json
+from pathlib import Path
+from urllib import parse
+
+import typer
+from owslib.wms import WebMapService
+
+from .settings import log
+
+app = typer.Typer(help="Scripts related to the generation of NINA maps")
+
+
+@app.command(help="convert a wms resource to a configuration for NINA maps")
+def wms_to_map(
+    get_capabilities_url: str = typer.Argument(
+        help="The URL of the get capabilities of a WMS service"
+    ),
+    output: str = typer.Option(
+        default="output.json", help="Path of the output json file"
+    ),
+):
+    wms = WebMapService(get_capabilities_url)
+    log.info(wms.contents)
+
+    parsed = parse.urlparse(get_capabilities_url)
+    qs = parse.parse_qs(parsed.query)
+    map_param = qs.get("map", [None])[0]
+
+    items = {}
+
+    cn = wms.__class__.__name__
+
+    for key, value in wms.contents.items():
+        log.info(key, value=value)
+
+        req = getattr(wms, f"_{cn}__build_getmap_request")(
+            layers=[key],
+            styles=[""],
+            srs="EPSG:3857",
+            format="image/png",
+            size=(256, 256),
+            transparent=True,
+            bbox=(0, 0, 0, 0),
+        )
+        if map_param:
+            req["map"] = map_param
+
+        del req["bbox"]
+
+        legend = {
+            "service": "WMS",
+            "version": wms.version,
+            "request": "GetLegendGraphic",
+            "layer": key,
+            "style": "",
+            "format": "image/png",
+            "transparent": "true",
+        }
+
+        if map_param:
+            legend["map"] = map_param
+
+        items[key] = {
+            "name": value.title,
+            "type": "layer",
+            "layer": {
+                "type": "raster",
+                "tiles": [
+                    parse.urlunsplit(
+                        (
+                            parsed.scheme,
+                            parsed.netloc,
+                            parsed.path,
+                            parse.urlencode(req, doseq=True) + "&bbox={bbox-epsg-3857}",
+                            "",
+                        )
+                    )
+                ],
+                "legend": {
+                    "type": "image",
+                    "url": parse.urlunsplit(
+                        (
+                            parsed.scheme,
+                            parsed.netloc,
+                            parsed.path,
+                            parse.urlencode(
+                                legend,
+                                doseq=True,
+                            ),
+                            "",
+                        )
+                    ),
+                },
+            },
+        }
+
+    conf = {
+        "id": parsed.hostname,
+        "title": wms.identification.title,
+        "subtitle": "",
+        "description": wms.identification.abstract,
+        "icon": "https://s3-ext-1.nina.no/dms/maps/logosmall-BFfoJdyr.png",
+        "baseMap": "positron",
+        "layerOrder": [],
+        # TODO: make this configurable or automatic based on the resource
+        "viewState": {"longitude": 17.8, "latitude": 65.6, "zoom": 2},
+        "expandedItems": [],
+        "items": {
+            "root": {
+                "name": "Layers",
+                "type": "folder",
+                "children": list(items.keys()),
+            },
+            **items,
+        },
+        "config": {
+            "titiler_api_url": "/titiler",
+            "theme": "nina",
+            "language": "en",
+            "footer": {
+                "items": [
+                    "![](https://s3-ext-1.nina.no/dms/maps/logowhite-6HSLYLYZ.png)",
+                    "Norsk institutt for naturforskning - [www.nina.no](https://www.nina.no)",
+                ],
+                "justify": "justify-between",
+                "align": "items-baseline",
+            },
+        },
+    }
+
+    with Path(output).open("w") as f:
+        json.dump(conf, f, indent=4)