org.tkit.quarkus.rs.context.tenant.RestContextTenantResolverService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tkit-quarkus-rest-context Show documentation
There is a newer version: 2.36.0
Show newest version
package org.tkit.quarkus.rs.context.tenant;

import jakarta.enterprise.context.RequestScoped;
import jakarta.inject.Inject;
import jakarta.ws.rs.container.ContainerRequestContext;
import jakarta.ws.rs.container.ResourceInfo;

import org.eclipse.microprofile.jwt.JsonWebToken;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.tkit.quarkus.rs.context.RestContextException;
import org.tkit.quarkus.rs.context.runtime.RestContextTenantRecorder;

@RequestScoped
public class RestContextTenantResolverService {

    private static final Logger log = LoggerFactory.getLogger(RestContextTenantResolverService.class);

    @Inject
    RestContextTenantIdConfig config;

    @Inject
    TenantCustomResolver customTenantResolver;

    @jakarta.ws.rs.core.Context
    ResourceInfo resourceInfo;

    public String getTenantId(JsonWebToken principalToken, ContainerRequestContext containerRequestContext) {

        if (!config.enabled()) {
            return null;
        }

        // check @TenantExclude annotation
        if (RestContextTenantRecorder.tenantExcludeClassMethod(resourceInfo.getResourceClass(),
                resourceInfo.getResourceMethod())) {
            return null;
        }

        var tenantId = get(principalToken, containerRequestContext);
        if (tenantId == null && config.mandatory()) {
            throw new TenantRequiredException();
        }

        return tenantId;
    }

    public String get(JsonWebToken principalToken, ContainerRequestContext containerRequestContext) {

        // check mock service
        if (config.mock().enabled()) {
            return getMockTenantId(principalToken);
        }

        if (config.token().enabled()) {
            // check principal name from token
            if (principalToken != null) {
                String tenantId = principalToken.getClaim(config.token().claimTenantParam());
                if (tenantId != null && !tenantId.isBlank()) {
                    return tenantId;
                }
            }
        }

        // get tenant-id from custom service
        if (config.enabledCustomService()) {
            try {
                String tenantId = customTenantResolver.getTenantId(principalToken, containerRequestContext);
                if (tenantId != null && !tenantId.isBlank()) {
                    return tenantId;
                }
            } catch (Exception ex) {
                log.error("Failed to call custom tenant resolver service, error: " + ex.getMessage(), ex);
                throw new RestContextException(ErrorKeys.ERROR_CALL_CUSTOM_TENANT_SERVICE,
                        "Failed to call custom tenant resolver service, error: " + ex.getMessage(), ex);
            }
        }

        // check tenant-id from header
        if (config.headerParamEnabled()) {
            String tenantId = containerRequestContext.getHeaders().getFirst(config.headerParamName());
            if (tenantId != null && !tenantId.isBlank()) {
                return tenantId;
            }
        }

        return config.defaultTenantId().orElse(null);
    }

    private String getMockTenantId(JsonWebToken principalToken) {

        var mockConfig = config.mock();
        String organization = null;
        if (principalToken != null) {
            organization = principalToken.getClaim(mockConfig.claimOrgId());
        }

        if (organization == null) {
            return mockConfig.defaultTenant();
        }

        var tenantId = mockConfig.data().get(organization);
        if (tenantId == null) {
            return mockConfig.defaultTenant();
        }

        return tenantId;
    }

    public enum ErrorKeys {

        ERROR_CALL_CUSTOM_TENANT_SERVICE;
    }
}