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

package notifier

import (
	"context"
	"errors"

	"github.com/grafana/alerting/alerting/notifier/channels"

	"github.com/grafana/grafana/pkg/services/ngalert/models"
	"github.com/grafana/grafana/pkg/services/ngalert/store"
)

type imageStore struct {
	store store.ImageStore
}

func newImageStore(store store.ImageStore) channels.ImageStore {
	return &imageStore{
		store: store,
	}
}

func (i imageStore) GetImage(ctx context.Context, token string) (*channels.Image, error) {
	image, err := i.store.GetImage(ctx, token)
	if err != nil {
		if errors.Is(err, models.ErrImageNotFound) {
			err = channels.ErrImageNotFound
		}
	}
	var result *channels.Image
	if image != nil {
		result = &channels.Image{
			Token:     image.Token,
			Path:      image.Path,
			URL:       image.URL,
			CreatedAt: image.CreatedAt,
		}
	}
	return result, err
}