Added export function for mam module

diff --git a/src/ejd2sql.erl b/src/ejd2sql.erl
index c775cd7bfab4073ac79cabb2d0697ffb97c808e3..343c3518a370803a499a653169d755a632c0ea4c 100644 (file)
--- a/src/ejd2sql.erl
+++ b/src/ejd2sql.erl
@@ -53,6 +53,7 @@ modules() ->
      mod_caps,
      mod_irc,
      mod_last,
+     mod_mam,
      mod_muc,
      mod_offline,
      mod_privacy,

diff --git a/src/mod_mam.erl b/src/mod_mam.erl
index 947fb5fbeee0631c0251df44063186c9950f62cb..c84cb907b0969b55661e53e0cf0a6564aae93174 100644 (file)
--- a/src/mod_mam.erl
+++ b/src/mod_mam.erl
@@ -37,7 +37,7 @@
         remove_user/2, remove_room/3, mod_opt_type/1, muc_process_iq/2,
         muc_filter_message/3, message_is_archived/3, delete_old_messages/2,
         get_commands_spec/0, msg_to_el/4, get_room_config/4, set_room_option/3,
-        offline_message/1]).
+        offline_message/1, export/1]).
 
 -include("xmpp.hrl").
 -include("logger.hrl").
@@ -457,6 +457,10 @@ delete_old_messages(TypeBin, Days) when TypeBin == <<"chat">>;
 delete_old_messages(_TypeBin, _Days) ->
     unsupported_type.
 
+export(LServer) ->
+    Mod = gen_mod:db_mod(LServer, ?MODULE),
+    Mod:export(LServer).
+
 %%%===================================================================
 %%% Internal functions
 %%%===================================================================

diff --git a/src/mod_mam_sql.erl b/src/mod_mam_sql.erl
index d2fd48a9fff6c80543aced38c0ccb7aaf671639a..7ae3bb87fe26ca9e7e31c4a9a8619be74cd4cbe0 100644 (file)
--- a/src/mod_mam_sql.erl
+++ b/src/mod_mam_sql.erl
@@ -30,7 +30,7 @@
 
 %% API
 -export([init/2, remove_user/2, remove_room/3, delete_old_messages/3,
-        extended_fields/0, store/7, write_prefs/4, get_prefs/2, select/6]).
+        extended_fields/0, store/7, write_prefs/4, get_prefs/2, select/6, export/1]).
 
 -include_lib("stdlib/include/ms_transform.hrl").
 -include("xmpp.hrl").
@@ -181,6 +181,47 @@ select(LServer, JidRequestor, #jid{luser = LUser} = JidArchive,
            {[], false, 0}
     end.
 
+export(_Server) ->
+    [{archive_prefs,
+      fun(Host, #archive_prefs{us =
+                {LUser, LServer},
+                default = Default,
+                always = Always,
+                never = Never})
+          when LServer == Host ->
+                SDefault = erlang:atom_to_binary(Default, utf8),
+                SAlways = misc:term_to_expr(Always),
+                SNever = misc:term_to_expr(Never),
+                [?SQL("insert into archive_prefs (username, def, always, never) values"
+                "(%(LUser)s, %(SDefault)s, %(SAlways)s, %(SNever)s)")];
+          (_Host, _R) ->
+              []
+      end},
+     {archive_msg,
+      fun(Host, #archive_msg{us ={_LUser, LServer},
+                id = _ID, timestamp = TS, peer = Peer,
+                bare_peer = {PUser, PServer, <<>>},
+                type = Type, nick = Nick, packet = Pkt})
+          when LServer == Host ->
+                TStmp = now_to_usec(TS),
+                SUser = case Type of
+                      chat -> PUser;
+                      groupchat -> jid:to_string({PUser, PServer, <<>>})
+                    end,
+                BarePeer = jid:to_string(jid:tolower(jid:remove_resource(Peer))),
+                LPeer = jid:to_string(jid:tolower(Peer)),
+                XML = fxml:element_to_binary(Pkt),
+                Body = fxml:get_subtag_cdata(Pkt, <<"body">>),
+                SType = jlib:atom_to_binary(Type),
+                [?SQL("insert into archive (username, timestamp, "
+                                "peer, bare_peer, xml, txt, kind, nick) "
+                                "values (%(SUser)s, %(TStmp)d, %(LPeer)s, "
+                                "%(BarePeer)s, %(XML)s, %(Body)s, %(SType)s, "
+                                "%(Nick)s)")];
+         (_Host, _R) ->
+              []
+      end}].
+
 %%%===================================================================
 %%% Internal functions
 %%%===================================================================