mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 11:34:26 +00:00
4a0d61d75a
* feat: use database as asset storage * being only uploading assets if allowed * tests * fixes * cleanup after merge * renaming * various fixes * fix: change to repository event types and removed unused code * feat: set default features * error handling * error handling and naming * fix tests * fix tests * fix merge * rename
238 lines
10 KiB
Go
238 lines
10 KiB
Go
package projection
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/caos/zitadel/internal/domain"
|
|
|
|
"github.com/caos/zitadel/internal/errors"
|
|
"github.com/caos/zitadel/internal/eventstore"
|
|
"github.com/caos/zitadel/internal/eventstore/handler"
|
|
"github.com/caos/zitadel/internal/eventstore/handler/crdb"
|
|
"github.com/caos/zitadel/internal/repository/features"
|
|
"github.com/caos/zitadel/internal/repository/instance"
|
|
"github.com/caos/zitadel/internal/repository/org"
|
|
)
|
|
|
|
const (
|
|
FeatureTable = "projections.features"
|
|
|
|
FeatureAggregateIDCol = "aggregate_id"
|
|
FeatureInstanceIDCol = "instance_id"
|
|
FeatureChangeDateCol = "change_date"
|
|
FeatureSequenceCol = "sequence"
|
|
FeatureIsDefaultCol = "is_default"
|
|
FeatureTierNameCol = "tier_name"
|
|
FeatureTierDescriptionCol = "tier_description"
|
|
FeatureStateCol = "state"
|
|
FeatureStateDescriptionCol = "state_description"
|
|
FeatureAuditLogRetentionCol = "audit_log_retention"
|
|
FeatureLoginPolicyFactorsCol = "login_policy_factors"
|
|
FeatureLoginPolicyIDPCol = "login_policy_idp"
|
|
FeatureLoginPolicyPasswordlessCol = "login_policy_passwordless"
|
|
FeatureLoginPolicyRegistrationCol = "login_policy_registration"
|
|
FeatureLoginPolicyUsernameLoginCol = "login_policy_username_login"
|
|
FeatureLoginPolicyPasswordResetCol = "login_policy_password_reset"
|
|
FeaturePasswordComplexityPolicyCol = "password_complexity_policy"
|
|
FeatureLabelPolicyPrivateLabelCol = "label_policy_private_label"
|
|
FeatureLabelPolicyWatermarkCol = "label_policy_watermark"
|
|
FeatureCustomDomainCol = "custom_domain"
|
|
FeaturePrivacyPolicyCol = "privacy_policy"
|
|
FeatureMetadataUserCol = "metadata_user"
|
|
FeatureCustomTextMessageCol = "custom_text_message"
|
|
FeatureCustomTextLoginCol = "custom_text_login"
|
|
FeatureLockoutPolicyCol = "lockout_policy"
|
|
FeatureActionsAllowedCol = "actions_allowed"
|
|
FeatureMaxActionsCol = "max_actions"
|
|
)
|
|
|
|
type FeatureProjection struct {
|
|
crdb.StatementHandler
|
|
}
|
|
|
|
func NewFeatureProjection(ctx context.Context, config crdb.StatementHandlerConfig) *FeatureProjection {
|
|
p := new(FeatureProjection)
|
|
config.ProjectionName = FeatureTable
|
|
config.Reducers = p.reducers()
|
|
config.InitCheck = crdb.NewTableCheck(
|
|
crdb.NewTable([]*crdb.Column{
|
|
crdb.NewColumn(FeatureAggregateIDCol, crdb.ColumnTypeText),
|
|
crdb.NewColumn(FeatureInstanceIDCol, crdb.ColumnTypeText),
|
|
crdb.NewColumn(FeatureChangeDateCol, crdb.ColumnTypeTimestamp),
|
|
crdb.NewColumn(FeatureSequenceCol, crdb.ColumnTypeInt64),
|
|
crdb.NewColumn(FeatureIsDefaultCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureTierNameCol, crdb.ColumnTypeText),
|
|
crdb.NewColumn(FeatureTierDescriptionCol, crdb.ColumnTypeText, crdb.Nullable()),
|
|
crdb.NewColumn(FeatureStateCol, crdb.ColumnTypeEnum, crdb.Default(0)),
|
|
crdb.NewColumn(FeatureStateDescriptionCol, crdb.ColumnTypeText, crdb.Nullable()),
|
|
crdb.NewColumn(FeatureAuditLogRetentionCol, crdb.ColumnTypeInt64, crdb.Default(0)),
|
|
crdb.NewColumn(FeatureLoginPolicyFactorsCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureLoginPolicyIDPCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureLoginPolicyPasswordlessCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureLoginPolicyRegistrationCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureLoginPolicyUsernameLoginCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureLoginPolicyPasswordResetCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeaturePasswordComplexityPolicyCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureLabelPolicyPrivateLabelCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureLabelPolicyWatermarkCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureCustomDomainCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeaturePrivacyPolicyCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureMetadataUserCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureCustomTextMessageCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureCustomTextLoginCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureLockoutPolicyCol, crdb.ColumnTypeBool, crdb.Default(false)),
|
|
crdb.NewColumn(FeatureActionsAllowedCol, crdb.ColumnTypeEnum, crdb.Default(0)),
|
|
crdb.NewColumn(FeatureMaxActionsCol, crdb.ColumnTypeInt64, crdb.Default(0)),
|
|
},
|
|
crdb.NewPrimaryKey(FeatureInstanceIDCol, FeatureAggregateIDCol),
|
|
),
|
|
)
|
|
p.StatementHandler = crdb.NewStatementHandler(ctx, config)
|
|
return p
|
|
}
|
|
|
|
func (p *FeatureProjection) reducers() []handler.AggregateReducer {
|
|
return []handler.AggregateReducer{
|
|
{
|
|
Aggregate: org.AggregateType,
|
|
EventRedusers: []handler.EventReducer{
|
|
{
|
|
Event: org.FeaturesSetEventType,
|
|
Reduce: p.reduceFeatureSet,
|
|
},
|
|
{
|
|
Event: org.FeaturesRemovedEventType,
|
|
Reduce: p.reduceFeatureRemoved,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Aggregate: instance.AggregateType,
|
|
EventRedusers: []handler.EventReducer{
|
|
{
|
|
Event: instance.FeaturesSetEventType,
|
|
Reduce: p.reduceFeatureSet,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (p *FeatureProjection) reduceFeatureSet(event eventstore.Event) (*handler.Statement, error) {
|
|
var featureEvent features.FeaturesSetEvent
|
|
var isDefault bool
|
|
switch e := event.(type) {
|
|
case *instance.FeaturesSetEvent:
|
|
featureEvent = e.FeaturesSetEvent
|
|
isDefault = true
|
|
case *org.FeaturesSetEvent:
|
|
featureEvent = e.FeaturesSetEvent
|
|
isDefault = false
|
|
default:
|
|
return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-K0erf", "reduce.wrong.event.type %v", []eventstore.EventType{org.FeaturesSetEventType, instance.FeaturesSetEventType})
|
|
}
|
|
|
|
cols := []handler.Column{
|
|
handler.NewCol(FeatureAggregateIDCol, featureEvent.Aggregate().ID),
|
|
handler.NewCol(FeatureInstanceIDCol, featureEvent.Aggregate().InstanceID),
|
|
handler.NewCol(FeatureChangeDateCol, featureEvent.CreationDate()),
|
|
handler.NewCol(FeatureSequenceCol, featureEvent.Sequence()),
|
|
handler.NewCol(FeatureIsDefaultCol, isDefault),
|
|
}
|
|
if featureEvent.TierName != nil {
|
|
cols = append(cols, handler.NewCol(FeatureTierNameCol, *featureEvent.TierName))
|
|
}
|
|
if featureEvent.TierDescription != nil {
|
|
cols = append(cols, handler.NewCol(FeatureTierDescriptionCol, *featureEvent.TierDescription))
|
|
}
|
|
if featureEvent.State != nil {
|
|
cols = append(cols, handler.NewCol(FeatureStateCol, *featureEvent.State))
|
|
}
|
|
if featureEvent.StateDescription != nil {
|
|
cols = append(cols, handler.NewCol(FeatureStateDescriptionCol, *featureEvent.StateDescription))
|
|
}
|
|
if featureEvent.AuditLogRetention != nil {
|
|
cols = append(cols, handler.NewCol(FeatureAuditLogRetentionCol, *featureEvent.AuditLogRetention))
|
|
}
|
|
if featureEvent.LoginPolicyFactors != nil {
|
|
cols = append(cols, handler.NewCol(FeatureLoginPolicyFactorsCol, *featureEvent.LoginPolicyFactors))
|
|
}
|
|
if featureEvent.LoginPolicyIDP != nil {
|
|
cols = append(cols, handler.NewCol(FeatureLoginPolicyIDPCol, *featureEvent.LoginPolicyIDP))
|
|
}
|
|
if featureEvent.LoginPolicyPasswordless != nil {
|
|
cols = append(cols, handler.NewCol(FeatureLoginPolicyPasswordlessCol, *featureEvent.LoginPolicyPasswordless))
|
|
}
|
|
if featureEvent.LoginPolicyRegistration != nil {
|
|
cols = append(cols, handler.NewCol(FeatureLoginPolicyRegistrationCol, *featureEvent.LoginPolicyRegistration))
|
|
}
|
|
if featureEvent.LoginPolicyUsernameLogin != nil {
|
|
cols = append(cols, handler.NewCol(FeatureLoginPolicyUsernameLoginCol, *featureEvent.LoginPolicyUsernameLogin))
|
|
}
|
|
if featureEvent.LoginPolicyPasswordReset != nil {
|
|
cols = append(cols, handler.NewCol(FeatureLoginPolicyPasswordResetCol, *featureEvent.LoginPolicyPasswordReset))
|
|
}
|
|
if featureEvent.PasswordComplexityPolicy != nil {
|
|
cols = append(cols, handler.NewCol(FeaturePasswordComplexityPolicyCol, *featureEvent.PasswordComplexityPolicy))
|
|
}
|
|
if featureEvent.LabelPolicyPrivateLabel != nil || featureEvent.LabelPolicy != nil {
|
|
var value bool
|
|
if featureEvent.LabelPolicyPrivateLabel != nil {
|
|
value = *featureEvent.LabelPolicyPrivateLabel
|
|
} else {
|
|
value = *featureEvent.LabelPolicy
|
|
}
|
|
cols = append(cols, handler.NewCol(FeatureLabelPolicyPrivateLabelCol, value))
|
|
}
|
|
if featureEvent.LabelPolicyWatermark != nil {
|
|
cols = append(cols, handler.NewCol(FeatureLabelPolicyWatermarkCol, *featureEvent.LabelPolicyWatermark))
|
|
}
|
|
if featureEvent.CustomDomain != nil {
|
|
cols = append(cols, handler.NewCol(FeatureCustomDomainCol, *featureEvent.CustomDomain))
|
|
}
|
|
if featureEvent.PrivacyPolicy != nil {
|
|
cols = append(cols, handler.NewCol(FeaturePrivacyPolicyCol, *featureEvent.PrivacyPolicy))
|
|
}
|
|
if featureEvent.MetadataUser != nil {
|
|
cols = append(cols, handler.NewCol(FeatureMetadataUserCol, *featureEvent.MetadataUser))
|
|
}
|
|
if featureEvent.CustomTextMessage != nil {
|
|
cols = append(cols, handler.NewCol(FeatureCustomTextMessageCol, *featureEvent.CustomTextMessage))
|
|
}
|
|
if featureEvent.CustomTextLogin != nil {
|
|
cols = append(cols, handler.NewCol(FeatureCustomTextLoginCol, *featureEvent.CustomTextLogin))
|
|
}
|
|
if featureEvent.LockoutPolicy != nil {
|
|
cols = append(cols, handler.NewCol(FeatureLockoutPolicyCol, *featureEvent.LockoutPolicy))
|
|
}
|
|
if featureEvent.Actions != nil {
|
|
actionsAllowed := domain.ActionsNotAllowed
|
|
if *featureEvent.Actions {
|
|
actionsAllowed = domain.ActionsAllowedUnlimited
|
|
}
|
|
cols = append(cols, handler.NewCol(FeatureActionsAllowedCol, actionsAllowed))
|
|
}
|
|
if featureEvent.ActionsAllowed != nil {
|
|
cols = append(cols, handler.NewCol(FeatureActionsAllowedCol, *featureEvent.ActionsAllowed))
|
|
}
|
|
if featureEvent.MaxActions != nil {
|
|
cols = append(cols, handler.NewCol(FeatureMaxActionsCol, *featureEvent.MaxActions))
|
|
}
|
|
return crdb.NewUpsertStatement(
|
|
&featureEvent,
|
|
cols), nil
|
|
}
|
|
|
|
func (p *FeatureProjection) reduceFeatureRemoved(event eventstore.Event) (*handler.Statement, error) {
|
|
e, ok := event.(*org.FeaturesRemovedEvent)
|
|
if !ok {
|
|
return nil, errors.ThrowInvalidArgumentf(nil, "HANDL-0p4rf", "reduce.wrong.event.type %s", org.FeaturesRemovedEventType)
|
|
}
|
|
return crdb.NewDeleteStatement(
|
|
e,
|
|
[]handler.Condition{
|
|
handler.NewCond(FeatureAggregateIDCol, e.Aggregate().ID),
|
|
},
|
|
), nil
|
|
}
|