zitadel/internal/admin/repository/eventsourcing/setup/setup.go

package setup

import (
	"context"
	policy_model "github.com/caos/zitadel/internal/policy/model"
	policy_event "github.com/caos/zitadel/internal/policy/repository/eventsourcing"
	"time"

	"github.com/caos/logging"
	"github.com/caos/zitadel/internal/api/auth"
	"github.com/caos/zitadel/internal/config/systemdefaults"
	"github.com/caos/zitadel/internal/config/types"
	caos_errs "github.com/caos/zitadel/internal/errors"
	"github.com/caos/zitadel/internal/eventstore/models"
	iam_model "github.com/caos/zitadel/internal/iam/model"
	iam_event "github.com/caos/zitadel/internal/iam/repository/eventsourcing"
	org_model "github.com/caos/zitadel/internal/org/model"
	org_event "github.com/caos/zitadel/internal/org/repository/eventsourcing"
	proj_model "github.com/caos/zitadel/internal/project/model"
	proj_event "github.com/caos/zitadel/internal/project/repository/eventsourcing"
	usr_model "github.com/caos/zitadel/internal/user/model"
	usr_event "github.com/caos/zitadel/internal/user/repository/eventsourcing"
)

type Setup struct {
	repos       EventstoreRepos
	iamID       string
	setUpConfig types.IAMSetUp
}

type EventstoreRepos struct {
	IamEvents     *iam_event.IamEventstore
	OrgEvents     *org_event.OrgEventstore
	UserEvents    *usr_event.UserEventstore
	ProjectEvents *proj_event.ProjectEventstore
	PolicyEvents  *policy_event.PolicyEventstore
}

type initializer struct {
	*Setup
	createdUsers       map[string]*usr_model.User
	createdOrgs        map[string]*org_model.Org
	createdProjects    map[string]*proj_model.Project
	pwComplexityPolicy *policy_model.PasswordComplexityPolicy
}

const (
	OrgOwnerRole                     = "ORG_OWNER"
	SETUP_USER                       = "SETUP"
	OIDCResponseType_CODE            = "CODE"
	OIDCResponseType_ID_TOKEN        = "ID_TOKEN"
	OIDCResponseType_TOKEN           = "TOKEN"
	OIDCGrantType_AUTHORIZATION_CODE = "AUTHORIZATION_CODE"
	OIDCGrantType_IMPLICIT           = "IMPLICIT"
	OIDCGrantType_REFRESH_TOKEN      = "REFRESH_TOKEN"
	OIDCApplicationType_NATIVE       = "NATIVE"
	OIDCApplicationType_USER_AGENT   = "USER_AGENT"
	OIDCApplicationType_WEB          = "WEB"
	OIDCAuthMethodType_NONE          = "NONE"
	OIDCAuthMethodType_BASIC         = "BASIC"
	OIDCAuthMethodType_POST          = "POST"
	DEFAULT_POLICY                   = "0"
)

func StartSetup(sd systemdefaults.SystemDefaults, repos EventstoreRepos) *Setup {
	return &Setup{
		repos:       repos,
		iamID:       sd.IamID,
		setUpConfig: sd.SetUp,
	}
}

func (s *Setup) Execute(ctx context.Context) error {
	iam, err := s.repos.IamEvents.IamByID(ctx, s.iamID)
	if err != nil && !caos_errs.IsNotFound(err) {
		return err
	}
	if iam != nil && iam.SetUpDone {
		return nil
	}

	if iam != nil && iam.SetUpStarted {
		return s.waitForSetupDone(ctx)
	}

	logging.Log("SETUP-hwG32").Info("starting setup")
	ctx = setSetUpContextData(ctx, s.iamID)
	iam, err = s.repos.IamEvents.StartSetup(ctx, s.iamID)
	if err != nil {
		return err
	}

	setUp := &initializer{
		Setup:           s,
		createdUsers:    make(map[string]*usr_model.User),
		createdOrgs:     make(map[string]*org_model.Org),
		createdProjects: make(map[string]*proj_model.Project),
	}

	pwComplexityPolicy, err := s.repos.PolicyEvents.GetPasswordComplexityPolicy(ctx, DEFAULT_POLICY)
	if err != nil {
		logging.Log("SETUP-9osWF").WithError(err).Error("unable to read complexity policy")
		return err
	}
	setUp.pwComplexityPolicy = pwComplexityPolicy

	err = setUp.orgs(ctx, s.setUpConfig.Orgs)
	if err != nil {
		logging.Log("SETUP-p4oWq").WithError(err).Error("unable to set up orgs")
		return err
	}

	ctx = setSetUpContextData(ctx, s.iamID)
	err = setUp.iamOwners(ctx, s.setUpConfig.Owners)
	if err != nil {
		logging.Log("SETUP-WHr01").WithError(err).Error("unable to set up iam owners")
		return err
	}

	err = setUp.setGlobalOrg(ctx)
	if err != nil {
		logging.Log("SETUP-0874m").WithError(err).Error("unable to set global org")
		return err
	}

	err = setUp.setIamProject(ctx)
	if err != nil {
		logging.Log("SETUP-kaWjq").WithError(err).Error("unable to set zitadel project")
		return err
	}

	iam, err = s.repos.IamEvents.SetupDone(ctx, s.iamID)
	if err != nil {
		logging.Log("SETUP-de342").WithError(err).Error("unable to finish setup")
		return err
	}
	logging.Log("SETUP-ds31h").Info("setup done")
	return nil
}

