zot/pkg/extensions/extension_events.go

//go:build events
// +build events

package extensions

import (
	zerr "zotregistry.dev/zot/errors"
	"zotregistry.dev/zot/pkg/api/config"
	eventsconfig "zotregistry.dev/zot/pkg/extensions/config/events"
	"zotregistry.dev/zot/pkg/extensions/events"
	"zotregistry.dev/zot/pkg/log"
)

func NewEventRecorder(config *config.Config, log log.Logger) (events.Recorder, error) {
	if !config.IsEventRecorderEnabled() {
		log.Info().Msg("events disabled in configuration")

		return nil, zerr.ErrExtensionNotEnabled
	}

	eventConfig := config.Extensions.Events

	if eventConfig.Sinks == nil || len(eventConfig.Sinks) == 0 {
		log.Info().Msg("no sinks provided, skipping events extension setup")

		return nil, zerr.ErrExtensionNotEnabled
	}

	var sinks []events.Sink

	log.Info().Msg("setting up event sinks")

	for _, sinkConfig := range eventConfig.Sinks {
		switch sinkConfig.Type {
		case eventsconfig.HTTP:
			sink, err := events.NewHTTPSink(sinkConfig)
			if err != nil {
				return nil, err
			}

			sinks = append(sinks, sink)
		case eventsconfig.NATS:
			sink, err := events.NewNATSSink(sinkConfig)
			if err != nil {
				return nil, err
			}

			sinks = append(sinks, sink)
		default:
			log.Warn().Msgf("skipping unsupported sink type: %s", sinkConfig.Type)
		}
	}

	if len(sinks) == 0 {
		log.Warn().Msg("no sinks provided, skipping events extension setup")

		return nil, zerr.ErrExtensionNotEnabled
	}

	return events.NewRecorder(log, sinks...)
}