Adding upstream version 1.1.0.upstream/1.1.0upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

3 files changed, 180 insertions, 0 deletions

diff --git a/tests/internal/value/notification_states.go b/tests/internal/value/notification_states.go
new file mode 100644
index 0000000..3c5479e
--- /dev/null
+++ b/tests/internal/value/notification_states.go
@@ -0,0 +1,37 @@
+package value
+
+import "fmt"
+
+type NotificationStates []string
+
+func (s NotificationStates) IcingaDbValue() interface{} {
+	v := uint(0)
+
+	for _, s := range s {
+		if bit, ok := notificationStateMap[s]; ok {
+			v |= bit
+		} else {
+			panic(fmt.Errorf("unknown notification state %q", s))
+		}
+	}
+
+	return v
+}
+
+func (s NotificationStates) Icinga2ConfigValue() string {
+	return ToIcinga2Config([]string(s))
+}
+
+func (s NotificationStates) Icinga2ApiValue() interface{} {
+	return ToIcinga2Api([]string(s))
+}
+
+// https://github.com/Icinga/icinga2/blob/a8f98cf72115d50152137bc924277b426f483a3f/lib/icinga/notification.hpp#L20-L32
+var notificationStateMap = map[string]uint{
+	"OK":       1,
+	"Warning":  2,
+	"Critical": 4,
+	"Unknown":  8,
+	"Up":       16,
+	"Down":     32,
+}
diff --git a/tests/internal/value/notification_types.go b/tests/internal/value/notification_types.go
new file mode 100644
index 0000000..f368ed5
--- /dev/null
+++ b/tests/internal/value/notification_types.go
@@ -0,0 +1,40 @@
+package value
+
+import "fmt"
+
+type NotificationTypes []string
+
+func (t NotificationTypes) IcingaDbValue() interface{} {
+	v := uint(0)
+
+	for _, s := range t {
+		if bit, ok := notificationTypeMap[s]; ok {
+			v |= bit
+		} else {
+			panic(fmt.Errorf("unknown notification type %q", s))
+		}
+	}
+
+	return v
+}
+
+func (t NotificationTypes) Icinga2ConfigValue() string {
+	return ToIcinga2Config([]string(t))
+}
+
+func (t NotificationTypes) Icinga2ApiValue() interface{} {
+	return ToIcinga2Api([]string(t))
+}
+
+// https://github.com/Icinga/icinga2/blob/a8f98cf72115d50152137bc924277b426f483a3f/lib/icinga/notification.hpp#L34-L50
+var notificationTypeMap = map[string]uint{
+	"DowntimeStart":   1,
+	"DowntimeEnd":     2,
+	"DowntimeRemoved": 4,
+	"Custom":          8,
+	"Acknowledgement": 16,
+	"Problem":         32,
+	"Recovery":        64,
+	"FlappingStart":   128,
+	"FlappingEnd":     256,
+}
diff --git a/tests/internal/value/value.go b/tests/internal/value/value.go
new file mode 100644
index 0000000..87027dd
--- /dev/null
+++ b/tests/internal/value/value.go
@@ -0,0 +1,103 @@
+package value
+
+import (
+	"fmt"
+	"reflect"
+)
+
+func ToIcinga2Config(value interface{}) string {
+	if value == nil {
+		return "null"
+	}
+
+	refVal := reflect.ValueOf(value)
+	switch refVal.Kind() {
+	case reflect.Slice:
+		vs := ""
+		for i := 0; i < refVal.Len(); i++ {
+			vs += ToIcinga2Config(refVal.Index(i).Interface()) + ","
+		}
+		return "[" + vs + "]"
+	case reflect.Map:
+		kvs := ""
+		iter := refVal.MapRange()
+		for iter.Next() {
+			kvs += ToIcinga2Config(iter.Key().Interface()) + "=" + ToIcinga2Config(iter.Value().Interface()) + ","
+		}
+		return "{" + kvs + "}"
+	}
+
+	switch v := value.(type) {
+	case interface{ Icinga2ConfigValue() string }:
+		return v.Icinga2ConfigValue()
+	case string:
+		// TODO(jb): probably not perfect quoting, but good enough for now
+		return fmt.Sprintf("%q", v)
+	case *string:
+		if v != nil {
+			return ToIcinga2Config(*v)
+		} else {
+			return "null"
+		}
+	case bool, int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64, float32, float64:
+		return fmt.Sprintf("%v", v)
+	default:
+		panic(fmt.Errorf("ToIcinga2Config called on unknown type %T", value))
+	}
+}
+
+func ToIcinga2Api(value interface{}) interface{} {
+	if value == nil {
+		return nil
+	}
+
+	refVal := reflect.ValueOf(value)
+	switch refVal.Kind() {
+	case reflect.Slice:
+		r := make([]interface{}, refVal.Len())
+		for i := range r {
+			r[i] = ToIcinga2Api(refVal.Index(i).Interface())
+		}
+		return r
+	case reflect.Map:
+		r := make(map[string]interface{})
+		iter := refVal.MapRange()
+		for iter.Next() {
+			// TODO: perform a better check than the type assertion
+			r[ToIcinga2Api(iter.Key().Interface()).(string)] = ToIcinga2Api(iter.Value().Interface())
+		}
+		return r
+	}
+
+	switch v := value.(type) {
+	case interface{ Icinga2ApiValue() interface{} }:
+		return v.Icinga2ApiValue()
+	case string, []string, bool, int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64, float32, float64:
+		return v
+	case *string:
+		if v != nil {
+			return ToIcinga2Api(*v)
+		} else {
+			return nil
+		}
+	default:
+		panic(fmt.Errorf("ToIcinga2Api called on unknown type %T", value))
+	}
+}
+
+func ToIcingaDb(value interface{}) interface{} {
+	switch v := value.(type) {
+	case interface{ IcingaDbValue() interface{} }:
+		return v.IcingaDbValue()
+	case bool:
+		if v {
+			return "y"
+		} else {
+			return "n"
+		}
+	case string, *string, int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64, float32, float64:
+		return v
+	default:
+		panic(fmt.Errorf("ToIcinga2Api called on unknown type %T", value))
+	}
+}