Implement MUC MAM

profanity.doap

@@ -360,7 +360,7 @@
         <xmpp:status>partial</xmpp:status>
         <xmpp:version>1.1.0</xmpp:version>
         <xmpp:since>0.14.0</xmpp:since>
-        <xmpp:note xml:lang='en'>No MUC MAM</xmpp:note>
+        <xmpp:note xml:lang='en'>MAM since 0.14.0. MUC MAM since DEV</xmpp:note>
       </xmpp:SupportedXep>
     </implements>
 

src/command/cmd_ac.c

@@ -112,6 +112,7 @@ static char* _strophe_autocomplete(ProfWin* window, const char* const input, gbo
 static char* _stamp_autocomplete(ProfWin* window, const char* const input, gboolean previous);
 static char* _adhoc_cmd_autocomplete(ProfWin* window, const char* const input, gboolean previous);
 static char* _vcard_autocomplete(ProfWin* window, const char* const input, gboolean previous);
+static char* _mam_autocomplete(ProfWin* window, const char* const input, gboolean previous);
 
 static char* _script_autocomplete_func(const char* const prefix, gboolean previous, void* context);
 
@@ -276,6 +277,7 @@ static Autocomplete vcard_name_ac;
 static Autocomplete vcard_set_param_ac;
 static Autocomplete vcard_togglable_param_ac;
 static Autocomplete vcard_address_type_ac;
+static Autocomplete mam_ac;
 
 static char* last_filepath_input = NULL;
 
@@ -434,6 +436,7 @@ static Autocomplete* all_acs[] = {
     &vcard_set_param_ac,
     &vcard_togglable_param_ac,
     &vcard_address_type_ac,
+    &mam_ac,
 };
 
 static GHashTable* ac_funcs = NULL;
@@ -1345,6 +1348,10 @@ cmd_ac_init(void)
     autocomplete_add(vcard_address_type_ac, "domestic");
     autocomplete_add(vcard_address_type_ac, "international");
 
+    autocomplete_add(mam_ac, "on");
+    autocomplete_add(mam_ac, "off");
+    autocomplete_add(mam_ac, "verify");
+
     if (ac_funcs != NULL)
         g_hash_table_destroy(ac_funcs);
     ac_funcs = g_hash_table_new(g_str_hash, g_str_equal);
@@ -1376,6 +1383,7 @@ cmd_ac_init(void)
     g_hash_table_insert(ac_funcs, "/lastactivity", _lastactivity_autocomplete);
     g_hash_table_insert(ac_funcs, "/log", _log_autocomplete);
     g_hash_table_insert(ac_funcs, "/logging", _logging_autocomplete);
+    g_hash_table_insert(ac_funcs, "/mam", _mam_autocomplete);
     g_hash_table_insert(ac_funcs, "/privacy", _privacy_autocomplete);
     g_hash_table_insert(ac_funcs, "/mood", _mood_autocomplete);
     g_hash_table_insert(ac_funcs, "/notify", _notify_autocomplete);
@@ -1771,6 +1779,16 @@ _spellcheck_autocomplete(ProfWin* window, const char* const input, gboolean prev
     return result;
 }
 
+static char*
+_mam_autocomplete(ProfWin* window, const char* const input, gboolean previous)
+{
+    char* result = NULL;
+
+    result = autocomplete_param_with_ac(input, "/mam", mam_ac, TRUE, previous);
+
+    return result;
+}
+
 static char*
 _cmd_ac_complete_params(ProfWin* window, const char* const input, gboolean previous)
 {
@@ -1781,7 +1799,7 @@ _cmd_ac_complete_params(ProfWin* window, const char* const input, gboolean previ
     // autocomplete boolean settings
     gchar* boolean_choices[] = { "/beep", "/states", "/outtype", "/flash", "/splash",
                                  "/history", "/vercheck", "/privileges", "/wrap",
-                                 "/carbons", "/slashguard", "/mam", "/silence" };
+                                 "/carbons", "/slashguard", "/silence" };
 
     for (size_t i = 0; i < ARRAY_SIZE(boolean_choices); i++) {
         result = autocomplete_param_with_func(input, boolean_choices[i], prefs_autocomplete_boolean_choice, previous, NULL);

src/command/cmd_defs.c

@@ -2615,19 +2615,21 @@ static const struct cmd_t command_defs[] = {
     },
 
     { CMD_PREAMBLE("/mam",
-                   parse_args, 1, 1, &cons_mam_setting)
-      CMD_MAINFUNC(cmd_mam)
-      CMD_TAGS(
-              CMD_TAG_CHAT)
-      CMD_SYN(
-              "/mam <on>|<off>")
-      CMD_DESC(
-              "Enable/Disable Message Archive Management (XEP-0313) "
-              "Currently MAM in groupchats (MUCs) is not supported. "
-              "Use the PG UP key to load more history.")
-      CMD_ARGS(
-              { "on|off", "Enable or disable MAM" })
-    },
+                    parse_args, 1, 1, &cons_mam_setting)
+       CMD_MAINFUNC(cmd_mam)
+       CMD_TAGS(
+               CMD_TAG_CHAT,
+               CMD_TAG_GROUPCHAT)
+       CMD_SYN(
+               "/mam <on>|<off>",
+               "/mam verify")
+       CMD_DESC(
+               "Enable or disable Message Archive Management (XEP-0313), or verify loaded history. "
+               "Use the PG UP key to load more history.")
+       CMD_ARGS(
+               { "on|off", "Enable or disable MAM" },
+               { "verify", "Verify history between the oldest and newest loaded messages in the current window and fetch any missing ones from the server" })
+     },
 
     { CMD_PREAMBLE("/changepassword",
                    parse_args, 0, 0, NULL)

src/command/cmd_funcs.c

@@ -9648,6 +9648,61 @@ cmd_executable_vcard_photo(ProfWin* window, const char* const command, gchar** a
 gboolean
 cmd_mam(ProfWin* window, const char* const command, gchar** args)
 {
+    if (g_strcmp0(args[0], "verify") == 0) {
+        if (connection_get_status() != JABBER_CONNECTED) {
+            cons_show("You are not currently connected.");
+            return TRUE;
+        }
+        if (window->type != WIN_CHAT && window->type != WIN_MUC) {
+            cons_show("MAM verification can only be performed in chat or groupchat windows.");
+            return TRUE;
+        }
+        if (!prefs_get_boolean(PREF_MAM)) {
+            cons_show("Message Archive Management (MAM) is currently disabled.");
+            return TRUE;
+        }
+        if (!connection_supports(XMPP_FEATURE_MAM2)) {
+            cons_show("Your server does not support Message Archive Management (MAM2).");
+            return TRUE;
+        }
+
+        unsigned int buf_size = buffer_size(window->layout->buffer);
+        if (buf_size == 0) {
+            cons_show("No loaded history in this window to verify.");
+            return TRUE;
+        }
+
+        ProfBuffEntry* first_msg = NULL;
+        ProfBuffEntry* last_msg = NULL;
+
+        for (unsigned int i = 0; i < buf_size; i++) {
+            ProfBuffEntry* entry = buffer_get_entry(window->layout->buffer, i);
+            if (entry && entry->message && entry->message[0] != '-') {
+                if (!first_msg) {
+                    first_msg = entry;
+                }
+                last_msg = entry;
+            }
+        }
+
+        if (!first_msg || !last_msg) {
+            cons_show("No messages found in the active scrollback buffer.");
+            return TRUE;
+        }
+
+        GDateTime* utc_start = g_date_time_to_timezone(first_msg->time, g_time_zone_new_utc());
+        GDateTime* utc_end = g_date_time_to_timezone(last_msg->time, g_time_zone_new_utc());
+        auto_gchar gchar* start_str = g_date_time_format(utc_start, "%FT%T.%f%:z");
+        auto_gchar gchar* end_str = g_date_time_format(utc_end, "%FT%T.%f%:z");
+        g_date_time_unref(utc_start);
+        g_date_time_unref(utc_end);
+
+        cons_show("Verifying history from %s to %s...", start_str, end_str);
+        iq_mam_verify_request(window, start_str, end_str);
+
+        return TRUE;
+    }
+
     _cmd_set_boolean_preference(args[0], "Message Archive Management", PREF_MAM);
 
     return TRUE;