chore: move the go code into a subfolder

apps/api/internal/command/project_grant_model.go

@@ -0,0 +1,198 @@
+package command
+
+import (
+	"github.com/zitadel/zitadel/internal/domain"
+	"github.com/zitadel/zitadel/internal/eventstore"
+	"github.com/zitadel/zitadel/internal/repository/org"
+	"github.com/zitadel/zitadel/internal/repository/project"
+)
+
+type ProjectGrantWriteModel struct {
+	eventstore.WriteModel
+
+	GrantID      string
+	GrantedOrgID string
+	RoleKeys     []string
+	State        domain.ProjectGrantState
+}
+
+func NewProjectGrantWriteModel(grantID, grantedOrgID, projectID, resourceOwner string) *ProjectGrantWriteModel {
+	return &ProjectGrantWriteModel{
+		WriteModel: eventstore.WriteModel{
+			AggregateID:   projectID,
+			ResourceOwner: resourceOwner,
+		},
+		GrantID:      grantID,
+		GrantedOrgID: grantedOrgID,
+	}
+}
+
+func (wm *ProjectGrantWriteModel) AppendEvents(events ...eventstore.Event) {
+	for _, event := range events {
+		switch e := event.(type) {
+		case *project.GrantAddedEvent:
+			if (wm.GrantID != "" && e.GrantID == wm.GrantID) ||
+				(wm.GrantedOrgID != "" && e.GrantedOrgID == wm.GrantedOrgID) {
+				wm.WriteModel.AppendEvents(e)
+			}
+		case *project.GrantChangedEvent:
+			if wm.GrantID != "" && e.GrantID == wm.GrantID {
+				wm.WriteModel.AppendEvents(e)
+			}
+		case *project.GrantCascadeChangedEvent:
+			if wm.GrantID != "" && e.GrantID == wm.GrantID {
+				wm.WriteModel.AppendEvents(e)
+			}
+		case *project.GrantDeactivateEvent:
+			if wm.GrantID != "" && e.GrantID == wm.GrantID {
+				wm.WriteModel.AppendEvents(e)
+			}
+		case *project.GrantReactivatedEvent:
+			if wm.GrantID != "" && e.GrantID == wm.GrantID {
+				wm.WriteModel.AppendEvents(e)
+			}
+		case *project.GrantRemovedEvent:
+			if wm.GrantID != "" && e.GrantID == wm.GrantID {
+				wm.WriteModel.AppendEvents(e)
+			}
+		case *project.ProjectRemovedEvent:
+			wm.WriteModel.AppendEvents(e)
+		}
+	}
+}
+
+func (wm *ProjectGrantWriteModel) Reduce() error {
+	for _, event := range wm.Events {
+		switch e := event.(type) {
+		case *project.GrantAddedEvent:
+			wm.GrantID = e.GrantID
+			wm.GrantedOrgID = e.GrantedOrgID
+			wm.RoleKeys = e.RoleKeys
+			wm.State = domain.ProjectGrantStateActive
+		case *project.GrantChangedEvent:
+			wm.RoleKeys = e.RoleKeys
+		case *project.GrantCascadeChangedEvent:
+			wm.RoleKeys = e.RoleKeys
+		case *project.GrantDeactivateEvent:
+			if wm.State == domain.ProjectGrantStateRemoved {
+				continue
+			}
+			wm.State = domain.ProjectGrantStateInactive
+		case *project.GrantReactivatedEvent:
+			if wm.State == domain.ProjectGrantStateRemoved {
+				continue
+			}
+			wm.State = domain.ProjectGrantStateActive
+		case *project.GrantRemovedEvent:
+			wm.State = domain.ProjectGrantStateRemoved
+		case *project.ProjectRemovedEvent:
+			wm.State = domain.ProjectGrantStateRemoved
+		}
+	}
+	return wm.WriteModel.Reduce()
+}
+
+func (wm *ProjectGrantWriteModel) Query() *eventstore.SearchQueryBuilder {
+	query := eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
+		AddQuery().
+		AggregateTypes(project.AggregateType).
+		AggregateIDs(wm.AggregateID).
+		EventTypes(
+			project.GrantAddedType,
+			project.GrantChangedType,
+			project.GrantCascadeChangedType,
+			project.GrantDeactivatedType,
+			project.GrantReactivatedType,
+			project.GrantRemovedType,
+			project.ProjectRemovedType).
+		Builder()
+
+	if wm.ResourceOwner != "" {
+		query.ResourceOwner(wm.ResourceOwner)
+	}
+	return query
+}
+
+type ProjectGrantPreConditionReadModel struct {
+	eventstore.WriteModel
+
+	ProjectResourceOwner string
+	ProjectID            string
+	GrantedOrgID         string
+	ProjectExists        bool
+	GrantedOrgExists     bool
+	ExistingRoleKeys     []string
+}
+
+func NewProjectGrantPreConditionReadModel(projectID, grantedOrgID, resourceOwner string) *ProjectGrantPreConditionReadModel {
+	return &ProjectGrantPreConditionReadModel{
+		WriteModel:           eventstore.WriteModel{},
+		ProjectResourceOwner: resourceOwner,
+		ProjectID:            projectID,
+		GrantedOrgID:         grantedOrgID,
+	}
+}
+
+func (wm *ProjectGrantPreConditionReadModel) Reduce() error {
+	for _, event := range wm.Events {
+		switch e := event.(type) {
+		case *project.ProjectAddedEvent:
+			if wm.ProjectResourceOwner == "" {
+				wm.ProjectResourceOwner = e.Aggregate().ResourceOwner
+			}
+			if wm.ProjectResourceOwner != e.Aggregate().ResourceOwner {
+				continue
+			}
+			wm.ProjectExists = true
+		case *project.ProjectRemovedEvent:
+			if wm.ProjectResourceOwner != e.Aggregate().ResourceOwner {
+				continue
+			}
+			wm.ProjectResourceOwner = ""
+			wm.ProjectExists = false
+		case *project.RoleAddedEvent:
+			if e.Aggregate().ResourceOwner != wm.ProjectResourceOwner {
+				continue
+			}
+			wm.ExistingRoleKeys = append(wm.ExistingRoleKeys, e.Key)
+		case *project.RoleRemovedEvent:
+			if e.Aggregate().ResourceOwner != wm.ProjectResourceOwner {
+				continue
+			}
+			for i, key := range wm.ExistingRoleKeys {
+				if key == e.Key {
+					copy(wm.ExistingRoleKeys[i:], wm.ExistingRoleKeys[i+1:])
+					wm.ExistingRoleKeys[len(wm.ExistingRoleKeys)-1] = ""
+					wm.ExistingRoleKeys = wm.ExistingRoleKeys[:len(wm.ExistingRoleKeys)-1]
+					continue
+				}
+			}
+		case *org.OrgAddedEvent:
+			wm.GrantedOrgExists = true
+		case *org.OrgRemovedEvent:
+			wm.GrantedOrgExists = false
+		}
+	}
+	return wm.WriteModel.Reduce()
+}
+
+func (wm *ProjectGrantPreConditionReadModel) Query() *eventstore.SearchQueryBuilder {
+	query := eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
+		AddQuery().
+		AggregateTypes(project.AggregateType).
+		AggregateIDs(wm.ProjectID).
+		EventTypes(
+			project.ProjectAddedType,
+			project.ProjectRemovedType,
+			project.RoleAddedType,
+			project.RoleRemovedType).
+		Or().
+		AggregateTypes(org.AggregateType).
+		AggregateIDs(wm.GrantedOrgID).
+		EventTypes(
+			org.OrgAddedEventType,
+			org.OrgRemovedEventType).
+		Builder()
+
+	return query
+}