summaryrefslogtreecommitdiffstats
path: root/tests/internal/value
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--tests/internal/value/notification_states.go37
-rw-r--r--tests/internal/value/notification_types.go40
-rw-r--r--tests/internal/value/value.go103
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))
+ }
+}