/*******************************************************************************
* Copyright (c) 2013 VMware, Inc.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* VMware, Inc. - initial API and implementation
*******************************************************************************/
package org.springframework.ide.eclipse.wizard.gettingstarted.github.auth;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLConnection;
import java.util.List;
import java.util.regex.Pattern;
import org.apache.commons.codec.binary.Base64;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpRequest;
import org.springframework.http.client.ClientHttpRequestExecution;
import org.springframework.http.client.ClientHttpRequestInterceptor;
import org.springframework.http.client.ClientHttpResponse;
import org.springframework.ide.eclipse.wizard.WizardPlugin;
import org.springframework.web.client.RestTemplate;
/**
* Uses basic authentication username and passwd to access github rest api.
*
* @author Kris De Volder
*/
public class BasicAuthCredentials extends Credentials {
private final Pattern host;
private final String username;
private final String passwd;
public BasicAuthCredentials(Pattern host, String username, String passwd) {
this.host = host;
this.username = username;
this.passwd = passwd;
}
@Override
public String toString() {
return "BasicAuthCredentials("+username+")";
}
private String computeAuthString() throws UnsupportedEncodingException {
String authorisation = username + ":" + passwd;
byte[] encodedAuthorisation = Base64.encodeBase64(authorisation.getBytes("utf8"));
String authString = "Basic " + new String(encodedAuthorisation);
return authString;
}
@Override
public RestTemplate apply(RestTemplate rest) {
List<ClientHttpRequestInterceptor> interceptors = rest.getInterceptors();
interceptors.add(new ClientHttpRequestInterceptor() {
public ClientHttpResponse intercept(HttpRequest request, byte[] body,
ClientHttpRequestExecution execution) throws IOException {
if (matchHost(request.getURI().getHost())) {
HttpHeaders headers = request.getHeaders();
if (!headers.containsKey("Authorization")) {
String authString = computeAuthString();
headers.add("Authorization", authString);
}
}
return execution.execute(request, body);
}
});
return rest;
}
@Override
public void apply(URLConnection conn) {
try {
if (matchHost(conn.getURL().getHost())) {
conn.setRequestProperty("Authorization", computeAuthString());
}
} catch (UnsupportedEncodingException e) {
//Shouldn't really be possible...
WizardPlugin.log(e);
}
}
private boolean matchHost(String host) {
if (this.host!=null) {
return this.host.matcher(host).matches();
} else {
return true;
}
}
}