kubernetes-operator/pkg/notifications/msteams/msteams.go

148 lines
3.5 KiB
Go

package msteams
import (
"bytes"
"context"
"encoding/json"
"fmt"
"net/http"
"strings"
"github.com/jenkinsci/kubernetes-operator/api/v1alpha2"
"github.com/jenkinsci/kubernetes-operator/pkg/notifications/event"
"github.com/jenkinsci/kubernetes-operator/pkg/notifications/provider"
"github.com/pkg/errors"
corev1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/types"
k8sclient "sigs.k8s.io/controller-runtime/pkg/client"
)
const (
infoColor = "439FE0"
warningColor = "E81123"
defaultColor = "C8C8C8"
)
// Teams is a Microsoft MicrosoftTeams notification service
type Teams struct {
httpClient http.Client
k8sClient k8sclient.Client
config v1alpha2.Notification
}
// New returns instance of Teams
func New(k8sClient k8sclient.Client, config v1alpha2.Notification, httpClient http.Client) *Teams {
return &Teams{k8sClient: k8sClient, config: config, httpClient: httpClient}
}
// Message is representation of json message structure
type Message struct {
Type string `json:"@type"`
Context string `json:"@context"`
ThemeColor event.StatusColor `json:"themeColor"`
Title string `json:"title"`
Sections []Section `json:"sections"`
Summary string `json:"summary"`
}
// Section is MS Teams message section
type Section struct {
Facts []Fact `json:"facts"`
Text string `json:"text"`
}
// Fact is field where we can put content
type Fact struct {
Name string `json:"name"`
Value string `json:"value"`
}
func (t Teams) getStatusColor(logLevel v1alpha2.NotificationLevel) event.StatusColor {
switch logLevel {
case v1alpha2.NotificationLevelInfo:
return infoColor
case v1alpha2.NotificationLevelWarning:
return warningColor
default:
return defaultColor
}
}
func (t Teams) generateMessage(e event.Event) Message {
var reason string
if t.config.Verbose {
reason = strings.Join(e.Reason.Verbose(), "\n\n - ")
} else {
reason = strings.Join(e.Reason.Short(), "\n\n - ")
}
tm := Message{
Title: provider.NotificationTitle(e),
Type: "MessageCard",
Context: "https://schema.org/extensions",
ThemeColor: t.getStatusColor(e.Level),
Sections: []Section{
{
Facts: []Fact{
{
Name: provider.CrNameFieldName,
Value: e.Jenkins.Name,
},
{
Name: provider.NamespaceFieldName,
Value: e.Jenkins.Namespace,
},
{
Name: provider.PhaseFieldName,
Value: string(e.Phase),
},
},
Text: reason,
},
},
Summary: reason,
}
return tm
}
// Send is function for sending directly to API
func (t Teams) Send(e event.Event) error {
secret := &corev1.Secret{}
selector := t.config.Teams.WebHookURLSecretKeySelector
err := t.k8sClient.Get(context.TODO(), types.NamespacedName{Name: selector.Name, Namespace: e.Jenkins.Namespace}, secret)
if err != nil {
return errors.WithStack(err)
}
secretValue := string(secret.Data[selector.Key])
if secretValue == "" {
return errors.Errorf("Microsoft Teams WebHook URL is empty in secret '%s/%s[%s]", e.Jenkins.Namespace, selector.Name, selector.Key)
}
msg, err := json.Marshal(t.generateMessage(e))
if err != nil {
return errors.WithStack(err)
}
request, err := http.NewRequest("POST", secretValue, bytes.NewBuffer(msg))
if err != nil {
return errors.WithStack(err)
}
resp, err := t.httpClient.Do(request)
if err != nil {
return errors.WithStack(err)
}
if resp.StatusCode != http.StatusOK {
return errors.New(fmt.Sprintf("Invalid response from server: %s", resp.Status))
}
defer func() { _ = resp.Body.Close() }()
return nil
}