func (s *Setup) waitForSetupDone(ctx context.Context) error {
	logging.Log("SETUP-hws22").Info("waiting for setup to be done")
	ctx, cancel := context.WithDeadline(ctx, time.Now().UTC().Add(10*time.Second))
	defer cancel()

	for {
		select {
		case <-time.After(1 * time.Second):
			iam, _ := s.repos.IamEvents.IamByID(ctx, s.iamID)
			if iam != nil && iam.SetUpDone {
				return nil
			}
			logging.Log("SETUP-d23g1").Info("setup not done yet")
		case <-ctx.Done():
			return caos_errs.ThrowInternal(ctx.Err(), "SETUP-dsjg3", "Timeout exceeded for setup")
		}
	}
}

func (setUp *initializer) orgs(ctx context.Context, orgs []types.Org) error {
	logging.Log("SETUP-dsTh3").Info("setting up orgs")
	for _, iamOrg := range orgs {
		org, err := setUp.org(ctx, iamOrg)
		if err != nil {
			logging.LogWithFields("SETUP-IlLif", "Org", iamOrg.Name).WithError(err).Error("unable to create org")
			return err
		}
		setUp.createdOrgs[iamOrg.Name] = org

		ctx = setSetUpContextData(ctx, org.AggregateID)
		err = setUp.users(ctx, iamOrg.Users)
		if err != nil {
			logging.LogWithFields("SETUP-8zfwz", "Org", iamOrg.Name).WithError(err).Error("unable to set up org users")
			return err
		}

		err = setUp.orgOwners(ctx, org, iamOrg.Owners)
		if err != nil {
			logging.LogWithFields("SETUP-0874m", "Org", iamOrg.Name).WithError(err).Error("unable to set up org owners")
			return err
		}

		err = setUp.projects(ctx, iamOrg.Projects)
		if err != nil {
			logging.LogWithFields("SETUP-wUzqY", "Org", iamOrg.Name).WithError(err).Error("unable to set up org projects")
			return err
		}
	}
	logging.Log("SETUP-dgjT4").Info("orgs set up")
	return nil
}

func (setUp *initializer) org(ctx context.Context, org types.Org) (*org_model.Org, error) {
	ctx = setSetUpContextData(ctx, "")
	createOrg := &org_model.Org{
		Name:   org.Name,
		Domain: org.Domain,
	}
	return setUp.repos.OrgEvents.CreateOrg(ctx, createOrg)
}

func (setUp *initializer) iamOwners(ctx context.Context, owners []string) error {
	logging.Log("SETUP-dtxfj").Info("setting iam owners")
	for _, iamOwner := range owners {
		user, ok := setUp.createdUsers[iamOwner]
		if !ok {
			logging.LogWithFields("SETUP-8siew", "Owner", iamOwner).Error("unable to add user to iam members")
			return caos_errs.ThrowPreconditionFailedf(nil, "SETUP-su6L3", "unable to add user to iam members")
		}
		_, err := setUp.repos.IamEvents.AddIamMember(ctx, &iam_model.IamMember{ObjectRoot: models.ObjectRoot{AggregateID: setUp.iamID}, UserID: user.AggregateID, Roles: []string{"IAM_OWNER"}})
		if err != nil {
			logging.Log("SETUP-LM7rI").WithError(err).Error("unable to add iam administrator to iam members as owner")
			return err
		}
	}
	logging.Log("SETUP-fg5aq").Info("iam owners set")
	return nil
}

