diff options
Diffstat (limited to 'xmpp-vala/src/module/roster/versioning_module.vala')
| -rw-r--r-- | xmpp-vala/src/module/roster/versioning_module.vala | 75 |
1 files changed, 75 insertions, 0 deletions
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 |