package com.reucon.openfire.plugin.archive.xep0136;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;

import org.dom4j.Element;
import org.jivesoftware.openfire.auth.UnauthorizedException;
import org.jivesoftware.openfire.disco.ServerFeaturesProvider;
import org.xmpp.packet.IQ;
import org.xmpp.packet.JID;

import com.reucon.openfire.plugin.archive.model.Conversation;
import com.reucon.openfire.plugin.archive.util.XmppDateUtil;
import com.reucon.openfire.plugin.archive.xep.AbstractIQHandler;
import com.reucon.openfire.plugin.archive.xep0059.XmppResultSet;

/**
 * Message Archiving List Handler.
 */
public class IQListHandler extends AbstractIQHandler implements
        ServerFeaturesProvider {

    private static final String NAMESPACE = "urn:xmpp:archive";
    private static final String NAMESPACE_MANAGE = "urn:xmpp:archive:manage";

    public IQListHandler() {
        super("Message Archiving List Handler", "list", NAMESPACE);
    }

    public IQ handleIQ(IQ packet) throws UnauthorizedException {
        IQ reply = IQ.createResultIQ(packet);
        ListRequest listRequest = new ListRequest(packet.getChildElement());
        JID from = packet.getFrom();

        Element listElement = reply.setChildElement("list", NAMESPACE);
        Collection<Conversation> conversations = list(from, listRequest);
        XmppResultSet resultSet = listRequest.getResultSet();

        for (Conversation conversation : conversations) {
            addChatElement(listElement, conversation);
        }

        if (resultSet != null) {
            listElement.add(resultSet.createResultElement());
        }

        return reply;
    }

    private Collection<Conversation> list(JID from, ListRequest request) {
        return getPersistenceManager(from).findConversations(request.getStart(),
                request.getEnd(), from.toBareJID(), request.getWith(),
                request.getResultSet());
    }

    private Element addChatElement(Element listElement,
            Conversation conversation) {
        Element chatElement = listElement.addElement("chat");

        chatElement.addAttribute("with", conversation.getWithJid());
        chatElement.addAttribute("start",
                XmppDateUtil.formatDate(conversation.getStart()));

        return chatElement;
    }

    public Iterator<String> getFeatures() {
        ArrayList<String> features = new ArrayList<String>();
        features.add(NAMESPACE_MANAGE);
        return features.iterator();
    }

}
