2024-09-26 14:45:04 +02:00
|
|
|
// Copyright Jan Dittberner
|
|
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
//
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
|
|
// it under the terms of the GNU General Public License as published by
|
|
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
|
|
// (at your option) any later version.
|
|
|
|
//
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
// GNU General Public License for more details.
|
|
|
|
//
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
2024-09-26 17:18:32 +02:00
|
|
|
|
2024-09-26 14:45:04 +02:00
|
|
|
package icinga2
|
|
|
|
|
|
|
|
import (
|
|
|
|
"errors"
|
|
|
|
"fmt"
|
|
|
|
"net/url"
|
2024-09-26 17:18:32 +02:00
|
|
|
"strings"
|
2024-09-26 14:45:04 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type URLValue struct {
|
|
|
|
URL *url.URL
|
|
|
|
}
|
|
|
|
|
|
|
|
func (u *URLValue) Set(value string) error {
|
|
|
|
var err error
|
|
|
|
|
|
|
|
u.URL, err = url.Parse(value)
|
|
|
|
if err != nil {
|
|
|
|
return fmt.Errorf("could not parse URL: %w", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (u *URLValue) String() string {
|
|
|
|
if u.URL != nil {
|
|
|
|
return u.URL.String()
|
|
|
|
}
|
|
|
|
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
2024-09-26 17:18:32 +02:00
|
|
|
type HostStateValue string
|
|
|
|
|
|
|
|
func (v *HostStateValue) Set(value string) error {
|
|
|
|
if strings.EqualFold(value, "Up") || strings.EqualFold(value, "Down") {
|
|
|
|
*v = HostStateValue(strings.ToUpper(value))
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
return fmt.Errorf("invalid host state value: %s", value)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (v *HostStateValue) String() string {
|
|
|
|
return string(*v)
|
|
|
|
}
|
|
|
|
|
|
|
|
type ServiceStateValue string
|
|
|
|
|
|
|
|
func (v *ServiceStateValue) Set(value string) error {
|
|
|
|
allowedValues := []string{"OK", "Warning", "Critical", "Unknown"}
|
|
|
|
|
|
|
|
for _, allowed := range allowedValues {
|
|
|
|
if strings.EqualFold(value, allowed) {
|
|
|
|
*v = ServiceStateValue(strings.ToUpper(value))
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return fmt.Errorf("invalid service state value: %s", value)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (v *ServiceStateValue) String() string {
|
|
|
|
return string(*v)
|
|
|
|
}
|
|
|
|
|
|
|
|
type NotificationTypeValue string
|
|
|
|
|
|
|
|
func (v *NotificationTypeValue) Set(value string) error {
|
|
|
|
allowedValues := []string{
|
|
|
|
"DowntimeStart", "DowntimeEnd", "DownTimeRemoved", "Custom", "Acknowledgement", "Problem", "Recovery",
|
|
|
|
"FlappingStart", "FlappingEnd",
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, allowed := range allowedValues {
|
|
|
|
if strings.EqualFold(value, allowed) {
|
|
|
|
*v = NotificationTypeValue(strings.ToUpper(value))
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return fmt.Errorf("invalid notification type value: %s", value)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (v *NotificationTypeValue) String() string {
|
|
|
|
return string(*v)
|
|
|
|
}
|
|
|
|
|
2024-09-26 14:45:04 +02:00
|
|
|
type MatrixParameters struct {
|
|
|
|
MatrixRoom string
|
|
|
|
MatrixServer URLValue
|
|
|
|
MatrixToken string
|
|
|
|
}
|
|
|
|
|
|
|
|
func (m *MatrixParameters) HasRequired() bool {
|
2024-09-26 17:18:32 +02:00
|
|
|
return !(m.MatrixRoom == "" || m.MatrixServer.URL == nil || m.MatrixToken == "")
|
2024-09-26 14:45:04 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
type BaseParameters struct {
|
|
|
|
NotificationAuthorName string
|
|
|
|
NotificationComment string
|
2024-09-26 17:18:32 +02:00
|
|
|
NotificationType NotificationTypeValue
|
2024-09-26 14:45:04 +02:00
|
|
|
IcingaWeb2URL URLValue
|
|
|
|
LongDateTime string
|
|
|
|
Hostname string
|
|
|
|
HostDisplayName string
|
|
|
|
HostAddress string
|
|
|
|
HostAddress6 string
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p BaseParameters) HasRequired() bool {
|
2024-09-26 17:18:32 +02:00
|
|
|
return !(p.LongDateTime == "" || p.Hostname == "" || p.HostDisplayName == "" || p.NotificationType == "")
|
2024-09-26 14:45:04 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
type HostParameters struct {
|
|
|
|
HostOutput string
|
2024-09-26 17:18:32 +02:00
|
|
|
HostState HostStateValue
|
2024-09-26 14:45:04 +02:00
|
|
|
BaseParameters
|
|
|
|
MatrixParameters
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *HostParameters) ValidateRequired() error {
|
|
|
|
if !p.BaseParameters.HasRequired() || !p.MatrixParameters.HasRequired() ||
|
|
|
|
p.HostOutput == "" || p.HostState == "" {
|
|
|
|
return errors.New("missing required fields")
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
type ServiceParameters struct {
|
|
|
|
ServiceName string
|
|
|
|
ServiceDisplayName string
|
|
|
|
ServiceOutput string
|
2024-09-26 17:18:32 +02:00
|
|
|
ServiceState ServiceStateValue
|
2024-09-26 14:45:04 +02:00
|
|
|
BaseParameters
|
|
|
|
MatrixParameters
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *ServiceParameters) ValidateRequired() error {
|
|
|
|
if !p.BaseParameters.HasRequired() || !p.MatrixParameters.HasRequired() ||
|
|
|
|
p.ServiceName == "" || p.ServiceDisplayName == "" || p.ServiceState == "" || p.ServiceOutput == "" {
|
|
|
|
return errors.New("missing required fields")
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|