1 files changed, 81 insertions, 0 deletions
diff --git a/vala-xmpp/src/module/xep/0333_chat_markers.vala b/vala-xmpp/src/module/xep/0333_chat_markers.vala
new file mode 100644
index 00000000..0dc0e637
--- /dev/null
+++ b/vala-xmpp/src/module/xep/0333_chat_markers.vala
@@ -0,0 +1,81 @@
+using Gee;
+
+using Xmpp.Core;
+
+namespace Xmpp.Xep.ChatMarkers {
+private const string NS_URI = "urn:xmpp:chat-markers:0";
+
+public const string MARKER_RECEIVED = "received";
+public const string MARKER_DISPLAYED = "displayed";
+public const string MARKER_ACKNOWLEDGED = "acknowledged";
+
+private const string[] MARKERS = {MARKER_RECEIVED, MARKER_DISPLAYED, MARKER_ACKNOWLEDGED};
+
+public class Module : XmppStreamModule {
+    public const string ID = "0333_chat_markers";
+
+    public signal void marker_received(XmppStream stream, string jid, string marker, string id);
+
+    public void send_marker(XmppStream stream, string jid, string message_id, string type_, string marker) {
+        Message.Stanza received_message = new Message.Stanza();
+        received_message.to = jid;
+        received_message.type_ = type_;
+        received_message.stanza.put_node(new StanzaNode.build(marker, NS_URI).add_self_xmlns().put_attribute("id", message_id));
+        Message.Module.get_module(stream).send_message(stream, received_message);
+    }
+
+    public static bool requests_marking(Message.Stanza message) {
+        StanzaNode markable_node = message.stanza.get_subnode("markable", NS_URI);
+        return markable_node != null;
+    }
+
+    public override void attach(XmppStream stream) {
+        Iq.Module.require(stream);
+        Message.Module.require(stream);
+        ServiceDiscovery.Module.require(stream);
+
+        ServiceDiscovery.Module.get_module(stream).add_feature(stream, NS_URI);
+        Message.Module.get_module(stream).pre_send_message.connect(on_pre_send_message);
+        Message.Module.get_module(stream).received_message.connect(on_received_message);
+    }
+
+    public override void detach(XmppStream stream) {
+        Message.Module.get_module(stream).pre_send_message.disconnect(on_pre_send_message);
+        Message.Module.get_module(stream).received_message.disconnect(on_received_message);
+    }
+
+    public static Module? get_module(XmppStream stream) {
+        return (Module?) stream.get_module(NS_URI, ID);
+    }
+
+    public static void require(XmppStream stream) {
+        if (get_module(stream) == null) stream.add_module(new ChatMarkers.Module());
+    }
+
+    public override string get_ns() { return NS_URI; }
+    public override string get_id() { return ID; }
+
+    private void on_received_message(XmppStream stream, Message.Stanza message) {
+        if (message.type_ != Message.Stanza.TYPE_CHAT) return;
+        if (requests_marking(message)) {
+            send_marker(stream, message.from, message.id, message.type_, MARKER_RECEIVED);
+            return;
+        }
+        ArrayList<StanzaNode> nodes = message.stanza.get_all_subnodes();
+        foreach (StanzaNode node in nodes) {
+            if (node.ns_uri == NS_URI && node.name in MARKERS) {
+                marker_received(stream, message.from, node.name, node.get_attribute("id", NS_URI));
+            }
+        }
+    }
+
+    private void on_pre_send_message(XmppStream stream, Message.Stanza message) {
+        StanzaNode? received_node = message.stanza.get_subnode("received", NS_URI);
+        if (received_node != null) return;
+        if (message.body == null) return;
+        if (message.type_ != Message.Stanza.TYPE_CHAT) return;
+        message.stanza.put_node(new StanzaNode.build("markable", NS_URI).add_self_xmlns());
+    }
+}
+
+}