using Gee; using Qlite; using Xmpp; using Xmpp.Xep; using Dino.Entities; public class Dino.FallbackBody : StreamInteractionModule, Object { public static ModuleIdentity IDENTITY = new ModuleIdentity("fallback-body"); public string id { get { return IDENTITY.id; } } private StreamInteractor stream_interactor; private Database db; private ReceivedMessageListener received_message_listener; public static void start(StreamInteractor stream_interactor, Database db) { FallbackBody m = new FallbackBody(stream_interactor, db); stream_interactor.add_module(m); } private FallbackBody(StreamInteractor stream_interactor, Database db) { this.stream_interactor = stream_interactor; this.db = db; this.received_message_listener = new ReceivedMessageListener(stream_interactor, db); stream_interactor.get_module(MessageProcessor.IDENTITY).received_pipeline.connect(received_message_listener); } private class ReceivedMessageListener : MessageListener { public string[] after_actions_const = new string[]{ "STORE" }; public override string action_group { get { return "Quote"; } } public override string[] after_actions { get { return after_actions_const; } } private StreamInteractor stream_interactor; private Database db; public ReceivedMessageListener(StreamInteractor stream_interactor, Database db) { this.stream_interactor = stream_interactor; this.db = db; } public override async bool run(Entities.Message message, Xmpp.MessageStanza stanza, Conversation conversation) { Gee.List fallbacks = Xep.FallbackIndication.get_fallbacks(stanza); if (fallbacks.is_empty) return false; foreach (var fallback in fallbacks) { if (fallback.ns_uri != Xep.Replies.NS_URI) continue; foreach (var location in fallback.locations) { db.body_meta.insert() .value(db.body_meta.message_id, message.id) .value(db.body_meta.info_type, Xep.FallbackIndication.NS_URI) .value(db.body_meta.info, fallback.ns_uri) .value(db.body_meta.from_char, location.from_char) .value(db.body_meta.to_char, location.to_char) .perform(); } message.set_fallbacks(fallbacks); } return false; } } }