org.finos.tracdap.common.auth.internal.ClientAuthProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tracdap-lib-common Show documentation

TRAC D.A.P. common library, interfaces and utilities used across all TRAC components

There is a newer version: 0.6.3

/*
 * Copyright 2023 Accenture Global Solutions Limited
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.finos.tracdap.common.auth.internal;

import io.grpc.CallCredentials;
import io.grpc.CallOptions;
import io.grpc.Metadata;
import io.grpc.Status;
import io.grpc.stub.AbstractStub;

import java.util.concurrent.Executor;


public class ClientAuthProvider extends CallCredentials {

    private final String token;

    public static >  T applyIfAvailable(T client, String token) {

        if (token != null && !token.isBlank())
            return client.withCallCredentials((new ClientAuthProvider(token)));
        else
            return client;
    }

    public static CallOptions applyIfAvailable(CallOptions callOptions, String token) {

        if (token != null && !token.isBlank())
            return callOptions.withCallCredentials((new ClientAuthProvider(token)));
        else
            return callOptions;
    }

    private ClientAuthProvider(String token) {
        this.token = token;
    }

    @Override
    public void applyRequestMetadata(RequestInfo requestInfo, Executor appExecutor, MetadataApplier applier) {

        if (token == null || token.isBlank()) {

            applier.fail(Status.UNAUTHENTICATED);
        }
        else {

            var authHeaders = new Metadata();
            authHeaders.put(AuthConstants.TRAC_AUTH_TOKEN_KEY, token);

            applier.apply(authHeaders);
        }
    }
}