added status servlet to pdf-as-web

1 files changed, 87 insertions, 0 deletions

diff --git a/pdf-as-web-status/src/main/java/at/gv/egiz/status/content/JsonGenerator.java b/pdf-as-web-status/src/main/java/at/gv/egiz/status/content/JsonGenerator.java
new file mode 100644
index 00000000..f26c0885
--- /dev/null
+++ b/pdf-as-web-status/src/main/java/at/gv/egiz/status/content/JsonGenerator.java
@@ -0,0 +1,87 @@
+package at.gv.egiz.status.content;
+
+import java.io.IOException;
+import java.util.Iterator;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.apache.commons.lang3.StringEscapeUtils;
+
+import at.gv.egiz.status.TestResult;
+import at.gv.egiz.status.TestStatus;
+import at.gv.egiz.status.impl.TestStatusString;
+
+public class JsonGenerator implements ContentGenerator {
+
+	@Override
+	public void generate(HttpServletRequest request,
+			HttpServletResponse response, Map<String, TestResult> results,
+			boolean details) throws IOException {
+		boolean allOk = true;
+		
+		Iterator<TestResult> testIterator = results.values().iterator();
+		while(testIterator.hasNext()) {
+			TestResult result = testIterator.next();
+			if(!result.getStatus().equals(TestStatus.OK)){
+				allOk = false;
+				break;
+			}
+		}
+		
+		if(!allOk) {
+			response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
+		} else {
+			response.setStatus(HttpServletResponse.SC_OK);
+		}
+		
+		response.setContentType("application/json");
+		
+		StringBuilder sb = new StringBuilder();
+		
+		sb.append("{");
+
+		Iterator<Entry<String,TestResult>> testResultIterator = results.entrySet().iterator();
+		while(testResultIterator.hasNext()) {
+			Entry<String,TestResult> entry = testResultIterator.next();
+			TestResult result = entry.getValue();
+			String testName = entry.getKey();
+			
+			sb.append("\"");
+			sb.append(StringEscapeUtils.escapeJson(testName));
+			sb.append("\": {\"Status\": \"");
+			sb.append(StringEscapeUtils.escapeJson(TestStatusString.getString(result.getStatus())));
+			sb.append("\"");
+
+			if(details) {
+				sb.append(", \"Detail\": \"");
+				
+				StringBuilder detail = new StringBuilder();
+				
+				Iterator<String> detailStringIt = result.getDetails().iterator();
+				
+				while(detailStringIt.hasNext()) {
+					String detailString = detailStringIt.next();
+					detail.append(StringEscapeUtils.escapeJson(detailString));
+					detail.append(" ");
+				}
+				
+				sb.append(detail.toString());
+				sb.append("\"");
+			} 
+			
+			sb.append("}");
+			if(testResultIterator.hasNext()) {
+				sb.append(", ");
+			}
+		}
+		
+		sb.append("}");
+		
+		response.getOutputStream().write(sb.toString().getBytes("UTF-8"));
+		response.getOutputStream().close();
+	}
+
+}