package ca.uhn.hl7v2.hoh.llp;
   
   import java.io.DataOutputStream;
   import java.io.IOException;
   import java.io.OutputStream;
   import java.nio.charset.Charset;
   
   import ca.uhn.hl7v2.hoh.api.EncodeException;
   import ca.uhn.hl7v2.hoh.encoder.AbstractHl7OverHttpEncoder;
  import ca.uhn.hl7v2.hoh.encoder.Hl7OverHttpRequestEncoder;
  import ca.uhn.hl7v2.hoh.encoder.Hl7OverHttpResponseEncoder;
  import ca.uhn.hl7v2.hoh.util.ServerRoleEnum;
  import ca.uhn.hl7v2.llp.HL7Writer;
  import ca.uhn.hl7v2.llp.LLPException;
  
  class HohLlpWriter implements HL7Writer {
  
  	private OutputStream myOutputStream;
  	private Charset myPreferredCharset;
  	private final Hl7OverHttpLowerLayerProtocol myProtocol;
  	private Charset myCharsetForNextMessage;
  
  	/**
  	 * Constructor
  	 */
  	public HohLlpWriter(Hl7OverHttpLowerLayerProtocol theProtocol) {
  		myProtocol = theProtocol;
  	}
  
  	/**
  	 * {@inheritDoc}
  	 */
  	public void close() throws IOException {
  		myOutputStream.close();
  	}
  
  	OutputStream getOutputStream() {
  		return myOutputStream;
  	}
  
  	/**
  	 * @return the preferredCharset
  	 */
  	public Charset getPreferredCharset() {
  		return myPreferredCharset;
  	}
  
  	/**
  	 * {@inheritDoc}
  	 */
  	@Override
  	public void setOutputStream(OutputStream theOutputStream) throws IOException {
  		myOutputStream = theOutputStream;
  	}
  
  	/**
  	 * @param thePreferredCharset
  	 *            the preferredCharset to set
  	 */
  	public void setPreferredCharset(Charset thePreferredCharset) {
  		myPreferredCharset = thePreferredCharset;
  	}
  
  	/**
  	 * {@inheritDoc}
  	 */
  	@Override
  	public void writeMessage(String theRawMessage) throws LLPException, IOException {
  
  		AbstractHl7OverHttpEncoder e;
  		if (myProtocol.getRole() == ServerRoleEnum.CLIENT) {
  			Hl7OverHttpRequestEncoder encoder = new Hl7OverHttpRequestEncoder();
  			if (myProtocol.getHost() != null) {
  				encoder.setHost(myProtocol.getHost());
  			}
  			e = encoder;
  			if (myProtocol.getAuthorizationClientCallback() != null) {
  				e.setUsername(myProtocol.getAuthorizationClientCallback().provideUsername(myProtocol.getUriPath()));
  				e.setPassword(myProtocol.getAuthorizationClientCallback().providePassword(myProtocol.getUriPath()));
  			}
  		} else {
  			e = new Hl7OverHttpResponseEncoder();
  		}
  
  		if (myProtocol.getRole() == ServerRoleEnum.CLIENT) {
  			e.setSigner(myProtocol.getSigner());
  		}
  		
  		e.setMessage(theRawMessage);
  		if (myCharsetForNextMessage != null) {
  			e.setCharset(myCharsetForNextMessage);
  			myCharsetForNextMessage = null;
  		} else if (getPreferredCharset() != null) {
  			e.setCharset(getPreferredCharset());
  		}
  
  		e.setPath(myProtocol.getUriPath());
  		DataOutputStream dos = new DataOutputStream(myOutputStream);
  		try {
 			e.encodeToOutputStream(dos);
 		} catch (EncodeException e1) {
 			throw new LLPException("Failed to encode message", e1);
 		}
 
 		dos.flush();
 
 	}
 
 	void setCharsetForNextMessage(Charset theCharset) {
 		myCharsetForNextMessage = theCharset;
 	}
 
 }