/**
* Write the content type header to the specified output stream
*
* @param out The output stream
* @throws IOException If an IO problem occurs.
*/
protected void sendContentTypeHeader(OutputStream out) throws IOException {
LOG.trace("enter sendContentTypeHeader(OutputStream out)");
String contentType = getContentType();
if (contentType != null) {
out.write(CRLF_BYTES);
out.write(CONTENT_TYPE_BYTES);
out.write(EncodingUtil.getAsciiBytes(contentType));
String charSet = getCharSet();
if (charSet != null) {
out.write(CHARSET_BYTES);
out.write(EncodingUtil.getAsciiBytes(charSet));
}
}
}
public static void sendDispositionHeader(
String name, String filename, String headerCharset, OutputStream out) throws IOException {
out.write(CONTENT_DISPOSITION_BYTES);
out.write(QUOTE_BYTES);
if (StringUtil.isAscci(name)) out.write(EncodingUtil.getAsciiBytes(name));
else out.write(name.getBytes(headerCharset));
out.write(QUOTE_BYTES);
if (filename != null) {
out.write(FILE_NAME_BYTES);
out.write(QUOTE_BYTES);
if (StringUtil.isAscci(filename)) out.write(EncodingUtil.getAsciiBytes(filename));
else out.write(filename.getBytes(headerCharset));
out.write(QUOTE_BYTES);
}
}
/**
* Write the content disposition header to the specified output stream
*
* @param out The output stream
* @throws IOException If an IO problem occurs.
*/
protected void sendDispositionHeader(OutputStream out) throws IOException {
LOG.trace("enter sendDispositionHeader(OutputStream out)");
out.write(CONTENT_DISPOSITION_BYTES);
out.write(QUOTE_BYTES);
out.write(EncodingUtil.getAsciiBytes(getName()));
out.write(QUOTE_BYTES);
}
/**
* Write the content transfer encoding header to the specified output stream
*
* @param out The output stream
* @throws IOException If an IO problem occurs.
*/
protected void sendTransferEncodingHeader(OutputStream out) throws IOException {
LOG.trace("enter sendTransferEncodingHeader(OutputStream out)");
String transferEncoding = getTransferEncoding();
if (transferEncoding != null) {
out.write(CRLF_BYTES);
out.write(CONTENT_TRANSFER_ENCODING_BYTES);
out.write(EncodingUtil.getAsciiBytes(transferEncoding));
}
}
/**
* Call the GetMethod.
*
* @param url The URL for the HTTP GET method.
* @return GetMethod
* @throws WebserverSystemException If connection failed.
*/
public GetMethod get(final String url, final HashMap<String, String> params)
throws RepositoryException {
GetMethod get = null;
try {
get = new GetMethod(url);
Set<String> paramKeys = params.keySet();
NameValuePair[] paramsArray = new NameValuePair[paramKeys.size()];
Iterator<String> iterator = paramKeys.iterator();
int i = 0;
while (iterator.hasNext()) {
String key = iterator.next();
String value = params.get(key);
paramsArray[i] = new NameValuePair(key, value);
i++;
}
if (params.size() > 0) {
String queryString = EncodingUtil.formUrlEncode(paramsArray, "UTF-8");
get.setQueryString(queryString);
}
// get.setQueryString(paramsArray);
int responseCode = getHttpClient().executeMethod(get);
if ((responseCode / 100) != (HTTP_RESPONSE_OK / 100)) {
String message = get.getResponseBodyAsString();
if (message == null) {
Header header = get.getResponseHeader("eSciDocException");
String value = header.getValue();
if (value != null) {
message = "GET-Request with url " + url + " results with Exception:" + value + " .";
} else {
message = "Connection to '" + url + "' failed with response code " + responseCode;
}
}
get.releaseConnection();
log.info(message);
throw new RepositoryException(message);
}
} catch (HttpException e) {
throw new RepositoryException(e.getMessage(), e);
} catch (IOException e) {
throw new RepositoryException(e.getMessage(), e);
}
return get;
}
public class ResourcePart extends FilePart {
protected static final String FILE_NAME = "; filename=";
/** Attachment's file name as a byte array */
private static final byte[] FILE_NAME_BYTES = EncodingUtil.getAsciiBytes(FILE_NAME);
private Resource resource;
private String headerCharset;
/*public ResourcePart(String name, ResourcePartSource partSource, String contentType, String charset) {
super(name, partSource, contentType, charset==null?"":charset);
this.resource=partSource.getResource();
}*/
public ResourcePart(
String name, ResourcePartSource partSource, String contentType, String headerCharset) {
super(name, partSource, contentType, "");
this.resource = partSource.getResource();
this.headerCharset = headerCharset;
}
/** @return the resource */
public Resource getResource() {
return resource;
}
/** @see org.apache.commons.httpclient.methods.multipart.PartBase#getCharSet() */
public String getCharSet() {
String cs = super.getCharSet();
if (StringUtil.isEmpty(cs)) return null;
return cs;
}
/**
* Write the disposition header to the output stream
*
* @param out The output stream
* @throws IOException If an IO problem occurs
* @see Part#sendDispositionHeader(OutputStream)
*/
protected void sendDispositionHeader(OutputStream out) throws IOException {
sendDispositionHeader(getName(), getSource().getFileName(), headerCharset, out);
}
public static void sendDispositionHeader(
String name, String filename, String headerCharset, OutputStream out) throws IOException {
out.write(CONTENT_DISPOSITION_BYTES);
out.write(QUOTE_BYTES);
if (StringUtil.isAscci(name)) out.write(EncodingUtil.getAsciiBytes(name));
else out.write(name.getBytes(headerCharset));
out.write(QUOTE_BYTES);
if (filename != null) {
out.write(FILE_NAME_BYTES);
out.write(QUOTE_BYTES);
if (StringUtil.isAscci(filename)) out.write(EncodingUtil.getAsciiBytes(filename));
else out.write(filename.getBytes(headerCharset));
out.write(QUOTE_BYTES);
}
}
}
/** * Abstract class for one Part of a multipart post object. * * @author <a href="mailto:[email protected]">Matthew Albright</a> * @author <a href="mailto:[email protected]">Jeff Dever</a> * @author <a href="mailto:[email protected]">Adrian Sutton</a> * @author <a href="mailto:[email protected]">Mike Bowler</a> * @author <a href="mailto:[email protected]">Oleg Kalnichevski</a> * @since 2.0 */ public abstract class Part { /** Log object for this class. */ private static final Log LOG = LogFactory.getLog(Part.class); /** * The boundary * * @deprecated use {@link * org.apache.commons.httpclient.params.HttpMethodParams#MULTIPART_BOUNDARY} */ protected static final String BOUNDARY = "----------------314159265358979323846"; /** * The boundary as a byte array. * * @deprecated */ protected static final byte[] BOUNDARY_BYTES = EncodingUtil.getAsciiBytes(BOUNDARY); /** * The default boundary to be used if {@link #setBoundaryBytes(byte[])) has not * been called. */ private static final byte[] DEFAULT_BOUNDARY_BYTES = BOUNDARY_BYTES; /** Carriage return/linefeed */ protected static final String CRLF = "\r\n"; /** Carriage return/linefeed as a byte array */ protected static final byte[] CRLF_BYTES = EncodingUtil.getAsciiBytes(CRLF); /** Content dispostion characters */ protected static final String QUOTE = "\""; /** Content dispostion as a byte array */ protected static final byte[] QUOTE_BYTES = EncodingUtil.getAsciiBytes(QUOTE); /** Extra characters */ protected static final String EXTRA = "--"; /** Extra characters as a byte array */ protected static final byte[] EXTRA_BYTES = EncodingUtil.getAsciiBytes(EXTRA); /** Content dispostion characters */ protected static final String CONTENT_DISPOSITION = "Content-Disposition: form-data; name="; /** Content dispostion as a byte array */ protected static final byte[] CONTENT_DISPOSITION_BYTES = EncodingUtil.getAsciiBytes(CONTENT_DISPOSITION); /** Content type header */ protected static final String CONTENT_TYPE = "Content-Type: "; /** Content type header as a byte array */ protected static final byte[] CONTENT_TYPE_BYTES = EncodingUtil.getAsciiBytes(CONTENT_TYPE); /** Content charset */ protected static final String CHARSET = "; charset="; /** Content charset as a byte array */ protected static final byte[] CHARSET_BYTES = EncodingUtil.getAsciiBytes(CHARSET); /** Content type header */ protected static final String CONTENT_TRANSFER_ENCODING = "Content-Transfer-Encoding: "; /** Content type header as a byte array */ protected static final byte[] CONTENT_TRANSFER_ENCODING_BYTES = EncodingUtil.getAsciiBytes(CONTENT_TRANSFER_ENCODING); /** * Return the boundary string. * * @return the boundary string * @deprecated uses a constant string. Rather use {@link #getPartBoundary} */ public static String getBoundary() { return BOUNDARY; } /** The ASCII bytes to use as the multipart boundary. */ private byte[] boundaryBytes; /** * Return the name of this part. * * @return The name. */ public abstract String getName(); /** * Returns the content type of this part. * * @return the content type, or <code>null</code> to exclude the content type header */ public abstract String getContentType(); /** * Return the character encoding of this part. * * @return the character encoding, or <code>null</code> to exclude the character encoding header */ public abstract String getCharSet(); /** * Return the transfer encoding of this part. * * @return the transfer encoding, or <code>null</code> to exclude the transfer encoding header */ public abstract String getTransferEncoding(); /** * Gets the part boundary to be used. * * @return the part boundary as an array of bytes. * @since 3.0 */ protected byte[] getPartBoundary() { if (boundaryBytes == null) { // custom boundary bytes have not been set, use the default. return DEFAULT_BOUNDARY_BYTES; } else { return boundaryBytes; } } /** * Sets the part boundary. Only meant to be used by {@link Part#sendParts(OutputStream, Part[], * byte[])} and {@link Part#getLengthOfParts(Part[], byte[])} * * @param boundaryBytes An array of ASCII bytes. * @since 3.0 */ void setPartBoundary(byte[] boundaryBytes) { this.boundaryBytes = boundaryBytes; } /** * Tests if this part can be sent more than once. * * @return <code>true</code> if {@link #sendData(OutputStream)} can be successfully called more * than once. * @since 3.0 */ public boolean isRepeatable() { return true; } /** * Write the start to the specified output stream * * @param out The output stream * @throws IOException If an IO problem occurs. */ protected void sendStart(OutputStream out) throws IOException { LOG.trace("enter sendStart(OutputStream out)"); out.write(EXTRA_BYTES); out.write(getPartBoundary()); out.write(CRLF_BYTES); } /** * Write the content disposition header to the specified output stream * * @param out The output stream * @throws IOException If an IO problem occurs. */ protected void sendDispositionHeader(OutputStream out) throws IOException { LOG.trace("enter sendDispositionHeader(OutputStream out)"); out.write(CONTENT_DISPOSITION_BYTES); out.write(QUOTE_BYTES); out.write(EncodingUtil.getAsciiBytes(getName())); out.write(QUOTE_BYTES); } /** * Write the content type header to the specified output stream * * @param out The output stream * @throws IOException If an IO problem occurs. */ protected void sendContentTypeHeader(OutputStream out) throws IOException { LOG.trace("enter sendContentTypeHeader(OutputStream out)"); String contentType = getContentType(); if (contentType != null) { out.write(CRLF_BYTES); out.write(CONTENT_TYPE_BYTES); out.write(EncodingUtil.getAsciiBytes(contentType)); String charSet = getCharSet(); if (charSet != null) { out.write(CHARSET_BYTES); out.write(EncodingUtil.getAsciiBytes(charSet)); } } } /** * Write the content transfer encoding header to the specified output stream * * @param out The output stream * @throws IOException If an IO problem occurs. */ protected void sendTransferEncodingHeader(OutputStream out) throws IOException { LOG.trace("enter sendTransferEncodingHeader(OutputStream out)"); String transferEncoding = getTransferEncoding(); if (transferEncoding != null) { out.write(CRLF_BYTES); out.write(CONTENT_TRANSFER_ENCODING_BYTES); out.write(EncodingUtil.getAsciiBytes(transferEncoding)); } } /** * Write the end of the header to the output stream * * @param out The output stream * @throws IOException If an IO problem occurs. */ protected void sendEndOfHeader(OutputStream out) throws IOException { LOG.trace("enter sendEndOfHeader(OutputStream out)"); out.write(CRLF_BYTES); out.write(CRLF_BYTES); } /** * Write the data to the specified output stream * * @param out The output stream * @throws IOException If an IO problem occurs. */ protected abstract void sendData(OutputStream out) throws IOException; /** * Return the length of the main content * * @return long The length. * @throws IOException If an IO problem occurs */ protected abstract long lengthOfData() throws IOException; /** * Write the end data to the output stream. * * @param out The output stream * @throws IOException If an IO problem occurs. */ protected void sendEnd(OutputStream out) throws IOException { LOG.trace("enter sendEnd(OutputStream out)"); out.write(CRLF_BYTES); } /** * Write all the data to the output stream. If you override this method make sure to override * #length() as well * * @param out The output stream * @throws IOException If an IO problem occurs. */ public void send(OutputStream out) throws IOException { LOG.trace("enter send(OutputStream out)"); sendStart(out); sendDispositionHeader(out); sendContentTypeHeader(out); sendTransferEncodingHeader(out); sendEndOfHeader(out); sendData(out); sendEnd(out); } /** * Return the full length of all the data. If you override this method make sure to override * #send(OutputStream) as well * * @return long The length. * @throws IOException If an IO problem occurs */ public long length() throws IOException { LOG.trace("enter length()"); if (lengthOfData() < 0) { return -1; } ByteArrayOutputStream overhead = new ByteArrayOutputStream(); sendStart(overhead); sendDispositionHeader(overhead); sendContentTypeHeader(overhead); sendTransferEncodingHeader(overhead); sendEndOfHeader(overhead); sendEnd(overhead); return overhead.size() + lengthOfData(); } /** * Return a string representation of this object. * * @return A string representation of this object. * @see java.lang.Object#toString() */ public String toString() { return this.getName(); } /** * Write all parts and the last boundary to the specified output stream. * * @param out The stream to write to. * @param parts The parts to write. * @throws IOException If an I/O error occurs while writing the parts. */ public static void sendParts(OutputStream out, final Part[] parts) throws IOException { sendParts(out, parts, DEFAULT_BOUNDARY_BYTES); } /** * Write all parts and the last boundary to the specified output stream. * * @param out The stream to write to. * @param parts The parts to write. * @param partBoundary The ASCII bytes to use as the part boundary. * @throws IOException If an I/O error occurs while writing the parts. * @since 3.0 */ public static void sendParts(OutputStream out, Part[] parts, byte[] partBoundary) throws IOException { if (parts == null) { throw new IllegalArgumentException("Parts may not be null"); } if (partBoundary == null || partBoundary.length == 0) { throw new IllegalArgumentException("partBoundary may not be empty"); } for (int i = 0; i < parts.length; i++) { // set the part boundary before the part is sent parts[i].setPartBoundary(partBoundary); parts[i].send(out); } out.write(EXTRA_BYTES); out.write(partBoundary); out.write(EXTRA_BYTES); out.write(CRLF_BYTES); } /** * Return the total sum of all parts and that of the last boundary * * @param parts The parts. * @return The total length * @throws IOException If an I/O error occurs while writing the parts. */ public static long getLengthOfParts(Part[] parts) throws IOException { return getLengthOfParts(parts, DEFAULT_BOUNDARY_BYTES); } /** * Gets the length of the multipart message including the given parts. * * @param parts The parts. * @param partBoundary The ASCII bytes to use as the part boundary. * @return The total length * @throws IOException If an I/O error occurs while writing the parts. * @since 3.0 */ public static long getLengthOfParts(Part[] parts, byte[] partBoundary) throws IOException { LOG.trace("getLengthOfParts(Parts[])"); if (parts == null) { throw new IllegalArgumentException("Parts may not be null"); } long total = 0; for (int i = 0; i < parts.length; i++) { // set the part boundary before we calculate the part's length parts[i].setPartBoundary(partBoundary); long l = parts[i].length(); if (l < 0) { return -1; } total += l; } total += EXTRA_BYTES.length; total += partBoundary.length; total += EXTRA_BYTES.length; total += CRLF_BYTES.length; return total; } }
/**
* @deprecated Use {@link #printLine(String, String)}
* <p>Writes the specified String (as bytes), followed by <tt>"\r\n".getBytes()</tt> to the
* output stream.
* @param data the data to be written
* @throws IllegalStateException if the connection is not open
* @throws IOException if an I/O problem occurs
*/
@Deprecated
public void printLine(String data) throws IOException, IllegalStateException {
LOG.trace("enter HttpConnection.printLine(String)");
writeLine(EncodingUtil.getBytes(data, "ISO-8859-1"));
}
/**
* Writes the specified String (as bytes) to the output stream.
*
* @param data the string to be written
* @param charset the charset to use for writing the data
* @throws IllegalStateException if the connection is not open
* @throws IOException if an I/O problem occurs
* @since 3.0
*/
public void print(String data, String charset) throws IOException, IllegalStateException {
LOG.trace("enter HttpConnection.print(String)");
write(EncodingUtil.getBytes(data, charset));
}
/** Guts of the implementation, returns a complete Choices result, or null for a failure. */
private Choices queryPerson(String text, int start, int limit) {
// punt if there is no query text
if (text == null || text.trim().length() == 0) return new Choices(true);
// 1. build CQL query
DCPersonName pn = new DCPersonName(text);
StringBuilder query = new StringBuilder();
query
.append("local.FirstName = \"")
.append(pn.getFirstNames())
.append("\" and local.FamilyName = \"")
.append(pn.getLastName())
.append("\"");
// XXX arbitrary default limit - should be configurable?
if (limit == 0) limit = 50;
NameValuePair args[] = new NameValuePair[6];
args[0] = new NameValuePair("operation", "searchRetrieve");
args[1] = new NameValuePair("version", "1.1");
args[2] = new NameValuePair("recordSchema", "info:srw/schema/1/marcxml-v1.1");
args[3] = new NameValuePair("query", query.toString());
args[4] = new NameValuePair("maximumRecords", String.valueOf(limit));
args[5] = new NameValuePair("startRecord", String.valueOf(start + 1));
HttpClient hc = new HttpClient();
String srUrl = url + "?" + EncodingUtil.formUrlEncode(args, "UTF8");
GetMethod get = new GetMethod(srUrl);
log.debug("Trying SRU query, URL=" + srUrl);
// 2. web request
try {
int status = hc.executeMethod(get);
if (status == 200) {
SAXParserFactory spf = SAXParserFactory.newInstance();
SAXParser sp = spf.newSAXParser();
XMLReader xr = sp.getXMLReader();
SRUHandler handler = new SRUHandler();
// XXX FIXME: should turn off validation here explicitly, but
// it seems to be off by default.
xr.setFeature("http://xml.org/sax/features/namespaces", true);
xr.setContentHandler(handler);
xr.setErrorHandler(handler);
xr.parse(new InputSource(get.getResponseBodyAsStream()));
// this probably just means more results available..
if (handler.hits != handler.result.size())
log.warn(
"Discrepency in results, result.length="
+ handler.result.size()
+ ", yet expected results="
+ handler.hits);
boolean more = handler.hits > (start + handler.result.size());
// XXX add non-auth option; perhaps the UI should do this?
// XXX it's really a policy matter if they allow unauth result.
// XXX good, stop it.
// handler.result.add(new Choice("", text, "Non-Authority: \""+text+"\""));
int confidence;
if (handler.hits == 0) confidence = Choices.CF_NOTFOUND;
else if (handler.hits == 1) confidence = Choices.CF_UNCERTAIN;
else confidence = Choices.CF_AMBIGUOUS;
return new Choices(
handler.result.toArray(new Choice[handler.result.size()]),
start,
handler.hits,
confidence,
more);
}
} catch (HttpException e) {
log.error("SRU query failed: ", e);
return new Choices(true);
} catch (IOException e) {
log.error("SRU query failed: ", e);
return new Choices(true);
} catch (ParserConfigurationException e) {
log.warn("Failed parsing SRU result: ", e);
return new Choices(true);
} catch (SAXException e) {
log.warn("Failed parsing SRU result: ", e);
return new Choices(true);
} finally {
get.releaseConnection();
}
return new Choices(true);
}