Started on creating config maps for legend generation

Author: Daimanta

api/v2beta1/wms_conversion.go

@@ -52,6 +52,10 @@ func V3HubFromV2(src *WMS, target *pdoknlv3.WMS) {
 	dst := target
 
 	dst.ObjectMeta = src.ObjectMeta
+	if dst.Annotations == nil {
+		dst.Annotations = make(map[string]string)
+	}
+
 	dst.Annotations[SERVICE_METADATA_IDENTIFIER_ANNOTATION] = src.Spec.Service.MetadataIdentifier
 
 	// Set LifeCycle if defined
@@ -322,6 +326,7 @@ func (v2Layer WMSLayer) MapToV3(v2Service WMSService) pdoknlv3.Layer {
 		BoundingBoxes:       []pdoknlv3.WMSBoundingBox{},
 		MinScaleDenominator: nil,
 		MaxScaleDenominator: nil,
+		Visible:             &v2Layer.Visible,
 	}
 
 	if v2Layer.SourceMetadataIdentifier != nil {

internal/controller/legendgenerator/legend-fixer.sh

@@ -0,0 +1,37 @@
+#!/usr/bin/env bash
+set -eo pipefail
+echo "creating legends for root and group layers by concatenating data layers"
+input_filepath="/input/input"
+remove_filepath="/input/remove"
+config_filepath="/input/ogc-webservice-proxy-config.yaml"
+legend_dir="/var/www/legend"
+< "${input_filepath}" xargs -n 2 echo | while read -r layer style; do
+  export layer
+  # shellcheck disable=SC2016 # dollar is for yq
+  if ! < "${config_filepath}" yq -e 'env(layer) as $layer | .grouplayers | keys | contains([$layer])' &>/dev/null; then
+    continue
+  fi
+  export grouplayer="${layer}"
+  grouplayer_style_filepath="${legend_dir}/${grouplayer}/${style}.png"
+  # shellcheck disable=SC2016 # dollar is for yq
+  datalayers=$(< "${config_filepath}" yq 'env(grouplayer) as $foo | .grouplayers[$foo][]')
+  datalayer_style_filepaths=()
+  for datalayer in $datalayers; do
+    datalayer_style_filepath="${legend_dir}/${datalayer}/${style}.png"
+    if [[ -f "${datalayer_style_filepath}" ]]; then
+      datalayer_style_filepaths+=("${datalayer_style_filepath}")
+    fi
+  done
+  if [[ -n "${datalayer_style_filepaths[*]}" ]]; then
+    echo "concatenating ${grouplayer_style_filepath}"
+    gm convert -append "${datalayer_style_filepaths[@]}" "${grouplayer_style_filepath}"
+  else
+    echo "no data for ${grouplayer_style_filepath}"
+  fi
+done
+< "${remove_filepath}" xargs -n 2 echo | while read -r layer style; do
+  remove_legend_file="${legend_dir}/${layer}/${style}.png"
+  echo removing $remove_legend_file
+  rm $remove_legend_file
+done
+echo "done"

internal/controller/legendgenerator/mapper.go

@@ -0,0 +1,112 @@
+package legendgenerator
+
+import (
+	"fmt"
+	pdoknlv3 "github.com/pdok/mapserver-operator/api/v3"
+	"github.com/pdok/mapserver-operator/internal/controller"
+	smoothoperatorutils "github.com/pdok/smooth-operator/pkg/util"
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"os"
+	"sigs.k8s.io/yaml"
+	"strings"
+)
+
+const (
+	defaultMapserverConf = `CONFIG
+  ENV
+    MS_MAP_NO_PATH "true"
+  END
+END
+`
+)
+
+type LegendReference struct {
+	Layer string `yaml:"layer" json:"layer"`
+	Style string `yaml:"style" json:"style"`
+}
+
+func getBareConfigMapLegendGenerator(obj metav1.Object) *corev1.ConfigMap {
+	return &corev1.ConfigMap{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      obj.GetName() + "-legend-generator",
+			Namespace: obj.GetNamespace(),
+		},
+	}
+}
+
+func GetLegendGeneratorConfigMap(wms *pdoknlv3.WMS) *corev1.ConfigMap {
+	result := getBareConfigMapLegendGenerator(wms)
+	labels := controller.AddCommonLabels(wms, smoothoperatorutils.CloneOrEmptyMap(wms.GetLabels()))
+	result.Labels = labels
+
+	result.Immutable = smoothoperatorutils.Pointer(true)
+	result.Data = map[string]string{}
+	result.Data["default_mapserver.conf"] = defaultMapserverConf
+
+	addLayerInput(wms, result.Data)
+
+	if wms.Spec.Options.RewriteGroupToDataLayers != nil && *wms.Spec.Options.RewriteGroupToDataLayers {
+		addLegendFixerConfig(wms, result.Data)
+	}
+
+	return result
+}
+
+func addLayerInput(wms *pdoknlv3.WMS, data map[string]string) {
+	legendReferences := make([]LegendReference, 0)
+
+	if wms.Spec.Service.Layer.Layers != nil {
+		for _, layer := range *wms.Spec.Service.Layer.Layers {
+			processLayer(&layer, &legendReferences)
+		}
+	}
+
+	sb := strings.Builder{}
+	for _, reference := range legendReferences {
+		sb.WriteString(fmt.Sprintf("\"%s\" \"%s\"\n", reference.Layer, reference.Style))
+	}
+
+	data["input"] = sb.String()
+	referencesYaml, err := yaml.Marshal(legendReferences)
+	if err == nil {
+		data["input2"] = string(referencesYaml)
+	}
+
+}
+
+func processLayer(layer *pdoknlv3.Layer, legendReferences *[]LegendReference) {
+	if layer.Visible == nil || !*layer.Visible {
+		return
+	}
+	for _, style := range layer.Styles {
+		if style.Legend == nil {
+			*legendReferences = append(*legendReferences, LegendReference{
+				Layer: layer.Name,
+				Style: style.Name,
+			})
+		}
+	}
+
+	if layer.Layers != nil {
+		for _, innerLayer := range *layer.Layers {
+			processLayer(&innerLayer, legendReferences)
+		}
+	}
+}
+
+func addLegendFixerConfig(wms *pdoknlv3.WMS, data map[string]string) {
+	fileBytes, err := os.ReadFile("./legend-fixer.sh")
+	if err == nil {
+		data["legend-fixer.sh"] = string(fileBytes)
+	}
+
+	//TODO: Identify middle layers
+	if wms.Spec.Service.Layer.Layers != nil {
+		for _, layer := range *wms.Spec.Service.Layer.Layers {
+			if layer.Layers != nil && len(*layer.Layers) > 0 {
+				println(layer.Name)
+			}
+		}
+	}
+}

internal/controller/legendgenerator/mapper_test.go

@@ -631,7 +631,7 @@ func AddCommonLabels[O pdoknlv3.WMSWFS](obj O, labels map[string]string) map[str
 	case *pdoknlv3.WFS:
 		inspire = any(obj).(*pdoknlv3.WFS).Spec.Service.Inspire != nil
 	case *pdoknlv3.WMS:
-		inspire = any(obj).(*pdoknlv3.WFS).Spec.Service.Inspire != nil
+		inspire = any(obj).(*pdoknlv3.WMS).Spec.Service.Inspire != nil
 	}
 
 	labels["inspire"] = strconv.FormatBool(inspire)