Add generic map to baaah

Signed-off-by: Darren Shepherd <[email protected]>

pkg/data/generic.go

@@ -0,0 +1,144 @@
+package data
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+
+	"github.com/acorn-io/baaah/pkg/typed"
+	"github.com/sirupsen/logrus"
+	"k8s.io/kube-openapi/pkg/common"
+	"k8s.io/kube-openapi/pkg/validation/spec"
+)
+
+type GenericMap struct {
+	// +optional
+	Data map[string]any `json:"-"`
+}
+
+func (GenericMap) OpenAPIDefinition() common.OpenAPIDefinition {
+	return common.OpenAPIDefinition{
+		Schema: spec.Schema{
+			VendorExtensible: spec.VendorExtensible{
+				Extensions: spec.Extensions{
+					"x-kubernetes-preserve-unknown-fields": "true",
+				},
+			},
+			SchemaProps: spec.SchemaProps{
+				Type: []string{"object"},
+			},
+		},
+	}
+}
+
+func (g *GenericMap) UnmarshalJSON(data []byte) error {
+	if g == nil {
+		return fmt.Errorf("%T: UnmarshalJSON on nil pointer", g)
+	}
+
+	dec := json.NewDecoder(bytes.NewBuffer(data))
+	dec.UseNumber()
+
+	d := map[string]any{}
+	if err := dec.Decode(&d); err != nil {
+		return fmt.Errorf("%T: Failed to decode data: %w", g, err)
+	}
+
+	if _, err := translateObject(d); err != nil {
+		return fmt.Errorf("%T: Failed to translate object: %w", g, err)
+	}
+
+	if len(d) > 0 {
+		// Consumers expect empty generic maps to have a nil Data field
+		g.Data = d
+	}
+
+	return nil
+}
+
+// GetData returns the underlying map[string]any and nil if the GenericMap is nil.
+func (g *GenericMap) GetData() map[string]any {
+	if g == nil {
+		return nil
+	}
+
+	return g.Data
+}
+
+// Merge merges the given map into this map, returning a new map, leaving the original unchanged.
+func (g *GenericMap) Merge(from *GenericMap) *GenericMap {
+	merged := typed.Concat(g.GetData(), from.GetData())
+	if merged == nil {
+		return nil
+	}
+
+	return &GenericMap{
+		Data: merged,
+	}
+}
+
+// MarshalJSON may get called on pointers or values, so implement MarshalJSON on value.
+func (g *GenericMap) MarshalJSON() ([]byte, error) {
+	return json.Marshal(g.GetData())
+}
+
+func translateObject(data any) (ret any, err error) {
+	switch t := data.(type) {
+	case map[string]any:
+		for k, v := range t {
+			if t[k], err = translateObject(v); err != nil {
+				return nil, err
+			}
+		}
+	case []any:
+		for i, v := range t {
+			if t[i], err = translateObject(v); err != nil {
+				return nil, err
+			}
+		}
+	case json.Number:
+		i, err := t.Int64()
+		if err == nil {
+			return i, nil
+		}
+		return t.Float64()
+	}
+	return data, nil
+}
+
+func (in *GenericMap) DeepCopyInto(out *GenericMap) {
+	var err error
+	if *out, err = Mapify(in.GetData()); err != nil {
+		logrus.WithError(err).Errorf("failed to deep copy into [%T]", out)
+	}
+}
+
+func (in GenericMap) DeepCopy() GenericMap {
+	var out GenericMap
+	in.DeepCopyInto(&out)
+	return out
+}
+
+func NewGenericMap(data map[string]any) *GenericMap {
+	if data == nil {
+		return nil
+	}
+
+	return &GenericMap{
+		Data: data,
+	}
+}
+
+func Mapify(data any) (GenericMap, error) {
+	marshaled, err := json.Marshal(data)
+	if err != nil {
+		return GenericMap{}, err
+	}
+
+	gm := &GenericMap{}
+	if err := gm.UnmarshalJSON(marshaled); err != nil {
+		return GenericMap{}, err
+	}
+
+	return *gm, nil
+}