zitadel/internal/api/grpc/feature/v2/converter.go

package feature

import (
	"net/url"

	"github.com/muhlemmer/gu"

	"github.com/zitadel/zitadel/internal/api/grpc/object/v2"
	"github.com/zitadel/zitadel/internal/command"
	"github.com/zitadel/zitadel/internal/feature"
	"github.com/zitadel/zitadel/internal/query"
	feature_pb "github.com/zitadel/zitadel/pkg/grpc/feature/v2"
)

func systemFeaturesToCommand(req *feature_pb.SetSystemFeaturesRequest) (*command.SystemFeatures, error) {
	loginV2, err := loginV2ToDomain(req.GetLoginV2())
	if err != nil {
		return nil, err
	}
	return &command.SystemFeatures{
		LoginDefaultOrg:                req.LoginDefaultOrg,
		UserSchema:                     req.UserSchema,
		TokenExchange:                  req.OidcTokenExchange,
		ImprovedPerformance:            improvedPerformanceListToDomain(req.ImprovedPerformance),
		OIDCSingleV1SessionTermination: req.OidcSingleV1SessionTermination,
		DisableUserTokenEvent:          req.DisableUserTokenEvent,
		EnableBackChannelLogout:        req.EnableBackChannelLogout,
		LoginV2:                        loginV2,
		PermissionCheckV2:              req.PermissionCheckV2,
	}, nil
}

func systemFeaturesToPb(f *query.SystemFeatures) *feature_pb.GetSystemFeaturesResponse {
	return &feature_pb.GetSystemFeaturesResponse{
		Details:                        object.DomainToDetailsPb(f.Details),
		LoginDefaultOrg:                featureSourceToFlagPb(&f.LoginDefaultOrg),
		UserSchema:                     featureSourceToFlagPb(&f.UserSchema),
		OidcTokenExchange:              featureSourceToFlagPb(&f.TokenExchange),
		ImprovedPerformance:            featureSourceToImprovedPerformanceFlagPb(&f.ImprovedPerformance),
		OidcSingleV1SessionTermination: featureSourceToFlagPb(&f.OIDCSingleV1SessionTermination),
		DisableUserTokenEvent:          featureSourceToFlagPb(&f.DisableUserTokenEvent),
		EnableBackChannelLogout:        featureSourceToFlagPb(&f.EnableBackChannelLogout),
		LoginV2:                        loginV2ToLoginV2FlagPb(f.LoginV2),
		PermissionCheckV2:              featureSourceToFlagPb(&f.PermissionCheckV2),
	}
}

func instanceFeaturesToCommand(req *feature_pb.SetInstanceFeaturesRequest) (*command.InstanceFeatures, error) {
	loginV2, err := loginV2ToDomain(req.GetLoginV2())
	if err != nil {
		return nil, err
	}
	return &command.InstanceFeatures{
		LoginDefaultOrg:                req.LoginDefaultOrg,
		UserSchema:                     req.UserSchema,
		TokenExchange:                  req.OidcTokenExchange,
		ImprovedPerformance:            improvedPerformanceListToDomain(req.ImprovedPerformance),
		DebugOIDCParentError:           req.DebugOidcParentError,
		OIDCSingleV1SessionTermination: req.OidcSingleV1SessionTermination,
		DisableUserTokenEvent:          req.DisableUserTokenEvent,
		EnableBackChannelLogout:        req.EnableBackChannelLogout,
		LoginV2:                        loginV2,
		PermissionCheckV2:              req.PermissionCheckV2,
		ConsoleUseV2UserApi:            req.ConsoleUseV2UserApi,
	}, nil
}

func instanceFeaturesToPb(f *query.InstanceFeatures) *feature_pb.GetInstanceFeaturesResponse {
	return &feature_pb.GetInstanceFeaturesResponse{
		Details:                        object.DomainToDetailsPb(f.Details),
		LoginDefaultOrg:                featureSourceToFlagPb(&f.LoginDefaultOrg),
		UserSchema:                     featureSourceToFlagPb(&f.UserSchema),
		OidcTokenExchange:              featureSourceToFlagPb(&f.TokenExchange),
		ImprovedPerformance:            featureSourceToImprovedPerformanceFlagPb(&f.ImprovedPerformance),
		DebugOidcParentError:           featureSourceToFlagPb(&f.DebugOIDCParentError),
		OidcSingleV1SessionTermination: featureSourceToFlagPb(&f.OIDCSingleV1SessionTermination),
		DisableUserTokenEvent:          featureSourceToFlagPb(&f.DisableUserTokenEvent),
		EnableBackChannelLogout:        featureSourceToFlagPb(&f.EnableBackChannelLogout),
		LoginV2:                        loginV2ToLoginV2FlagPb(f.LoginV2),
		PermissionCheckV2:              featureSourceToFlagPb(&f.PermissionCheckV2),
		ConsoleUseV2UserApi:            featureSourceToFlagPb(&f.ConsoleUseV2UserApi),
	}
}

func featureSourceToImprovedPerformanceFlagPb(fs *query.FeatureSource[[]feature.ImprovedPerformanceType]) *feature_pb.ImprovedPerformanceFeatureFlag {
	return &feature_pb.ImprovedPerformanceFeatureFlag{
		ExecutionPaths: improvedPerformanceTypesToPb(fs.Value),
		Source:         featureLevelToSourcePb(fs.Level),
	}
}

