/*
    BEEM is a videoconference application on the Android Platform.

    Copyright (C) 2009 by Frederic-Charles Barthelery,
                          Jean-Manuel Da Silva,
                          Nikita Kozlov,
                          Philippe Lago,
                          Jean Baptiste Vergely,
                          Vincent Veronis.

    This file is part of BEEM.

    BEEM is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    BEEM 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 Public License for more details.

    You should have received a copy of the GNU General Public License
    along with BEEM.  If not, see <http://www.gnu.org/licenses/>.

    Please send bug reports with examples or suggestions to
    contact@beem-project.com or http://dev.beem-project.com/

    Epitech, hereby disclaims all copyright interest in the program "Beem"
    written by Frederic-Charles Barthelery,
               Jean-Manuel Da Silva,
               Nikita Kozlov,
               Philippe Lago,
               Jean Baptiste Vergely,
               Vincent Veronis.

    Nicolas Sadirac, November 26, 2009
    President of Epitech.

    Flavien Astraud, November 26, 2009
    Head of the EIP Laboratory.

*/
package com.beem.project.beem.smack.pep;

import java.util.ArrayList;
import java.util.List;

import org.jivesoftware.smack.Connection;
import org.jivesoftware.smack.filter.PacketExtensionFilter;
import org.jivesoftware.smack.filter.PacketFilter;
import org.jivesoftware.smack.PacketListener;
import org.jivesoftware.smack.packet.Packet;
import org.jivesoftware.smackx.pubsub.EventElement;
import org.jivesoftware.smackx.pubsub.EventElementType;
import org.jivesoftware.smackx.pubsub.Item;
import org.jivesoftware.smackx.pubsub.ItemsExtension;
import org.jivesoftware.smackx.pubsub.PubSubManager;

/**
 * Little extension of {@link PubSubManager} which allows to add {@link PEPListener}.
 *
 */
public class PepSubManager extends PubSubManager {
    private List<PEPListener> mPepListeners = new ArrayList<PEPListener>();
    private PacketFilter mPacketFilter = new PacketExtensionFilter("event", "http://jabber.org/protocol/pubsub#event");

    /**
     * Create a PepSubManager.
     *
     * @param connection the connection
     */
    public PepSubManager(final Connection connection) {
	super(connection);
	init(connection);
    }

    /**
     * Create a PepSubManager associated to the specified connection where the pubsub
     * requests require a specific to address for packets.
     *
     * @param connection the connection
     * @param toAddress The pubsub specific to address (required for some servers)
     */
    public PepSubManager(final Connection connection, final String toAddress) {
	super(connection, toAddress);
	init(connection);
    }

    /**
     * Add a listener to PEP event.
     *
     * @param listener the listener
     */
    public void addPEPListener(PEPListener listener) {
	if (!mPepListeners.contains(listener))
	    mPepListeners.add(listener);
    }

    /**
     * Remove a listener to PEP event.
     *
     * @param listener the listener
     */
    public void removePEPListener(PEPListener listener) {
	mPepListeners.remove(listener);
    }

    /**
     * Initialize the PepSubManager.
     *
     * @param con the connection
     */
    private void init(Connection con) {
	PacketListener packetListener = new PacketListener() {

	    @Override
	    public void processPacket(Packet packet) {
		EventElement e = (EventElement) packet.getExtension("event", "http://jabber.org/protocol/pubsub#event");
		if (e.getEventType() != EventElementType.items)
		    return;
		ItemsExtension it = (ItemsExtension) e.getEvent();
		if (it.getItemsElementType() != ItemsExtension.ItemsElementType.items)
		    return;
		List<Item> items = (List<Item>) it.getItems();
		firePEPListeners(packet.getFrom(), it.getNode(), items);
	    }
	};
	con.addPacketListener(packetListener, mPacketFilter);
    }

    /**
     * Fire the PEP listeners.
     *
     * @param from the JID of the user who send the event
     * @param node the node of the items in the event
     * @param items the different items of the event
     */
    private void firePEPListeners(String from, String node, List<Item> items) {
	for (PEPListener listener : mPepListeners) {
	    listener.eventReceived(from, node, items);
	}
    }
}