func (setUp *initializer) setGlobalOrg(ctx context.Context) error {
	logging.Log("SETUP-dsj75").Info("setting global org")
	globalOrg, ok := setUp.createdOrgs[setUp.setUpConfig.GlobalOrg]
	if !ok {
		logging.LogWithFields("SETUP-FBhs9", "GlobalOrg", setUp.setUpConfig.GlobalOrg).Error("global org not created")
		return caos_errs.ThrowPreconditionFailedf(nil, "SETUP-4GwU7", "global org not created: %v", setUp.setUpConfig.GlobalOrg)
	}

	if _, err := setUp.repos.IamEvents.SetGlobalOrg(ctx, setUp.iamID, globalOrg.AggregateID); err != nil {
		logging.Log("SETUP-uGMA3").WithError(err).Error("unable to set global org on iam")
		return err
	}
	logging.Log("SETUP-d32h1").Info("global org set")
	return nil
}

func (setUp *initializer) setIamProject(ctx context.Context) error {
	logging.Log("SETUP-HE3qa").Info("setting iam project")
	iamProject, ok := setUp.createdProjects[setUp.setUpConfig.IAMProject]
	if !ok {
		logging.LogWithFields("SETUP-SJFWP", "Iam Project", setUp.setUpConfig.IAMProject).Error("iam project created")
		return caos_errs.ThrowPreconditionFailedf(nil, "SETUP-sGmQt", "iam project not created: %v", setUp.setUpConfig.IAMProject)
	}

	if _, err := setUp.repos.IamEvents.SetIamProject(ctx, setUp.iamID, iamProject.AggregateID); err != nil {
		logging.Log("SETUP-i1pNh").WithError(err).Error("unable to set iam project on iam")
		return err
	}
	logging.Log("SETUP-d7WEU").Info("iam project set")
	return nil
}

func (setUp *initializer) users(ctx context.Context, users []types.User) error {
	for _, user := range users {
		created, err := setUp.user(ctx, user)
		if err != nil {
			logging.LogWithFields("SETUP-9soer", "Email", user.Email).WithError(err).Error("unable to create iam user")
			return err
		}
		setUp.createdUsers[user.Email] = created
	}
	return nil
}

func (setUp *initializer) user(ctx context.Context, user types.User) (*usr_model.User, error) {
	createUser := &usr_model.User{
		Profile: &usr_model.Profile{
			UserName:  user.UserName,
			FirstName: user.FirstName,
			LastName:  user.LastName,
		},
		Email: &usr_model.Email{
			EmailAddress:    user.Email,
			IsEmailVerified: true,
		},
		Password: &usr_model.Password{
			SecretString: user.Password,
		},
	}
	return setUp.repos.UserEvents.CreateUser(ctx, createUser, setUp.pwComplexityPolicy)
}

func (setUp *initializer) orgOwners(ctx context.Context, org *org_model.Org, owners []string) error {
	for _, orgOwner := range owners {
		user, ok := setUp.createdUsers[orgOwner]
		if !ok {
			logging.LogWithFields("SETUP-s9ilr", "Owner", orgOwner).Error("unable to add user to org members")
			return caos_errs.ThrowPreconditionFailedf(nil, "SETUP-s0prs", "unable to add user to org members: %v", orgOwner)
		}
		err := setUp.orgOwner(ctx, org, user)
		if err != nil {
			logging.Log("SETUP-s90oe").WithError(err).Error("unable to add global org admin to members of global org")
			return err
		}
	}
	return nil
}

func (setUp *initializer) orgOwner(ctx context.Context, org *org_model.Org, user *usr_model.User) error {
	addMember := &org_model.OrgMember{
		ObjectRoot: models.ObjectRoot{AggregateID: org.AggregateID},
		UserID:     user.AggregateID,
		Roles:      []string{OrgOwnerRole},
	}
	_, err := setUp.repos.OrgEvents.AddOrgMember(ctx, addMember)
	return err
}

func (setUp *initializer) projects(ctx context.Context, projects []types.Project) error {
	for _, project := range projects {
		createdProject, err := setUp.project(ctx, project)
		if err != nil {
			return err
		}
		setUp.createdProjects[createdProject.Name] = createdProject
		for _, oidc := range project.OIDCApps {
			app, err := setUp.oidcApp(ctx, createdProject, oidc)
			if err != nil {
				return err
			}
			logging.LogWithFields("SETUP-asd32f", "name", app.Name, "clientID", app.OIDCConfig.ClientID).Info("created OIDC application")
		}
	}
	return nil
}

