void sendHTML(
HttpServletRequest request,
HttpServletResponse response,
String script,
Exception scriptError,
Object scriptResult,
StringBuffer scriptOutput,
boolean capture)
throws IOException {
// Format the output using a simple templating utility
SimpleTemplate st = new SimpleTemplate(BshServlet.class.getResource("page.template"));
st.replace("version", getBshVersion());
// String requestURI = HttpUtils.getRequestURL( request ).toString()
// I was told this should work
String requestURI = request.getRequestURI();
st.replace("servletURL", requestURI);
if (script != null) st.replace("script", script);
else st.replace("script", exampleScript);
if (capture) st.replace("captureOutErr", "CHECKED");
else st.replace("captureOutErr", "");
if (script != null)
st.replace(
"scriptResult", formatScriptResultHTML(script, scriptResult, scriptError, scriptOutput));
response.setContentType("text/html");
PrintWriter out = response.getWriter();
st.write(out);
out.flush();
}
private static void sendServerProblemResponse(HttpServletResponse res, String s)
throws IOException {
// Commenting out the status code because we want the client to eval the error() call
// res.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
res.setContentType("x-marklogic/xquery; charset=UTF-8");
if (s != null && s.length() > 4096) { // Cap super long errors
s = s.substring(0, 2048) + " ...[trimmed]... " + s.substring(s.length() - 2048);
}
Writer writer = res.getWriter();
writer.write("error('" + escapeSingleQuotes(s) + "')");
writer.flush();
}
void sendRaw(
HttpServletRequest request,
HttpServletResponse response,
Exception scriptError,
Object scriptResult,
StringBuffer scriptOutput)
throws IOException {
response.setContentType("text/plain");
PrintWriter out = response.getWriter();
if (scriptError != null) out.println("Script Error:\n" + scriptError);
else out.println(scriptOutput.toString());
out.flush();
}
public void doGet(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
String script = request.getParameter("bsh.script");
String client = request.getParameter("bsh.client");
String output = request.getParameter("bsh.servlet.output");
String captureOutErr = request.getParameter("bsh.servlet.captureOutErr");
boolean capture = false;
if (captureOutErr != null && captureOutErr.equalsIgnoreCase("true")) capture = true;
Object scriptResult = null;
Exception scriptError = null;
StringBuffer scriptOutput = new StringBuffer();
if (script != null) {
try {
scriptResult = evalScript(script, scriptOutput, capture, request, response);
} catch (Exception e) {
scriptError = e;
}
}
response.setHeader("Bsh-Return", String.valueOf(scriptResult));
if ((output != null && output.equalsIgnoreCase("raw"))
|| (client != null && client.equals("Remote")))
sendRaw(request, response, scriptError, scriptResult, scriptOutput);
else sendHTML(request, response, script, scriptError, scriptResult, scriptOutput, capture);
}
private static void sendXQueryResponse(HttpServletResponse res, Object o) throws IOException {
// Make sure to leave the status code alone. It defaults to 200, but sometimes
// callers of this method will have set it to a custom code.
res.setContentType("x-marklogic/xquery; charset=UTF-8");
// res.setContentType("text/plain");
Writer writer = res.getWriter(); // care to handle errors later?
if (o == null) {
writer.write("()");
} else if (o instanceof byte[]) {
writer.write("binary {'");
writer.write(hexEncode((byte[]) o));
writer.write("'}");
} else if (o instanceof Object[]) {
Object[] arr = (Object[]) o;
writer.write("(");
for (int i = 0; i < arr.length; i++) {
sendXQueryResponse(res, arr[i]);
if (i + 1 < arr.length) writer.write(", ");
}
writer.write(")");
} else if (o instanceof String) {
writer.write("'");
writer.write(escapeSingleQuotes(o.toString()));
writer.write("'");
} else if (o instanceof Integer) {
writer.write("xs:int(");
writer.write(o.toString());
writer.write(")");
} else if (o instanceof Long) {
writer.write("xs:integer(");
writer.write(o.toString());
writer.write(")");
} else if (o instanceof Float) {
Float flt = (Float) o;
writer.write("xs:float(");
if (flt.equals(Float.POSITIVE_INFINITY)) {
writer.write("'INF'");
} else if (flt.equals(Float.NEGATIVE_INFINITY)) {
writer.write("'-INF'");
} else if (flt.equals(Float.NaN)) {
writer.write("fn:number(())"); // poor man's way to write NaN
} else {
writer.write(o.toString());
}
writer.write(")");
} else if (o instanceof Double) {
Double dbl = (Double) o;
writer.write("xs:double(");
if (dbl.equals(Double.POSITIVE_INFINITY)) {
writer.write("'INF'");
} else if (dbl.equals(Double.NEGATIVE_INFINITY)) {
writer.write("'-INF'");
} else if (dbl.equals(Double.NaN)) {
writer.write("fn:number(())"); // poor man's way to write NaN
} else {
writer.write(o.toString());
}
writer.write(")");
} else if (o instanceof Boolean) {
writer.write("xs:boolean('");
writer.write(o.toString());
writer.write("')");
} else if (o instanceof BigDecimal) {
writer.write("xs:decimal(");
writer.write(o.toString());
writer.write(")");
} else if (o instanceof Date) {
// We want something like: 2006-04-30T01:28:30.499-07:00
// We format to get: 2006-04-30T01:28:30.499-0700
// Then we add in the colon
writer.write("xs:dateTime('");
String d = dateFormat.format((Date) o);
writer.write(d.substring(0, d.length() - 2));
writer.write(":");
writer.write(d.substring(d.length() - 2));
writer.write("')");
} else if (o instanceof XMLGregorianCalendar) {
XMLGregorianCalendar greg = (XMLGregorianCalendar) o;
QName type = greg.getXMLSchemaType();
if (type.equals(DatatypeConstants.DATETIME)) {
writer.write("xs:dateTime('");
} else if (type.equals(DatatypeConstants.DATE)) {
writer.write("xs:date('");
} else if (type.equals(DatatypeConstants.TIME)) {
writer.write("xs:time('");
} else if (type.equals(DatatypeConstants.GYEARMONTH)) {
writer.write("xs:gYearMonth('");
} else if (type.equals(DatatypeConstants.GMONTHDAY)) {
writer.write("xs:gMonthDay('");
} else if (type.equals(DatatypeConstants.GYEAR)) {
writer.write("xs:gYear('");
} else if (type.equals(DatatypeConstants.GMONTH)) {
writer.write("xs:gMonth('");
} else if (type.equals(DatatypeConstants.GDAY)) {
writer.write("xs:gDay('");
}
writer.write(greg.toXMLFormat());
writer.write("')");
} else if (o instanceof Duration) {
Duration dur = (Duration) o;
/*
// The following fails on Xerces
QName type = dur.getXMLSchemaType();
if (type.equals(DatatypeConstants.DURATION)) {
writer.write("xs:duration('");
}
else if (type.equals(DatatypeConstants.DURATION_DAYTIME)) {
writer.write("xdt:dayTimeDuration('");
}
else if (type.equals(DatatypeConstants.DURATION_YEARMONTH)) {
writer.write("xdt:yearMonthDuration('");
}
*/
// If no years or months, must be DURATION_DAYTIME
if (dur.getYears() == 0 && dur.getMonths() == 0) {
writer.write("xdt:dayTimeDuration('");
}
// If has years or months but nothing else, must be DURATION_YEARMONTH
else if (dur.getDays() == 0
&& dur.getHours() == 0
&& dur.getMinutes() == 0
&& dur.getSeconds() == 0) {
writer.write("xdt:yearMonthDuration('");
} else {
writer.write("xs:duration('");
}
writer.write(dur.toString());
writer.write("')");
} else if (o instanceof org.jdom.Element) {
org.jdom.Element elt = (org.jdom.Element) o;
writer.write("xdmp:unquote('");
// Because "<" in XQuery is the same as "<" I need to double escape any ampersands
writer.write(
new org.jdom.output.XMLOutputter()
.outputString(elt)
.replaceAll("&", "&")
.replaceAll("'", "''"));
writer.write("')/*"); // make sure to return the root elt
} else if (o instanceof org.jdom.Document) {
org.jdom.Document doc = (org.jdom.Document) o;
writer.write("xdmp:unquote('");
writer.write(
new org.jdom.output.XMLOutputter()
.outputString(doc)
.replaceAll("&", "&")
.replaceAll("'", "''"));
writer.write("')");
} else if (o instanceof org.jdom.Text) {
org.jdom.Text text = (org.jdom.Text) o;
writer.write("text {'");
writer.write(escapeSingleQuotes(text.getText()));
writer.write("'}");
} else if (o instanceof org.jdom.Attribute) {
// <fake xmlns:pref="http://uri.com" pref:attrname="attrvalue"/>/@*:attrname
// <fake xmlns="http://uri.com" attrname="attrvalue"/>/@*:attrname
org.jdom.Attribute attr = (org.jdom.Attribute) o;
writer.write("<fake xmlns");
if ("".equals(attr.getNamespacePrefix())) {
writer.write("=\"");
} else {
writer.write(":" + attr.getNamespacePrefix() + "=\"");
}
writer.write(attr.getNamespaceURI());
writer.write("\" ");
writer.write(attr.getQualifiedName());
writer.write("=\"");
writer.write(escapeSingleQuotes(attr.getValue()));
writer.write("\"/>/@*:");
writer.write(attr.getName());
} else if (o instanceof org.jdom.Comment) {
org.jdom.Comment com = (org.jdom.Comment) o;
writer.write("comment {'");
writer.write(escapeSingleQuotes(com.getText()));
writer.write("'}");
} else if (o instanceof org.jdom.ProcessingInstruction) {
org.jdom.ProcessingInstruction pi = (org.jdom.ProcessingInstruction) o;
writer.write("processing-instruction ");
writer.write(pi.getTarget());
writer.write(" {'");
writer.write(escapeSingleQuotes(pi.getData()));
writer.write("'}");
} else if (o instanceof QName) {
QName q = (QName) o;
writer.write("fn:expanded-QName('");
writer.write(escapeSingleQuotes(q.getNamespaceURI()));
writer.write("','");
writer.write(q.getLocalPart());
writer.write("')");
} else {
writer.write(
"error('XQuery tried to retrieve unsupported type: " + o.getClass().getName() + "')");
}
writer.flush();
}
private static void sendStringResponse(HttpServletResponse res, String s) throws IOException {
res.setContentType("text/plain; charset=UTF-8");
Writer w = res.getWriter();
w.write(s);
w.flush();
}
private static void sendBinaryResponse(HttpServletResponse res, byte[] bytes) throws IOException {
res.setContentType("application/binary-encoded");
OutputStream out = res.getOutputStream(); // care to handle errors later?
out.write(bytes);
out.flush();
}
private static void sendNoResponse(HttpServletResponse res) {
res.setStatus(HttpServletResponse.SC_NO_CONTENT);
}