| 1 |
|
%%%------------------------------------------------------------------- |
| 2 |
|
%%% @author Dominik Stanaszek dominik.stanaszek@erlang-solutions.com |
| 3 |
|
%%% @copyright (C) 2018, Erlang-Solutions |
| 4 |
|
%%% @doc |
| 5 |
|
%%% |
| 6 |
|
%%% @end |
| 7 |
|
%%% Created : 6.07.2018 |
| 8 |
|
%%%------------------------------------------------------------------- |
| 9 |
|
-module(mod_inbox_muc). |
| 10 |
|
-author("dominik.stanaszek@erlang-solutions.com"). |
| 11 |
|
-include("jlib.hrl"). |
| 12 |
|
-include("mongoose.hrl"). |
| 13 |
|
|
| 14 |
|
-export([update_inbox_for_muc/3, start/1, stop/1]). |
| 15 |
|
|
| 16 |
|
%% User jid example is "alice@localhost" |
| 17 |
|
-type user_jid() :: jid:jid(). |
| 18 |
|
-type receiver_bare_user_jid() :: user_jid(). |
| 19 |
|
-type room_bare_jid() :: jid:jid(). |
| 20 |
|
-type packet() :: exml:element(). |
| 21 |
|
|
| 22 |
|
start(HostType) -> |
| 23 |
:-( |
gen_hook:add_handlers(hooks(HostType)), |
| 24 |
|
% TODO check options: if system messages stored -> |
| 25 |
|
% add hook handler for system messages on hook ie. invitation_sent |
| 26 |
:-( |
ok. |
| 27 |
|
|
| 28 |
|
stop(HostType) -> |
| 29 |
:-( |
gen_hook:delete_handlers(hooks(HostType)), |
| 30 |
:-( |
ok. |
| 31 |
|
|
| 32 |
|
hooks(HostType) -> |
| 33 |
:-( |
[{update_inbox_for_muc, HostType, fun ?MODULE:update_inbox_for_muc/3, #{}, 90}]. |
| 34 |
|
|
| 35 |
|
-spec update_inbox_for_muc(Acc, Params, Extra) -> {ok, Acc} when |
| 36 |
|
Acc :: mod_muc_room:update_inbox_for_muc_payload(), |
| 37 |
|
Params :: map(), |
| 38 |
|
Extra :: gen_hook:extra(). |
| 39 |
|
update_inbox_for_muc( |
| 40 |
|
#{host_type := HostType, |
| 41 |
|
room_jid := Room, |
| 42 |
|
from_jid := From, |
| 43 |
|
from_room_jid := FromRoomJid, |
| 44 |
|
packet := Packet, |
| 45 |
|
affiliations_map := AffsMap} = Acc, _, _) -> |
| 46 |
:-( |
F = fun(AffLJID, Affiliation) -> |
| 47 |
:-( |
case is_allowed_affiliation(Affiliation) of |
| 48 |
|
true -> |
| 49 |
:-( |
To = jid:to_bare(jid:make(AffLJID)), |
| 50 |
|
%% Guess direction based on user JIDs |
| 51 |
:-( |
Direction = direction(From, To), |
| 52 |
:-( |
Packet2 = jlib:replace_from_to(FromRoomJid, To, Packet), |
| 53 |
:-( |
update_inbox_for_user(HostType, Direction, Room, To, Packet2); |
| 54 |
|
false -> |
| 55 |
:-( |
ok |
| 56 |
|
end |
| 57 |
|
end, |
| 58 |
:-( |
mongoose_lib:maps_foreach(F, AffsMap), |
| 59 |
:-( |
{ok, Acc}. |
| 60 |
|
|
| 61 |
|
-spec is_allowed_affiliation(mod_muc:affiliation()) -> boolean(). |
| 62 |
:-( |
is_allowed_affiliation(outcast) -> false; |
| 63 |
:-( |
is_allowed_affiliation(_) -> true. |
| 64 |
|
|
| 65 |
|
-spec update_inbox_for_user(HostType, Direction, Room, To, Packet) -> term() when |
| 66 |
|
HostType :: mongooseim:host_type(), |
| 67 |
|
Direction :: incoming | outgoing, |
| 68 |
|
Room :: room_bare_jid(), |
| 69 |
|
To :: receiver_bare_user_jid(), |
| 70 |
|
Packet :: packet(). |
| 71 |
|
update_inbox_for_user(HostType, Direction, Room, To, Packet) -> |
| 72 |
:-( |
ReceiverDomain = To#jid.lserver, |
| 73 |
:-( |
MucDomain = mod_muc:server_host_to_muc_host(HostType, ReceiverDomain), |
| 74 |
:-( |
case Room#jid.lserver of |
| 75 |
|
MucDomain -> |
| 76 |
:-( |
handle_message(HostType, Room, To, Packet, Direction); |
| 77 |
|
_ -> |
| 78 |
|
%% We ignore inbox for users on the remote (s2s) hosts |
| 79 |
|
%% We ignore inbox for components (also known as services or bots) |
| 80 |
:-( |
ok |
| 81 |
|
end. |
| 82 |
|
|
| 83 |
|
handle_message(HostType, Room, To, Packet, outgoing) -> |
| 84 |
:-( |
handle_outgoing_message(HostType, Room, To, Packet); |
| 85 |
|
handle_message(HostType, Room, To, Packet, incoming) -> |
| 86 |
:-( |
handle_incoming_message(HostType, Room, To, Packet). |
| 87 |
|
|
| 88 |
|
-spec direction(From :: user_jid(), To :: user_jid()) -> incoming | outgoing. |
| 89 |
|
direction(From, To) -> |
| 90 |
:-( |
case jid:are_bare_equal(From, To) of |
| 91 |
:-( |
true -> outgoing; |
| 92 |
:-( |
false -> incoming |
| 93 |
|
end. |
| 94 |
|
|
| 95 |
|
%% Sender and receiver is the same user |
| 96 |
|
-spec handle_outgoing_message(HostType, Room, To, Packet) -> term() when |
| 97 |
|
HostType :: mongooseim:host_type(), |
| 98 |
|
Room :: room_bare_jid(), |
| 99 |
|
To :: receiver_bare_user_jid(), |
| 100 |
|
Packet :: packet(). |
| 101 |
|
handle_outgoing_message(HostType, Room, To, Packet) -> |
| 102 |
:-( |
Acc = mongoose_acc:new(#{location => ?LOCATION, lserver => To#jid.lserver, host_type => HostType}), |
| 103 |
:-( |
maybe_reset_unread_count(HostType, To, Room, Packet, Acc), |
| 104 |
:-( |
maybe_write_to_inbox(HostType, To, Room, Packet, Acc, fun write_to_sender_inbox/5). |
| 105 |
|
|
| 106 |
|
-spec handle_incoming_message(HostType, Room, To, Packet) -> term() when |
| 107 |
|
HostType :: mongooseim:host_type(), |
| 108 |
|
Room :: room_bare_jid(), |
| 109 |
|
To :: receiver_bare_user_jid(), |
| 110 |
|
Packet :: packet(). |
| 111 |
|
handle_incoming_message(HostType, Room, To, Packet) -> |
| 112 |
:-( |
Acc = mongoose_acc:new(#{location => ?LOCATION, lserver => To#jid.lserver, host_type => HostType}), |
| 113 |
:-( |
maybe_write_to_inbox(HostType, Room, To, Packet, Acc, fun write_to_receiver_inbox/5). |
| 114 |
|
|
| 115 |
|
maybe_reset_unread_count(HostType, User, Room, Packet, Acc) -> |
| 116 |
:-( |
mod_inbox_utils:maybe_reset_unread_count(HostType, User, Room, Packet, Acc). |
| 117 |
|
|
| 118 |
|
maybe_write_to_inbox(HostType, User, Remote, Packet, Acc, WriteF) -> |
| 119 |
:-( |
mod_inbox_utils:maybe_write_to_inbox(HostType, User, Remote, Packet, Acc, WriteF). |
| 120 |
|
|
| 121 |
|
write_to_sender_inbox(Server, User, Remote, Packet, Acc) -> |
| 122 |
:-( |
mod_inbox_utils:write_to_sender_inbox(Server, User, Remote, Packet, Acc). |
| 123 |
|
|
| 124 |
|
write_to_receiver_inbox(Server, User, Remote, Packet, Acc) -> |
| 125 |
:-( |
mod_inbox_utils:write_to_receiver_inbox(Server, User, Remote, Packet, Acc). |