From bcb96909c9b53c2ca5287433f2fef103b0ddf317 Mon Sep 17 00:00:00 2001
From: fiaxh <git@mx.ax.lt>
Date: Sun, 21 May 2017 23:30:30 +0200
Subject: Roster versioning

---
 xmpp-vala/src/module/roster/versioning_module.vala | 75 ++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 xmpp-vala/src/module/roster/versioning_module.vala

(limited to 'xmpp-vala/src/module/roster/versioning_module.vala')

diff --git a/xmpp-vala/src/module/roster/versioning_module.vala b/xmpp-vala/src/module/roster/versioning_module.vala
new file mode 100644
index 00000000..fbdb6cef
--- /dev/null
+++ b/xmpp-vala/src/module/roster/versioning_module.vala
@@ -0,0 +1,75 @@
+using Gee;
+
+using Xmpp.Core;
+
+namespace Xmpp.Roster {
+
+public class VersioningModule : XmppStreamModule {
+    private const string NS_URI_FEATURE = "urn:xmpp:features:rosterver";
+
+    public static ModuleIdentity<Module> IDENTITY = new ModuleIdentity<Module>(NS_URI, "roster_versioning");
+
+    private Storage storage;
+
+    public VersioningModule(Storage storage) {
+        this.storage = storage;
+    }
+
+    public override void attach(XmppStream stream) {
+        Module.require(stream);
+        stream.get_module(Module.IDENTITY).pre_get_roster.connect(on_pre_get_roster);
+        stream.get_module(Module.IDENTITY).received_roster.connect(on_received_roster);
+        stream.get_module(Module.IDENTITY).item_updated.connect(on_item_updated);
+        stream.get_module(Module.IDENTITY).item_removed.connect(on_item_removed);
+    }
+
+    public override void detach(XmppStream stream) {
+        stream.get_module(Module.IDENTITY).pre_get_roster.disconnect(on_pre_get_roster);
+    }
+
+    internal override string get_ns() { return NS_URI; }
+    internal override string get_id() { return IDENTITY.id; }
+
+    private void on_pre_get_roster(XmppStream stream, Iq.Stanza iq) {
+        StanzaNode? ver_feature = stream.features.get_subnode("ver", NS_URI_FEATURE);
+        if (ver_feature != null) {
+            iq.stanza.get_subnode("query", NS_URI).set_attribute("ver", storage.get_roster_version() ?? "");
+        }
+    }
+
+    private void on_received_roster(XmppStream stream, Collection<Item> roster, Iq.Stanza iq) {
+        string? ver = iq.stanza.get_deep_attribute(NS_URI + ":query", NS_URI + ":ver");
+        if (ver != null) storage.set_roster_version(ver);
+        if (iq.stanza.get_subnode("query", NS_URI) != null) {
+            storage.set_roster(roster);
+        } else {
+            Flag flag = stream.get_flag(Flag.IDENTITY);
+            foreach (Item item in storage.get_roster()) {
+                flag.roster_items[item.jid] = item;
+            }
+        }
+    }
+
+    private void on_item_updated(XmppStream stream, Item item, Iq.Stanza iq) {
+        string? ver = iq.stanza.get_deep_attribute(NS_URI + ":query", NS_URI + ":ver");
+        if (ver != null) storage.set_roster_version(ver);
+        storage.set_item(item);
+    }
+
+    private void on_item_removed(XmppStream stream, Item item, Iq.Stanza iq) {
+        string? ver = iq.stanza.get_deep_attribute(NS_URI + ":query", NS_URI + ":ver");
+        if (ver != null) storage.set_roster_version(ver);
+        storage.remove_item(item);
+    }
+}
+
+public interface Storage : Object {
+    public abstract string? get_roster_version();
+    public abstract Collection<Roster.Item> get_roster();
+    public abstract void set_roster_version(string version);
+    public abstract void set_roster(Collection<Roster.Item> items);
+    public abstract void set_item(Roster.Item item);
+    public abstract void remove_item(Roster.Item item);
+}
+
+}
\ No newline at end of file
-- 
cgit v1.2.3-54-g00ecf