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

import com.reucon.openfire.plugin.archive.model.ArchivedMessage;
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;
import org.dom4j.Element;
import org.jivesoftware.openfire.auth.UnauthorizedException;
import org.xmpp.packet.IQ;
import org.xmpp.packet.JID;
import org.xmpp.packet.PacketError;

import java.util.List;

/**
 * Message Archiving Retrieve Handler.
 */
public class IQRetrieveHandler extends AbstractIQHandler {

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

	public IQRetrieveHandler() {
		super("Message Archiving Retrieve Handler", "retrieve", NAMESPACE);
	}

	public IQ handleIQ(IQ packet) throws UnauthorizedException {
		final IQ reply = IQ.createResultIQ(packet);
		final RetrieveRequest retrieveRequest = new RetrieveRequest(
				packet.getChildElement());
		int fromIndex; // inclusive
		int toIndex; // exclusive
		int max;

		final Conversation conversation = retrieve(packet.getFrom(),
				retrieveRequest);
		if (conversation == null) {
			return error(packet, PacketError.Condition.item_not_found);
		}

		final Element chatElement = reply.setChildElement("chat", NAMESPACE);
		chatElement.addAttribute("with", conversation.getWithJid());
		chatElement.addAttribute("start",
				XmppDateUtil.formatDate(conversation.getStart()));

		max = conversation.getMessages().size();
		fromIndex = 0;
		toIndex = max > 0 ? max : 0;

		final XmppResultSet resultSet = retrieveRequest.getResultSet();
		if (resultSet != null) {
			if (resultSet.getMax() != null && resultSet.getMax() <= max) {
				max = resultSet.getMax();
				toIndex = fromIndex + max;
			}

			if (resultSet.getIndex() != null) {
				fromIndex = resultSet.getIndex();
				toIndex = fromIndex + max;
			} else if (resultSet.getAfter() != null) {
				fromIndex = resultSet.getAfter().intValue() + 1;
				toIndex = fromIndex + max;
			} else if (resultSet.getBefore() != null) {
				toIndex = resultSet.getBefore().intValue();
				fromIndex = toIndex - max;
			}
		}
		fromIndex = fromIndex < 0 ? 0 : fromIndex;
		toIndex = toIndex > conversation.getMessages().size() ? conversation
				.getMessages().size() : toIndex;
		toIndex = toIndex < fromIndex ? fromIndex : toIndex;

		final List<ArchivedMessage> messages = conversation.getMessages()
				.subList(fromIndex, toIndex);
		for (ArchivedMessage message : messages) {
			addMessageElement(chatElement, conversation, message);
		}

		if (resultSet != null && messages.size() > 0) {
			resultSet.setFirst((long) fromIndex);
			resultSet.setFirstIndex(fromIndex);
			resultSet.setLast((long) toIndex - 1);
			resultSet.setCount(conversation.getMessages().size());
			chatElement.add(resultSet.createResultElement());
		}

		return reply;
	}

	private Conversation retrieve(JID from, RetrieveRequest request) {
		return getPersistenceManager().getConversation(from.toBareJID(),
				request.getWith(), request.getStart());
	}

	private Element addMessageElement(Element parentElement,
			Conversation conversation, ArchivedMessage message) {
		final Element messageElement;
		final long secs;

		secs = (message.getTime().getTime() - conversation.getStart().getTime()) / 1000;
		messageElement = parentElement.addElement(message.getDirection()
				.toString());
		messageElement.addAttribute("secs", Long.toString(secs));
		if (message.getWithJid() != null) {
			messageElement.addAttribute("jid", message.getWithJid().toBareJID());
		}
		messageElement.addElement("body").setText(message.getBody());

		return messageElement;
	}
}

@@ -2655,20 +2655,27 @@
  != null
- && 
+) %7B%0A%09%09%09if (
 messages
@@ -2685,25 +2685,40 @@
 ze() %3E 0) %7B%0A
-%09
+                
 %09%09resultSet.
@@ -2737,33 +2737,48 @@
 ng) fromIndex);%0A
-%09
+                
 %09%09resultSet.setF
@@ -2799,17 +2799,32 @@
 Index);%0A
-%09
+                
 %09%09result
@@ -2856,17 +2856,44 @@
 x - 1);%0A
-%09
+            %09%09%7D%0A            
 %09%09result
@@ -2941,17 +2941,28 @@
 ize());%0A
-%09
+            
 %09%09chatEl

Fix request collection count request on XEP-0059