package org.apache.maven.wagon.shared.http;
import java.net.URI;
import org.apache.http.HttpEntityEnclosingRequest;
import org.apache.http.HttpRequest;
import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.ProtocolException;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpHead;
import org.apache.http.client.methods.HttpPut;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.client.methods.RequestBuilder;
import org.apache.http.impl.client.DefaultRedirectStrategy;
import org.apache.http.protocol.HttpContext;
import org.apache.http.util.Args;
import org.apache.maven.wagon.events.TransferEvent;
import org.apache.maven.wagon.shared.http.AbstractHttpClientWagon.WagonHttpEntity;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class WagonRedirectStrategy extends DefaultRedirectStrategy
{
private static final Logger LOGGER = LoggerFactory.getLogger( WagonRedirectStrategy.class );
private static final int SC_PERMANENT_REDIRECT = 308;
public WagonRedirectStrategy()
{
super( new String[] {
HttpGet.METHOD_NAME,
HttpHead.METHOD_NAME,
HttpPut.METHOD_NAME,
"MKCOL" } );
}
@Override
public boolean isRedirected( final HttpRequest request, final HttpResponse response,
final HttpContext context ) throws ProtocolException
{
Args.notNull( request, "HTTP request" );
Args.notNull( response, "HTTP response" );
final int statusCode = response.getStatusLine().getStatusCode();
final String method = request.getRequestLine().getMethod();
switch ( statusCode )
{
case HttpStatus.SC_MOVED_TEMPORARILY:
case HttpStatus.SC_MOVED_PERMANENTLY:
case HttpStatus.SC_SEE_OTHER:
case HttpStatus.SC_TEMPORARY_REDIRECT:
case SC_PERMANENT_REDIRECT:
return isRedirectable( method );
default:
return false;
}
}
@Override
public HttpUriRequest getRedirect( final HttpRequest request, final HttpResponse response,
final HttpContext context ) throws ProtocolException
{
final URI uri = getLocationURI( request, response, context );
if ( request instanceof HttpEntityEnclosingRequest )
{
HttpEntityEnclosingRequest encRequest = (HttpEntityEnclosingRequest) request;
if ( encRequest.getEntity() instanceof AbstractHttpClientWagon.WagonHttpEntity )
{
AbstractHttpClientWagon.WagonHttpEntity whe = (WagonHttpEntity) encRequest.getEntity();
if ( whe.getWagon() instanceof AbstractHttpClientWagon )
{
AbstractHttpClientWagon httpWagon = (AbstractHttpClientWagon) whe.getWagon();
TransferEvent transferEvent =
new TransferEvent( httpWagon, whe.getResource(),
TransferEvent.TRANSFER_STARTED, TransferEvent.REQUEST_PUT );
transferEvent.setTimestamp( System.currentTimeMillis() );
transferEvent.setLocalFile( whe.getSource() );
httpWagon.getTransferEventSupport().fireDebug(
String.format( "Following redirect from '%s' to '%s'",
request.getRequestLine().getUri(), uri.toASCIIString() ) );
httpWagon.getTransferEventSupport().fireTransferStarted( transferEvent );
}
else
{
LOGGER.warn( "Cannot properly handle redirect transfer event, wagon has unexpected class: {}",
whe.getWagon().getClass() );
}
}
}
return RequestBuilder.copy( request ).setUri( uri ).build();
}
}