Initial QSfera import

Server/vendor/github.com/open-policy-agent/opa/v1/ast/map.go

@@ -0,0 +1,108 @@
+// Copyright 2016 The OPA Authors.  All rights reserved.
+// Use of this source code is governed by an Apache2
+// license that can be found in the LICENSE file.
+
+package ast
+
+import (
+	"encoding/json"
+
+	"github.com/open-policy-agent/opa/v1/util"
+)
+
+// ValueMap represents a key/value map between AST term values. Any type of term
+// can be used as a key in the map.
+type ValueMap struct {
+	hashMap *util.TypedHashMap[Value, Value]
+}
+
+// NewValueMap returns a new ValueMap.
+func NewValueMap() *ValueMap {
+	return &ValueMap{
+		hashMap: util.NewTypedHashMap(ValueEqual, ValueEqual, Value.Hash, Value.Hash, nil),
+	}
+}
+
+// MarshalJSON provides a custom marshaller for the ValueMap which
+// will include the key, value, and value type.
+func (vs *ValueMap) MarshalJSON() ([]byte, error) {
+	var tmp []map[string]any
+	vs.Iter(func(k Value, v Value) bool {
+		tmp = append(tmp, map[string]any{
+			"name":  k.String(),
+			"type":  ValueName(v),
+			"value": v,
+		})
+		return false
+	})
+	return json.Marshal(tmp)
+}
+
+// Equal returns true if this ValueMap equals the other.
+func (vs *ValueMap) Equal(other *ValueMap) bool {
+	if vs == nil {
+		return other == nil || other.Len() == 0
+	}
+	if other == nil {
+		return vs.Len() == 0
+	}
+	return vs.hashMap.Equal(other.hashMap)
+}
+
+// Len returns the number of elements in the map.
+func (vs *ValueMap) Len() int {
+	if vs == nil {
+		return 0
+	}
+	return vs.hashMap.Len()
+}
+
+// Get returns the value in the map for k.
+func (vs *ValueMap) Get(k Value) Value {
+	if vs != nil {
+		if v, ok := vs.hashMap.Get(k); ok {
+			return v
+		}
+	}
+	return nil
+}
+
+// Hash returns a hash code for this ValueMap.
+func (vs *ValueMap) Hash() int {
+	if vs == nil {
+		return 0
+	}
+	return vs.hashMap.Hash()
+}
+
+// Iter calls the iter function for each key/value pair in the map. If the iter
+// function returns true, iteration stops.
+func (vs *ValueMap) Iter(iter func(Value, Value) bool) bool {
+	if vs == nil {
+		return false
+	}
+	return vs.hashMap.Iter(iter)
+}
+
+// Put inserts a key k into the map with value v.
+func (vs *ValueMap) Put(k, v Value) {
+	if vs == nil {
+		panic("put on nil value map")
+	}
+	vs.hashMap.Put(k, v)
+}
+
+// Delete removes a key k from the map.
+func (vs *ValueMap) Delete(k Value) {
+	if vs == nil {
+		return
+	}
+	vs.hashMap.Delete(k)
+}
+
+func (vs *ValueMap) String() string {
+	if vs == nil {
+		return "{}"
+	}
+	return vs.hashMap.String()
+}