package eu.siacs.conversations.services; import android.util.Log; import java.math.BigInteger; import java.util.HashSet; import java.util.List; import eu.siacs.conversations.Config; import eu.siacs.conversations.entities.Account; import eu.siacs.conversations.entities.Conversation; import eu.siacs.conversations.xml.Element; import eu.siacs.conversations.xmpp.OnAdvancedStreamFeaturesLoaded; import eu.siacs.conversations.xmpp.OnIqPacketReceived; import eu.siacs.conversations.xmpp.jid.Jid; import eu.siacs.conversations.xmpp.stanzas.IqPacket; public class MessageArchiveService implements OnAdvancedStreamFeaturesLoaded { private final XmppConnectionService mXmppConnectionService; private final HashSet queries = new HashSet(); public MessageArchiveService(final XmppConnectionService service) { this.mXmppConnectionService = service; } public void query(final Conversation conversation) { synchronized (this.queries) { final Account account = conversation.getAccount(); long start = conversation.getLastMessageTransmitted(); long end = account.getXmppConnection().getLastSessionEstablished(); if (end - start >= Config.MAX_HISTORY_AGE) { start = end - Config.MAX_HISTORY_AGE; } final Query query = new Query(conversation, start, end); this.queries.add(query); IqPacket packet = this.mXmppConnectionService.getIqGenerator().queryMessageArchiveManagement(query); this.mXmppConnectionService.sendIqPacket(account, packet, new OnIqPacketReceived() { @Override public void onIqPacketReceived(Account account, IqPacket packet) { if (packet.getType() == IqPacket.TYPE_ERROR) { finalizeQuery(query); } } }); } } private void finalizeQuery(Query query) { synchronized (this.queries) { this.queries.remove(query); } final Conversation conversation = query.getConversation(); conversation.sort(); if (conversation.setLastMessageTransmitted(query.getEnd())) { this.mXmppConnectionService.databaseBackend.updateConversation(conversation); } this.mXmppConnectionService.updateConversationUi(); } public void processFin(Element fin) { if (fin == null) { return; } Query query = findQuery(fin.getAttribute("queryid")); if (query == null) { return; } boolean complete = fin.getAttributeAsBoolean("complete"); Element set = fin.findChild("set","http://jabber.org/protocol/rsm"); Element last = set == null ? null : set.findChild("last"); if (complete || last == null) { final Account account = query.getConversation().getAccount(); Log.d(Config.LOGTAG,account.getJid().toBareJid().toString()+": completed mam query for "+query.getWith().toString()); this.finalizeQuery(query); } else { final Query nextQuery = query.next(last == null ? null : last.getContent()); IqPacket packet = this.mXmppConnectionService.getIqGenerator().queryMessageArchiveManagement(nextQuery); synchronized (this.queries) { this.queries.remove(query); this.queries.add(nextQuery); } this.mXmppConnectionService.sendIqPacket(query.getConversation().getAccount(),packet,new OnIqPacketReceived() { @Override public void onIqPacketReceived(Account account, IqPacket packet) { if (packet.getType() == IqPacket.TYPE_ERROR) { finalizeQuery(nextQuery); } } }); } } private Query findQuery(String id) { if (id == null) { return null; } synchronized (this.queries) { for(Query query : this.queries) { if (query.getQueryId().equals(id)) { return query; } } return null; } } @Override public void onAdvancedStreamFeaturesAvailable(Account account) { if (account.getXmppConnection() != null && account.getXmppConnection().getFeatures().mam()) { List conversations = mXmppConnectionService.getConversations(); for (Conversation conversation : conversations) { if (conversation.getMode() == Conversation.MODE_SINGLE && conversation.getAccount() == account) { this.query(conversation); } } } else { Log.d(Config.LOGTAG,"no mam available"); } } public class Query { private long start; private long end; private Jid with; private String queryId; private String after = null; private Conversation conversation; public Query(Conversation conversation, long start, long end) { this.conversation = conversation; this.with = conversation.getContactJid().toBareJid(); this.start = start; this.end = end; this.queryId = new BigInteger(50, mXmppConnectionService.getRNG()).toString(32); } public Query next(String after) { Query query = new Query(this.conversation,this.start,this.end); query.after = after; return query; } public String getAfter() { return after; } public String getQueryId() { return queryId; } public Jid getWith() { return with; } public long getStart() { return start; } public long getEnd() { return end; } public Conversation getConversation() { return conversation; } } }