./ct_report/coverage/mongoose_admin_api_messages.COVER.html

1 -module(mongoose_admin_api_messages).
2
3 -behaviour(mongoose_admin_api).
4 -export([routes/1]).
5
6 -behaviour(cowboy_rest).
7 -export([init/2,
8 is_authorized/2,
9 content_types_provided/2,
10 content_types_accepted/2,
11 allowed_methods/2,
12 to_json/2,
13 from_json/2]).
14
15 -ignore_xref([to_json/2, from_json/2]).
16
17 -import(mongoose_admin_api, [parse_body/1, parse_qs/1, try_handle_request/3, throw_error/2]).
18
19 -type req() :: cowboy_req:req().
20 -type state() :: mongoose_admin_api:state().
21
22 -spec routes(state()) -> mongoose_http_handler:routes().
23 routes(State) ->
24 4 [{"/messages/:owner/:with", ?MODULE, State},
25 {"/messages/[:owner]", ?MODULE, State}].
26
27 -spec init(req(), state()) -> {cowboy_rest, req(), state()}.
28 init(Req, State) ->
29
:-(
mongoose_admin_api:init(Req, State).
30
31 -spec is_authorized(req(), state()) -> {true | {false, iodata()}, req(), state()}.
32 is_authorized(Req, State) ->
33
:-(
mongoose_admin_api:is_authorized(Req, State).
34
35 -spec content_types_provided(req(), state()) ->
36 {[{{binary(), binary(), '*'}, atom()}], req(), state()}.
37 content_types_provided(Req, State) ->
38
:-(
{[
39 {{<<"application">>, <<"json">>, '*'}, to_json}
40 ], Req, State}.
41
42 -spec content_types_accepted(req(), state()) ->
43 {[{{binary(), binary(), '*'}, atom()}], req(), state()}.
44 content_types_accepted(Req, State) ->
45
:-(
{[
46 {{<<"application">>, <<"json">>, '*'}, from_json}
47 ], Req, State}.
48
49 -spec allowed_methods(req(), state()) -> {[binary()], req(), state()}.
50 allowed_methods(Req, State) ->
51
:-(
{[<<"OPTIONS">>, <<"GET">>, <<"POST">>], Req, State}.
52
53 %% @doc Called for a method of type "GET"
54 -spec to_json(req(), state()) -> {iodata() | stop, req(), state()}.
55 to_json(Req, State) ->
56
:-(
try_handle_request(Req, State, fun handle_get/2).
57
58 %% @doc Called for a method of type "POST"
59 -spec from_json(req(), state()) -> {true | stop, req(), state()}.
60 from_json(Req, State) ->
61
:-(
try_handle_request(Req, State, fun handle_post/2).
62
63 %% Internal functions
64
65 handle_get(Req, State) ->
66
:-(
Bindings = cowboy_req:bindings(Req),
67
:-(
OwnerJid = get_owner_jid(Bindings),
68
:-(
WithJid = get_with_jid(Bindings),
69
:-(
Args = parse_qs(Req),
70
:-(
Limit = get_limit(Args),
71
:-(
Before = get_before(Args),
72
:-(
case mongoose_stanza_api:lookup_recent_messages(OwnerJid, WithJid, Before, Limit, true) of
73 {ok, {Rows, _Limit}} ->
74
:-(
Messages = lists:map(fun row_to_map/1, Rows),
75
:-(
{jiffy:encode(Messages), Req, State};
76 {unknown_user, Msg} ->
77
:-(
throw_error(bad_request, Msg)
78 end.
79
80 handle_post(Req, State) ->
81
:-(
Args = parse_body(Req),
82
:-(
From = get_caller(Args),
83
:-(
To = get_to(Args),
84
:-(
Body = get_body(Args),
85
:-(
case mongoose_stanza_api:send_chat_message(undefined, From, To, Body) of
86 {ok, _} ->
87
:-(
{true, Req, State};
88 {_Error, Msg} ->
89
:-(
throw_error(bad_request, Msg)
90 end.
91
92 -spec row_to_map(mod_mam:message_row()) -> map().
93 row_to_map(#{id := Id, jid := From, packet := Msg}) ->
94
:-(
Jbin = jid:to_binary(From),
95
:-(
{Msec, _} = mod_mam_utils:decode_compact_uuid(Id),
96
:-(
MsgId = case xml:get_tag_attr(<<"id">>, Msg) of
97
:-(
{value, MId} -> MId;
98
:-(
false -> <<"">>
99 end,
100
:-(
Body = exml_query:path(Msg, [{element, <<"body">>}, cdata]),
101
:-(
#{sender => Jbin, timestamp => round(Msec / 1000000), message_id => MsgId, body => Body}.
102
103 get_limit(#{limit := LimitBin}) ->
104
:-(
try
105
:-(
Limit = binary_to_integer(LimitBin),
106
:-(
true = Limit >= 0,
107
:-(
Limit
108 catch
109
:-(
_:_ -> throw_error(bad_request, <<"Invalid limit">>)
110 end;
111
:-(
get_limit(#{}) -> 100.
112
113 get_before(#{before := BeforeBin}) ->
114
:-(
try
115
:-(
1000000 * binary_to_integer(BeforeBin)
116 catch
117
:-(
_:_ -> throw_error(bad_request, <<"Invalid value of 'before'">>)
118 end;
119
:-(
get_before(#{}) -> undefined.
120
121 get_owner_jid(#{owner := Owner}) ->
122
:-(
case jid:from_binary(Owner) of
123
:-(
error -> throw_error(bad_request, <<"Invalid owner JID">>);
124
:-(
OwnerJid -> OwnerJid
125 end;
126
:-(
get_owner_jid(#{}) -> throw_error(not_found, <<"Missing owner JID">>).
127
128 get_with_jid(#{with := With}) ->
129
:-(
case jid:from_binary(With) of
130
:-(
error -> throw_error(bad_request, <<"Invalid interlocutor JID">>);
131
:-(
WithJid -> WithJid
132 end;
133
:-(
get_with_jid(#{}) -> undefined.
134
135 get_caller(#{caller := Caller}) ->
136
:-(
case jid:from_binary(Caller) of
137
:-(
error -> throw_error(bad_request, <<"Invalid sender JID">>);
138
:-(
CallerJid -> CallerJid
139 end;
140
:-(
get_caller(#{}) -> throw_error(bad_request, <<"Missing sender JID">>).
141
142 get_to(#{to := To}) ->
143
:-(
case jid:from_binary(To) of
144
:-(
error -> throw_error(bad_request, <<"Invalid recipient JID">>);
145
:-(
ToJid -> ToJid
146 end;
147
:-(
get_to(#{}) -> throw_error(bad_request, <<"Missing recipient JID">>).
148
149
:-(
get_body(#{body := Body}) -> Body;
150
:-(
get_body(#{}) -> throw_error(bad_request, <<"Missing message body">>).
Line Hits Source