func (setUp *initializer) project(ctx context.Context, project types.Project) (*proj_model.Project, error) {
	addProject := &proj_model.Project{
		Name: project.Name,
	}
	return setUp.repos.ProjectEvents.CreateProject(ctx, addProject)
}

func (setUp *initializer) oidcApp(ctx context.Context, project *proj_model.Project, oidc types.OIDCApp) (*proj_model.Application, error) {
	addOIDCApp := &proj_model.Application{
		ObjectRoot: models.ObjectRoot{AggregateID: project.AggregateID},
		Name:       oidc.Name,
		OIDCConfig: &proj_model.OIDCConfig{
			RedirectUris:           oidc.RedirectUris,
			ResponseTypes:          getOIDCResponseTypes(oidc.ResponseTypes),
			GrantTypes:             getOIDCGrantTypes(oidc.GrantTypes),
			ApplicationType:        getOIDCApplicationType(oidc.ApplicationType),
			AuthMethodType:         getOIDCAuthMethod(oidc.AuthMethodType),
			PostLogoutRedirectUris: oidc.PostLogoutRedirectUris,
		},
	}
	return setUp.repos.ProjectEvents.AddApplication(ctx, addOIDCApp)
}

func getOIDCResponseTypes(responseTypes []string) []proj_model.OIDCResponseType {
	types := make([]proj_model.OIDCResponseType, len(responseTypes))
	for i, t := range responseTypes {
		types[i] = getOIDCResponseType(t)
	}
	return types
}

func getOIDCResponseType(responseType string) proj_model.OIDCResponseType {
	switch responseType {
	case OIDCResponseType_CODE:
		return proj_model.OIDCRESPONSETYPE_CODE
	case OIDCResponseType_ID_TOKEN:
		return proj_model.OIDCRESPONSETYPE_ID_TOKEN
	case OIDCResponseType_TOKEN:
		return proj_model.OIDCRESPONSETYPE_TOKEN
	}
	return proj_model.OIDCRESPONSETYPE_CODE
}

func getOIDCGrantTypes(grantTypes []string) []proj_model.OIDCGrantType {
	types := make([]proj_model.OIDCGrantType, len(grantTypes))
	for i, t := range grantTypes {
		types[i] = getOIDCGrantType(t)
	}
	return types
}

func getOIDCGrantType(grantTypes string) proj_model.OIDCGrantType {
	switch grantTypes {
	case OIDCGrantType_AUTHORIZATION_CODE:
		return proj_model.OIDCGRANTTYPE_AUTHORIZATION_CODE
	case OIDCGrantType_IMPLICIT:
		return proj_model.OIDCGRANTTYPE_IMPLICIT
	case OIDCGrantType_REFRESH_TOKEN:
		return proj_model.OIDCGRANTTYPE_REFRESH_TOKEN
	}
	return proj_model.OIDCGRANTTYPE_AUTHORIZATION_CODE
}

func getOIDCApplicationType(appType string) proj_model.OIDCApplicationType {
	switch appType {
	case OIDCApplicationType_NATIVE:
		return proj_model.OIDCAPPLICATIONTYPE_NATIVE
	case OIDCApplicationType_USER_AGENT:
		return proj_model.OIDCAPPLICATIONTYPE_USER_AGENT
	case OIDCApplicationType_WEB:
		return proj_model.OIDCAPPLICATIONTYPE_WEB
	}
	return proj_model.OIDCAPPLICATIONTYPE_WEB
}

func getOIDCAuthMethod(authMethod string) proj_model.OIDCAuthMethodType {
	switch authMethod {
	case OIDCAuthMethodType_NONE:
		return proj_model.OIDCAUTHMETHODTYPE_NONE
	case OIDCAuthMethodType_BASIC:
		return proj_model.OIDCAUTHMETHODTYPE_BASIC
	case OIDCAuthMethodType_POST:
		return proj_model.OIDCAUTHMETHODTYPE_POST
	}
	return proj_model.OIDCAUTHMETHODTYPE_NONE
}

func setSetUpContextData(ctx context.Context, orgID string) context.Context {
	return auth.SetCtxData(ctx, auth.CtxData{UserID: SETUP_USER, OrgID: orgID})
}