From 8470e5251685cf8bc388d5ef9661b55950d2037a Mon Sep 17 00:00:00 2001
From: Madhur Garg <madhurgarg96@gmail.com>
Date: Tue, 9 Jul 2019 16:05:25 +0530
Subject: Added check for MAM support in disco#info

---
 poezio/mam.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/poezio/mam.py b/poezio/mam.py
index 7600f137..7716e851 100644
--- a/poezio/mam.py
+++ b/poezio/mam.py
@@ -41,6 +41,10 @@ async def query(self, remote_jid, start, end, top):
     self.start_date = start
     self.end_date = end
     text_buffer = self._text_buffer
+    def callback(iq):
+        if 'urn:xmpp:mam:2' not in iq['disco_info'].get_features():
+            return self.core.information("This MUC doesn't support MAM.", "Error")
+    self.core.xmpp.plugin['xep_0030'].get_info(jid=remote_jid, callback=callback)
     results = self.core.xmpp['xep_0313'].retrieve(jid=self.remote_jid,
     iterator=True, reverse=top, start=self.start_date, end=self.end_date)
     msg_count = 0
-- 
cgit v1.2.3