2020-11-06 21:09:19 +00:00
|
|
|
package iam
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/caos/zitadel/internal/eventstore/v2"
|
|
|
|
"github.com/caos/zitadel/internal/v2/repository/member"
|
2020-11-11 16:51:44 +00:00
|
|
|
"github.com/caos/zitadel/internal/v2/repository/policy"
|
2020-11-06 21:09:19 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type ReadModel struct {
|
|
|
|
eventstore.ReadModel
|
|
|
|
|
|
|
|
SetUpStarted Step
|
|
|
|
SetUpDone Step
|
|
|
|
|
2020-11-11 16:51:44 +00:00
|
|
|
Members MembersReadModel
|
2020-11-26 12:14:07 +00:00
|
|
|
IDPs IDPConfigsReadModel
|
2020-11-06 21:09:19 +00:00
|
|
|
|
|
|
|
GlobalOrgID string
|
|
|
|
ProjectID string
|
2020-11-11 16:51:44 +00:00
|
|
|
|
2020-11-12 21:50:01 +00:00
|
|
|
DefaultLoginPolicy LoginPolicyReadModel
|
|
|
|
DefaultLabelPolicy LabelPolicyReadModel
|
|
|
|
DefaultOrgIAMPolicy OrgIAMPolicyReadModel
|
|
|
|
DefaultPasswordComplexityPolicy PasswordComplexityPolicyReadModel
|
|
|
|
DefaultPasswordAgePolicy PasswordAgePolicyReadModel
|
|
|
|
DefaultPasswordLockoutPolicy PasswordLockoutPolicyReadModel
|
2020-11-06 21:09:19 +00:00
|
|
|
}
|
|
|
|
|
2020-11-25 13:12:44 +00:00
|
|
|
func NewReadModel(id string) *ReadModel {
|
|
|
|
return &ReadModel{
|
|
|
|
ReadModel: eventstore.ReadModel{
|
|
|
|
AggregateID: id,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-11-26 12:14:07 +00:00
|
|
|
func (rm *ReadModel) IDPByID(idpID string) *IDPConfigReadModel {
|
|
|
|
_, config := rm.IDPs.ConfigByID(idpID)
|
|
|
|
if config == nil {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
return &IDPConfigReadModel{ConfigReadModel: *config}
|
|
|
|
}
|
|
|
|
|
2020-11-25 13:12:44 +00:00
|
|
|
func (rm *ReadModel) AppendEvents(events ...eventstore.EventReader) {
|
2020-11-06 21:09:19 +00:00
|
|
|
rm.ReadModel.AppendEvents(events...)
|
|
|
|
for _, event := range events {
|
|
|
|
switch event.(type) {
|
2020-11-26 12:14:07 +00:00
|
|
|
case *member.AddedEvent,
|
|
|
|
*member.ChangedEvent,
|
|
|
|
*member.RemovedEvent:
|
|
|
|
|
2020-11-11 16:51:44 +00:00
|
|
|
rm.Members.AppendEvents(event)
|
2020-11-26 12:14:07 +00:00
|
|
|
case *IDPConfigAddedEvent,
|
|
|
|
*IDPConfigChangedEvent,
|
|
|
|
*IDPConfigDeactivatedEvent,
|
|
|
|
*IDPConfigReactivatedEvent,
|
|
|
|
*IDPConfigRemovedEvent,
|
|
|
|
*IDPOIDCConfigAddedEvent,
|
|
|
|
*IDPOIDCConfigChangedEvent:
|
|
|
|
|
|
|
|
rm.IDPs.AppendEvents(event)
|
|
|
|
case *policy.LabelPolicyAddedEvent,
|
|
|
|
*policy.LabelPolicyChangedEvent:
|
|
|
|
|
2020-11-11 16:51:44 +00:00
|
|
|
rm.DefaultLabelPolicy.AppendEvents(event)
|
2020-11-26 12:14:07 +00:00
|
|
|
case *policy.LoginPolicyAddedEvent,
|
|
|
|
*policy.LoginPolicyChangedEvent:
|
|
|
|
|
2020-11-11 16:51:44 +00:00
|
|
|
rm.DefaultLoginPolicy.AppendEvents(event)
|
|
|
|
case *policy.OrgIAMPolicyAddedEvent:
|
|
|
|
rm.DefaultOrgIAMPolicy.AppendEvents(event)
|
2020-11-26 12:14:07 +00:00
|
|
|
case *policy.PasswordComplexityPolicyAddedEvent,
|
|
|
|
*policy.PasswordComplexityPolicyChangedEvent:
|
|
|
|
|
2020-11-11 16:51:44 +00:00
|
|
|
rm.DefaultPasswordComplexityPolicy.AppendEvents(event)
|
2020-11-26 12:14:07 +00:00
|
|
|
case *policy.PasswordAgePolicyAddedEvent,
|
|
|
|
*policy.PasswordAgePolicyChangedEvent:
|
|
|
|
|
2020-11-11 16:51:44 +00:00
|
|
|
rm.DefaultPasswordAgePolicy.AppendEvents(event)
|
2020-11-26 12:14:07 +00:00
|
|
|
case *policy.PasswordLockoutPolicyAddedEvent,
|
|
|
|
*policy.PasswordLockoutPolicyChangedEvent:
|
|
|
|
|
2020-11-11 16:51:44 +00:00
|
|
|
rm.DefaultPasswordLockoutPolicy.AppendEvents(event)
|
2020-11-06 21:09:19 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (rm *ReadModel) Reduce() (err error) {
|
|
|
|
for _, event := range rm.Events {
|
|
|
|
switch e := event.(type) {
|
|
|
|
case *ProjectSetEvent:
|
|
|
|
rm.ProjectID = e.ProjectID
|
|
|
|
case *GlobalOrgSetEvent:
|
|
|
|
rm.GlobalOrgID = e.OrgID
|
|
|
|
case *SetupStepEvent:
|
|
|
|
if e.Done {
|
|
|
|
rm.SetUpDone = e.Step
|
|
|
|
} else {
|
|
|
|
rm.SetUpStarted = e.Step
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-11-11 16:51:44 +00:00
|
|
|
for _, reduce := range []func() error{
|
|
|
|
rm.Members.Reduce,
|
2020-11-26 12:14:07 +00:00
|
|
|
rm.IDPs.Reduce,
|
2020-11-11 16:51:44 +00:00
|
|
|
rm.DefaultLoginPolicy.Reduce,
|
|
|
|
rm.DefaultLabelPolicy.Reduce,
|
|
|
|
rm.DefaultOrgIAMPolicy.Reduce,
|
|
|
|
rm.DefaultPasswordComplexityPolicy.Reduce,
|
|
|
|
rm.DefaultPasswordAgePolicy.Reduce,
|
|
|
|
rm.DefaultPasswordLockoutPolicy.Reduce,
|
|
|
|
rm.ReadModel.Reduce,
|
|
|
|
} {
|
|
|
|
if err = reduce(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2020-11-06 21:09:19 +00:00
|
|
|
}
|
2020-11-11 16:51:44 +00:00
|
|
|
|
|
|
|
return nil
|
2020-11-06 21:09:19 +00:00
|
|
|
}
|
2020-11-12 21:50:01 +00:00
|
|
|
|
|
|
|
func (rm *ReadModel) AppendAndReduce(events ...eventstore.EventReader) error {
|
2020-11-25 19:04:32 +00:00
|
|
|
rm.AppendEvents(events...)
|
2020-11-12 21:50:01 +00:00
|
|
|
return rm.Reduce()
|
|
|
|
}
|
2020-11-25 13:12:44 +00:00
|
|
|
|
2020-12-01 13:44:19 +00:00
|
|
|
func (rm *ReadModel) Query() *eventstore.SearchQueryBuilder {
|
|
|
|
return eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent, AggregateType).AggregateIDs(rm.AggregateID)
|
2020-11-25 13:12:44 +00:00
|
|
|
}
|