/*
 * Copyright 2005-2016 the original author or authors.
 *
 * 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 org.springframework.ws.transport.xmpp;

import java.io.IOException;

import org.jivesoftware.smack.SmackException;
import org.jivesoftware.smack.StanzaListener;
import org.jivesoftware.smack.XMPPConnection;
import org.jivesoftware.smack.XMPPException;
import org.jivesoftware.smack.filter.StanzaFilter;
import org.jivesoftware.smack.filter.StanzaTypeFilter;
import org.jivesoftware.smack.packet.Message;
import org.jivesoftware.smack.packet.Stanza;
import org.jivesoftware.smack.tcp.XMPPTCPConnection;

import org.springframework.ws.transport.support.AbstractStandaloneMessageReceiver;

/**
 * Server-side component for receiving XMPP (Jabber) messages.	Requires a {@linkplain #setConnection(XMPPConnection)
 * connection} to be set, in addition to the {@link #setMessageFactory(org.springframework.ws.WebServiceMessageFactory)
 * messageFactory} and {@link #setMessageReceiver(org.springframework.ws.transport.WebServiceMessageReceiver)
 * messageReceiver} required by the base class.
 *
 * @author Gildas Cuisinier
 * @author Arjen Poutsma
 * @author Greg Turnquist
 * @see org.springframework.ws.transport.xmpp.support.XmppConnectionFactoryBean
 * @since 2.0
 */
public class XmppMessageReceiver extends AbstractStandaloneMessageReceiver {

	/** Default encoding used to read from and write to {@link org.jivesoftware.smack.packet.Message} messages. */
	public static final String DEFAULT_MESSAGE_ENCODING = "UTF-8";

	private XMPPTCPConnection connection;

	private WebServicePacketListener packetListener;

	private String messageEncoding = DEFAULT_MESSAGE_ENCODING;

	public XmppMessageReceiver() {
	}

	/** Sets the {@code XMPPConnection} to use. Setting this property is required. */
	public void setConnection(XMPPTCPConnection connection) {
		this.connection = connection;
	}

	@Override
	protected void onActivate() throws XMPPException, IOException, SmackException {
		if (!connection.isConnected()) {
			connection.connect();
		}
	}

	@Override
	protected void onStart() {
		if (logger.isInfoEnabled()) {
			logger.info("Starting XMPP receiver [" + connection.getUser() + "]");
		}
		packetListener = new WebServicePacketListener();
		StanzaFilter packetFilter = new StanzaTypeFilter(Message.class);
		connection.addAsyncStanzaListener(packetListener, packetFilter);
	}

	@Override
	protected void onStop() {
		if (logger.isInfoEnabled()) {
			logger.info("Stopping XMPP receiver [" + connection.getUser() + "]");
		}
		connection.removeAsyncStanzaListener(packetListener);
		packetListener = null;
	}

	@Override
	protected void onShutdown() {
		if (logger.isInfoEnabled()) {
			logger.info("Shutting down XMPP receiver [" + connection.getUser() + "]");
		}
		if (connection.isConnected()) {
			connection.disconnect();
		}
	}

	private class WebServicePacketListener implements StanzaListener {

		@Override
		public void processPacket(Stanza packet) {
			logger.info("Received " + packet);
			if (packet instanceof Message) {
				Message message = (Message) packet;
				try {
					XmppReceiverConnection wsConnection = new XmppReceiverConnection(connection, message);
					wsConnection.setMessageEncoding(messageEncoding);
					handleConnection(wsConnection);
				}
				catch (Exception ex) {
					logger.error(ex);
				}
			}
		}
	}

}