2003-01-05 21:24:59 +01:00
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
%%% File : mod_offline.erl
|
2007-08-07 18:43:02 +02:00
|
|
|
%%% Author : Alexey Shchepin <alexey@process-one.net>
|
|
|
|
%%% Purpose : Store and manage offline messages in Mnesia database.
|
|
|
|
%%% Created : 5 Jan 2003 by Alexey Shchepin <alexey@process-one.net>
|
2007-12-24 13:58:05 +01:00
|
|
|
%%%
|
|
|
|
%%%
|
2008-09-16 16:39:57 +02:00
|
|
|
%%% ejabberd, Copyright (C) 2002-2008 ProcessOne
|
2007-12-24 13:58:05 +01:00
|
|
|
%%%
|
|
|
|
%%% This program is free software; you can redistribute it and/or
|
|
|
|
%%% modify it under the terms of the GNU General Public License as
|
|
|
|
%%% published by the Free Software Foundation; either version 2 of the
|
|
|
|
%%% License, or (at your option) any later version.
|
|
|
|
%%%
|
|
|
|
%%% This program is distributed in the hope that it will be useful,
|
|
|
|
%%% but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
%%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
%%% General Public License for more details.
|
|
|
|
%%%
|
|
|
|
%%% You should have received a copy of the GNU General Public License
|
|
|
|
%%% along with this program; if not, write to the Free Software
|
|
|
|
%%% Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
|
|
|
|
%%% 02111-1307 USA
|
|
|
|
%%%
|
2003-01-05 21:24:59 +01:00
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
|
|
|
|
-module(mod_offline).
|
2007-08-07 18:43:02 +02:00
|
|
|
-author('alexey@process-one.net').
|
2003-01-05 21:24:59 +01:00
|
|
|
|
2003-01-24 21:18:33 +01:00
|
|
|
-behaviour(gen_mod).
|
|
|
|
|
2005-06-20 05:18:13 +02:00
|
|
|
-export([start/2,
|
2007-08-13 19:28:26 +02:00
|
|
|
init/1,
|
2005-06-20 05:18:13 +02:00
|
|
|
stop/1,
|
2003-01-05 21:24:59 +01:00
|
|
|
store_packet/3,
|
2005-04-17 20:08:34 +02:00
|
|
|
resend_offline_messages/2,
|
|
|
|
pop_offline_messages/3,
|
2004-09-10 22:57:00 +02:00
|
|
|
remove_expired_messages/0,
|
2003-10-24 21:21:07 +02:00
|
|
|
remove_old_messages/1,
|
2007-08-23 02:51:54 +02:00
|
|
|
remove_user/2,
|
|
|
|
webadmin_page/3,
|
|
|
|
webadmin_user/4]).
|
2003-01-05 21:24:59 +01:00
|
|
|
|
2008-08-06 15:44:58 +02:00
|
|
|
-include_lib("exmpp/include/exmpp.hrl").
|
|
|
|
|
2004-09-10 22:57:00 +02:00
|
|
|
-include("ejabberd.hrl").
|
2008-07-17 17:33:50 +02:00
|
|
|
-include("web/ejabberd_http.hrl").
|
|
|
|
-include("web/ejabberd_web_admin.hrl").
|
2003-01-05 21:24:59 +01:00
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
-record(offline_msg, {us, timestamp, expire, from, to, packet}).
|
2003-01-05 21:24:59 +01:00
|
|
|
|
2003-02-13 20:39:13 +01:00
|
|
|
-define(PROCNAME, ejabberd_offline).
|
2004-08-22 23:54:14 +02:00
|
|
|
-define(OFFLINE_TABLE_LOCK_THRESHOLD, 1000).
|
2003-01-05 21:24:59 +01:00
|
|
|
|
2008-08-06 15:44:58 +02:00
|
|
|
% These are the namespace already declared by the stream opening. This is
|
|
|
|
% used at serialization time.
|
|
|
|
-define(DEFAULT_NS, ?NS_JABBER_CLIENT).
|
|
|
|
-define(PREFIXED_NS, [{?NS_XMPP, ?NS_XMPP_pfx}]).
|
|
|
|
|
2007-08-13 19:28:26 +02:00
|
|
|
start(Host, Opts) ->
|
2003-01-05 21:24:59 +01:00
|
|
|
mnesia:create_table(offline_msg,
|
|
|
|
[{disc_only_copies, [node()]},
|
|
|
|
{type, bag},
|
2003-02-13 20:39:13 +01:00
|
|
|
{attributes, record_info(fields, offline_msg)}]),
|
2004-09-10 22:57:00 +02:00
|
|
|
update_table(),
|
2005-06-20 05:18:13 +02:00
|
|
|
ejabberd_hooks:add(offline_message_hook, Host,
|
2004-08-08 21:07:55 +02:00
|
|
|
?MODULE, store_packet, 50),
|
2005-06-20 05:18:13 +02:00
|
|
|
ejabberd_hooks:add(resend_offline_messages_hook, Host,
|
2004-08-08 21:07:55 +02:00
|
|
|
?MODULE, pop_offline_messages, 50),
|
2005-06-20 05:18:13 +02:00
|
|
|
ejabberd_hooks:add(remove_user, Host,
|
2004-12-19 21:47:35 +01:00
|
|
|
?MODULE, remove_user, 50),
|
2007-05-12 18:28:34 +02:00
|
|
|
ejabberd_hooks:add(anonymous_purge_hook, Host,
|
|
|
|
?MODULE, remove_user, 50),
|
2007-08-23 02:51:54 +02:00
|
|
|
ejabberd_hooks:add(webadmin_page_host, Host,
|
|
|
|
?MODULE, webadmin_page, 50),
|
|
|
|
ejabberd_hooks:add(webadmin_user, Host,
|
|
|
|
?MODULE, webadmin_user, 50),
|
2007-08-20 20:02:12 +02:00
|
|
|
MaxOfflineMsgs = gen_mod:get_opt(user_max_messages, Opts, infinity),
|
2005-06-20 05:18:13 +02:00
|
|
|
register(gen_mod:get_module_proc(Host, ?PROCNAME),
|
2007-08-20 20:02:12 +02:00
|
|
|
spawn(?MODULE, init, [MaxOfflineMsgs])).
|
2003-02-13 20:39:13 +01:00
|
|
|
|
2007-08-20 20:02:12 +02:00
|
|
|
%% MaxOfflineMsgs is either infinity of integer > 0
|
2007-08-13 19:28:26 +02:00
|
|
|
init(infinity) ->
|
|
|
|
loop(infinity);
|
2007-08-20 20:02:12 +02:00
|
|
|
init(MaxOfflineMsgs)
|
|
|
|
when integer(MaxOfflineMsgs), MaxOfflineMsgs > 0 ->
|
|
|
|
loop(MaxOfflineMsgs).
|
2003-02-13 20:39:13 +01:00
|
|
|
|
2007-08-20 20:02:12 +02:00
|
|
|
loop(MaxOfflineMsgs) ->
|
2003-02-13 20:39:13 +01:00
|
|
|
receive
|
2007-08-07 18:43:02 +02:00
|
|
|
#offline_msg{us=US} = Msg ->
|
|
|
|
Msgs = receive_all(US, [Msg]),
|
2004-08-22 23:54:14 +02:00
|
|
|
Len = length(Msgs),
|
2003-02-13 20:39:13 +01:00
|
|
|
F = fun() ->
|
2007-08-08 11:48:41 +02:00
|
|
|
%% Only count messages if needed:
|
2007-08-20 20:02:12 +02:00
|
|
|
Count = if MaxOfflineMsgs =/= infinity ->
|
2007-08-08 11:48:41 +02:00
|
|
|
Len + p1_mnesia:count_records(
|
|
|
|
offline_msg,
|
|
|
|
#offline_msg{us=US, _='_'});
|
|
|
|
true ->
|
|
|
|
0
|
|
|
|
end,
|
2004-08-22 23:54:14 +02:00
|
|
|
if
|
2007-08-20 20:02:12 +02:00
|
|
|
Count > MaxOfflineMsgs ->
|
2007-08-13 12:27:28 +02:00
|
|
|
discard_warn_sender(Msgs);
|
2004-08-22 23:54:14 +02:00
|
|
|
true ->
|
2007-08-07 18:43:02 +02:00
|
|
|
if
|
|
|
|
Len >= ?OFFLINE_TABLE_LOCK_THRESHOLD ->
|
|
|
|
mnesia:write_lock_table(offline_msg);
|
|
|
|
true ->
|
|
|
|
ok
|
|
|
|
end,
|
|
|
|
lists:foreach(fun(M) ->
|
|
|
|
mnesia:write(M)
|
|
|
|
end, Msgs)
|
|
|
|
end
|
2003-02-13 20:39:13 +01:00
|
|
|
end,
|
|
|
|
mnesia:transaction(F),
|
2007-08-20 20:02:12 +02:00
|
|
|
loop(MaxOfflineMsgs);
|
2003-02-13 20:39:13 +01:00
|
|
|
_ ->
|
2007-08-20 20:02:12 +02:00
|
|
|
loop(MaxOfflineMsgs)
|
2003-02-13 20:39:13 +01:00
|
|
|
end.
|
|
|
|
|
2007-08-07 18:43:02 +02:00
|
|
|
receive_all(US, Msgs) ->
|
2003-02-13 20:39:13 +01:00
|
|
|
receive
|
2007-08-07 18:43:02 +02:00
|
|
|
#offline_msg{us=US} = Msg ->
|
|
|
|
receive_all(US, [Msg | Msgs])
|
2003-02-13 20:39:13 +01:00
|
|
|
after 0 ->
|
|
|
|
Msgs
|
|
|
|
end.
|
|
|
|
|
2003-01-05 21:24:59 +01:00
|
|
|
|
2005-06-20 05:18:13 +02:00
|
|
|
stop(Host) ->
|
|
|
|
ejabberd_hooks:delete(offline_message_hook, Host,
|
2004-08-08 21:07:55 +02:00
|
|
|
?MODULE, store_packet, 50),
|
2005-06-20 05:18:13 +02:00
|
|
|
ejabberd_hooks:delete(resend_offline_messages_hook, Host,
|
2004-08-08 21:07:55 +02:00
|
|
|
?MODULE, pop_offline_messages, 50),
|
2005-06-20 05:18:13 +02:00
|
|
|
ejabberd_hooks:delete(remove_user, Host,
|
2004-12-19 21:47:35 +01:00
|
|
|
?MODULE, remove_user, 50),
|
2007-05-12 18:28:34 +02:00
|
|
|
ejabberd_hooks:delete(anonymous_purge_hook, Host,
|
|
|
|
?MODULE, remove_user, 50),
|
2007-08-23 02:51:54 +02:00
|
|
|
ejabberd_hooks:delete(webadmin_page_host, Host,
|
|
|
|
?MODULE, webadmin_page, 50),
|
|
|
|
ejabberd_hooks:delete(webadmin_user, Host,
|
|
|
|
?MODULE, webadmin_user, 50),
|
2005-06-20 05:18:13 +02:00
|
|
|
Proc = gen_mod:get_module_proc(Host, ?PROCNAME),
|
|
|
|
exit(whereis(Proc), stop),
|
|
|
|
{wait, Proc}.
|
2003-01-05 21:24:59 +01:00
|
|
|
|
|
|
|
store_packet(From, To, Packet) ->
|
2008-08-06 15:44:58 +02:00
|
|
|
Type = exmpp_stanza:get_type(Packet),
|
2004-08-08 21:07:55 +02:00
|
|
|
if
|
2005-12-07 14:01:33 +01:00
|
|
|
(Type /= "error") and (Type /= "groupchat") and
|
|
|
|
(Type /= "headline") ->
|
2004-08-08 21:07:55 +02:00
|
|
|
case check_event(From, To, Packet) of
|
|
|
|
true ->
|
2008-08-06 15:44:58 +02:00
|
|
|
#jid{lnode = LUser, ldomain = LServer} = To,
|
2004-08-08 21:07:55 +02:00
|
|
|
TimeStamp = now(),
|
2008-08-06 15:44:58 +02:00
|
|
|
Expire = find_x_expire(TimeStamp, Packet#xmlel.children),
|
|
|
|
% XXX OLD FORMAT: Packet is stored in the old format.
|
|
|
|
PacketOld = exmpp_xml:xmlel_to_xmlelement(Packet,
|
|
|
|
[?DEFAULT_NS], ?PREFIXED_NS),
|
|
|
|
gen_mod:get_module_proc(To#jid.ldomain, ?PROCNAME) !
|
2005-06-30 13:29:04 +02:00
|
|
|
#offline_msg{us = {LUser, LServer},
|
|
|
|
timestamp = TimeStamp,
|
|
|
|
expire = Expire,
|
|
|
|
from = From,
|
|
|
|
to = To,
|
2008-08-06 15:44:58 +02:00
|
|
|
packet = PacketOld},
|
2004-08-08 21:07:55 +02:00
|
|
|
stop;
|
|
|
|
_ ->
|
|
|
|
ok
|
|
|
|
end;
|
2003-01-05 21:24:59 +01:00
|
|
|
true ->
|
|
|
|
ok
|
|
|
|
end.
|
|
|
|
|
|
|
|
check_event(From, To, Packet) ->
|
2008-08-06 15:44:58 +02:00
|
|
|
case find_x_event(Packet#xmlel.children) of
|
2003-01-05 21:24:59 +01:00
|
|
|
false ->
|
|
|
|
true;
|
|
|
|
El ->
|
2008-08-06 15:44:58 +02:00
|
|
|
case exmpp_xml:get_element(El, 'id') of
|
|
|
|
undefined ->
|
|
|
|
case exmpp_xml:get_element(El, 'offline') of
|
|
|
|
undefined ->
|
2003-01-05 21:24:59 +01:00
|
|
|
true;
|
|
|
|
_ ->
|
2008-08-06 15:44:58 +02:00
|
|
|
ID = case exmpp_stanza:get_id(Packet) of
|
|
|
|
undefined ->
|
|
|
|
#xmlel{ns = ?NS_MESSAGE_EVENT, name = 'id'};
|
2003-01-05 21:24:59 +01:00
|
|
|
S ->
|
2008-08-06 15:44:58 +02:00
|
|
|
#xmlel{ns = ?NS_MESSAGE_EVENT, name = 'id',
|
|
|
|
children = [#xmlcdata{cdata =
|
|
|
|
list_to_binary(S)}]}
|
2003-01-05 21:24:59 +01:00
|
|
|
end,
|
2008-08-06 15:44:58 +02:00
|
|
|
X = #xmlel{ns = ?NS_MESSAGE_EVENT, name = 'x', children =
|
|
|
|
[ID, #xmlel{ns = ?NS_MESSAGE_EVENT, name = 'offline'}]},
|
2003-01-05 21:24:59 +01:00
|
|
|
ejabberd_router:route(
|
2008-08-06 15:44:58 +02:00
|
|
|
To, From, exmpp_xml:set_children(Packet, [X])),
|
2003-01-05 21:24:59 +01:00
|
|
|
true
|
2005-09-29 03:04:24 +02:00
|
|
|
end;
|
2003-01-05 21:24:59 +01:00
|
|
|
_ ->
|
|
|
|
false
|
|
|
|
end
|
|
|
|
end.
|
|
|
|
|
|
|
|
find_x_event([]) ->
|
|
|
|
false;
|
2008-08-06 15:44:58 +02:00
|
|
|
find_x_event([#xmlel{ns = ?NS_MESSAGE_EVENT} = El | _Els]) ->
|
|
|
|
El;
|
|
|
|
find_x_event([_ | Els]) ->
|
|
|
|
find_x_event(Els).
|
2003-01-05 21:24:59 +01:00
|
|
|
|
2004-09-10 22:57:00 +02:00
|
|
|
find_x_expire(_, []) ->
|
|
|
|
never;
|
2008-08-06 15:44:58 +02:00
|
|
|
find_x_expire(TimeStamp, [#xmlel{ns = ?NS_MESSAGE_EXPIRE} = El | _Els]) ->
|
|
|
|
Val = exmpp_xml:get_attribute(El, 'seconds', ""),
|
|
|
|
case catch list_to_integer(Val) of
|
|
|
|
{'EXIT', _} ->
|
|
|
|
never;
|
|
|
|
Int when Int > 0 ->
|
|
|
|
{MegaSecs, Secs, MicroSecs} = TimeStamp,
|
|
|
|
S = MegaSecs * 1000000 + Secs + Int,
|
|
|
|
MegaSecs1 = S div 1000000,
|
|
|
|
Secs1 = S rem 1000000,
|
|
|
|
{MegaSecs1, Secs1, MicroSecs};
|
2004-09-10 22:57:00 +02:00
|
|
|
_ ->
|
2008-08-06 15:44:58 +02:00
|
|
|
never
|
|
|
|
end;
|
|
|
|
find_x_expire(TimeStamp, [_ | Els]) ->
|
|
|
|
find_x_expire(TimeStamp, Els).
|
2004-09-10 22:57:00 +02:00
|
|
|
|
2003-01-05 21:24:59 +01:00
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
resend_offline_messages(User, Server) ->
|
2008-08-06 15:44:58 +02:00
|
|
|
LUser = exmpp_stringprep:nodeprep(User),
|
|
|
|
LServer = exmpp_stringprep:nameprep(Server),
|
2005-04-17 20:08:34 +02:00
|
|
|
US = {LUser, LServer},
|
2003-01-05 21:24:59 +01:00
|
|
|
F = fun() ->
|
2005-04-17 20:08:34 +02:00
|
|
|
Rs = mnesia:wread({offline_msg, US}),
|
|
|
|
mnesia:delete({offline_msg, US}),
|
2003-01-05 21:24:59 +01:00
|
|
|
Rs
|
|
|
|
end,
|
|
|
|
case mnesia:transaction(F) of
|
|
|
|
{atomic, Rs} ->
|
|
|
|
lists:foreach(
|
|
|
|
fun(R) ->
|
2008-08-06 15:44:58 +02:00
|
|
|
Packet = case R#offline_msg.packet of
|
|
|
|
#xmlelement{} = P ->
|
|
|
|
exmpp_xml:xmlelement_to_xmlel(P,
|
|
|
|
[?DEFAULT_NS], ?PREFIXED_NS);
|
|
|
|
#xmlel{} = P ->
|
|
|
|
P
|
|
|
|
end,
|
|
|
|
% XXX OLD FORMAT: Convert From & To.
|
2003-01-05 21:24:59 +01:00
|
|
|
ejabberd_sm !
|
|
|
|
{route,
|
2008-08-06 15:44:58 +02:00
|
|
|
jlib:from_old_jid(R#offline_msg.from),
|
|
|
|
jlib:from_old_jid(R#offline_msg.to),
|
|
|
|
exmpp_xml:append_child(Packet,
|
|
|
|
jlib:timestamp_to_xml(
|
2003-01-11 21:25:11 +01:00
|
|
|
calendar:now_to_universal_time(
|
2008-08-06 15:44:58 +02:00
|
|
|
R#offline_msg.timestamp)))}
|
2003-01-05 21:24:59 +01:00
|
|
|
end,
|
|
|
|
lists:keysort(#offline_msg.timestamp, Rs));
|
|
|
|
_ ->
|
|
|
|
ok
|
|
|
|
end.
|
2003-01-26 21:16:53 +01:00
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
pop_offline_messages(Ls, User, Server) ->
|
2008-08-06 15:44:58 +02:00
|
|
|
LUser = exmpp_stringprep:nodeprep(User),
|
|
|
|
LServer = exmpp_stringprep:nameprep(Server),
|
2005-04-17 20:08:34 +02:00
|
|
|
US = {LUser, LServer},
|
2004-01-18 21:42:09 +01:00
|
|
|
F = fun() ->
|
2005-04-17 20:08:34 +02:00
|
|
|
Rs = mnesia:wread({offline_msg, US}),
|
|
|
|
mnesia:delete({offline_msg, US}),
|
2004-01-18 21:42:09 +01:00
|
|
|
Rs
|
|
|
|
end,
|
|
|
|
case mnesia:transaction(F) of
|
|
|
|
{atomic, Rs} ->
|
2004-09-10 22:57:00 +02:00
|
|
|
TS = now(),
|
|
|
|
Ls ++ lists:map(
|
|
|
|
fun(R) ->
|
2008-08-06 15:44:58 +02:00
|
|
|
Packet = case R#offline_msg.packet of
|
|
|
|
#xmlelement{} = P ->
|
|
|
|
exmpp_xml:xmlelement_to_xmlel(P,
|
|
|
|
[?DEFAULT_NS], ?PREFIXED_NS);
|
|
|
|
#xmlel{} = P ->
|
|
|
|
P
|
|
|
|
end,
|
|
|
|
% XXX OLD FORMAT: Convert From & To.
|
2004-09-10 22:57:00 +02:00
|
|
|
{route,
|
2008-08-06 15:44:58 +02:00
|
|
|
jlib:from_old_jid(R#offline_msg.from),
|
|
|
|
jlib:from_old_jid(R#offline_msg.to),
|
|
|
|
exmpp_xml:append_child(Packet,
|
|
|
|
jlib:timestamp_to_xml(
|
2004-09-10 22:57:00 +02:00
|
|
|
calendar:now_to_universal_time(
|
2008-08-06 15:44:58 +02:00
|
|
|
R#offline_msg.timestamp)))}
|
2004-09-10 22:57:00 +02:00
|
|
|
end,
|
|
|
|
lists:filter(
|
|
|
|
fun(R) ->
|
|
|
|
case R#offline_msg.expire of
|
|
|
|
never ->
|
|
|
|
true;
|
|
|
|
TimeStamp ->
|
|
|
|
TS < TimeStamp
|
|
|
|
end
|
|
|
|
end,
|
|
|
|
lists:keysort(#offline_msg.timestamp, Rs)));
|
2004-01-18 21:42:09 +01:00
|
|
|
_ ->
|
2004-08-08 21:07:55 +02:00
|
|
|
Ls
|
2004-01-18 21:42:09 +01:00
|
|
|
end.
|
|
|
|
|
2004-09-10 22:57:00 +02:00
|
|
|
remove_expired_messages() ->
|
|
|
|
TimeStamp = now(),
|
|
|
|
F = fun() ->
|
|
|
|
mnesia:write_lock_table(offline_msg),
|
|
|
|
mnesia:foldl(
|
|
|
|
fun(Rec, _Acc) ->
|
|
|
|
case Rec#offline_msg.expire of
|
|
|
|
never ->
|
|
|
|
ok;
|
|
|
|
TS ->
|
|
|
|
if
|
|
|
|
TS < TimeStamp ->
|
|
|
|
mnesia:delete_object(Rec);
|
|
|
|
true ->
|
|
|
|
ok
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end, ok, offline_msg)
|
|
|
|
end,
|
|
|
|
mnesia:transaction(F).
|
|
|
|
|
2003-10-24 21:21:07 +02:00
|
|
|
remove_old_messages(Days) ->
|
|
|
|
{MegaSecs, Secs, _MicroSecs} = now(),
|
|
|
|
S = MegaSecs * 1000000 + Secs - 60 * 60 * 24 * Days,
|
|
|
|
MegaSecs1 = S div 1000000,
|
|
|
|
Secs1 = S rem 1000000,
|
|
|
|
TimeStamp = {MegaSecs1, Secs1, 0},
|
|
|
|
F = fun() ->
|
2004-08-08 21:07:55 +02:00
|
|
|
mnesia:write_lock_table(offline_msg),
|
2003-10-24 21:21:07 +02:00
|
|
|
mnesia:foldl(
|
|
|
|
fun(#offline_msg{timestamp = TS} = Rec, _Acc)
|
|
|
|
when TS < TimeStamp ->
|
|
|
|
mnesia:delete_object(Rec);
|
|
|
|
(_Rec, _Acc) -> ok
|
|
|
|
end, ok, offline_msg)
|
|
|
|
end,
|
|
|
|
mnesia:transaction(F).
|
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
remove_user(User, Server) ->
|
2008-08-06 15:44:58 +02:00
|
|
|
LUser = exmpp_stringprep:nodeprep(User),
|
|
|
|
LServer = exmpp_stringprep:nameprep(Server),
|
2005-04-17 20:08:34 +02:00
|
|
|
US = {LUser, LServer},
|
2003-01-26 21:16:53 +01:00
|
|
|
F = fun() ->
|
2005-04-17 20:08:34 +02:00
|
|
|
mnesia:delete({offline_msg, US})
|
2003-01-26 21:16:53 +01:00
|
|
|
end,
|
|
|
|
mnesia:transaction(F).
|
2004-09-10 22:57:00 +02:00
|
|
|
|
|
|
|
update_table() ->
|
|
|
|
Fields = record_info(fields, offline_msg),
|
|
|
|
case mnesia:table_info(offline_msg, attributes) of
|
|
|
|
Fields ->
|
|
|
|
ok;
|
2005-04-17 20:08:34 +02:00
|
|
|
[user, timestamp, expire, from, to, packet] ->
|
|
|
|
?INFO_MSG("Converting offline_msg table from "
|
|
|
|
"{user, timestamp, expire, from, to, packet} format", []),
|
|
|
|
Host = ?MYNAME,
|
|
|
|
{atomic, ok} = mnesia:create_table(
|
|
|
|
mod_offline_tmp_table,
|
|
|
|
[{disc_only_copies, [node()]},
|
|
|
|
{type, bag},
|
|
|
|
{local_content, true},
|
|
|
|
{record_name, offline_msg},
|
|
|
|
{attributes, record_info(fields, offline_msg)}]),
|
|
|
|
mnesia:transform_table(offline_msg, ignore, Fields),
|
|
|
|
F1 = fun() ->
|
|
|
|
mnesia:write_lock_table(mod_offline_tmp_table),
|
|
|
|
mnesia:foldl(
|
2008-08-06 15:44:58 +02:00
|
|
|
fun(#offline_msg{us = U, packet = P} = R, _) ->
|
|
|
|
New_R = R#offline_msg{
|
|
|
|
us = {U, Host},
|
|
|
|
packet = exmpp_xml:xmlelement_to_xmlel(P,
|
|
|
|
[?DEFAULT_NS], ?PREFIXED_NS)
|
|
|
|
},
|
2005-04-17 20:08:34 +02:00
|
|
|
mnesia:dirty_write(
|
|
|
|
mod_offline_tmp_table,
|
2008-08-06 15:44:58 +02:00
|
|
|
New_R)
|
2005-04-17 20:08:34 +02:00
|
|
|
end, ok, offline_msg)
|
|
|
|
end,
|
|
|
|
mnesia:transaction(F1),
|
|
|
|
mnesia:clear_table(offline_msg),
|
|
|
|
F2 = fun() ->
|
|
|
|
mnesia:write_lock_table(offline_msg),
|
|
|
|
mnesia:foldl(
|
|
|
|
fun(R, _) ->
|
|
|
|
mnesia:dirty_write(R)
|
|
|
|
end, ok, mod_offline_tmp_table)
|
|
|
|
end,
|
|
|
|
mnesia:transaction(F2),
|
|
|
|
mnesia:delete_table(mod_offline_tmp_table);
|
2004-09-10 22:57:00 +02:00
|
|
|
[user, timestamp, from, to, packet] ->
|
|
|
|
?INFO_MSG("Converting offline_msg table from "
|
|
|
|
"{user, timestamp, from, to, packet} format", []),
|
2005-04-17 20:08:34 +02:00
|
|
|
Host = ?MYNAME,
|
|
|
|
{atomic, ok} = mnesia:create_table(
|
|
|
|
mod_offline_tmp_table,
|
|
|
|
[{disc_only_copies, [node()]},
|
|
|
|
{type, bag},
|
|
|
|
{local_content, true},
|
|
|
|
{record_name, offline_msg},
|
|
|
|
{attributes, record_info(fields, offline_msg)}]),
|
2004-09-10 22:57:00 +02:00
|
|
|
mnesia:transform_table(
|
|
|
|
offline_msg,
|
|
|
|
fun({_, U, TS, F, T, P}) ->
|
2008-08-06 15:44:58 +02:00
|
|
|
Expire = find_x_expire(TS, P#xmlelement.children),
|
2005-04-17 20:08:34 +02:00
|
|
|
#offline_msg{us = U,
|
2004-09-10 22:57:00 +02:00
|
|
|
timestamp = TS,
|
|
|
|
expire = Expire,
|
|
|
|
from = F,
|
|
|
|
to = T,
|
|
|
|
packet = P}
|
2005-04-17 20:08:34 +02:00
|
|
|
end, Fields),
|
|
|
|
F1 = fun() ->
|
|
|
|
mnesia:write_lock_table(mod_offline_tmp_table),
|
|
|
|
mnesia:foldl(
|
|
|
|
fun(#offline_msg{us = U} = R, _) ->
|
|
|
|
mnesia:dirty_write(
|
|
|
|
mod_offline_tmp_table,
|
|
|
|
R#offline_msg{us = {U, Host}})
|
|
|
|
end, ok, offline_msg)
|
|
|
|
end,
|
|
|
|
mnesia:transaction(F1),
|
|
|
|
mnesia:clear_table(offline_msg),
|
|
|
|
F2 = fun() ->
|
|
|
|
mnesia:write_lock_table(offline_msg),
|
|
|
|
mnesia:foldl(
|
|
|
|
fun(R, _) ->
|
|
|
|
mnesia:dirty_write(R)
|
|
|
|
end, ok, mod_offline_tmp_table)
|
|
|
|
end,
|
|
|
|
mnesia:transaction(F2),
|
|
|
|
mnesia:delete_table(mod_offline_tmp_table);
|
2004-09-10 22:57:00 +02:00
|
|
|
_ ->
|
|
|
|
?INFO_MSG("Recreating offline_msg table", []),
|
2005-04-17 20:08:34 +02:00
|
|
|
mnesia:transform_table(offline_msg, ignore, Fields)
|
2004-09-10 22:57:00 +02:00
|
|
|
end.
|
2007-08-13 12:27:28 +02:00
|
|
|
|
|
|
|
|
|
|
|
%% Helper functions:
|
|
|
|
|
|
|
|
%% Warn senders that their messages have been discarded:
|
|
|
|
discard_warn_sender(Msgs) ->
|
|
|
|
lists:foreach(
|
2008-08-06 15:44:58 +02:00
|
|
|
fun(#offline_msg{from=From, to=To, packet=Packet0}) ->
|
|
|
|
Packet = case Packet0 of
|
|
|
|
#xmlelement{} = P ->
|
|
|
|
exmpp_xml:xmlelement_to_xmlel(P,
|
|
|
|
[?DEFAULT_NS], ?PREFIXED_NS);
|
|
|
|
#xmlel{} = P ->
|
|
|
|
P
|
|
|
|
end,
|
2007-08-13 12:27:28 +02:00
|
|
|
ErrText = "Your contact offline message queue is full. The message has been discarded.",
|
2008-08-06 15:44:58 +02:00
|
|
|
Error = exmpp_stanza:error('resource-constraint',
|
|
|
|
{"en", ErrText}),
|
|
|
|
Err = exmpp_stanza:reply_with_error(Packet, Error),
|
2007-08-13 12:27:28 +02:00
|
|
|
ejabberd_router:route(
|
2008-08-06 15:44:58 +02:00
|
|
|
jlib:from_old_jid(To),
|
|
|
|
jlib:from_old_jid(From), Err)
|
2007-08-13 12:27:28 +02:00
|
|
|
end, Msgs).
|
2007-08-23 02:51:54 +02:00
|
|
|
|
|
|
|
|
|
|
|
webadmin_page(_, Host,
|
|
|
|
#request{us = _US,
|
|
|
|
path = ["user", U, "queue"],
|
|
|
|
q = Query,
|
|
|
|
lang = Lang} = _Request) ->
|
|
|
|
Res = user_queue(U, Host, Query, Lang),
|
|
|
|
{stop, Res};
|
|
|
|
|
|
|
|
webadmin_page(Acc, _, _) -> Acc.
|
|
|
|
|
|
|
|
user_queue(User, Server, Query, Lang) ->
|
2008-08-06 15:44:58 +02:00
|
|
|
US = {exmpp_stringprep:nodeprep(User), exmpp_stringprep:nameprep(Server)},
|
2007-08-23 02:51:54 +02:00
|
|
|
Res = user_queue_parse_query(US, Query),
|
|
|
|
Msgs = lists:keysort(#offline_msg.timestamp,
|
|
|
|
mnesia:dirty_read({offline_msg, US})),
|
|
|
|
FMsgs =
|
|
|
|
lists:map(
|
|
|
|
fun(#offline_msg{timestamp = TimeStamp, from = From, to = To,
|
2008-08-06 15:44:58 +02:00
|
|
|
packet = Packet} = Msg) ->
|
2007-08-23 02:51:54 +02:00
|
|
|
ID = jlib:encode_base64(binary_to_list(term_to_binary(Msg))),
|
|
|
|
{{Year, Month, Day}, {Hour, Minute, Second}} =
|
|
|
|
calendar:now_to_local_time(TimeStamp),
|
|
|
|
Time = lists:flatten(
|
|
|
|
io_lib:format(
|
|
|
|
"~w-~.2.0w-~.2.0w ~.2.0w:~.2.0w:~.2.0w",
|
|
|
|
[Year, Month, Day, Hour, Minute, Second])),
|
2008-08-06 15:51:42 +02:00
|
|
|
SFrom = exmpp_jid:jid_to_list(jlib:from_old_jid(From)),
|
|
|
|
STo = exmpp_jid:jid_to_list(jlib:from_old_jid(To)),
|
2008-08-06 15:44:58 +02:00
|
|
|
Packet0 = exmpp_xml:xmlelement_to_xmlel(Packet,
|
|
|
|
[?DEFAULT_NS], ?PREFIXED_NS),
|
|
|
|
Packet1 = exmpp_stanza:set_jids(Packet0, SFrom, STo),
|
|
|
|
FPacket = exmpp_xml:node_to_list(
|
|
|
|
exmpp_xml:indent_document(Packet1, <<" ">>),
|
|
|
|
[?DEFAULT_NS], ?PREFIXED_NS),
|
2007-08-23 02:51:54 +02:00
|
|
|
?XE("tr",
|
|
|
|
[?XAE("td", [{"class", "valign"}], [?INPUT("checkbox", "selected", ID)]),
|
|
|
|
?XAC("td", [{"class", "valign"}], Time),
|
|
|
|
?XAC("td", [{"class", "valign"}], SFrom),
|
|
|
|
?XAC("td", [{"class", "valign"}], STo),
|
|
|
|
?XAE("td", [{"class", "valign"}], [?XC("pre", FPacket)])]
|
|
|
|
)
|
|
|
|
end, Msgs),
|
|
|
|
[?XC("h1", io_lib:format(?T("~s's Offline Messages Queue"),
|
|
|
|
[us_to_list(US)]))] ++
|
|
|
|
case Res of
|
|
|
|
ok -> [?CT("Submitted"), ?P];
|
|
|
|
nothing -> []
|
|
|
|
end ++
|
|
|
|
[?XAE("form", [{"action", ""}, {"method", "post"}],
|
|
|
|
[?XE("table",
|
|
|
|
[?XE("thead",
|
|
|
|
[?XE("tr",
|
|
|
|
[?X("td"),
|
|
|
|
?XCT("td", "Time"),
|
|
|
|
?XCT("td", "From"),
|
|
|
|
?XCT("td", "To"),
|
|
|
|
?XCT("td", "Packet")
|
|
|
|
])]),
|
|
|
|
?XE("tbody",
|
|
|
|
if
|
|
|
|
FMsgs == [] ->
|
|
|
|
[?XE("tr",
|
|
|
|
[?XAC("td", [{"colspan", "4"}], " ")]
|
|
|
|
)];
|
|
|
|
true ->
|
|
|
|
FMsgs
|
|
|
|
end
|
|
|
|
)]),
|
|
|
|
?BR,
|
|
|
|
?INPUTT("submit", "delete", "Delete Selected")
|
|
|
|
])].
|
|
|
|
|
|
|
|
user_queue_parse_query(US, Query) ->
|
|
|
|
case lists:keysearch("delete", 1, Query) of
|
|
|
|
{value, _} ->
|
|
|
|
Msgs = lists:keysort(#offline_msg.timestamp,
|
|
|
|
mnesia:dirty_read({offline_msg, US})),
|
|
|
|
F = fun() ->
|
|
|
|
lists:foreach(
|
|
|
|
fun(Msg) ->
|
|
|
|
ID = jlib:encode_base64(
|
|
|
|
binary_to_list(term_to_binary(Msg))),
|
|
|
|
case lists:member({"selected", ID}, Query) of
|
|
|
|
true ->
|
|
|
|
mnesia:delete_object(Msg);
|
|
|
|
false ->
|
|
|
|
ok
|
|
|
|
end
|
|
|
|
end, Msgs)
|
|
|
|
end,
|
|
|
|
mnesia:transaction(F),
|
|
|
|
ok;
|
|
|
|
false ->
|
|
|
|
nothing
|
|
|
|
end.
|
|
|
|
|
|
|
|
us_to_list({User, Server}) ->
|
2008-08-06 15:51:42 +02:00
|
|
|
exmpp_jid:jid_to_list(User, Server).
|
2007-08-23 02:51:54 +02:00
|
|
|
|
|
|
|
webadmin_user(Acc, User, Server, Lang) ->
|
2008-08-06 15:44:58 +02:00
|
|
|
US = {exmpp_stringprep:nodeprep(User), exmpp_stringprep:nameprep(Server)},
|
2007-08-23 02:51:54 +02:00
|
|
|
QueueLen = length(mnesia:dirty_read({offline_msg, US})),
|
|
|
|
FQueueLen = [?AC("queue/",
|
|
|
|
integer_to_list(QueueLen))],
|
|
|
|
Acc ++ [?XCT("h3", "Offline Messages:")] ++ FQueueLen.
|