/*
* Copyright 1998-2009 University Corporation for Atmospheric Research/Unidata
*
* Portions of this software were developed by the Unidata Program at the
* University Corporation for Atmospheric Research.
*
* Access and use of this software shall impose the following obligations
* and understandings on the user. The user is granted the right, without
* any fee or cost, to use, copy, modify, alter, enhance and distribute
* this software, and any derivative works thereof, and its supporting
* documentation for any purpose whatsoever, provided that this entire
* notice appears in all copies of the software, derivative works and
* supporting documentation. Further, UCAR requests that the user credit
* UCAR/Unidata in any publications that result from the use of this
* software or in any product that includes this software. The names UCAR
* and/or Unidata, however, may not be used in any advertising or publicity
* to endorse or promote any products or commercial entity unless specific
* written permission is obtained from UCAR/Unidata. The user also
* understands that UCAR/Unidata is not obligated to provide the user with
* any support, consulting, training or assistance of any kind with regard
* to the use, operation and performance of this software nor to provide
* the user with any updates, revisions, new versions or "bug fixes."
*
* THIS SOFTWARE IS PROVIDED BY UCAR/UNIDATA "AS IS" AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL UCAR/UNIDATA BE LIABLE FOR ANY SPECIAL,
* INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING
* FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
* NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION
* WITH THE ACCESS, USE OR PERFORMANCE OF THIS SOFTWARE.
*/
package thredds.util;
import opendap.dap.http.HTTPException;
import opendap.dap.http.HTTPMethod;
import opendap.dap.http.HTTPSession;
import org.apache.commons.httpclient.Header;
import java.io.*;
import java.net.URI;
import java.util.zip.GZIPInputStream;
import java.util.zip.InflaterInputStream;
import java.util.Map;
import java.util.HashMap;
/**
* _more_
*
* @author edavis
* @since 4.0
*/
public class HttpUriResolver
{
private org.slf4j.Logger logger =
org.slf4j.LoggerFactory.getLogger( HttpUriResolver.class );
private URI uri;
private long connectionTimeout;
private int socketTimeout;
private String contentEncoding = "gzip,deflate";
private boolean allowContentEncoding;
private boolean followRedirects;
private HTTPMethod method = null;
private HTTPSession session = null;
private Map<String,String> respHeaders;
HttpUriResolver( URI uri, long connectionTimeout, int socketTimeout,
boolean allowContentEncoding,
boolean followRedirects )
{
if ( ! uri.getScheme().equalsIgnoreCase( "http" ) )
throw new IllegalArgumentException( "Given a Non-HTTP URI [" + uri.toString() + "].");
this.uri = uri;
this.connectionTimeout = connectionTimeout;
this.socketTimeout = socketTimeout;
this.allowContentEncoding = allowContentEncoding;
this.followRedirects = followRedirects;
}
public void close()
{
if(method != null) method.close();
if(session != null) session.close();
}
public URI getUri() { return this.uri; }
public long getConnectionTimeout() { return this.connectionTimeout; }
public int getSocketTimeout() { return this.socketTimeout; }
public String getContentEncoding() { return this.contentEncoding; }
public boolean getAllowContentEncoding() { return this.allowContentEncoding; }
public boolean getFollowRedirects() { return this.followRedirects; }
public void makeRequest()
throws IOException
{
if ( method != null )
throw new IllegalStateException( "Request already made.");
this.method = getHttpResponse( uri );
}
public int getResponseStatusCode()
{
if ( method == null )
throw new IllegalStateException( "Request has not been made." );
return this.method.getStatusCode();
}
public String getResponseStatusText()
{
if ( method == null )
throw new IllegalStateException( "Request has not been made." );
return this.method.getStatusText();
}
public Map<String,String> getResponseHeaders()
{
if ( method == null )
throw new IllegalStateException( "Request has not been made." );
if ( this.respHeaders == null )
{
this.respHeaders = new HashMap<String,String>();
Header[] headers = this.method.getResponseHeaders();
for ( Header h : headers )
this.respHeaders.put( h.getName(), h.getValue() );
}
return respHeaders;
}
public String getResponseHeaderValue( String name )
{
if ( method == null )
throw new IllegalStateException( "Request has not been made." );
Header responseHeader = this.method.getResponseHeader( name );
return responseHeader == null ? null : responseHeader.getValue();
}
public InputStream getResponseBodyAsInputStream()
throws IOException
{
if ( method == null )
throw new IllegalStateException( "Request has not been made." );
InputStream is = method.getResponseAsStream();
Header contentEncodingHeader = method.getResponseHeader( "Content-Encoding" );
if ( contentEncodingHeader != null )
{
String contentEncoding = contentEncodingHeader.getValue();
if ( contentEncoding != null )
{
if ( contentEncoding.equalsIgnoreCase( "gzip" ) )
return new GZIPInputStream( is );
else if ( contentEncoding.equalsIgnoreCase( "deflate" ) )
return new InflaterInputStream( is );
}
}
return is;
}
private HTTPMethod getHttpResponse( URI uri )
throws IOException, HTTPException
{
if(session == null)
session = new HTTPSession();
session.setConnectionManagerTimeout( this.connectionTimeout );
session.setSoTimeout( this.socketTimeout );
HTTPMethod method = session.newMethodGet( uri.toString() );
method.setFollowRedirects( this.followRedirects );
method.setRequestHeader( "Accept-Encoding", this.contentEncoding );
method.execute();
int statusCode = method.getStatusCode();
if ( statusCode == 200 || statusCode == 201 )
{
return method;
}
return null; // ToDo throw exception with some informative inforamtion.
}
}