import java.io.ByteArrayOutputStream;
import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
private int _maxDelay;
private NameValuePair[] _params;
+
+ private NameValuePair[] _headers;
private String _xslt;
* Maximum delay before executing a request.
* @param aParams
* Request parameters to use.
+ * @param aHeaders
+ * Request headers to use.
* @param aXslt
* XSLT used to convert the response.
*/
protected AbstractPageRequest(int aMaxTries, int aMaxDelay,
- NameValuePair[] aParams, String aXslt, XslTransformer aTransformer) {
+ NameValuePair[] aParams, NameValuePair[] aHeaders, String aXslt, XslTransformer aTransformer) {
if (aParams == null) {
throw new IllegalArgumentException("aParams is null");
}
+ if (aHeaders == null) {
+ throw new IllegalArgumentException("aHeaders is null");
+ }
if (aXslt == null) {
throw new IllegalArgumentException("aXslt is null");
}
_maxTries = aMaxTries;
_maxDelay = aMaxDelay;
_params = aParams;
+ _headers = aHeaders;
_xslt = aXslt;
_transformer = aTransformer;
}
/**
* Gets the parameters for the request.
*
+ * @param aParams Additional parameters to use, obtained from another page, most likely as
+ * hidden form fields.
* @return Request parameters.
*/
- protected NameValuePair[] getParameters() {
- return _params;
+ protected NameValuePair[] getParameters(NameValuePair[] aParams) {
+ List<NameValuePair> params = new ArrayList<NameValuePair>();
+ params.addAll(Arrays.asList(_params));
+ params.addAll(Arrays.asList(aParams));
+ return params.toArray(new NameValuePair[0]);
+ }
+
+ /**
+ * Gets the headers for the request.
+ * @return Request headers.
+ */
+ protected NameValuePair[] getHeaders() {
+ return _headers;
}
/**
*/
protected Document executeMethod(HttpClient aClient, HttpMethod aMethod)
throws IOException, TransformerException {
+
+ for (NameValuePair header: getHeaders()) {
+ aMethod.setRequestHeader(header.getName(), header.getValue());
+ }
+
int triesLeft = _maxTries;
while (triesLeft > 0) {
triesLeft--;
try {
aMethod = executeWithRedirects(aClient, aMethod);
byte[] xhtmlData = getXhtml(aMethod);
-
+
+
Document transformed = _transformer.transform(xhtmlData,
_transformer.resolve(_xslt));
ByteArrayOutputStream os = new ByteArrayOutputStream();
LOG.debug("Transformed result is \n" + os.toString());
return transformed;
} catch (TransformerConfigurationException e) {
- throw new RuntimeException(e.getMessage(), e);
+ throw new TransformerException("Transformer configuration problem", e);
} finally {
// Release the connection.
aMethod.releaseConnection();
// recursion.
}
default: {
- throw new RuntimeException("Method failed: "
+ throw new IOException("Method failed: "
+ aMethod.getStatusLine());
}
}