package ca.uhn.hl7v2.hoh.raw.server;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.Enumeration;
import java.util.LinkedHashMap;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import ca.uhn.hl7v2.hoh.api.DecodeException;
import ca.uhn.hl7v2.hoh.api.IAuthorizationServerCallback;
import ca.uhn.hl7v2.hoh.api.IMessageHandler;
import ca.uhn.hl7v2.hoh.api.IResponseSendable;
import ca.uhn.hl7v2.hoh.api.MessageMetadataKeys;
import ca.uhn.hl7v2.hoh.api.MessageProcessingException;
import ca.uhn.hl7v2.hoh.encoder.AuthorizationFailureException;
import ca.uhn.hl7v2.hoh.encoder.Hl7OverHttpRequestDecoder;
import ca.uhn.hl7v2.hoh.raw.api.RawReceivable;
import ca.uhn.hl7v2.hoh.sign.ISigner;
import ca.uhn.hl7v2.hoh.sign.SignatureVerificationException;
import ca.uhn.hl7v2.hoh.util.HTTPUtils;
public class HohRawServlet extends HttpServlet {
private static final org.slf4j.Logger ourLog = org.slf4j.LoggerFactory.getLogger(HohRawServlet.class);
private static final long serialVersionUID = 1L;
private IAuthorizationServerCallback myAuthorizationCallback;
private IMessageHandler<String> myMessageHandler;
private ISigner mySigner;
/**
* {@inheritDoc}
*/
@Override
protected void doGet(HttpServletRequest theReq, HttpServletResponse theResp) throws ServletException, IOException {
theResp.setStatus(400);
theResp.setContentType("text/html");
String message = "GET method is not supported by this server";
HTTPUtils.write400BadRequest(theResp.getOutputStream(), message, false);
}
/**
* {@inheritDoc}
*/
@Override
protected void doPost(HttpServletRequest theReq, HttpServletResponse theResp) throws ServletException, IOException {
Hl7OverHttpRequestDecoder decoder = new Hl7OverHttpRequestDecoder();
decoder.setHeaders(new LinkedHashMap<String, String>());
Enumeration<?> headerNames = theReq.getHeaderNames();
while (headerNames.hasMoreElements()) {
String nextName = (String) headerNames.nextElement();
decoder.getHeaders().put(nextName, theReq.getHeader(nextName));
}
decoder.setUri(theReq.getRequestURI());
decoder.setAuthorizationCallback(myAuthorizationCallback);
decoder.setSigner(mySigner);
try {
decoder.readContentsFromInputStreamAndDecode(theReq.getInputStream());
} catch (AuthorizationFailureException e) {
ourLog.error("Authorization failed on request for {}", theReq.getRequestURI());
theResp.setStatus(HttpServletResponse.SC_UNAUTHORIZED);
HTTPUtils.write401Unauthorized(theResp.getOutputStream(), false);
return;
} catch (DecodeException e) {
ourLog.error("Request failure for " + theReq.getRequestURI(), e.getMessage(), e);
theResp.setStatus(HttpServletResponse.SC_BAD_REQUEST);
HTTPUtils.write400BadRequest(theResp.getOutputStream(), e.getMessage(), false);
return;
} catch (SignatureVerificationException e) {
ourLog.error("Signature verification failed on request for {}", theReq.getRequestURI());
theResp.setStatus(HttpServletResponse.SC_BAD_REQUEST);
HTTPUtils.write400SignatureVerificationFailed(theResp.getOutputStream(), false);
return;
}
Charset charset = decoder.getCharset();
ourLog.debug("Message charset is {}", charset.displayName());
RawReceivable rawMessage = new RawReceivable(decoder.getMessage());
rawMessage.addMetadata(MessageMetadataKeys.REMOTE_HOST_ADDRESS.name(), theReq.getRemoteAddr());
IResponseSendable<String> response;
try {
response = myMessageHandler.messageReceived(rawMessage);
} catch (MessageProcessingException e) {
ourLog.error("Processing problem for " + theReq.getRequestURI(), e.getMessage(), e);
theResp.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
HTTPUtils.write500InternalServerError(theResp.getOutputStream(), e.getMessage(), false);
return;
}
theResp.setCharacterEncoding(charset.name());
theResp.setContentType(response.getEncodingStyle().getContentType());
theResp.setStatus(response.getResponseCode().getCode());
// n.b. don't ask for the writer until headers are set
response.writeMessage(theResp.getWriter());
theResp.flushBuffer();
}
/**
* If set, provides a callback which will be used to validate incoming
* credentials
*/
public void setAuthorizationCallback(IAuthorizationServerCallback theAuthorizationCallback) {
myAuthorizationCallback = theAuthorizationCallback;
}
/**
* @param theMessageHandler
* the messageHandler to set
*/
public void setMessageHandler(IMessageHandler<String> theMessageHandler) {
myMessageHandler = theMessageHandler;
}
/**
* Sets the message signer if signature profile is being used
*/
public void setSigner(ISigner theSigner) {
mySigner = theSigner;
}
}