zitadel/internal/api/scim/metadata/context.go

package metadata

import (
	"context"
	"strings"

	"github.com/zitadel/zitadel/internal/zerrors"
)

const bulkIDPrefix = "bulkid:"

type scimContextKeyType struct{}

var scimContextKey scimContextKeyType

type ScimContextData struct {
	ProvisioningDomain          string
	ExternalIDScopedMetadataKey ScopedKey
	bulkIDMapping               map[string]string
}

func NewScimContextData() ScimContextData {
	return ScimContextData{
		ExternalIDScopedMetadataKey: ScopedKey(KeyExternalId),
		bulkIDMapping:               make(map[string]string),
	}
}

func SetScimContextData(ctx context.Context, data ScimContextData) context.Context {
	return context.WithValue(ctx, scimContextKey, data)
}

func GetScimContextData(ctx context.Context) ScimContextData {
	data, _ := ctx.Value(scimContextKey).(ScimContextData)
	return data
}

func SetScimBulkIDMapping(ctx context.Context, bulkID, zitadelID string) context.Context {
	data := GetScimContextData(ctx)
	data.bulkIDMapping[bulkID] = zitadelID
	return ctx
}

func ResolveScimBulkIDIfNeeded(ctx context.Context, resourceID string) (string, error) {
	lowerResourceID := strings.ToLower(resourceID)
	if !strings.HasPrefix(lowerResourceID, bulkIDPrefix) {
		return resourceID, nil
	}

	bulkID := strings.TrimPrefix(lowerResourceID, bulkIDPrefix)
	data := GetScimContextData(ctx)
	zitadelID, ok := data.bulkIDMapping[bulkID]
	if !ok {
		return bulkID, zerrors.ThrowInvalidArgumentf(nil, "SCIM-BLK4", "Could not resolve bulkID %v to created ID", bulkID)
	}

	return zitadelID, nil
}