func loginV2ToDomain(loginV2 *feature_pb.LoginV2) (_ *feature.LoginV2, err error) {
	if loginV2 == nil {
		return nil, nil
	}
	var baseURI *url.URL
	if loginV2.GetBaseUri() != "" {
		baseURI, err = url.Parse(loginV2.GetBaseUri())
		if err != nil {
			return nil, err
		}
	}
	return &feature.LoginV2{
		Required: loginV2.GetRequired(),
		BaseURI:  baseURI,
	}, nil
}

func loginV2ToLoginV2FlagPb(f query.FeatureSource[*feature.LoginV2]) *feature_pb.LoginV2FeatureFlag {
	var required bool
	var baseURI *string
	if f.Value != nil {
		required = f.Value.Required
		if f.Value.BaseURI != nil && f.Value.BaseURI.String() != "" {
			baseURI = gu.Ptr(f.Value.BaseURI.String())
		}
	}
	return &feature_pb.LoginV2FeatureFlag{
		Required: required,
		BaseUri:  baseURI,
		Source:   featureLevelToSourcePb(f.Level),
	}
}

func featureSourceToFlagPb(fs *query.FeatureSource[bool]) *feature_pb.FeatureFlag {
	return &feature_pb.FeatureFlag{
		Enabled: fs.Value,
		Source:  featureLevelToSourcePb(fs.Level),
	}
}

func featureLevelToSourcePb(level feature.Level) feature_pb.Source {
	switch level {
	case feature.LevelUnspecified:
		return feature_pb.Source_SOURCE_UNSPECIFIED
	case feature.LevelSystem:
		return feature_pb.Source_SOURCE_SYSTEM
	case feature.LevelInstance:
		return feature_pb.Source_SOURCE_INSTANCE
	case feature.LevelOrg:
		return feature_pb.Source_SOURCE_ORGANIZATION
	case feature.LevelProject:
		return feature_pb.Source_SOURCE_PROJECT
	case feature.LevelApp:
		return feature_pb.Source_SOURCE_APP
	case feature.LevelUser:
		return feature_pb.Source_SOURCE_USER
	default:
		return feature_pb.Source(level)
	}
}

func improvedPerformanceTypesToPb(types []feature.ImprovedPerformanceType) []feature_pb.ImprovedPerformance {
	res := make([]feature_pb.ImprovedPerformance, len(types))

	for i, typ := range types {
		res[i] = improvedPerformanceTypeToPb(typ)
	}

	return res
}

func improvedPerformanceTypeToPb(typ feature.ImprovedPerformanceType) feature_pb.ImprovedPerformance {
	switch typ {
	case feature.ImprovedPerformanceTypeUnspecified:
		return feature_pb.ImprovedPerformance_IMPROVED_PERFORMANCE_UNSPECIFIED
	case feature.ImprovedPerformanceTypeProjectGrant:
		return feature_pb.ImprovedPerformance_IMPROVED_PERFORMANCE_PROJECT_GRANT
	case feature.ImprovedPerformanceTypeProject:
		return feature_pb.ImprovedPerformance_IMPROVED_PERFORMANCE_PROJECT
	case feature.ImprovedPerformanceTypeUserGrant:
		return feature_pb.ImprovedPerformance_IMPROVED_PERFORMANCE_USER_GRANT
	case feature.ImprovedPerformanceTypeOrgDomainVerified:
		return feature_pb.ImprovedPerformance_IMPROVED_PERFORMANCE_ORG_DOMAIN_VERIFIED
	default:
		return feature_pb.ImprovedPerformance(typ)
	}
}

func improvedPerformanceListToDomain(list []feature_pb.ImprovedPerformance) []feature.ImprovedPerformanceType {
	if list == nil {
		return nil
	}
	res := make([]feature.ImprovedPerformanceType, len(list))

	for i, typ := range list {
		res[i] = improvedPerformanceToDomain(typ)
	}

	return res
}

func improvedPerformanceToDomain(typ feature_pb.ImprovedPerformance) feature.ImprovedPerformanceType {
	switch typ {
	case feature_pb.ImprovedPerformance_IMPROVED_PERFORMANCE_UNSPECIFIED:
		return feature.ImprovedPerformanceTypeUnspecified
	case feature_pb.ImprovedPerformance_IMPROVED_PERFORMANCE_PROJECT_GRANT:
		return feature.ImprovedPerformanceTypeProjectGrant
	case feature_pb.ImprovedPerformance_IMPROVED_PERFORMANCE_PROJECT:
		return feature.ImprovedPerformanceTypeProject
	case feature_pb.ImprovedPerformance_IMPROVED_PERFORMANCE_USER_GRANT:
		return feature.ImprovedPerformanceTypeUserGrant
	case feature_pb.ImprovedPerformance_IMPROVED_PERFORMANCE_ORG_DOMAIN_VERIFIED:
		return feature.ImprovedPerformanceTypeOrgDomainVerified
	default:
		return feature.ImprovedPerformanceTypeUnspecified
	}
}