org.zalando.riptide.auth.AuthorizationPlugin Maven / Gradle / Ivy

Go to download
package org.zalando.riptide.auth;

import lombok.AllArgsConstructor;
import org.zalando.riptide.Plugin;
import org.zalando.riptide.RequestArguments;
import org.zalando.riptide.RequestExecution;

import java.io.IOException;

import static com.google.common.net.HttpHeaders.AUTHORIZATION;

@AllArgsConstructor
public final class AuthorizationPlugin implements Plugin {

    private final AuthorizationProvider provider;

    @Override
    public RequestExecution aroundNetwork(final RequestExecution execution) {
        return arguments -> execution.execute(withAuthorizationIfAbsent(arguments));
    }

    private RequestArguments withAuthorizationIfAbsent(final RequestArguments arguments) throws IOException {
        if (arguments.getHeaders().containsKey(AUTHORIZATION)) {
            return arguments;
        }

        return arguments.withHeader(AUTHORIZATION, provider.get());
    }

}