net.oneandone.shared.artifactory.PreemptiveRequestInterceptor.java Source code

Introduction

Here is the source code for net.oneandone.shared.artifactory.PreemptiveRequestInterceptor.java

Source

/**
 * Copyright 1&1 Internet AG, https://github.com/1and1/
 *
 * 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 net.oneandone.shared.artifactory;

import com.google.common.base.Charsets;
import com.google.common.io.BaseEncoding;
import java.io.IOException;
import java.util.HashMap;
import org.apache.http.HttpException;
import org.apache.http.HttpHost;
import org.apache.http.HttpRequest;
import org.apache.http.HttpRequestInterceptor;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.protocol.ExecutionContext;
import org.apache.http.protocol.HttpContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * {@link HttpRequestInterceptor} which adds preemptively basic auth credentials depending on the host.
 *
 * @author Mirko Friedenhagen
 */
public class PreemptiveRequestInterceptor implements HttpRequestInterceptor {

    private static final Logger LOG = LoggerFactory.getLogger(PreemptiveRequestInterceptor.class);

    /** AUTHORIZATION_HEADER */
    static final String AUTHORIZATION_HEADER = "Authorization";

    /** Holds the credentials for hosts. */
    private final HashMap<String, UsernamePasswordCredentials> credentialsMap = new HashMap<String, UsernamePasswordCredentials>();

    /**
     * Adds userName and password for one host.
     *
     * @param hostName for which userName and password are added .
     * @param userName for the host.
     * @param password for the host.
     */
    public void addCredentialsForHost(String hostName, String userName, String password) {
        credentialsMap.put(hostName, new UsernamePasswordCredentials(userName, password));
    }

    @Override
    public void process(HttpRequest request, HttpContext context) throws HttpException, IOException {
        final HttpHost httpHost = (HttpHost) context.getAttribute(ExecutionContext.HTTP_TARGET_HOST);
        final String httpHostString = httpHost.toHostString();
        final UsernamePasswordCredentials credentials = credentialsMap.get(httpHostString);
        if (credentials != null) {
            final String userName = credentials.getUserName();
            final String auth = userName + ":" + credentials.getPassword();
            LOG.debug("Adding authorization for host {}, userName={}", httpHostString, userName);
            request.addHeader(AUTHORIZATION_HEADER,
                    "Basic " + BaseEncoding.base64().encode(auth.getBytes(Charsets.UTF_8)));
        } else {
            LOG.debug("No authorization for host {}", httpHostString);
        }
    }

}