/*
 * Copyright 2003,2004,2005 Colin Crist
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

package hermes.renderers;

import hermes.util.MessageUtils;
import hermes.util.XmlUtils;

import java.nio.charset.Charset;

import javax.jms.BytesMessage;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.TextMessage;
import javax.swing.JComponent;
import javax.swing.JEditorPane;
import javax.swing.JScrollPane;

import org.apache.log4j.Logger;

/**
 * A renderer that displays toString() on a JMS message in a text area.
 * 
 * @author colincrist@hermesjms.com
 * @version $Id: XMLMessageRenderer.java,v 1.3 2007/02/18 16:13:41 colincrist
 *          Exp $
 */

public class XMLMessageRenderer extends AbstractMessageRenderer {
	private static final Logger log = Logger.getLogger(XMLMessageRenderer.class);

	public class MyConfig extends AbstractMessageRenderer.BasicConfig {
		private String encoding = Charset.defaultCharset().name();

		public String getEncoding() {
			return encoding;
		}

		public void setEncoding(String encoding) {
			this.encoding = encoding;
		}
	}

	public XMLMessageRenderer() {
		super();
		setConfig(createConfig()); // force using instance of MyConfig
	}

	@Override
	public JComponent render(JScrollPane parent, Message m) {
		//
		// Raw Panel.

		JEditorPane pane = new JEditorPane();
		parent.setViewportView(pane);

		pane.setEditable(false);

		try {
			String string;
			if (m instanceof BytesMessage) {
				string = new String(MessageUtils.asBytes(m), getConfig().getEncoding());
			} else {
				string = new String(MessageUtils.asString(m).getBytes(Charset.defaultCharset()), getConfig().getEncoding());
			}

			pane.setContentType("text/xml");
			pane.setText(XmlUtils.prettyPrintXml(string));

		} catch (Throwable e) {
			pane.setText(e.getMessage());
			log.error("exception converting message to byte[]: ", e);
		}

		pane.setCaretPosition(0);

		return pane;
	}

	@Override
	public MyConfig createConfig() {
		return new MyConfig();
	}

	@Override
	public MyConfig getConfig() {
		return (MyConfig) super.getConfig();
	}

	/**
	 * Any JMS message is rederable.
	 */
	@Override
	public boolean canRender(Message message) {
		try {
			if (message instanceof TextMessage) {
				final String text = ((TextMessage) message).getText();

				return XmlUtils.isXML(text);
			}

			if (message instanceof BytesMessage) {
				BytesMessage bytesMsg = (BytesMessage) message;
				bytesMsg.reset();

				final byte[] decl = new byte["<?xml".length()];
				bytesMsg.readBytes(decl);

				return XmlUtils.isXML(new String(decl));
			}
		} catch (JMSException e) {
			log.error("error getting text: " + e.getMessage(), e);
		}

		return false;
	}

	@Override
	public String getDisplayName() {
		return "XML";
	}
}