MessageWriter

/*
  (C) Copyright IBM Corp. 2005, 2013

  THIS FILE IS PROVIDED UNDER THE TERMS OF THE ECLIPSE PUBLIC LICENSE
  ("AGREEMENT"). ANY USE, REPRODUCTION OR DISTRIBUTION OF THIS FILE
  CONSTITUTES RECIPIENTS ACCEPTANCE OF THE AGREEMENT.

  You can obtain a current copy of the Eclipse Public License from
  http://www.opensource.org/licenses/eclipse-1.0.php

  @author : Roberto Pineiro, IBM, roberto.pineiro@us.ibm.com
 * @author : Chung-hao Tan, IBM, chungtan@us.ibm.com
 *
 *
 * Change History
 * Flag       Date        Prog         Description
 *-------------------------------------------------------------------------------
 * 17620      2005-06-29  thschaef     eliminate ASCIIPrintStream1 in import statement
 * 1535756    2006-08-07  lupusalex    Make code warning free
 * 1565892    2006-11-28  lupusalex    Make SBLIM client JSR48 compliant
 * 2003590    2008-06-30  blaschke-oss Change licensing from CPL to EPL
 * 2524131    2009-01-21  raman_arora  Upgrade client to JDK 1.5 (Phase 1)
 * 3304058    2011-05-20  blaschke-oss Use same date format in change history
 * 3601894    2013-01-23  blaschke-oss Enhance HTTP and CIM-XML tracing
 *    2620    2013-02-23  blaschke-oss Chunked output broken
 */

package org.sentrysoftware.wbem.sblim.cimclient.internal.http;

/*-
 * ╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲
 * WBEM Java Client
 * ჻჻჻჻჻჻
 * Copyright (C) 2023 Sentry Software
 * ჻჻჻჻჻჻
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Lesser Public License for more details.
 *
 * You should have received a copy of the GNU General Lesser Public
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/lgpl-3.0.html>.
 * ╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱╲╱
 */

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.logging.Level;

import org.sentrysoftware.wbem.sblim.cimclient.internal.http.io.ASCIIPrintStream;
import org.sentrysoftware.wbem.sblim.cimclient.internal.http.io.ChunkedOutputStream;
import org.sentrysoftware.wbem.sblim.cimclient.internal.http.io.PersistentOutputStream;
import org.sentrysoftware.wbem.sblim.cimclient.internal.logging.LogAndTraceBroker;

/**
 * Class MessageWriter is responsible for creating http messages
 *
 */
public class MessageWriter {

	HttpHeader iHeader = null;

	HttpServerMethod iMethod = null;

	HttpHeader iTrailer = null;

	boolean iChunked = false;

	boolean iPersistent = false;

	ASCIIPrintStream iRealOS;

	ASCIIPrintStream iClientOS;

	ByteArrayOutputStream iBufferedOS;

	/**
	 * Ctor.
	 *
	 * @param pStream
	 * @param pPersistent
	 * @param pChunked
	 */
	public MessageWriter(OutputStream pStream, boolean pPersistent, boolean pChunked) {
		this.iRealOS = new ASCIIPrintStream(pStream);
		this.iChunked = pChunked;
		this.iPersistent = pPersistent;
		this.iBufferedOS = new ByteArrayOutputStream();
		if (pChunked) {
			this.iClientOS = new ASCIIPrintStream(new ChunkedOutputStream(
					new PersistentOutputStream(this.iBufferedOS, pPersistent), 512));
		} else {
			this.iClientOS = new ASCIIPrintStream(new PersistentOutputStream(this.iBufferedOS,
					pPersistent));
		}
		this.iHeader = new HttpHeader();
		this.iMethod = new HttpServerMethod(HttpConnectionHandler.MAJOR_VERSION,
				HttpConnectionHandler.MINOR_VERSION, 200, "OK");
	}

	/**
	 * Resets the stream
	 */
	public void reset() {
		this.iBufferedOS.reset();
	}

	/**
	 * Sets the http header
	 *
	 * @param header
	 *            The new value
	 */
	public void setHeader(HttpHeader header) {
		this.iHeader = header;
	}

	/**
	 * Sets the http server method
	 *
	 * @param method
	 *            The new value
	 */
	public void setMethod(HttpServerMethod method) {
		this.iMethod = method;
	}

	/**
	 * Returns the http header
	 *
	 * @return The http header
	 */
	public HttpHeader getHeader() {
		return this.iHeader;
	}

	/**
	 * Returns the http server method
	 *
	 * @return The http server method
	 */
	public HttpServerMethod getMethod() {
		return this.iMethod;
	}

	/**
	 * Returns the output stream
	 *
	 * @return The output stream
	 */
	public ASCIIPrintStream getOutputStream() {
		return this.iClientOS;
	}

	/**
	 * Write the message and flushes the streams
	 *
	 * @throws IOException
	 */
	public void close() throws IOException {
		this.iMethod.write(this.iRealOS);
		this.iRealOS.flush();
		if (!this.iChunked) this.iHeader.removeField("Transfer-Encoding");
		else this.iHeader.addField("Transfer-Encoding", "chunked");
		if (this.iPersistent) this.iHeader.addField("Connection", "Keep-iAlive");
		else this.iHeader.addField("Connection", "close");

		this.iHeader.addField("Content-Type", "application/xml;charset=\"utf-8\"");
		if (!this.iChunked) this.iHeader.addField("Content-length", Integer
				.toString(this.iBufferedOS.size()));
		LogAndTraceBroker.getBroker().trace(Level.FINER,
				"Indication Response HTTP Headers= " + this.iHeader.toString());
		this.iHeader.write(this.iRealOS);
		this.iRealOS.flush();
		if (this.iChunked) this.iClientOS.close();
		this.iBufferedOS.writeTo(this.iRealOS);
		if (this.iChunked && (this.iTrailer != null)) {
			LogAndTraceBroker.getBroker().trace(Level.FINER,
					"Indication Response HTTP Trailer Headers= " + this.iTrailer.toString());
			this.iTrailer.write(this.iRealOS);
		}
		this.iRealOS.flush();
	}

	/**
	 * Sets the trailer
	 *
	 * @param pTrailer
	 *            The new value
	 */
	public void setTrailer(HttpHeader pTrailer) {
		this.iTrailer = pTrailer;
	}
}