grafana/pkg/services/ngalert/notifier/sender.go

package notifier

import (
	"context"

	"github.com/grafana/grafana/pkg/models"
	"github.com/grafana/grafana/pkg/services/ngalert/notifier/channels"
	"github.com/grafana/grafana/pkg/services/notifications"
)

type sender struct {
	ns notifications.Service
}

func (s sender) SendWebhook(ctx context.Context, cmd *channels.SendWebhookSettings) error {
	return s.ns.SendWebhookSync(ctx, &models.SendWebhookSync{
		Url:         cmd.Url,
		User:        cmd.User,
		Password:    cmd.Password,
		Body:        cmd.Body,
		HttpMethod:  cmd.HttpMethod,
		HttpHeader:  cmd.HttpHeader,
		ContentType: cmd.ContentType,
		Validation:  cmd.Validation,
	})
}

func (s sender) SendEmail(ctx context.Context, cmd *channels.SendEmailSettings) error {
	var attached []*models.SendEmailAttachFile
	if cmd.AttachedFiles != nil {
		attached = make([]*models.SendEmailAttachFile, 0, len(cmd.AttachedFiles))
		for _, file := range cmd.AttachedFiles {
			attached = append(attached, &models.SendEmailAttachFile{
				Name:    file.Name,
				Content: file.Content,
			})
		}
	}
	return s.ns.SendEmailCommandHandlerSync(ctx, &models.SendEmailCommandSync{
		SendEmailCommand: models.SendEmailCommand{
			To:            cmd.To,
			SingleEmail:   cmd.SingleEmail,
			Template:      cmd.Template,
			Subject:       cmd.Subject,
			Data:          cmd.Data,
			Info:          cmd.Info,
			ReplyTo:       cmd.ReplyTo,
			EmbeddedFiles: cmd.EmbeddedFiles,
			AttachedFiles: attached,
		},
	})
}

func NewNotificationSender(ns notifications.Service) channels.NotificationSender {
	return &sender{ns: ns}
}