package model import ( "encoding/json" "time" "github.com/caos/logging" "github.com/caos/zitadel/internal/domain" caos_errs "github.com/caos/zitadel/internal/errors" "github.com/caos/zitadel/internal/eventstore/v1/models" features_model "github.com/caos/zitadel/internal/features/model" iam_repo "github.com/caos/zitadel/internal/repository/iam" org_repo "github.com/caos/zitadel/internal/repository/org" ) const ( FeaturesKeyAggregateID = "aggregate_id" FeaturesKeyDefault = "default_features" ) type FeaturesView struct { AggregateID string `json:"-" gorm:"column:aggregate_id;primary_key"` CreationDate time.Time `json:"-" gorm:"column:creation_date"` ChangeDate time.Time `json:"-" gorm:"column:change_date"` Sequence uint64 `json:"-" gorm:"column:sequence"` Default bool `json:"-" gorm:"column:default_features"` TierName string `json:"tierName" gorm:"column:tier_name"` TierDescription string `json:"tierDescription" gorm:"column:tier_description"` State int32 `json:"state" gorm:"column:state"` StateDescription string `json:"stateDescription" gorm:"column:state_description"` AuditLogRetention time.Duration `json:"auditLogRetention" gorm:"column:audit_log_retention"` LoginPolicyFactors bool `json:"loginPolicyFactors" gorm:"column:login_policy_factors"` LoginPolicyIDP bool `json:"loginPolicyIDP" gorm:"column:login_policy_idp"` LoginPolicyPasswordless bool `json:"loginPolicyPasswordless" gorm:"column:login_policy_passwordless"` LoginPolicyRegistration bool `json:"loginPolicyRegistration" gorm:"column:login_policy_registration"` LoginPolicyUsernameLogin bool `json:"loginPolicyUsernameLogin" gorm:"column:login_policy_username_login"` LoginPolicyPasswordReset bool `json:"loginPolicyPasswordReset" gorm:"column:login_policy_password_reset"` PasswordComplexityPolicy bool `json:"passwordComplexityPolicy" gorm:"column:password_complexity_policy"` LabelPolicy *bool `json:"labelPolicy" gorm:"-"` LabelPolicyPrivateLabel bool `json:"labelPolicyPrivateLabel" gorm:"column:label_policy_private_label"` LabelPolicyWatermark bool `json:"labelPolicyWatermark" gorm:"column:label_policy_watermark"` CustomDomain bool `json:"customDomain" gorm:"column:custom_domain"` } func FeaturesToModel(features *FeaturesView) *features_model.FeaturesView { return &features_model.FeaturesView{ AggregateID: features.AggregateID, CreationDate: features.CreationDate, ChangeDate: features.ChangeDate, Sequence: features.Sequence, Default: features.Default, TierName: features.TierName, TierDescription: features.TierDescription, State: domain.FeaturesState(features.State), StateDescription: features.StateDescription, AuditLogRetention: features.AuditLogRetention, LoginPolicyFactors: features.LoginPolicyFactors, LoginPolicyIDP: features.LoginPolicyIDP, LoginPolicyPasswordless: features.LoginPolicyPasswordless, LoginPolicyRegistration: features.LoginPolicyRegistration, LoginPolicyUsernameLogin: features.LoginPolicyUsernameLogin, LoginPolicyPasswordReset: features.LoginPolicyPasswordReset, PasswordComplexityPolicy: features.PasswordComplexityPolicy, LabelPolicyPrivateLabel: features.LabelPolicyPrivateLabel, LabelPolicyWatermark: features.LabelPolicyWatermark, CustomDomain: features.CustomDomain, } } func (f *FeaturesView) AppendEvent(event *models.Event) (err error) { f.Sequence = event.Sequence f.ChangeDate = event.CreationDate switch string(event.Type) { case string(iam_repo.FeaturesSetEventType): f.SetRootData(event) f.CreationDate = event.CreationDate f.Default = true err = f.SetData(event) case string(org_repo.FeaturesSetEventType): f.SetRootData(event) f.CreationDate = event.CreationDate err = f.SetData(event) f.Default = false } return err } func (f *FeaturesView) SetRootData(event *models.Event) { if f.AggregateID == "" { f.AggregateID = event.AggregateID } } func (f *FeaturesView) SetData(event *models.Event) error { if err := json.Unmarshal(event.Data, f); err != nil { logging.Log("EVEN-DVsf2").WithError(err).Error("could not unmarshal event data") return caos_errs.ThrowInternal(err, "MODEL-Bfg31", "Could not unmarshal data") } if f.LabelPolicy != nil { f.LabelPolicyPrivateLabel = *f.LabelPolicy } return nil }