mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-15 12:27:59 +00:00
c2a5b785fb
* feat(instance): add remove instance event with projections cleanup * fix(instance): corrected used id to clean up projections * fix merge * fix: correct unit test projection names * fix: current sequence of lists and query for ensuring keypair based projections Co-authored-by: Livio Spring <livio.a@gmail.com> Co-authored-by: Fabi <38692350+hifabienne@users.noreply.github.com>
134 lines
3.4 KiB
Go
134 lines
3.4 KiB
Go
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 nil
|
|
}
|
|
|
|
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
|
|
}
|