package projection import ( "context" "github.com/zitadel/zitadel/internal/domain" "github.com/zitadel/zitadel/internal/repository/settings" "github.com/zitadel/zitadel/internal/errors" "github.com/zitadel/zitadel/internal/eventstore" "github.com/zitadel/zitadel/internal/eventstore/handler" "github.com/zitadel/zitadel/internal/eventstore/handler/crdb" "github.com/zitadel/zitadel/internal/repository/instance" ) const ( DebugNotificationProviderTable = "projections.notification_providers" DebugNotificationProviderAggIDCol = "aggregate_id" DebugNotificationProviderCreationDateCol = "creation_date" DebugNotificationProviderChangeDateCol = "change_date" DebugNotificationProviderSequenceCol = "sequence" DebugNotificationProviderResourceOwnerCol = "resource_owner" DebugNotificationProviderInstanceIDCol = "instance_id" DebugNotificationProviderStateCol = "state" DebugNotificationProviderTypeCol = "provider_type" DebugNotificationProviderCompactCol = "compact" ) type debugNotificationProviderProjection struct { crdb.StatementHandler } func newDebugNotificationProviderProjection(ctx context.Context, config crdb.StatementHandlerConfig) *debugNotificationProviderProjection { p := &debugNotificationProviderProjection{} config.ProjectionName = DebugNotificationProviderTable config.Reducers = p.reducers() config.InitCheck = crdb.NewTableCheck( crdb.NewTable([]*crdb.Column{ crdb.NewColumn(DebugNotificationProviderAggIDCol, crdb.ColumnTypeText), crdb.NewColumn(DebugNotificationProviderCreationDateCol, crdb.ColumnTypeTimestamp), crdb.NewColumn(DebugNotificationProviderChangeDateCol, crdb.ColumnTypeTimestamp), crdb.NewColumn(DebugNotificationProviderSequenceCol, crdb.ColumnTypeInt64), crdb.NewColumn(DebugNotificationProviderResourceOwnerCol, crdb.ColumnTypeText), crdb.NewColumn(DebugNotificationProviderInstanceIDCol, crdb.ColumnTypeText), crdb.NewColumn(DebugNotificationProviderStateCol, crdb.ColumnTypeEnum), crdb.NewColumn(DebugNotificationProviderTypeCol, crdb.ColumnTypeEnum), crdb.NewColumn(DebugNotificationProviderCompactCol, crdb.ColumnTypeBool), }, crdb.NewPrimaryKey(DebugNotificationProviderInstanceIDCol, DebugNotificationProviderAggIDCol, DebugNotificationProviderTypeCol), ), ) p.StatementHandler = crdb.NewStatementHandler(ctx, config) return p } func (p *debugNotificationProviderProjection) reducers() []handler.AggregateReducer { return []handler.AggregateReducer{ { Aggregate: instance.AggregateType, EventRedusers: []handler.EventReducer{ { Event: instance.DebugNotificationProviderFileAddedEventType, Reduce: p.reduceDebugNotificationProviderAdded, }, { Event: instance.DebugNotificationProviderFileChangedEventType, Reduce: p.reduceDebugNotificationProviderChanged, }, { Event: instance.DebugNotificationProviderFileRemovedEventType, Reduce: p.reduceDebugNotificationProviderRemoved, }, { Event: instance.DebugNotificationProviderLogAddedEventType, Reduce: p.reduceDebugNotificationProviderAdded, }, { Event: instance.DebugNotificationProviderLogChangedEventType, Reduce: p.reduceDebugNotificationProviderChanged, }, { Event: instance.DebugNotificationProviderLogRemovedEventType, Reduce: p.reduceDebugNotificationProviderRemoved, }, }, }, } } func (p *debugNotificationProviderProjection) reduceDebugNotificationProviderAdded(event eventstore.Event) (*handler.Statement, error) { var providerEvent settings.DebugNotificationProviderAddedEvent var providerType domain.NotificationProviderType switch e := event.(type) { case *instance.DebugNotificationProviderFileAddedEvent: providerEvent = e.DebugNotificationProviderAddedEvent providerType = domain.NotificationProviderTypeFile case *instance.DebugNotificationProviderLogAddedEvent: providerEvent = e.DebugNotificationProviderAddedEvent providerType = domain.NotificationProviderTypeLog default: return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-pYPxS", "reduce.wrong.event.type %v", []eventstore.EventType{instance.DebugNotificationProviderFileAddedEventType, instance.DebugNotificationProviderLogAddedEventType}) } return crdb.NewCreateStatement(&providerEvent, []handler.Column{ handler.NewCol(DebugNotificationProviderAggIDCol, providerEvent.Aggregate().ID), handler.NewCol(DebugNotificationProviderCreationDateCol, providerEvent.CreationDate()), handler.NewCol(DebugNotificationProviderChangeDateCol, providerEvent.CreationDate()), handler.NewCol(DebugNotificationProviderSequenceCol, providerEvent.Sequence()), handler.NewCol(DebugNotificationProviderResourceOwnerCol, providerEvent.Aggregate().ResourceOwner), handler.NewCol(DebugNotificationProviderInstanceIDCol, providerEvent.Aggregate().InstanceID), handler.NewCol(DebugNotificationProviderStateCol, domain.NotificationProviderStateActive), handler.NewCol(DebugNotificationProviderTypeCol, providerType), handler.NewCol(DebugNotificationProviderCompactCol, providerEvent.Compact), }), nil } func (p *debugNotificationProviderProjection) reduceDebugNotificationProviderChanged(event eventstore.Event) (*handler.Statement, error) { var providerEvent settings.DebugNotificationProviderChangedEvent var providerType domain.NotificationProviderType switch e := event.(type) { case *instance.DebugNotificationProviderFileChangedEvent: providerEvent = e.DebugNotificationProviderChangedEvent providerType = domain.NotificationProviderTypeFile case *instance.DebugNotificationProviderLogChangedEvent: providerEvent = e.DebugNotificationProviderChangedEvent providerType = domain.NotificationProviderTypeLog default: return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-pYPxS", "reduce.wrong.event.type %v", []eventstore.EventType{instance.DebugNotificationProviderFileChangedEventType, instance.DebugNotificationProviderLogChangedEventType}) } cols := []handler.Column{ handler.NewCol(DebugNotificationProviderChangeDateCol, providerEvent.CreationDate()), handler.NewCol(DebugNotificationProviderSequenceCol, providerEvent.Sequence()), } if providerEvent.Compact != nil { cols = append(cols, handler.NewCol(DebugNotificationProviderCompactCol, *providerEvent.Compact)) } return crdb.NewUpdateStatement( &providerEvent, cols, []handler.Condition{ handler.NewCond(DebugNotificationProviderAggIDCol, providerEvent.Aggregate().ID), handler.NewCond(DebugNotificationProviderTypeCol, providerType), }, ), nil } func (p *debugNotificationProviderProjection) reduceDebugNotificationProviderRemoved(event eventstore.Event) (*handler.Statement, error) { var providerEvent settings.DebugNotificationProviderRemovedEvent var providerType domain.NotificationProviderType switch e := event.(type) { case *instance.DebugNotificationProviderFileRemovedEvent: providerEvent = e.DebugNotificationProviderRemovedEvent providerType = domain.NotificationProviderTypeFile case *instance.DebugNotificationProviderLogRemovedEvent: providerEvent = e.DebugNotificationProviderRemovedEvent providerType = domain.NotificationProviderTypeLog default: return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-dow9f", "reduce.wrong.event.type %v", []eventstore.EventType{instance.DebugNotificationProviderFileRemovedEventType, instance.DebugNotificationProviderLogRemovedEventType}) } return crdb.NewDeleteStatement( &providerEvent, []handler.Condition{ handler.NewCond(DebugNotificationProviderAggIDCol, providerEvent.Aggregate().ID), handler.NewCond(DebugNotificationProviderTypeCol, providerType), }, ), nil }