diff options
Diffstat (limited to 'tests/internal/value')
-rw-r--r-- | tests/internal/value/notification_states.go | 37 | ||||
-rw-r--r-- | tests/internal/value/notification_types.go | 40 | ||||
-rw-r--r-- | tests/internal/value/value.go | 103 |
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)) + } +} |