chore: move the go code into a subfolder

2025-08-12 00:47:33 +00:00 · 2025-08-05 15:20:32 -07:00
parent 4ad22ba456
commit cd2921de26
2978 changed files with 373 additions and 300 deletions
--- a/apps/api/internal/command/user_schema.go
+++ b/apps/api/internal/command/user_schema.go
@@ -0,0 +1,194 @@
+package command
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+
+	"github.com/zitadel/zitadel/internal/domain"
+	domain_schema "github.com/zitadel/zitadel/internal/domain/schema"
+	"github.com/zitadel/zitadel/internal/repository/user/schema"
+	"github.com/zitadel/zitadel/internal/zerrors"
+)
+
+type CreateUserSchema struct {
+	Details *domain.ObjectDetails
+
+	ResourceOwner          string
+	Type                   string
+	Schema                 json.RawMessage
+	PossibleAuthenticators []domain.AuthenticatorType
+}
+
+func (s *CreateUserSchema) Valid() error {
+	if s.Type == "" {
+		return zerrors.ThrowInvalidArgument(nil, "COMMA-DGFj3", "Errors.UserSchema.Type.Missing")
+	}
+	if err := validateUserSchema(s.Schema); err != nil {
+		return err
+	}
+	for _, authenticator := range s.PossibleAuthenticators {
+		if authenticator == domain.AuthenticatorTypeUnspecified {
+			return zerrors.ThrowInvalidArgument(nil, "COMMA-Gh652", "Errors.UserSchema.Authenticator.Invalid")
+		}
+	}
+	return nil
+}
+
+type ChangeUserSchema struct {
+	Details *domain.ObjectDetails
+
+	ID                     string
+	ResourceOwner          string
+	Type                   *string
+	Schema                 json.RawMessage
+	PossibleAuthenticators []domain.AuthenticatorType
+}
+
+func (s *ChangeUserSchema) Valid() error {
+	if s.ID == "" {
+		return zerrors.ThrowInvalidArgument(nil, "COMMA-H5421", "Errors.IDMissing")
+	}
+	if s.Type != nil && *s.Type == "" {
+		return zerrors.ThrowInvalidArgument(nil, "COMMA-G43gn", "Errors.UserSchema.Type.Missing")
+	}
+	if err := validateUserSchema(s.Schema); err != nil {
+		return err
+	}
+	for _, authenticator := range s.PossibleAuthenticators {
+		if authenticator == domain.AuthenticatorTypeUnspecified {
+			return zerrors.ThrowInvalidArgument(nil, "COMMA-WF4hg", "Errors.UserSchema.Authenticator.Invalid")
+		}
+	}
+	return nil
+}
+
+func (c *Commands) CreateUserSchema(ctx context.Context, userSchema *CreateUserSchema) error {
+	if err := userSchema.Valid(); err != nil {
+		return err
+	}
+	if userSchema.ResourceOwner == "" {
+		return zerrors.ThrowInvalidArgument(nil, "COMMA-J3hhj", "Errors.ResourceOwnerMissing")
+	}
+	id, err := c.idGenerator.Next()
+	if err != nil {
+		return err
+	}
+	writeModel, err := c.getSchemaWriteModelByID(ctx, userSchema.ResourceOwner, id)
+	if err != nil {
+		return err
+	}
+	if err := c.pushAppendAndReduce(ctx, writeModel,
+		schema.NewCreatedEvent(ctx,
+			UserSchemaAggregateFromWriteModel(&writeModel.WriteModel),
+			userSchema.Type, userSchema.Schema, userSchema.PossibleAuthenticators,
+		),
+	); err != nil {
+		return err
+	}
+	userSchema.Details = writeModelToObjectDetails(&writeModel.WriteModel)
+	return nil
+}
+
+func (c *Commands) ChangeUserSchema(ctx context.Context, userSchema *ChangeUserSchema) error {
+	if err := userSchema.Valid(); err != nil {
+		return err
+	}
+	writeModel, err := c.getSchemaWriteModelByID(ctx, userSchema.ResourceOwner, userSchema.ID)
+	if err != nil {
+		return err
+	}
+	if writeModel.State != domain.UserSchemaStateActive {
+		return zerrors.ThrowPreconditionFailed(nil, "COMMA-HB3e1", "Errors.UserSchema.NotActive")
+	}
+	updatedEvent := writeModel.NewUpdatedEvent(
+		ctx,
+		UserSchemaAggregateFromWriteModel(&writeModel.WriteModel),
+		userSchema.Type,
+		userSchema.Schema,
+		userSchema.PossibleAuthenticators,
+	)
+	if updatedEvent == nil {
+		userSchema.Details = writeModelToObjectDetails(&writeModel.WriteModel)
+		return nil
+	}
+	if err := c.pushAppendAndReduce(ctx, writeModel, updatedEvent); err != nil {
+		return err
+	}
+	userSchema.Details = writeModelToObjectDetails(&writeModel.WriteModel)
+	return nil
+}
+
+func (c *Commands) DeactivateUserSchema(ctx context.Context, id, resourceOwner string) (*domain.ObjectDetails, error) {
+	if id == "" {
+		return nil, zerrors.ThrowInvalidArgument(nil, "COMMA-Vvf3w", "Errors.IDMissing")
+	}
+	writeModel, err := c.getSchemaWriteModelByID(ctx, resourceOwner, id)
+	if err != nil {
+		return nil, err
+	}
+	if writeModel.State != domain.UserSchemaStateActive {
+		return nil, zerrors.ThrowPreconditionFailed(nil, "COMMA-E4t4z", "Errors.UserSchema.NotActive")
+	}
+	if err := c.pushAppendAndReduce(ctx, writeModel,
+		schema.NewDeactivatedEvent(ctx, UserSchemaAggregateFromWriteModel(&writeModel.WriteModel)),
+	); err != nil {
+		return nil, err
+	}
+	return writeModelToObjectDetails(&writeModel.WriteModel), nil
+}
+
+func (c *Commands) ReactivateUserSchema(ctx context.Context, id, resourceOwner string) (*domain.ObjectDetails, error) {
+	if id == "" {
+		return nil, zerrors.ThrowInvalidArgument(nil, "COMMA-wq3Gw", "Errors.IDMissing")
+	}
+	writeModel, err := c.getSchemaWriteModelByID(ctx, resourceOwner, id)
+	if err != nil {
+		return nil, err
+	}
+	if writeModel.State != domain.UserSchemaStateInactive {
+		return nil, zerrors.ThrowPreconditionFailed(nil, "COMMA-DGzh5", "Errors.UserSchema.NotInactive")
+	}
+	if err := c.pushAppendAndReduce(ctx, writeModel,
+		schema.NewReactivatedEvent(ctx, UserSchemaAggregateFromWriteModel(&writeModel.WriteModel)),
+	); err != nil {
+		return nil, err
+	}
+	return writeModelToObjectDetails(&writeModel.WriteModel), nil
+}
+
+func (c *Commands) DeleteUserSchema(ctx context.Context, id, resourceOwner string) (*domain.ObjectDetails, error) {
+	if id == "" {
+		return nil, zerrors.ThrowInvalidArgument(nil, "COMMA-E22gg", "Errors.IDMissing")
+	}
+	writeModel, err := c.getSchemaWriteModelByID(ctx, resourceOwner, id)
+	if err != nil {
+		return nil, err
+	}
+	if !writeModel.Exists() {
+		return nil, zerrors.ThrowPreconditionFailed(nil, "COMMA-Grg41", "Errors.UserSchema.NotExists")
+	}
+	// TODO: check for users based on that schema; this is only possible with / after https://github.com/zitadel/zitadel/issues/7308
+	if err := c.pushAppendAndReduce(ctx, writeModel,
+		schema.NewDeletedEvent(ctx, UserSchemaAggregateFromWriteModel(&writeModel.WriteModel), writeModel.SchemaType),
+	); err != nil {
+		return nil, err
+	}
+	return writeModelToObjectDetails(&writeModel.WriteModel), nil
+}
+
+func validateUserSchema(userSchema json.RawMessage) error {
+	_, err := domain_schema.NewSchema(0, bytes.NewReader(userSchema))
+	if err != nil {
+		return zerrors.ThrowInvalidArgument(err, "COMMA-W21tg", "Errors.UserSchema.Schema.Invalid")
+	}
+	return nil
+}
+
+func (c *Commands) getSchemaWriteModelByID(ctx context.Context, resourceOwner, id string) (*UserSchemaWriteModel, error) {
+	writeModel := NewUserSchemaWriteModel(resourceOwner, id)
+	if err := c.eventstore.FilterToQueryReducer(ctx, writeModel); err != nil {
+		return nil, err
+	}
+	return writeModel, nil
+}