1 files changed, 24 insertions, 24 deletions

diff --git a/xmpp-vala/src/module/presence/flag.vala b/xmpp-vala/src/module/presence/flag.vala
index 8fb44e13..bb3562a4 100644
--- a/xmpp-vala/src/module/presence/flag.vala
+++ b/xmpp-vala/src/module/presence/flag.vala
@@ -1,60 +1,60 @@
 using Gee;
 
-using Xmpp.Core;
-
 namespace Xmpp.Presence {
 
 public class Flag : XmppStreamFlag {
     public static FlagIdentity<Flag> IDENTITY = new FlagIdentity<Flag>(NS_URI, "presence");
 
-    private HashMap<string, ConcurrentList<string>> resources = new HashMap<string, ConcurrentList<string>>();
-    private HashMap<string, Presence.Stanza> presences = new HashMap<string, Presence.Stanza>();
+    private HashMap<Jid, Gee.List<Jid>> resources = new HashMap<Jid, Gee.List<Jid>>(Jid.hash_bare_func, Jid.equals_bare_func);
+    private HashMap<Jid, Presence.Stanza> presences = new HashMap<Jid, Presence.Stanza>(Jid.hash_func, Jid.equals_func);
 
-    public Set<string> get_available_jids() {
+    public Set<Jid> get_available_jids() {
         return resources.keys;
     }
 
-    public Gee.List<string>? get_resources(string bare_jid) {
-        return resources[bare_jid];
+    public Gee.List<Jid>? get_resources(Jid jid) {
+        return resources[jid];
     }
 
-    public Presence.Stanza? get_presence(string full_jid) {
+    public Presence.Stanza? get_presence(Jid full_jid) {
         return presences[full_jid];
     }
 
     public void add_presence(Presence.Stanza presence) {
-        string bare_jid = get_bare_jid(presence.from);
-        if (!resources.has_key(bare_jid)) {
-            resources[bare_jid] = new ConcurrentList<string>();
+        if (!resources.has_key(presence.from)) {
+            resources[presence.from] = new ArrayList<Jid>(Jid.equals_func);
         }
-        if (resources[bare_jid].contains(presence.from)) {
-            resources[bare_jid].remove(presence.from);
+        if (resources[presence.from].contains(presence.from)) {
+            resources[presence.from].remove(presence.from);
         }
-        resources[bare_jid].add(presence.from);
+        resources[presence.from].add(presence.from);
         presences[presence.from] = presence;
     }
 
-    public void remove_presence(string jid) {
-        string bare_jid = get_bare_jid(jid);
-        if (resources.has_key(bare_jid)) {
-            if (is_bare_jid(jid)) {
-                foreach (string full_jid in resources[jid]) {
+    public void remove_presence(Jid jid) {
+        if (resources.has_key(jid)) {
+            if (jid.is_bare()) {
+                foreach (Jid full_jid in resources[jid]) {
                     presences.unset(full_jid);
                 }
                 resources.unset(jid);
             } else {
-                resources[bare_jid].remove(jid);
-                if (resources[bare_jid].size == 0) {
-                    resources.unset(bare_jid);
+                resources[jid].remove(jid);
+                if (resources[jid].size == 0) {
+                    resources.unset(jid);
                 }
                 presences.unset(jid);
             }
         }
     }
 
-    public override string get_ns() { return NS_URI; }
+    public override string get_ns() {
+        return NS_URI;
+    }
 
-    public override string get_id() { return IDENTITY.id; }
+    public override string get_id() {
+        return IDENTITY.id;
+    }
 }
 
 }
\ No newline at end of file