148 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			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
 | 
						|
}
 |