1
0
mirror of https://github.com/zitadel/zitadel.git synced 2024-12-23 00:07:31 +00:00
zitadel/internal/command/instance_domain_model.go

134 lines
3.4 KiB
Go
Raw Normal View History

package command
import (
"context"
"github.com/zitadel/zitadel/internal/api/authz"
"github.com/zitadel/zitadel/internal/domain"
"github.com/zitadel/zitadel/internal/eventstore"
"github.com/zitadel/zitadel/internal/repository/instance"
)
type InstanceDomainWriteModel struct {
eventstore.WriteModel
Domain string
Generated bool
State domain.InstanceDomainState
}
func NewInstanceDomainWriteModel(ctx context.Context, instanceDomain string) *InstanceDomainWriteModel {
return &InstanceDomainWriteModel{
WriteModel: eventstore.WriteModel{
AggregateID: authz.GetInstance(ctx).InstanceID(),
ResourceOwner: authz.GetInstance(ctx).InstanceID(),
},
Domain: instanceDomain,
}
}
func (wm *InstanceDomainWriteModel) AppendEvents(events ...eventstore.Event) {
for _, event := range events {
switch e := event.(type) {
case *instance.DomainAddedEvent:
if e.Domain != wm.Domain {
continue
}
wm.WriteModel.AppendEvents(e)
case *instance.DomainRemovedEvent:
if e.Domain != wm.Domain {
continue
}
wm.WriteModel.AppendEvents(e)
}
}
}
func (wm *InstanceDomainWriteModel) Reduce() error {
for _, event := range wm.Events {
switch e := event.(type) {
case *instance.DomainAddedEvent:
wm.Domain = e.Domain
wm.Generated = e.Generated
wm.State = domain.InstanceDomainStateActive
case *instance.DomainRemovedEvent:
wm.State = domain.InstanceDomainStateRemoved
}
}
return wm.WriteModel.Reduce()
}
func (wm *InstanceDomainWriteModel) Query() *eventstore.SearchQueryBuilder {
return eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
ResourceOwner(wm.ResourceOwner).
AddQuery().
AggregateTypes(instance.AggregateType).
AggregateIDs(wm.AggregateID).
EventTypes(
instance.InstanceDomainAddedEventType,
instance.InstanceDomainRemovedEventType).
Builder()
}
type InstanceDomainsWriteModel struct {
eventstore.WriteModel
Domains []string
}
func (wm *InstanceDomainsWriteModel) Query() *eventstore.SearchQueryBuilder {
return eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
ResourceOwner(wm.ResourceOwner).
AddQuery().
AggregateTypes(instance.AggregateType).
AggregateIDs(wm.AggregateID).
EventTypes(
instance.InstanceDomainAddedEventType,
instance.InstanceDomainRemovedEventType).
Builder()
}
func NewInstanceDomainsWriteModel(instanceID string) *InstanceDomainsWriteModel {
return &InstanceDomainsWriteModel{
WriteModel: eventstore.WriteModel{
AggregateID: instanceID,
ResourceOwner: instanceID,
},
Domains: []string{},
}
}
func (wm *InstanceDomainsWriteModel) AppendEvents(events ...eventstore.Event) {
for _, event := range events {
switch e := event.(type) {
case *instance.DomainAddedEvent:
wm.WriteModel.AppendEvents(e)
case *instance.DomainRemovedEvent:
wm.WriteModel.AppendEvents(e)
}
}
}
func (wm *InstanceDomainsWriteModel) Reduce() error {
for _, event := range wm.Events {
switch e := event.(type) {
case *instance.DomainAddedEvent:
wm.Domains = append(wm.Domains, e.Domain)
case *instance.DomainRemovedEvent:
wm.Domains = removeDomainFromDomains(wm.Domains, e.Domain)
}
}
return wm.WriteModel.Reduce()
}
func removeDomainFromDomains(items []string, domain string) []string {
for i := len(items) - 1; i >= 0; i-- {
if items[i] == domain {
items[i] = items[len(items)-1]
items[len(items)-1] = ""
items = items[:len(items)-1]
}
}
return items
}