# HG changeset patch
# User Kim Alvefur <zash@zash.se>
# Date 1631721358 -7200
# Node ID ecfffbbcbf4247a70be4be4a2eb122ad9f67b329
# Parent  05725276fac062caa8c82fc3c2fe6a07cb7d6669
mod_debug_omemo: Use archive single message retrieval method

Likely more efficient, where implemented.

diff -r 05725276fac0 -r ecfffbbcbf42 mod_debug_omemo/mod_debug_omemo.lua
--- a/mod_debug_omemo/mod_debug_omemo.lua	Wed Sep 15 17:39:37 2021 +0200
+++ b/mod_debug_omemo/mod_debug_omemo.lua	Wed Sep 15 17:55:58 2021 +0200
@@ -100,15 +100,24 @@
 	presence = "Contacts only";
 };
 
+local function get_message(username, message_id)
+	if mam.get then
+		return mam:get(username, message_id);
+	end
+	-- COMPAT
+	local message;
+	for _, result in mam:find(username, { key = message_id }) do
+		message = result;
+	end
+	return message;
+end
+
 local function render_message(event, path)
 	local username, message_id = path:match("^([^/]+)/(.+)$");
 	if not username then
 		return 400;
 	end
-	local message;
-	for _, result in mam:find(username, { key = message_id }) do
-		message = result;
-	end
+	local message = get_message(username, message_id);
 	if not message then
 		return 404;
 	end