mirror of
https://github.com/andwati/zitadel.git
synced 2026-11-12 17:15:15 +00:00
5349d96ce4
* sub queries * fix: tests * add builder to tests * new search query * rename searchquerybuilder to builder * remove comment from code * test with multiple queries * add filters test * fix(contibute): listing * add validate module * fix: search queries * remove unused event type in query * ignore query if error in marshal * go mod tidy * update privacy policy query * update queries Co-authored-by: Livio Amstutz <[email protected]>
102 lines
2.7 KiB
Go
102 lines
2.7 KiB
Go
package command
|
|
|
|
import (
|
|
"github.com/caos/zitadel/internal/domain"
|
|
"github.com/caos/zitadel/internal/eventstore"
|
|
"github.com/caos/zitadel/internal/repository/iam"
|
|
)
|
|
|
|
type IAMSecondFactorWriteModel struct {
|
|
SecondFactorWriteModel
|
|
}
|
|
|
|
func NewIAMSecondFactorWriteModel(factorType domain.SecondFactorType) *IAMSecondFactorWriteModel {
|
|
return &IAMSecondFactorWriteModel{
|
|
SecondFactorWriteModel{
|
|
WriteModel: eventstore.WriteModel{
|
|
AggregateID: domain.IAMID,
|
|
ResourceOwner: domain.IAMID,
|
|
},
|
|
MFAType: factorType,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (wm *IAMSecondFactorWriteModel) AppendEvents(events ...eventstore.EventReader) {
|
|
for _, event := range events {
|
|
switch e := event.(type) {
|
|
case *iam.LoginPolicySecondFactorAddedEvent:
|
|
if wm.MFAType == e.MFAType {
|
|
wm.WriteModel.AppendEvents(&e.SecondFactorAddedEvent)
|
|
}
|
|
case *iam.LoginPolicySecondFactorRemovedEvent:
|
|
if wm.MFAType == e.MFAType {
|
|
wm.WriteModel.AppendEvents(&e.SecondFactorRemovedEvent)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (wm *IAMSecondFactorWriteModel) Reduce() error {
|
|
return wm.SecondFactorWriteModel.Reduce()
|
|
}
|
|
|
|
func (wm *IAMSecondFactorWriteModel) Query() *eventstore.SearchQueryBuilder {
|
|
return eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
|
|
ResourceOwner(wm.ResourceOwner).
|
|
AddQuery().
|
|
AggregateTypes(iam.AggregateType).
|
|
AggregateIDs(wm.WriteModel.AggregateID).
|
|
EventTypes(
|
|
iam.LoginPolicySecondFactorAddedEventType,
|
|
iam.LoginPolicySecondFactorRemovedEventType).
|
|
Builder()
|
|
}
|
|
|
|
type IAMMultiFactorWriteModel struct {
|
|
MultiFactorWriteModel
|
|
}
|
|
|
|
func NewIAMMultiFactorWriteModel(factorType domain.MultiFactorType) *IAMMultiFactorWriteModel {
|
|
return &IAMMultiFactorWriteModel{
|
|
MultiFactorWriteModel{
|
|
WriteModel: eventstore.WriteModel{
|
|
AggregateID: domain.IAMID,
|
|
ResourceOwner: domain.IAMID,
|
|
},
|
|
MFAType: factorType,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (wm *IAMMultiFactorWriteModel) AppendEvents(events ...eventstore.EventReader) {
|
|
for _, event := range events {
|
|
switch e := event.(type) {
|
|
case *iam.LoginPolicyMultiFactorAddedEvent:
|
|
if wm.MFAType == e.MFAType {
|
|
wm.WriteModel.AppendEvents(&e.MultiFactorAddedEvent)
|
|
}
|
|
case *iam.LoginPolicyMultiFactorRemovedEvent:
|
|
if wm.MFAType == e.MFAType {
|
|
wm.WriteModel.AppendEvents(&e.MultiFactorRemovedEvent)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (wm *IAMMultiFactorWriteModel) Reduce() error {
|
|
return wm.MultiFactorWriteModel.Reduce()
|
|
}
|
|
|
|
func (wm *IAMMultiFactorWriteModel) Query() *eventstore.SearchQueryBuilder {
|
|
return eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
|
|
ResourceOwner(wm.ResourceOwner).
|
|
AddQuery().
|
|
AggregateTypes(iam.AggregateType).
|
|
AggregateIDs(wm.WriteModel.AggregateID).
|
|
EventTypes(
|
|
iam.LoginPolicyMultiFactorAddedEventType,
|
|
iam.LoginPolicyMultiFactorRemovedEventType).
|
|
Builder()
|
|
}
|