/*
    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.avatar;

import java.util.List;
import java.util.Arrays;
import org.jivesoftware.smack.Connection;
import org.jivesoftware.smack.XMPPException;
import org.jivesoftware.smackx.pubsub.PubSubManager;
import org.jivesoftware.smackx.pubsub.Node;
import org.jivesoftware.smackx.pubsub.LeafNode;
import org.jivesoftware.smackx.pubsub.Item;
import org.jivesoftware.smackx.pubsub.PayloadItem;

/**
 * An AvatarRetriever which retrieve the avatar over the XMPP connection.
 */
public class XmppAvatarRetriever implements AvatarRetriever {

    private static String AVATARDATANODE = "urn:xmpp:avatar:data";
    private PubSubManager mPubsub;
    private String mFrom;
    private String mId;

    public XmppAvatarRetriever(final Connection con, final String from, final String id) {
	mPubsub = new PubSubManager(con, from);
	mFrom = from;
	mId = id;
    }

    @Override
    public byte[] getAvatar() {
	try {
	    Node node = mPubsub.getNode(AVATARDATANODE);
	    if (node instanceof LeafNode) {
		LeafNode lnode = (LeafNode) node;
		List<Item> items = lnode.getItems(Arrays.asList(mId));
		// TODO the rest ^^
		PayloadItem<AvatarExtension> item = (PayloadItem<AvatarExtension>) items.get(0);
		AvatarExtension avatar = item.getPayload();
		return avatar.getData();
	    }
	} catch (XMPPException e) {
	    return null;
	}
	return null;
    }

}
