suggestion for implementing wms/wmts/wfs/... links in record.geojson

geopython · Dec 8, 2022 · a8ce743 · a8ce743
1 parent 817c7df
commit a8ce743
Show file tree

Hide file tree

Showing 2 changed files with 86 additions and 15 deletions.
diff --git a/pycsw/ogc/api/records.py b/pycsw/ogc/api/records.py
@@ -45,7 +45,7 @@
 from pycsw.core.pygeofilter_evaluate import to_filter
 from pycsw.core.util import bind_url, get_today_and_now, jsonify_links, load_custom_repo_mappings, wkt2geom
 from pycsw.ogc.api.oapi import gen_oapi
-from pycsw.ogc.api.util import match_env_var, render_j2_template, to_json
+from pycsw.ogc.api.util import match_env_var, render_j2_template, to_json, merge_qs
 
 LOGGER = logging.getLogger(__name__)
 
@@ -1048,22 +1048,70 @@ def record2json(record, url, collection, stac_item=False):
 
     if record.links:
         rdl = record_dict['links']
-
         for link in jsonify_links(record.links):
-            link2 = {
-                'href': link['url'],
-                'name': link['name'],
+            link2 = {   
+                'type': link['protocol'],
                 'description': link['description'],
-                'type': link['protocol']
-            }
-            if 'rel' in link:
-                link2['rel'] = link['rel']
-            elif link['protocol'] == 'WWW:LINK-1.0-http--image-thumbnail':
-                link2['rel'] = 'preview'
-            elif 'function' in link:
-                link2['rel'] = link['function']
-
-            rdl.append(link2)
+                'title': link['name'],
+                'href': link['url']
+                }
+            if link['protocol']:
+                if 'OGC:WMS' in link['protocol'].upper():
+                    link2['rel'] = 'map'
+                    link2['templated'] = 'true'
+                    # assumes link['url'] includes '&layers=...', else link['name'] contains layername(s) 
+                    link2['href'] = merge_qs(link['url'], { 
+                            'request': 'GetMap', 'service': 'WMS',   
+                            'width': '{width}', 'height': '{height}', 'bbox': '{bbox}'}, {
+                            'version': '1.3.0', 'crs': 'epsg:4326', 
+                            'layers': link['name'], 'format': 'image/png'})
+                    link2['variables'] = {
+                            'bbox': {
+                                'type': 'array', 
+                                'items': {'type': 'number', 'format': 'double'},
+                                'minItems': 4, 'maxItems': 4
+                                },
+                            'width': {'type': 'number', 'format': 'integer'},
+                            'height': {'type': 'number', 'format': 'integer'}
+                            }
+                    link2['type'] = 'image/png'
+
+                elif 'OGC:WMTS' in link['protocol'].upper():
+                    link2['rel'] = 'map'
+                    link2['templated'] = 'true'
+                    link2['href'] = merge_qs(link['url'], { 
+                        'service': 'WMTS', 'request': 'GetTile', 'version': '1.0.0',
+                        'TileMatrix': '{TileMatrix}', 'TileRow': '{TileRow}', 'TileCol': '{TileCol}'}, {
+                        'TileMatrixSet': 'default', 'Layer': link['name'], 
+                        'Style': 'default', 'Format': 'image/png'})
+                    link2['variables'] = { 
+                        'TileMatrix': {'type': 'number', 'format': 'integer'},
+                        'TileRow': {'type': 'number', 'format': 'integer'},
+                        'TileCol': {'type': 'number', 'format': 'integer'}
+                    }
+                    link2['type'] = 'image/png'
+
+                elif 'OGC:WFS' in link['protocol'].upper():   
+                    link2['rel'] = 'map'
+                    link2['templated'] = 'true'
+                    link2['href'] = merge_qs(link['url'], {
+                        'version': '2.0.0', 'request': 'GetFeature', 'service': 'WFS', 'count': '{count}'}, {
+                        'typenames': link['name'], 'srsName': 'urn:ogc:def:crs:EPSG::4326',
+                        'outputFormat':'application/gml+xml'})
+                    link2['variables'] = { 
+                        'count': {'type': 'number', 'format': 'integer'}}
+                    link2['type'] = 'application/gml+xml'
+
+                #elif 'OSGEO:TMS' in link['protocol'].upper():
+                #elif 'OGC:CSW' in link['protocol'].upper():   
+                #elif 'OGC:WCS' in link['protocol'].upper():   
+
+                elif 'rel' in link:
+                    link2['rel'] = link['rel']
+                elif 'function' in link:
+                    link2['rel'] = link['function']
+
+                rdl.append(link2)
 
     record_dict['links'].append({
         'rel': 'collection',

diff --git a/pycsw/ogc/api/util.py b/pycsw/ogc/api/util.py
@@ -38,6 +38,7 @@
 import mimetypes
 import os
 import re
+from urllib.parse import urlparse, parse_qsl, urlencode
 
 from jinja2 import Environment, FileSystemLoader
 from jinja2.exceptions import TemplateNotFound
@@ -139,6 +140,28 @@ class EnvVarLoader(yaml.SafeLoader):
 
     return yaml.load(fh, Loader=EnvVarLoader)
 
+def merge_qs(url, req, opt):
+    """
+    Merge required and optional parameters into a  querystring.
+    Typically replaces https://example.com/mapserver?map=foo.map&servic=wms&request=GetCapabilities&layers=trees for
+    https://example.com/mapserver?map=foo.map?servic=wms&request=GetMap&version=1.3.0&layers=trees&bbox={bbox}&format=...
+    
+    :param url: The original url to be substituted 
+    :param req: These arguments will be sustituted
+    :param opt: These arguments are substituted if they don't exist
+
+    :returns: url
+    """
+
+    parsed_url = urlparse(url)
+    args = dict(parse_qsl(parsed_url.query))
+    for k,v in req.items():
+        args[k] = v
+    for k,v in opt.items():
+        if k not in args.keys() or args[k] == '':
+            args[k] = v
+
+    return url.split('?')[0] + '?' + urlencode(args)
 
 def to_json(dict_, pretty=False):
     """