All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.jivesoftware.smack.UserRoster Maven / Gradle / Ivy

The newest version!
/**
 *
 * Copyright 2003-2007 Jive Software.
 *
 * All rights reserved. Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.jivesoftware.smack;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArrayList;

import org.jivesoftware.smack.filter.PacketFilter;
import org.jivesoftware.smack.filter.PacketIDFilter;
import org.jivesoftware.smack.filter.PacketTypeFilter;
import org.jivesoftware.smack.util.StringUtils;
import org.xmpp.packet.IQ;
import org.xmpp.packet.JID;
import org.xmpp.packet.Packet;
import org.xmpp.packet.Presence;
import org.xmpp.packet.Roster;
import org.xmpp.packet.Roster.Item;
import org.xmpp.packet.Roster.Subscription;

/**
 * Represents a user's roster, which is the collection of users a person
 * receives presence updates for. Roster items are categorized into groups for
 * easier management.
 * 

*

* Others users may attempt to subscribe to this user using a subscription * request. Three modes are supported for handling these requests: *

    *
  • {@link SubscriptionMode#accept_all accept_all} -- accept all subscription * requests.
  • *
  • {@link SubscriptionMode#reject_all reject_all} -- reject all subscription * requests.
  • *
  • {@link SubscriptionMode#manual manual} -- manually process all * subscription requests.
  • *
* * @author Matt Tucker * @see Connection#getRoster() */ public class UserRoster { /** * The default subscription processing mode to use when a Roster is created. * By default all subscription requests are automatically accepted. */ private static SubscriptionMode defaultSubscriptionMode = SubscriptionMode.accept_all; private final Connection connection; private final RosterStore rosterStore; private final Map groups; private final Map entries; private final List unfiledEntries; private final List rosterListeners; private Map> presenceMap; // The roster is marked as initialized when at least a single roster packet // has been received and processed. boolean rosterInitialized = false; private PresencePacketListener presencePacketListener; private SubscriptionMode subscriptionMode = getDefaultSubscriptionMode(); /** * Returns the default subscription processing mode to use when a new Roster * is created. The subscription processing mode dictates what action Smack * will take when subscription requests from other users are made. The * default subscription mode is {@link SubscriptionMode#accept_all}. * * @return the default subscription mode to use for new Rosters */ public static SubscriptionMode getDefaultSubscriptionMode() { return defaultSubscriptionMode; } /** * Sets the default subscription processing mode to use when a new Roster is * created. The subscription processing mode dictates what action Smack will * take when subscription requests from other users are made. The default * subscription mode is {@link SubscriptionMode#accept_all}. * * @param subscriptionMode * the default subscription mode to use for new Rosters. */ public static void setDefaultSubscriptionMode( SubscriptionMode subscriptionMode) { defaultSubscriptionMode = subscriptionMode; } /** * Creates a new roster. * * @param connection * an XMPP connection. */ UserRoster(final Connection connection) { this.connection = connection; rosterStore = connection.getConfiguration().getRosterStore(); groups = new ConcurrentHashMap(); unfiledEntries = new CopyOnWriteArrayList(); entries = new ConcurrentHashMap(); rosterListeners = new CopyOnWriteArrayList(); presenceMap = new ConcurrentHashMap>(); // Listen for any roster packets. PacketFilter rosterFilter = new PacketTypeFilter(Roster.class); connection.addPacketListener(new RosterPacketListener(), rosterFilter); // Listen for any presence packets. PacketFilter presenceFilter = new PacketTypeFilter(Presence.class); presencePacketListener = new PresencePacketListener(); connection.addPacketListener(presencePacketListener, presenceFilter); // Listen for connection events final ConnectionListener connectionListener = new AbstractConnectionListener() { public void connectionClosed() { // Changes the presence available contacts to unavailable setOfflinePresences(); } public void connectionClosedOnError(Exception e) { // Changes the presence available contacts to unavailable setOfflinePresences(); } }; // if not connected add listener after successful login if (!this.connection.isConnected()) { Connection .addConnectionCreationListener(new ConnectionCreationListener() { public void connectionCreated(Connection connection) { if (connection.equals(UserRoster.this.connection)) { UserRoster.this.connection .addConnectionListener(connectionListener); } } }); } else { connection.addConnectionListener(connectionListener); } } /** * Returns the subscription processing mode, which dictates what action * Smack will take when subscription requests from other users are made. The * default subscription mode is {@link SubscriptionMode#accept_all}. *

*

* If using the manual mode, a PacketListener should be registered that * listens for Presence packets that have a type of * {@link org.jivesoftware.smack.packet.Presence.Type#subscribe}. * * @return the subscription mode. */ public SubscriptionMode getSubscriptionMode() { return subscriptionMode; } /** * Sets the subscription processing mode, which dictates what action Smack * will take when subscription requests from other users are made. The * default subscription mode is {@link SubscriptionMode#accept_all}. *

*

* If using the manual mode, a PacketListener should be registered that * listens for Presence packets that have a type of * {@link org.jivesoftware.smack.packet.Presence.Type#subscribe}. * * @param subscriptionMode * the subscription mode. */ public void setSubscriptionMode(SubscriptionMode subscriptionMode) { this.subscriptionMode = subscriptionMode; } /** * Reloads the entire roster from the server. This is an asynchronous * operation, which means the method will return immediately, and the roster * will be reloaded at a later point when the server responds to the reload * request. * * @throws IllegalStateException * if connection is not logged in or logged in anonymously */ public void reload() { if (!connection.isAuthenticated()) { throw new IllegalStateException("Not logged in to server."); } if (connection.isAnonymous()) { throw new IllegalStateException( "Anonymous users can't have a roster."); } Roster packet = new Roster(); if (rosterStore != null && connection.isRosterVersioningSupported()) { packet.getElement().element("query") .addAttribute("ver", rosterStore.getRosterVersion()); PacketFilter filter = new PacketIDFilter(packet.getID()); connection.addPacketListener(new RosterResultListener(), filter); } connection.sendPacket(packet); } /** * Adds a listener to this roster. The listener will be fired anytime one or * more changes to the roster are pushed from the server. * * @param rosterListener * a roster listener. */ public void addRosterListener(RosterListener rosterListener) { if (!rosterListeners.contains(rosterListener)) { rosterListeners.add(rosterListener); } } /** * Removes a listener from this roster. The listener will be fired anytime * one or more changes to the roster are pushed from the server. * * @param rosterListener * a roster listener. */ public void removeRosterListener(RosterListener rosterListener) { rosterListeners.remove(rosterListener); } /** * Creates a new group. *

*

* Note: you must add at least one entry to the group for the group to be * kept after a logout/login. This is due to the way that XMPP stores group * information. * * @param name * the name of the group. * @return a new group. * @throws IllegalStateException * if connection is not logged in or logged in anonymously */ public RosterGroup createGroup(String name) { if (!connection.isAuthenticated()) { throw new IllegalStateException("Not logged in to server."); } if (connection.isAnonymous()) { throw new IllegalStateException( "Anonymous users can't have a roster."); } if (groups.containsKey(name)) { throw new IllegalArgumentException("Group with name " + name + " alread exists."); } RosterGroup group = new RosterGroup(name, connection); groups.put(name, group); return group; } /** * Creates a new roster entry and presence subscription. The server will * asynchronously update the roster with the subscription status. * * @param user * the user. (e.g. [email protected]) * @param name * the nickname of the user. * @param groups * the list of group names the entry will belong to, or * null if the the roster entry won't belong to a group. * @throws XMPPException * if an XMPP exception occurs. * @throws IllegalStateException * if connection is not logged in or logged in anonymously */ public void createEntry(String user, String name, String[] groups) throws XMPPException { if (!connection.isAuthenticated()) { throw new IllegalStateException("Not logged in to server."); } if (connection.isAnonymous()) { throw new IllegalStateException( "Anonymous users can't have a roster."); } // Create and send roster entry creation packet. Roster rosterPacket = new Roster(); rosterPacket.setType(IQ.Type.set); rosterPacket.addItem(new JID(user), name, null, null, Arrays.asList(groups)); // Wait up to a certain number of seconds for a reply from the server. PacketCollector collector = connection .createPacketCollector(new PacketIDFilter(rosterPacket.getID())); connection.sendPacket(rosterPacket); IQ response = (IQ) collector.nextResult(SmackConfiguration .getPacketReplyTimeout()); collector.cancel(); if (response == null) { throw new XMPPException("No response from the server."); } // If the server replied with an error, throw an exception. else if (response.getType() == IQ.Type.error) { throw new XMPPException(response.getError()); } // Create a presence subscription packet and send. Presence presencePacket = new Presence(Presence.Type.subscribe); presencePacket.setTo(user); connection.sendPacket(presencePacket); } /** * Removes a roster entry from the roster. The roster entry will also be * removed from the unfiled entries or from any roster group where it could * belong and will no longer be part of the roster. Note that this is an * asynchronous call -- Smack must wait for the server to send an updated * subscription status. * * @param entry * a roster entry. * @throws XMPPException * if an XMPP error occurs. * @throws IllegalStateException * if connection is not logged in or logged in anonymously */ public void removeEntry(RosterEntry entry) throws XMPPException { if (!connection.isAuthenticated()) { throw new IllegalStateException("Not logged in to server."); } if (connection.isAnonymous()) { throw new IllegalStateException( "Anonymous users can't have a roster."); } // Only remove the entry if it's in the entry list. // The actual removal logic takes place in // RosterPacketListenerprocess>>Packet(Packet) if (!entries.containsKey(entry.getUser())) { return; } Roster packet = new Roster(); packet.setType(IQ.Type.set); packet.addItem(new JID(entry.getUser()), entry.getName(), entry.getAsk(), Subscription.remove, entry.getGroupNames()); PacketCollector collector = connection .createPacketCollector(new PacketIDFilter(packet.getID())); connection.sendPacket(packet); IQ response = (IQ) collector.nextResult(SmackConfiguration .getPacketReplyTimeout()); collector.cancel(); if (response == null) { throw new XMPPException("No response from the server."); } // If the server replied with an error, throw an exception. else if (response.getType() == IQ.Type.error) { throw new XMPPException(response.getError()); } } /** * Returns a count of the entries in the roster. * * @return the number of entries in the roster. */ public int getEntryCount() { return getEntries().size(); } /** * Returns an unmodifiable collection of all entries in the roster, * including entries that don't belong to any groups. * * @return all entries in the roster. */ public Collection getEntries() { Set allEntries = new HashSet(); // Loop through all roster groups and add their entries to the answer for (RosterGroup rosterGroup : getGroups()) { allEntries.addAll(rosterGroup.getEntries()); } // Add the roster unfiled entries to the answer allEntries.addAll(unfiledEntries); return Collections.unmodifiableCollection(allEntries); } /** * Returns a count of the unfiled entries in the roster. An unfiled entry is * an entry that doesn't belong to any groups. * * @return the number of unfiled entries in the roster. */ public int getUnfiledEntryCount() { return unfiledEntries.size(); } /** * Returns an unmodifiable collection for the unfiled roster entries. An * unfiled entry is an entry that doesn't belong to any groups. * * @return the unfiled roster entries. */ public Collection getUnfiledEntries() { return Collections.unmodifiableList(unfiledEntries); } /** * Returns the roster entry associated with the given XMPP address or * null if the user is not an entry in the roster. * * @param user * the XMPP address of the user (eg "[email protected]"). The * address could be in any valid format (e.g. "domain/resource", * "user@domain" or "user@domain/resource"). * @return the roster entry or null if it does not exist. */ public RosterEntry getEntry(String user) { if (user == null) { return null; } return entries.get(user.toLowerCase()); } /** * Returns true if the specified XMPP address is an entry in the roster. * * @param user * the XMPP address of the user (eg "[email protected]"). The * address could be in any valid format (e.g. "domain/resource", * "user@domain" or "user@domain/resource"). * @return true if the XMPP address is an entry in the roster. */ public boolean contains(String user) { return getEntry(user) != null; } /** * Returns the roster group with the specified name, or null if the * group doesn't exist. * * @param name * the name of the group. * @return the roster group with the specified name. */ public RosterGroup getGroup(String name) { return groups.get(name); } /** * Returns the number of the groups in the roster. * * @return the number of groups in the roster. */ public int getGroupCount() { return groups.size(); } /** * Returns an unmodifiable collections of all the roster groups. * * @return an iterator for all roster groups. */ public Collection getGroups() { return Collections.unmodifiableCollection(groups.values()); } /** * Returns the presence info for a particular user. If the user is offline, * or if no presence data is available (such as when you are not subscribed * to the user's presence updates), unavailable presence will be returned. *

*

* If the user has several presences (one for each resource), then the * presence with highest priority will be returned. If multiple presences * have the same priority, the one with the "most available" presence mode * will be returned. In order, that's * {@link org.jivesoftware.smack.packet.Presence.Mode#chat free to chat}, * {@link org.jivesoftware.smack.packet.Presence.Mode#available available}, * {@link org.jivesoftware.smack.packet.Presence.Mode#away away}, * {@link org.jivesoftware.smack.packet.Presence.Mode#xa extended away}, and * {@link org.jivesoftware.smack.packet.Presence.Mode#dnd do not disturb}. *

*

* Note that presence information is received asynchronously. So, just after * logging in to the server, presence values for users in the roster may be * unavailable even if they are actually online. In other words, the value * returned by this method should only be treated as a snapshot in time, and * may not accurately reflect other user's presence instant by instant. If * you need to track presence over time, such as when showing a visual * representation of the roster, consider using a {@link RosterListener}. * * @param user * an XMPP ID. The address could be in any valid format (e.g. * "domain/resource", "user@domain" or "user@domain/resource"). * Any resource information that's part of the ID will be * discarded. * @return the user's current presence, or unavailable presence if the user * is offline or if no presence information is available.. */ public Presence getPresence(String user) { String key = getPresenceMapKey(StringUtils.parseBareAddress(user)); Map userPresences = presenceMap.get(key); if (userPresences == null) { Presence presence = new Presence(Presence.Type.unavailable); presence.setFrom(user); return presence; } else { // Find the resource with the highest priority // Might be changed to use the resource with the highest // availability instead. Presence presence = null; for (String resource : userPresences.keySet()) { Presence p = userPresences.get(resource); if (!p.isAvailable()) { continue; } // Chose presence with highest priority first. if (presence == null || p.getPriority() > presence.getPriority()) { presence = p; } // If equal priority, choose "most available" by the mode value. else if (p.getPriority() == presence.getPriority()) { Presence.Show pMode = p.getShow(); // Default to presence mode of available. if (pMode == null) { pMode = Presence.Show.chat; } Presence.Show presenceMode = presence.getShow(); // Default to presence mode of available. if (presenceMode == null) { presenceMode = Presence.Show.chat; } if (pMode.compareTo(presenceMode) < 0) { presence = p; } } } if (presence == null) { presence = new Presence(Presence.Type.unavailable); presence.setFrom(user); return presence; } else { return presence; } } } /** * Returns the presence info for a particular user's resource, or * unavailable presence if the user is offline or if no presence information * is available, such as when you are not subscribed to the user's presence * updates. * * @param userWithResource * a fully qualified XMPP ID including a resource * (user@domain/resource). * @return the user's current presence, or unavailable presence if the user * is offline or if no presence information is available. */ public Presence getPresenceResource(String userWithResource) { String key = getPresenceMapKey(userWithResource); String resource = StringUtils.parseResource(userWithResource); Map userPresences = presenceMap.get(key); if (userPresences == null) { Presence presence = new Presence(Presence.Type.unavailable); presence.setFrom(userWithResource); return presence; } else { Presence presence = userPresences.get(resource); if (presence == null) { presence = new Presence(Presence.Type.unavailable); presence.setFrom(userWithResource); return presence; } else { return presence; } } } /** * Returns an iterator (of Presence objects) for all of a user's current * presences or an unavailable presence if the user is unavailable (offline) * or if no presence information is available, such as when you are not * subscribed to the user's presence updates. * * @param user * a XMPP ID, e.g. [email protected]. * @return an iterator (of Presence objects) for all the user's current * presences, or an unavailable presence if the user is offline or * if no presence information is available. */ public Iterator getPresences(String user) { String key = getPresenceMapKey(user); Map userPresences = presenceMap.get(key); if (userPresences == null) { Presence presence = new Presence(Presence.Type.unavailable); presence.setFrom(user); return Arrays.asList(presence).iterator(); } else { Collection answer = new ArrayList(); for (Presence presence : userPresences.values()) { if (presence.isAvailable()) { answer.add(presence); } } if (!answer.isEmpty()) { return answer.iterator(); } else { Presence presence = new Presence(Presence.Type.unavailable); presence.setFrom(user); return Arrays.asList(presence).iterator(); } } } /** * Returns the key to use in the presenceMap for a fully qualified XMPP ID. * The roster can contain any valid address format such us * "domain/resource", "user@domain" or "user@domain/resource". If the roster * contains an entry associated with the fully qualified XMPP ID then use * the fully qualified XMPP ID as the key in presenceMap, otherwise use the * bare address. Note: When the key in presenceMap is a fully qualified XMPP * ID, the userPresences is useless since it will always contain one entry * for the user. * * @param user * the bare or fully qualified XMPP ID, e.g. [email protected] or * [email protected]/Work. * @return the key to use in the presenceMap for the fully qualified XMPP * ID. */ private String getPresenceMapKey(String user) { if (user == null) { return null; } String key = user; if (!contains(user)) { key = StringUtils.parseBareAddress(user); } return key.toLowerCase(); } /** * Changes the presence of available contacts offline by simulating an * unavailable presence sent from the server. After a disconnection, every * Presence is set to offline. */ private void setOfflinePresences() { Presence packetUnavailable; for (String user : presenceMap.keySet()) { Map resources = presenceMap.get(user); if (resources != null) { for (String resource : resources.keySet()) { packetUnavailable = new Presence(Presence.Type.unavailable); packetUnavailable.setFrom(user + "/" + resource); presencePacketListener.processPacket(packetUnavailable); } } } } /** * Fires roster changed event to roster listeners indicating that the * specified collections of contacts have been added, updated or deleted * from the roster. * * @param addedEntries * the collection of address of the added contacts. * @param updatedEntries * the collection of address of the updated contacts. * @param deletedEntries * the collection of address of the deleted contacts. */ private void fireRosterChangedEvent(Collection addedEntries, Collection updatedEntries, Collection deletedEntries) { for (RosterListener listener : rosterListeners) { if (!addedEntries.isEmpty()) { listener.entriesAdded(addedEntries); } if (!updatedEntries.isEmpty()) { listener.entriesUpdated(updatedEntries); } if (!deletedEntries.isEmpty()) { listener.entriesDeleted(deletedEntries); } } } /** * Fires roster presence changed event to roster listeners. * * @param presence * the presence change. */ private void fireRosterPresenceEvent(Presence presence) { for (RosterListener listener : rosterListeners) { listener.presenceChanged(presence); } } private void addUpdateEntry(Collection addedEntries, Collection updatedEntries, Roster.Item item, RosterEntry entry) { RosterEntry oldEntry = entries.put(item.getJID().toBareJID(), entry); if (oldEntry == null) { addedEntries.add(item.getJID().toBareJID()); } else { if (!oldEntry.equalsDeep(entry) || !item.getGroups().equals(oldEntry.getGroupNames())) { updatedEntries.add(item.getJID().toBareJID()); } } // Mark the entry as unfiled if it does not belong to any groups. if (item.getGroups().isEmpty()) { if (!unfiledEntries.contains(entry)) { unfiledEntries.add(entry); } } else { unfiledEntries.remove(entry); } // Add the user to the new groups // Add the entry to the groups List newGroupNames = new ArrayList(); for (String groupName : item.getGroups()) { // Add the group name to the list. newGroupNames.add(groupName); // Add the entry to the group. RosterGroup group = getGroup(groupName); if (group == null) { group = createGroup(groupName); groups.put(groupName, group); } // Add the entry. group.addEntryLocal(entry); } // Remove user from the remaining groups. List oldGroupNames = new ArrayList(); for (RosterGroup group : getGroups()) { oldGroupNames.add(group.getName()); } oldGroupNames.removeAll(newGroupNames); for (String groupName : oldGroupNames) { RosterGroup group = getGroup(groupName); group.removeEntryLocal(entry); if (group.getEntryCount() == 0) { groups.remove(groupName); } } } private void deleteEntry(Collection deletedEntries, RosterEntry entry) { String user = entry.getUser(); entries.remove(user); unfiledEntries.remove(entry); presenceMap.remove(StringUtils.parseBareAddress(user)); deletedEntries.add(user); for (Entry e : groups.entrySet()) { RosterGroup group = e.getValue(); group.removeEntryLocal(entry); if (group.getEntryCount() == 0) { groups.remove(e.getKey()); } } } /** * An enumeration for the subscription mode options. */ public enum SubscriptionMode { /** * Automatically accept all subscription and unsubscription requests. * This is the default mode and is suitable for simple client. More * complex client will likely wish to handle subscription requests * manually. */ accept_all, /** * Automatically reject all subscription requests. */ reject_all, /** * Subscription requests are ignored, which means they must be manually * processed by registering a listener for presence packets and then * looking for any presence requests that have the type * Presence.Type.SUBSCRIBE or Presence.Type.UNSUBSCRIBE. */ manual } /** * Listens for all presence packets and processes them. */ private class PresencePacketListener implements PacketListener { public void processPacket(Packet packet) { Presence presence = (Presence) packet; String from = presence.getFrom().toBareJID(); String key = getPresenceMapKey(from); // If an "available" presence, add it to the presence map. Each // presence // map will hold for a particular user a map with the presence // packets saved for each resource. if (presence.getType() == Presence.Type.subscribed) { Map userPresences; // Get the user presence map if (presenceMap.get(key) == null) { userPresences = new ConcurrentHashMap(); presenceMap.put(key, userPresences); } else { userPresences = presenceMap.get(key); } // See if an offline presence was being stored in the map. If // so, remove // it since we now have an online presence. userPresences.remove(""); // Add the new presence, using the resources as a key. userPresences.put(StringUtils.parseResource(from), presence); // If the user is in the roster, fire an event. RosterEntry entry = entries.get(key); if (entry != null) { fireRosterPresenceEvent(presence); } } // If an "unavailable" packet. else if (presence.getType() == Presence.Type.unavailable) { // If no resource, this is likely an offline presence as part of // a roster presence flood. In that case, we store it. if ("".equals(StringUtils.parseResource(from))) { Map userPresences; // Get the user presence map if (presenceMap.get(key) == null) { userPresences = new ConcurrentHashMap(); presenceMap.put(key, userPresences); } else { userPresences = presenceMap.get(key); } userPresences.put("", presence); } // Otherwise, this is a normal offline presence. else if (presenceMap.get(key) != null) { Map userPresences = presenceMap.get(key); // Store the offline presence, as it may include extra // information // such as the user being on vacation. userPresences .put(StringUtils.parseResource(from), presence); } // If the user is in the roster, fire an event. RosterEntry entry = entries.get(key); if (entry != null) { fireRosterPresenceEvent(presence); } } else if (presence.getType() == Presence.Type.subscribe) { if (subscriptionMode == SubscriptionMode.accept_all) { // Accept all subscription requests. Presence response = new Presence(Presence.Type.subscribed); response.setTo(presence.getFrom()); connection.sendPacket(response); } else if (subscriptionMode == SubscriptionMode.reject_all) { // Reject all subscription requests. Presence response = new Presence(Presence.Type.unsubscribed); response.setTo(presence.getFrom()); connection.sendPacket(response); } // Otherwise, in manual mode so ignore. } else if (presence.getType() == Presence.Type.unsubscribe) { if (subscriptionMode != SubscriptionMode.manual) { // Acknowledge and accept unsubscription notification so // that the // server will stop sending notifications saying that the // contact // has unsubscribed to our presence. Presence response = new Presence(Presence.Type.unsubscribed); response.setTo(presence.getFrom()); connection.sendPacket(response); } // Otherwise, in manual mode so ignore. } // Error presence packets from a bare JID mean we invalidate all // existing // presence info for the user. else if (presence.getType() == Presence.Type.error && "".equals(StringUtils.parseResource(from))) { Map userPresences; if (!presenceMap.containsKey(key)) { userPresences = new ConcurrentHashMap(); presenceMap.put(key, userPresences); } else { userPresences = presenceMap.get(key); // Any other presence data is invalidated by the error // packet. userPresences.clear(); } // Set the new presence using the empty resource as a key. userPresences.put("", presence); // If the user is in the roster, fire an event. RosterEntry entry = entries.get(key); if (entry != null) { fireRosterPresenceEvent(presence); } } } } /** * Handles the case of the empty IQ-result for roster versioning. * * Intended to listen for a concrete roster result and deregisters itself * after a processed packet. */ private class RosterResultListener implements PacketListener { @Override public void processPacket(Packet packet) { connection.removePacketListener(this); if (packet instanceof Roster) { // Non-empty roster results are processed by the // RosterPacketListener class return; } if (!(packet instanceof IQ)) { return; } IQ result = (IQ) packet; if (result.getType().equals(IQ.Type.result)) { Collection addedEntries = new ArrayList(); Collection updatedEntries = new ArrayList(); for (Roster.Item item : rosterStore.getEntries()) { RosterEntry entry = new RosterEntry(item.getJID() .toBareJID(), item.getName(), item.getSubscription(), item.getAsk(), UserRoster.this, connection); addUpdateEntry(addedEntries, updatedEntries, item, entry); } synchronized (UserRoster.this) { rosterInitialized = true; UserRoster.this.notifyAll(); } fireRosterChangedEvent(addedEntries, updatedEntries, Collections. emptyList()); } } } /** * Listens for all roster packets and processes them. */ private class RosterPacketListener implements PacketListener { public void processPacket(Packet packet) { Roster rosterPacket = (Roster) packet; Collection addedEntries = new ArrayList(); Collection updatedEntries = new ArrayList(); Collection deletedEntries = new ArrayList(); String version = rosterPacket.getElement().element("query") .attributeValue("ver"); if (rosterPacket.getType().equals(IQ.Type.set)) { // Roster push (RFC 6121, 2.1.6) // A roster push with a non-empty from not matching our address // MUST be ignored String jid = StringUtils.parseBareAddress(connection.getUser()); if (rosterPacket.getFrom() != null && !rosterPacket.getFrom().equals(jid)) { return; } // A roster push must contain exactly one entry Collection items = rosterPacket.getItems(); if (items.size() != 1) { return; } Item item = items.iterator().next(); processPushItem(addedEntries, updatedEntries, deletedEntries, version, item); connection.sendPacket(IQ.createResultIQ(rosterPacket)); } else { // Roster result (RFC 6121, 2.1.4) // Ignore items without valid subscription type ArrayList validItems = new ArrayList(); for (Roster.Item item : rosterPacket.getItems()) { if (hasValidSubscriptionType(item)) { validItems.add(item); } } processResult(addedEntries, updatedEntries, deletedEntries, version, validItems); } // Remove all the groups with no entries. We have to do this because // RosterGroup.removeEntry removes the entry immediately (locally) // and the // group could remain empty. // TODO Check the performance/logic for rosters with large number of // groups for (RosterGroup group : getGroups()) { if (group.getEntryCount() == 0) { groups.remove(group.getName()); } } // Mark the roster as initialized. synchronized (UserRoster.this) { rosterInitialized = true; UserRoster.this.notifyAll(); } // Fire event for roster listeners. fireRosterChangedEvent(addedEntries, updatedEntries, deletedEntries); } private void processPushItem(Collection addedEntries, Collection updatedEntries, Collection deletedEntries, String version, Item item) { RosterEntry entry = new RosterEntry(item.getJID().toBareJID(), item.getName(), item.getSubscription(), item.getAsk(), UserRoster.this, connection); if (item.getSubscription().equals(Roster.Subscription.remove)) { deleteEntry(deletedEntries, entry); if (rosterStore != null) { rosterStore.removeEntry(entry.getUser(), version); } } else if (hasValidSubscriptionType(item)) { addUpdateEntry(addedEntries, updatedEntries, item, entry); if (rosterStore != null) { rosterStore.addEntry(item, version); } } } private void processResult(Collection addedEntries, Collection updatedEntries, Collection deletedEntries, String version, Collection items) { for (Roster.Item item : items) { RosterEntry entry = new RosterEntry(item.getJID().toBareJID(), item.getName(), item.getSubscription(), item.getAsk(), UserRoster.this, connection); addUpdateEntry(addedEntries, updatedEntries, item, entry); } List toDelete = new ArrayList(); // Delete all entries which where not added or updated for (RosterEntry entry : entries.values()) { toDelete.add(entry.getUser()); } toDelete.removeAll(addedEntries); toDelete.removeAll(updatedEntries); for (String user : toDelete) { deleteEntry(deletedEntries, entries.get(user)); } if (rosterStore != null) { rosterStore.resetEntries(items, version); } } /* Ignore ItemTypes as of RFC 6121, 2.1.2.5. */ private boolean hasValidSubscriptionType(Roster.Item item) { return item.getSubscription().equals(Roster.Subscription.none) || item.getSubscription().equals(Roster.Subscription.from) || item.getSubscription().equals(Roster.Subscription.to) || item.getSubscription().equals(Roster.Subscription.both); } } }





© 2015 - 2025 Weber Informatics LLC | Privacy Policy