zitadel/internal/repository/iam/events_step.go

package iam

import (
	"context"
	"encoding/json"
	"strconv"

	"github.com/caos/zitadel/internal/eventstore"

	"github.com/caos/zitadel/internal/domain"
	"github.com/caos/zitadel/internal/errors"
	"github.com/caos/zitadel/internal/eventstore/repository"
)

const (
	UniqueStepStarted                          = "stepstarted"
	UniqueStepDone                             = "stepdone"
	SetupDoneEventType    eventstore.EventType = "iam.setup.done"
	SetupStartedEventType eventstore.EventType = "iam.setup.started"
)

type SetupStepEvent struct {
	eventstore.BaseEvent `json:"-"`

	Step domain.Step `json:"Step"`
	Done bool        `json:"-"`
}

func NewAddSetupStepStartedUniqueConstraint(step domain.Step) *eventstore.EventUniqueConstraint {
	return eventstore.NewAddEventUniqueConstraint(
		UniqueStepStarted,
		strconv.Itoa(int(step)),
		"Errors.Step.Started.AlreadyExists")
}

func NewAddSetupStepDoneUniqueConstraint(step domain.Step) *eventstore.EventUniqueConstraint {
	return eventstore.NewAddEventUniqueConstraint(
		UniqueStepDone,
		strconv.Itoa(int(step)),
		"Errors.Step.Done.AlreadyExists")
}

func (e *SetupStepEvent) Data() interface{} {
	return e
}

func (e *SetupStepEvent) UniqueConstraints() []*eventstore.EventUniqueConstraint {
	if e.Done {
		return []*eventstore.EventUniqueConstraint{NewAddSetupStepDoneUniqueConstraint(e.Step)}
	} else {
		return []*eventstore.EventUniqueConstraint{NewAddSetupStepStartedUniqueConstraint(e.Step)}
	}
}

func SetupStepMapper(event *repository.Event) (eventstore.Event, error) {
	step := &SetupStepEvent{
		BaseEvent: *eventstore.BaseEventFromRepo(event),
		Done:      eventstore.EventType(event.Type) == SetupDoneEventType,
		Step:      domain.Step1,
	}
	if len(event.Data) == 0 {
		return step, nil
	}
	err := json.Unmarshal(event.Data, step)
	if err != nil {
		return nil, errors.ThrowInternal(err, "IAM-O6rVg", "unable to unmarshal step")
	}

	return step, nil
}

func NewSetupStepDoneEvent(
	ctx context.Context,
	aggregate *eventstore.Aggregate,
	step domain.Step,
) *SetupStepEvent {

	return &SetupStepEvent{
		BaseEvent: *eventstore.NewBaseEventForPush(
			ctx,
			aggregate,
			SetupDoneEventType,
		),
		Step: step,
		Done: true,
	}
}

func NewSetupStepStartedEvent(
	ctx context.Context,
	aggregate *eventstore.Aggregate,
	step domain.Step,
) *SetupStepEvent {

	return &SetupStepEvent{
		BaseEvent: *eventstore.NewBaseEventForPush(
			ctx,
			aggregate,
			SetupStartedEventType,
		),
		Step: step,
	}
}