Add SMTP notification provider
This commit is contained in:
parent
3f801cd180
commit
dfd0b54579
|
|
@ -73,6 +73,7 @@ type Notification struct {
|
|||
Slack *Slack `json:"slack,omitempty"`
|
||||
Teams *MicrosoftTeams `json:"teams,omitempty"`
|
||||
Mailgun *Mailgun `json:"mailgun,omitempty"`
|
||||
SMTP *SMTP `json:"smtp,omitempty"`
|
||||
}
|
||||
|
||||
// Slack is handler for Slack notification channel
|
||||
|
|
@ -81,6 +82,16 @@ type Slack struct {
|
|||
WebHookURLSecretKeySelector SecretKeySelector `json:"webHookURLSecretKeySelector"`
|
||||
}
|
||||
|
||||
type SMTP struct {
|
||||
UsernameSecretKeySelector SecretKeySelector `json:"usernameSecretKeySelector"`
|
||||
PasswordSecretKeySelector SecretKeySelector `json:"passwordSecretKeySelector"`
|
||||
Port int `json:"port"`
|
||||
Server string `json:"server"`
|
||||
TLSInsecureSkipVerify bool `json:"tlsInsecureSkipVerify,omitempty"`
|
||||
From string `json:"from"`
|
||||
To string `json:"to"`
|
||||
}
|
||||
|
||||
// MicrosoftTeams is handler for Microsoft MicrosoftTeams notification channel
|
||||
type MicrosoftTeams struct {
|
||||
// The web hook URL to MicrosoftTeams App
|
||||
|
|
|
|||
|
|
@ -458,6 +458,11 @@ func (in *Notification) DeepCopyInto(out *Notification) {
|
|||
*out = new(Mailgun)
|
||||
**out = **in
|
||||
}
|
||||
if in.SMTP != nil {
|
||||
in, out := &in.SMTP, &out.SMTP
|
||||
*out = new(SMTP)
|
||||
**out = **in
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
|
|
@ -504,6 +509,24 @@ func (in *Restore) DeepCopy() *Restore {
|
|||
return out
|
||||
}
|
||||
|
||||
// DeepCopyInto is an autogenerated deepcopy function, copying the receiver, writing into out. in must be non-nil.
|
||||
func (in *SMTP) DeepCopyInto(out *SMTP) {
|
||||
*out = *in
|
||||
out.UsernameSecretKeySelector = in.UsernameSecretKeySelector
|
||||
out.PasswordSecretKeySelector = in.PasswordSecretKeySelector
|
||||
return
|
||||
}
|
||||
|
||||
// DeepCopy is an autogenerated deepcopy function, copying the receiver, creating a new SMTP.
|
||||
func (in *SMTP) DeepCopy() *SMTP {
|
||||
if in == nil {
|
||||
return nil
|
||||
}
|
||||
out := new(SMTP)
|
||||
in.DeepCopyInto(out)
|
||||
return out
|
||||
}
|
||||
|
||||
// DeepCopyInto is an autogenerated deepcopy function, copying the receiver, writing into out. in must be non-nil.
|
||||
func (in *SecretKeySelector) DeepCopyInto(out *SecretKeySelector) {
|
||||
*out = *in
|
||||
|
|
|
|||
|
|
@ -81,6 +81,8 @@ func Listen(events chan Event, k8sEvent event.Recorder, k8sClient k8sclient.Clie
|
|||
svc = Teams{k8sClient: k8sClient}
|
||||
} else if notificationConfig.Mailgun != nil {
|
||||
svc = MailGun{k8sClient: k8sClient}
|
||||
} else if notificationConfig.SMTP != nil {
|
||||
svc = SMTP{k8sClient: k8sClient}
|
||||
} else {
|
||||
logger.V(log.VWarn).Info(fmt.Sprintf("Unknown notification service `%+v`", notificationConfig))
|
||||
continue
|
||||
|
|
|
|||
|
|
@ -0,0 +1,93 @@
|
|||
package notifications
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/tls"
|
||||
"fmt"
|
||||
|
||||
"github.com/jenkinsci/kubernetes-operator/pkg/apis/jenkins/v1alpha2"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
"gopkg.in/gomail.v2"
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
"k8s.io/apimachinery/pkg/types"
|
||||
k8sclient "sigs.k8s.io/controller-runtime/pkg/client"
|
||||
)
|
||||
|
||||
const (
|
||||
mailSubject = "Jenkins Operator Notification"
|
||||
)
|
||||
|
||||
type SMTP struct {
|
||||
k8sClient k8sclient.Client
|
||||
}
|
||||
|
||||
func (s SMTP) Send(event Event, config v1alpha2.Notification) error {
|
||||
usernameSecret := &corev1.Secret{}
|
||||
passwordSecret := &corev1.Secret{}
|
||||
|
||||
usernameSelector := config.SMTP.UsernameSecretKeySelector
|
||||
passwordSelector := config.SMTP.PasswordSecretKeySelector
|
||||
|
||||
err := s.k8sClient.Get(context.TODO(), types.NamespacedName{Name: usernameSelector.Name, Namespace: event.Jenkins.Namespace}, usernameSecret)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = s.k8sClient.Get(context.TODO(), types.NamespacedName{Name: passwordSelector.Name, Namespace: event.Jenkins.Namespace}, passwordSecret)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
usernameSecretValue := string(usernameSecret.Data[usernameSelector.Key])
|
||||
if usernameSecretValue == "" {
|
||||
return errors.Errorf("SMTP username is empty in secret '%s/%s[%s]", event.Jenkins.Namespace, usernameSelector.Name, usernameSelector.Key)
|
||||
}
|
||||
|
||||
passwordSecretValue := string(passwordSecret.Data[passwordSelector.Key])
|
||||
if passwordSecretValue == "" {
|
||||
return errors.Errorf("SMTP password is empty in secret '%s/%s[%s]", event.Jenkins.Namespace, passwordSelector.Name, passwordSelector.Key)
|
||||
}
|
||||
|
||||
mailer := gomail.NewDialer(config.SMTP.Server, config.SMTP.Port, usernameSecretValue, passwordSecretValue)
|
||||
mailer.TLSConfig = &tls.Config{InsecureSkipVerify: config.SMTP.TLSInsecureSkipVerify}
|
||||
|
||||
var statusMessage string
|
||||
|
||||
if config.Verbose {
|
||||
message := event.Message + "<ul>"
|
||||
for _, msg := range event.MessagesVerbose {
|
||||
message = message + "<li>" + msg + "</li>"
|
||||
}
|
||||
message = message + "</ul>"
|
||||
statusMessage = message
|
||||
} else {
|
||||
statusMessage = event.Message
|
||||
}
|
||||
|
||||
htmlMessage := fmt.Sprintf(content, s.getStatusColor(event.LogLevel), notificationTitle(event), statusMessage, event.Jenkins.Name, event.Phase)
|
||||
message := gomail.NewMessage()
|
||||
|
||||
message.SetHeader("From", config.SMTP.From)
|
||||
message.SetHeader("To", config.SMTP.To)
|
||||
message.SetHeader("Subject", mailSubject)
|
||||
message.SetBody("text/html", htmlMessage)
|
||||
|
||||
if err := mailer.DialAndSend(message); err != nil {
|
||||
fmt.Print(err)
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s SMTP) getStatusColor(logLevel v1alpha2.NotificationLogLevel) StatusColor {
|
||||
switch logLevel {
|
||||
case v1alpha2.NotificationLogLevelInfo:
|
||||
return "blue"
|
||||
case v1alpha2.NotificationLogLevelWarning:
|
||||
return "red"
|
||||
default:
|
||||
return "gray"
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,152 @@
|
|||
package notifications
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/emersion/go-smtp"
|
||||
"github.com/jenkinsci/kubernetes-operator/pkg/apis/jenkins/v1alpha2"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"sigs.k8s.io/controller-runtime/pkg/client/fake"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
)
|
||||
|
||||
const (
|
||||
testSMTPUsername = "username"
|
||||
testSMTPPassword = "password"
|
||||
|
||||
testSMTPPort = 1025
|
||||
)
|
||||
|
||||
type testServer struct{}
|
||||
|
||||
// Login handles a login command with username and password.
|
||||
func (bkd *testServer) Login(state *smtp.ConnectionState, username, password string) (smtp.Session, error) {
|
||||
if username != testSMTPUsername || password != testSMTPPassword {
|
||||
return nil, errors.New("invalid username or password")
|
||||
}
|
||||
return &testSession{}, nil
|
||||
}
|
||||
|
||||
// AnonymousLogin requires clients to authenticate using SMTP AUTH before sending emails
|
||||
func (bkd *testServer) AnonymousLogin(state *smtp.ConnectionState) (smtp.Session, error) {
|
||||
return nil, smtp.ErrAuthRequired
|
||||
}
|
||||
|
||||
// A Session is returned after successful login.
|
||||
type testSession struct{}
|
||||
|
||||
func (s *testSession) Mail(from string) error {
|
||||
log.Println("Mail from:", from)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *testSession) Rcpt(to string) error {
|
||||
log.Println("Rcpt to:", to)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *testSession) Data(r io.Reader) error {
|
||||
if b, err := ioutil.ReadAll(r); err != nil {
|
||||
return err
|
||||
} else {
|
||||
log.Println("Data:", string(b))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *testSession) Reset() {}
|
||||
|
||||
func (s *testSession) Logout() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
func TestSMTP_Send(t *testing.T) {
|
||||
fakeClient := fake.NewFakeClient()
|
||||
testUsernameSelectorKeyName := "test-username-selector"
|
||||
testPasswordSelectorKeyName := "test-password-selector"
|
||||
testSecretName := "test-secret"
|
||||
|
||||
event := Event{
|
||||
Jenkins: v1alpha2.Jenkins{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: testCrName,
|
||||
Namespace: testNamespace,
|
||||
},
|
||||
},
|
||||
Phase: testPhase,
|
||||
Message: testMessage,
|
||||
MessagesVerbose: testMessageVerbose,
|
||||
LogLevel: testLoggingLevel,
|
||||
}
|
||||
|
||||
smtpClient := SMTP{k8sClient: fakeClient}
|
||||
|
||||
ts := &testServer{}
|
||||
|
||||
// Create fake SMTP server
|
||||
|
||||
s := smtp.NewServer(ts)
|
||||
|
||||
s.Addr = fmt.Sprintf(":%d", testSMTPPort)
|
||||
s.Domain = "localhost"
|
||||
s.ReadTimeout = 10 * time.Second
|
||||
s.WriteTimeout = 10 * time.Second
|
||||
s.MaxMessageBytes = 1024 * 1024
|
||||
s.MaxRecipients = 50
|
||||
s.AllowInsecureAuth = true
|
||||
|
||||
// Create secrets
|
||||
secret := &corev1.Secret{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: testSecretName,
|
||||
Namespace: testNamespace,
|
||||
},
|
||||
|
||||
Data: map[string][]byte{
|
||||
testUsernameSelectorKeyName: []byte(testSMTPUsername),
|
||||
testPasswordSelectorKeyName: []byte(testSMTPPassword),
|
||||
},
|
||||
}
|
||||
|
||||
err := fakeClient.Create(context.TODO(), secret)
|
||||
assert.NoError(t, err)
|
||||
|
||||
go func() {
|
||||
if err := s.ListenAndServe(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
err = smtpClient.Send(event, v1alpha2.Notification{
|
||||
SMTP: &v1alpha2.SMTP{
|
||||
Server: "localhost",
|
||||
From: "test@localhost",
|
||||
To: "test@localhost",
|
||||
TLSInsecureSkipVerify: true,
|
||||
Port: testSMTPPort,
|
||||
UsernameSecretKeySelector: v1alpha2.SecretKeySelector{
|
||||
LocalObjectReference: corev1.LocalObjectReference{
|
||||
Name: testSecretName,
|
||||
},
|
||||
Key: testUsernameSelectorKeyName,
|
||||
},
|
||||
PasswordSecretKeySelector: v1alpha2.SecretKeySelector{
|
||||
LocalObjectReference: corev1.LocalObjectReference{
|
||||
Name: testSecretName,
|
||||
},
|
||||
Key: testPasswordSelectorKeyName,
|
||||
},
|
||||
},
|
||||
})
|
||||
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
Loading…
Reference in New Issue