2020-05-13 12:22:29 +00:00
|
|
|
package eventsourcing
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"github.com/caos/zitadel/internal/errors"
|
|
|
|
es_models "github.com/caos/zitadel/internal/eventstore/models"
|
|
|
|
org_model "github.com/caos/zitadel/internal/org/model"
|
2020-06-05 05:50:04 +00:00
|
|
|
"github.com/caos/zitadel/internal/org/repository/eventsourcing/model"
|
2020-05-13 12:22:29 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func OrgByIDQuery(id string, latestSequence uint64) (*es_models.SearchQuery, error) {
|
|
|
|
if id == "" {
|
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-dke74", "id should be filled")
|
|
|
|
}
|
|
|
|
return OrgQuery(latestSequence).
|
|
|
|
AggregateIDFilter(id), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func OrgDomainUniqueQuery(domain string) *es_models.SearchQuery {
|
|
|
|
return es_models.NewSearchQuery().
|
2020-06-05 05:50:04 +00:00
|
|
|
AggregateTypeFilter(model.OrgDomainAggregate).
|
2020-05-13 12:22:29 +00:00
|
|
|
AggregateIDFilter(domain).
|
|
|
|
OrderDesc().
|
|
|
|
SetLimit(1)
|
|
|
|
}
|
|
|
|
|
|
|
|
func OrgNameUniqueQuery(name string) *es_models.SearchQuery {
|
|
|
|
return es_models.NewSearchQuery().
|
2020-06-05 05:50:04 +00:00
|
|
|
AggregateTypeFilter(model.OrgNameAggregate).
|
2020-05-13 12:22:29 +00:00
|
|
|
AggregateIDFilter(name).
|
|
|
|
OrderDesc().
|
|
|
|
SetLimit(1)
|
|
|
|
}
|
|
|
|
|
|
|
|
func OrgQuery(latestSequence uint64) *es_models.SearchQuery {
|
|
|
|
return es_models.NewSearchQuery().
|
2020-06-05 05:50:04 +00:00
|
|
|
AggregateTypeFilter(model.OrgAggregate).
|
2020-05-13 12:22:29 +00:00
|
|
|
LatestSequenceFilter(latestSequence)
|
|
|
|
}
|
|
|
|
|
|
|
|
func OrgAggregate(ctx context.Context, aggCreator *es_models.AggregateCreator, id string, sequence uint64) (*es_models.Aggregate, error) {
|
2020-06-05 05:50:04 +00:00
|
|
|
return aggCreator.NewAggregate(ctx, id, model.OrgAggregate, model.OrgVersion, sequence)
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
|
2020-06-05 05:50:04 +00:00
|
|
|
func orgCreatedAggregates(ctx context.Context, aggCreator *es_models.AggregateCreator, org *model.Org) (_ []*es_models.Aggregate, err error) {
|
2020-05-13 12:22:29 +00:00
|
|
|
if org == nil {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-kdie7", "Errors.Internal")
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
|
2020-06-16 09:40:18 +00:00
|
|
|
agg, err := aggCreator.NewAggregate(ctx, org.AggregateID, model.OrgAggregate, model.OrgVersion, org.Sequence, es_models.OverwriteResourceOwner(org.AggregateID))
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2020-06-16 09:40:18 +00:00
|
|
|
agg, err = agg.AppendEvent(model.OrgAdded, org)
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2020-06-16 09:40:18 +00:00
|
|
|
aggregates := make([]*es_models.Aggregate, 0)
|
|
|
|
aggregates, err = addDomainAggregateAndEvents(ctx, aggCreator, agg, aggregates, org)
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2020-06-16 09:40:18 +00:00
|
|
|
nameAggregate, err := reservedUniqueNameAggregate(ctx, aggCreator, org.AggregateID, org.Name)
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2020-06-16 09:40:18 +00:00
|
|
|
aggregates = append(aggregates, nameAggregate)
|
|
|
|
return append(aggregates, agg), nil
|
|
|
|
}
|
2020-05-13 12:22:29 +00:00
|
|
|
|
2020-06-16 09:40:18 +00:00
|
|
|
func addDomainAggregateAndEvents(ctx context.Context, aggCreator *es_models.AggregateCreator, orgAggregate *es_models.Aggregate, aggregates []*es_models.Aggregate, org *model.Org) ([]*es_models.Aggregate, error) {
|
|
|
|
for _, domain := range org.Domains {
|
|
|
|
orgAggregate, err := orgAggregate.AppendEvent(model.OrgDomainAdded, domain)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if domain.Verified {
|
|
|
|
domainAggregate, err := reservedUniqueDomainAggregate(ctx, aggCreator, org.AggregateID, domain.Domain)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
aggregates = append(aggregates, domainAggregate)
|
|
|
|
orgAggregate, err = orgAggregate.AppendEvent(model.OrgDomainVerified, domain)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if domain.Primary {
|
|
|
|
orgAggregate, err = orgAggregate.AppendEvent(model.OrgDomainPrimarySet, domain)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return aggregates, nil
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
|
2020-06-05 05:50:04 +00:00
|
|
|
func OrgUpdateAggregates(ctx context.Context, aggCreator *es_models.AggregateCreator, existing *model.Org, updated *model.Org) ([]*es_models.Aggregate, error) {
|
2020-05-13 12:22:29 +00:00
|
|
|
if existing == nil {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-dk83d", "Errors.Internal")
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
if updated == nil {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-dhr74", "Errors.Internal")
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
changes := existing.Changes(updated)
|
|
|
|
if len(changes) == 0 {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-E0hc5", "Errors.NoChangesFound")
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
aggregates := make([]*es_models.Aggregate, 0, 3)
|
|
|
|
|
|
|
|
if name, ok := changes["name"]; ok {
|
2020-06-16 09:40:18 +00:00
|
|
|
nameAggregate, err := reservedUniqueNameAggregate(ctx, aggCreator, "", name.(string))
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
aggregates = append(aggregates, nameAggregate)
|
2020-06-16 09:40:18 +00:00
|
|
|
nameReleasedAggregate, err := releasedUniqueNameAggregate(ctx, aggCreator, "", existing.Name)
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2020-06-16 09:40:18 +00:00
|
|
|
aggregates = append(aggregates, nameReleasedAggregate)
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
orgAggregate, err := OrgAggregate(ctx, aggCreator, existing.AggregateID, existing.Sequence)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2020-06-05 05:50:04 +00:00
|
|
|
orgAggregate, err = orgAggregate.AppendEvent(model.OrgChanged, changes)
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
aggregates = append(aggregates, orgAggregate)
|
|
|
|
|
|
|
|
return aggregates, nil
|
|
|
|
}
|
|
|
|
|
2020-06-05 05:50:04 +00:00
|
|
|
func orgDeactivateAggregate(aggCreator *es_models.AggregateCreator, org *model.Org) func(ctx context.Context) (*es_models.Aggregate, error) {
|
2020-05-13 12:22:29 +00:00
|
|
|
return func(ctx context.Context) (*es_models.Aggregate, error) {
|
|
|
|
if org == nil {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-R03z8", "Errors.Internal")
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
if org.State == int32(org_model.ORGSTATE_INACTIVE) {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowInvalidArgument(nil, "EVENT-mcPH0", "Errors.Internal.AlreadyDeactivated")
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
agg, err := OrgAggregate(ctx, aggCreator, org.AggregateID, org.Sequence)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2020-06-05 05:50:04 +00:00
|
|
|
return agg.AppendEvent(model.OrgDeactivated, nil)
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-06-05 05:50:04 +00:00
|
|
|
func orgReactivateAggregate(aggCreator *es_models.AggregateCreator, org *model.Org) func(ctx context.Context) (*es_models.Aggregate, error) {
|
2020-05-13 12:22:29 +00:00
|
|
|
return func(ctx context.Context) (*es_models.Aggregate, error) {
|
|
|
|
if org == nil {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-cTHLd", "Errors.Internal")
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
if org.State == int32(org_model.ORGSTATE_ACTIVE) {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowInvalidArgument(nil, "EVENT-pUSMs", "Errors.Org.AlreadyActive")
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
agg, err := OrgAggregate(ctx, aggCreator, org.AggregateID, org.Sequence)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2020-06-05 05:50:04 +00:00
|
|
|
return agg.AppendEvent(model.OrgReactivated, nil)
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-06-16 09:40:18 +00:00
|
|
|
func reservedUniqueDomainAggregate(ctx context.Context, aggCreator *es_models.AggregateCreator, resourceOwner, domain string) (*es_models.Aggregate, error) {
|
2020-06-05 05:50:04 +00:00
|
|
|
aggregate, err := aggCreator.NewAggregate(ctx, domain, model.OrgDomainAggregate, model.OrgVersion, 0)
|
2020-05-18 09:32:16 +00:00
|
|
|
if resourceOwner != "" {
|
2020-06-05 05:50:04 +00:00
|
|
|
aggregate, err = aggCreator.NewAggregate(ctx, domain, model.OrgDomainAggregate, model.OrgVersion, 0, es_models.OverwriteResourceOwner(resourceOwner))
|
2020-05-18 09:32:16 +00:00
|
|
|
}
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2020-06-05 05:50:04 +00:00
|
|
|
aggregate, err = aggregate.AppendEvent(model.OrgDomainReserved, nil)
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2020-06-16 09:40:18 +00:00
|
|
|
return aggregate.SetPrecondition(OrgDomainUniqueQuery(domain), isEventValidation(aggregate, model.OrgDomainReserved)), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func releasedUniqueDomainAggregate(ctx context.Context, aggCreator *es_models.AggregateCreator, resourceOwner, domain string) (*es_models.Aggregate, error) {
|
|
|
|
aggregate, err := aggCreator.NewAggregate(ctx, domain, model.OrgDomainAggregate, model.OrgVersion, 0)
|
|
|
|
if resourceOwner != "" {
|
|
|
|
aggregate, err = aggCreator.NewAggregate(ctx, domain, model.OrgDomainAggregate, model.OrgVersion, 0, es_models.OverwriteResourceOwner(resourceOwner))
|
|
|
|
}
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
aggregate, err = aggregate.AppendEvent(model.OrgDomainReleased, nil)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return aggregate.SetPrecondition(OrgDomainUniqueQuery(domain), isEventValidation(aggregate, model.OrgDomainReleased)), nil
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
|
2020-06-16 09:40:18 +00:00
|
|
|
func reservedUniqueNameAggregate(ctx context.Context, aggCreator *es_models.AggregateCreator, resourceOwner, name string) (aggregate *es_models.Aggregate, err error) {
|
2020-06-05 05:50:04 +00:00
|
|
|
aggregate, err = aggCreator.NewAggregate(ctx, name, model.OrgNameAggregate, model.OrgVersion, 0)
|
2020-05-18 09:32:16 +00:00
|
|
|
if resourceOwner != "" {
|
2020-06-05 05:50:04 +00:00
|
|
|
aggregate, err = aggCreator.NewAggregate(ctx, name, model.OrgNameAggregate, model.OrgVersion, 0, es_models.OverwriteResourceOwner(resourceOwner))
|
2020-05-18 09:32:16 +00:00
|
|
|
}
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2020-06-05 05:50:04 +00:00
|
|
|
aggregate, err = aggregate.AppendEvent(model.OrgNameReserved, nil)
|
2020-05-13 12:22:29 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2020-06-16 09:40:18 +00:00
|
|
|
return aggregate.SetPrecondition(OrgNameUniqueQuery(name), isEventValidation(aggregate, model.OrgNameReserved)), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func releasedUniqueNameAggregate(ctx context.Context, aggCreator *es_models.AggregateCreator, resourceOwner, name string) (aggregate *es_models.Aggregate, err error) {
|
|
|
|
aggregate, err = aggCreator.NewAggregate(ctx, name, model.OrgNameAggregate, model.OrgVersion, 0)
|
|
|
|
if resourceOwner != "" {
|
|
|
|
aggregate, err = aggCreator.NewAggregate(ctx, name, model.OrgNameAggregate, model.OrgVersion, 0, es_models.OverwriteResourceOwner(resourceOwner))
|
|
|
|
}
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
aggregate, err = aggregate.AppendEvent(model.OrgNameReleased, nil)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return aggregate.SetPrecondition(OrgNameUniqueQuery(name), isEventValidation(aggregate, model.OrgNameReleased)), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func OrgDomainAddedAggregate(aggCreator *es_models.AggregateCreator, existing *model.Org, domain *model.OrgDomain) func(ctx context.Context) (*es_models.Aggregate, error) {
|
|
|
|
return func(ctx context.Context) (*es_models.Aggregate, error) {
|
|
|
|
if domain == nil {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-OSid3", "Errors.Internal")
|
2020-06-16 09:40:18 +00:00
|
|
|
}
|
|
|
|
agg, err := OrgAggregate(ctx, aggCreator, existing.AggregateID, existing.Sequence)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return agg.AppendEvent(model.OrgDomainAdded, domain)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func OrgDomainVerifiedAggregate(aggCreator *es_models.AggregateCreator, existing *model.Org, domain *model.OrgDomain) func(ctx context.Context) ([]*es_models.Aggregate, error) {
|
|
|
|
return func(ctx context.Context) ([]*es_models.Aggregate, error) {
|
|
|
|
if domain == nil {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-DHs7s", "Errors.Internal")
|
2020-06-16 09:40:18 +00:00
|
|
|
}
|
|
|
|
agg, err := OrgAggregate(ctx, aggCreator, existing.AggregateID, existing.Sequence)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
aggregates := make([]*es_models.Aggregate, 0, 2)
|
|
|
|
agg, err = agg.AppendEvent(model.OrgDomainVerified, domain)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
domainAgregate, err := reservedUniqueDomainAggregate(ctx, aggCreator, existing.AggregateID, domain.Domain)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
aggregates = append(aggregates, domainAgregate)
|
|
|
|
return append(aggregates, agg), nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func OrgDomainSetPrimaryAggregate(aggCreator *es_models.AggregateCreator, existing *model.Org, domain *model.OrgDomain) func(ctx context.Context) (*es_models.Aggregate, error) {
|
|
|
|
return func(ctx context.Context) (*es_models.Aggregate, error) {
|
|
|
|
if domain == nil {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-PSw3j", "Errors.Internal")
|
2020-06-16 09:40:18 +00:00
|
|
|
}
|
|
|
|
agg, err := OrgAggregate(ctx, aggCreator, existing.AggregateID, existing.Sequence)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return agg.AppendEvent(model.OrgDomainPrimarySet, domain)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func OrgDomainRemovedAggregate(ctx context.Context, aggCreator *es_models.AggregateCreator, existing *model.Org, domain *model.OrgDomain) ([]*es_models.Aggregate, error) {
|
|
|
|
if domain == nil {
|
2020-06-22 11:51:44 +00:00
|
|
|
return nil, errors.ThrowPreconditionFailed(nil, "EVENT-si8dW", "Errors.Internal")
|
2020-06-16 09:40:18 +00:00
|
|
|
}
|
|
|
|
aggregates := make([]*es_models.Aggregate, 0, 2)
|
|
|
|
agg, err := OrgAggregate(ctx, aggCreator, existing.AggregateID, existing.Sequence)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
agg, err = agg.AppendEvent(model.OrgDomainRemoved, domain)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
aggregates = append(aggregates, agg)
|
|
|
|
domainAgregate, err := releasedUniqueDomainAggregate(ctx, aggCreator, existing.AggregateID, domain.Domain)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return append(aggregates, domainAgregate), nil
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
|
2020-06-16 09:40:18 +00:00
|
|
|
func isEventValidation(aggregate *es_models.Aggregate, eventType es_models.EventType) func(...*es_models.Event) error {
|
2020-05-13 12:22:29 +00:00
|
|
|
return func(events ...*es_models.Event) error {
|
|
|
|
if len(events) == 0 {
|
|
|
|
aggregate.PreviousSequence = 0
|
|
|
|
return nil
|
|
|
|
}
|
2020-06-16 09:40:18 +00:00
|
|
|
if events[0].Type == eventType {
|
|
|
|
return errors.ThrowPreconditionFailedf(nil, "EVENT-eJQqe", "user is already %v", eventType)
|
2020-05-13 12:22:29 +00:00
|
|
|
}
|
|
|
|
aggregate.PreviousSequence = events[0].Sequence
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|