mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 03:24:26 +00:00
103 lines
3.9 KiB
Go
103 lines
3.9 KiB
Go
package projection
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/zitadel/zitadel/internal/domain"
|
|
"github.com/zitadel/zitadel/internal/eventstore"
|
|
old_handler "github.com/zitadel/zitadel/internal/eventstore/handler"
|
|
"github.com/zitadel/zitadel/internal/eventstore/handler/v2"
|
|
"github.com/zitadel/zitadel/internal/repository/instance"
|
|
"github.com/zitadel/zitadel/internal/repository/restrictions"
|
|
)
|
|
|
|
const (
|
|
RestrictionsProjectionTable = "projections.restrictions2"
|
|
|
|
RestrictionsColumnAggregateID = "aggregate_id"
|
|
RestrictionsColumnCreationDate = "creation_date"
|
|
RestrictionsColumnChangeDate = "change_date"
|
|
RestrictionsColumnResourceOwner = "resource_owner"
|
|
RestrictionsColumnInstanceID = "instance_id"
|
|
RestrictionsColumnSequence = "sequence"
|
|
|
|
RestrictionsColumnDisallowPublicOrgRegistration = "disallow_public_org_registration"
|
|
RestrictionsColumnAllowedLanguages = "allowed_languages"
|
|
)
|
|
|
|
type restrictionsProjection struct{}
|
|
|
|
func newRestrictionsProjection(ctx context.Context, config handler.Config) *handler.Handler {
|
|
return handler.NewHandler(ctx, &config, &restrictionsProjection{})
|
|
}
|
|
|
|
func (*restrictionsProjection) Name() string {
|
|
return RestrictionsProjectionTable
|
|
}
|
|
|
|
func (*restrictionsProjection) Init() *old_handler.Check {
|
|
return handler.NewTableCheck(
|
|
handler.NewTable([]*handler.InitColumn{
|
|
handler.NewColumn(RestrictionsColumnAggregateID, handler.ColumnTypeText),
|
|
handler.NewColumn(RestrictionsColumnCreationDate, handler.ColumnTypeTimestamp),
|
|
handler.NewColumn(RestrictionsColumnChangeDate, handler.ColumnTypeTimestamp),
|
|
handler.NewColumn(RestrictionsColumnResourceOwner, handler.ColumnTypeText),
|
|
handler.NewColumn(RestrictionsColumnInstanceID, handler.ColumnTypeText),
|
|
handler.NewColumn(RestrictionsColumnSequence, handler.ColumnTypeInt64),
|
|
handler.NewColumn(RestrictionsColumnDisallowPublicOrgRegistration, handler.ColumnTypeBool, handler.Nullable()),
|
|
handler.NewColumn(RestrictionsColumnAllowedLanguages, handler.ColumnTypeTextArray, handler.Nullable()),
|
|
},
|
|
handler.NewPrimaryKey(RestrictionsColumnInstanceID, RestrictionsColumnResourceOwner),
|
|
),
|
|
)
|
|
}
|
|
|
|
func (p *restrictionsProjection) Reducers() []handler.AggregateReducer {
|
|
return []handler.AggregateReducer{
|
|
{
|
|
Aggregate: restrictions.AggregateType,
|
|
EventReducers: []handler.EventReducer{
|
|
{
|
|
Event: restrictions.SetEventType,
|
|
Reduce: p.reduceRestrictionsSet,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Aggregate: instance.AggregateType,
|
|
EventReducers: []handler.EventReducer{
|
|
{
|
|
Event: instance.InstanceRemovedEventType,
|
|
Reduce: reduceInstanceRemovedHelper(RestrictionsColumnInstanceID),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (p *restrictionsProjection) reduceRestrictionsSet(event eventstore.Event) (*handler.Statement, error) {
|
|
e, err := assertEvent[*restrictions.SetEvent](event)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
conflictCols := []handler.Column{
|
|
handler.NewCol(RestrictionsColumnInstanceID, e.Aggregate().InstanceID),
|
|
handler.NewCol(RestrictionsColumnResourceOwner, e.Aggregate().ResourceOwner),
|
|
}
|
|
updateCols := []handler.Column{
|
|
handler.NewCol(RestrictionsColumnInstanceID, e.Aggregate().InstanceID),
|
|
handler.NewCol(RestrictionsColumnResourceOwner, e.Aggregate().ResourceOwner),
|
|
handler.NewCol(RestrictionsColumnCreationDate, handler.OnlySetValueOnInsert(RestrictionsProjectionTable, e.CreationDate())),
|
|
handler.NewCol(RestrictionsColumnChangeDate, e.CreationDate()),
|
|
handler.NewCol(RestrictionsColumnSequence, e.Sequence()),
|
|
handler.NewCol(RestrictionsColumnAggregateID, e.Aggregate().ID),
|
|
}
|
|
if e.DisallowPublicOrgRegistration != nil {
|
|
updateCols = append(updateCols, handler.NewCol(RestrictionsColumnDisallowPublicOrgRegistration, *e.DisallowPublicOrgRegistration))
|
|
}
|
|
if e.AllowedLanguages != nil {
|
|
updateCols = append(updateCols, handler.NewCol(RestrictionsColumnAllowedLanguages, domain.LanguagesToStrings(*e.AllowedLanguages)))
|
|
}
|
|
return handler.NewUpsertStatement(e, conflictCols, updateCols), nil
|
|
}
|