mirror of
https://github.com/grafana/grafana.git
synced 2025-02-20 11:48:34 -06:00
9e8efaa459
Adds three functions: `withStoredImages` iterates over a list of models.Alerts, extracting a stored image's data from storage, if available, and executing a user-provided function. `withStoredImage` does this for an image attached to a specific alert. `openImage` finds and opens an image file on disk. Moves `store.Image` to `models.Image` Simplifies `channels.ImageStore` interface and updates notifiers that use it to use the simpler methods. Updates all pkg/alert/notifier/channels to use withStoredImage routines.
75 lines
2.4 KiB
Go
75 lines
2.4 KiB
Go
package channels
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"strings"
|
|
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
"github.com/grafana/grafana/pkg/services/notifications"
|
|
"github.com/prometheus/alertmanager/template"
|
|
)
|
|
|
|
type FactoryConfig struct {
|
|
Config *NotificationChannelConfig
|
|
NotificationService notifications.Service
|
|
DecryptFunc GetDecryptedValueFn
|
|
ImageStore ImageStore
|
|
// Used to retrieve image URLs for messages, or data for uploads.
|
|
Template *template.Template
|
|
}
|
|
|
|
type ImageStore interface {
|
|
GetImage(ctx context.Context, token string) (*models.Image, error)
|
|
}
|
|
|
|
func NewFactoryConfig(config *NotificationChannelConfig, notificationService notifications.Service,
|
|
decryptFunc GetDecryptedValueFn, template *template.Template, imageStore ImageStore) (FactoryConfig, error) {
|
|
if config.Settings == nil {
|
|
return FactoryConfig{}, errors.New("no settings supplied")
|
|
}
|
|
// not all receivers do need secure settings, we still might interact with
|
|
// them, so we make sure they are never nil
|
|
if config.SecureSettings == nil {
|
|
config.SecureSettings = map[string][]byte{}
|
|
}
|
|
|
|
if imageStore == nil {
|
|
imageStore = &UnavailableImageStore{}
|
|
}
|
|
return FactoryConfig{
|
|
Config: config,
|
|
NotificationService: notificationService,
|
|
DecryptFunc: decryptFunc,
|
|
Template: template,
|
|
ImageStore: imageStore,
|
|
}, nil
|
|
}
|
|
|
|
var receiverFactories = map[string]func(FactoryConfig) (NotificationChannel, error){
|
|
"prometheus-alertmanager": AlertmanagerFactory,
|
|
"dingding": DingDingFactory,
|
|
"discord": DiscordFactory,
|
|
"email": EmailFactory,
|
|
"googlechat": GoogleChatFactory,
|
|
"kafka": KafkaFactory,
|
|
"line": LineFactory,
|
|
"opsgenie": OpsgenieFactory,
|
|
"pagerduty": PagerdutyFactory,
|
|
"pushover": PushoverFactory,
|
|
"sensugo": SensuGoFactory,
|
|
"slack": SlackFactory,
|
|
"teams": TeamsFactory,
|
|
"telegram": TelegramFactory,
|
|
"threema": ThreemaFactory,
|
|
"victorops": VictorOpsFactory,
|
|
"webhook": WebHookFactory,
|
|
"wecom": WeComFactory,
|
|
}
|
|
|
|
func Factory(receiverType string) (func(FactoryConfig) (NotificationChannel, error), bool) {
|
|
receiverType = strings.ToLower(receiverType)
|
|
factory, exists := receiverFactories[receiverType]
|
|
return factory, exists
|
|
}
|