2016-07-18 14:01:32 +02:00
|
|
|
%%%-------------------------------------------------------------------
|
|
|
|
%%% @author Evgeny Khramtsov <ekhramtsov@process-one.net>
|
|
|
|
%%% @copyright (C) 2016, Evgeny Khramtsov
|
|
|
|
%%% @doc
|
|
|
|
%%%
|
|
|
|
%%% @end
|
|
|
|
%%% Created : 12 Jul 2016 by Evgeny Khramtsov <ekhramtsov@process-one.net>
|
|
|
|
%%%-------------------------------------------------------------------
|
|
|
|
-module(xmpp_util).
|
|
|
|
|
|
|
|
%% API
|
|
|
|
-export([add_delay_info/3, add_delay_info/4, unwrap_carbon/1,
|
2016-07-25 12:50:30 +02:00
|
|
|
is_standalone_chat_state/1, get_xdata_values/2,
|
2016-10-07 09:31:03 +02:00
|
|
|
has_xdata_var/2, make_adhoc_response/1, make_adhoc_response/2,
|
|
|
|
decode_timestamp/1, encode_timestamp/1]).
|
2016-07-18 14:01:32 +02:00
|
|
|
|
|
|
|
-include("xmpp.hrl").
|
|
|
|
|
|
|
|
%%%===================================================================
|
|
|
|
%%% API
|
|
|
|
%%%===================================================================
|
|
|
|
-spec add_delay_info(stanza(), jid(), erlang:timestamp()) -> stanza().
|
|
|
|
add_delay_info(Stz, From, Time) ->
|
|
|
|
add_delay_info(Stz, From, Time, <<"">>).
|
|
|
|
|
|
|
|
-spec add_delay_info(stanza(), jid(),
|
|
|
|
erlang:timestamp(), binary()) -> stanza().
|
|
|
|
|
|
|
|
add_delay_info(Stz, From, Time, Desc) ->
|
2016-11-13 08:44:53 +01:00
|
|
|
NewDelay = #delay{stamp = Time, from = From, desc = Desc},
|
2016-07-18 14:01:32 +02:00
|
|
|
case xmpp:get_subtag(Stz, #delay{}) of
|
2016-11-13 08:44:53 +01:00
|
|
|
#delay{from = OldFrom} ->
|
2016-07-18 14:01:32 +02:00
|
|
|
case jid:tolower(From) == jid:tolower(OldFrom) of
|
|
|
|
false ->
|
2016-11-13 08:44:53 +01:00
|
|
|
xmpp:set_subtag(Stz, NewDelay);
|
|
|
|
true ->
|
|
|
|
xmpp:append_subtags(Stz, [NewDelay])
|
2016-07-18 14:01:32 +02:00
|
|
|
end;
|
|
|
|
false ->
|
2016-11-13 08:44:53 +01:00
|
|
|
xmpp:append_subtags(Stz, [NewDelay])
|
2016-07-18 14:01:32 +02:00
|
|
|
end.
|
|
|
|
|
|
|
|
-spec unwrap_carbon(stanza()) -> xmpp_element().
|
|
|
|
unwrap_carbon(#message{} = Msg) ->
|
2016-11-13 11:41:04 +01:00
|
|
|
try
|
|
|
|
case xmpp:get_subtag(Msg, #carbons_sent{}) of
|
|
|
|
#carbons_sent{forwarded = #forwarded{xml_els = [El]}} ->
|
|
|
|
xmpp:decode(El, ?NS_CLIENT, [ignore_els]);
|
|
|
|
_ ->
|
|
|
|
case xmpp:get_subtag(Msg, #carbons_received{}) of
|
|
|
|
#carbons_received{forwarded = #forwarded{xml_els = [El]}} ->
|
|
|
|
xmpp:decode(El, ?NS_CLIENT, [ignore_els]);
|
|
|
|
_ ->
|
|
|
|
Msg
|
|
|
|
end
|
|
|
|
end
|
|
|
|
catch _:{xmpp_codec, _} ->
|
|
|
|
Msg
|
2016-07-18 14:01:32 +02:00
|
|
|
end;
|
|
|
|
unwrap_carbon(Stanza) -> Stanza.
|
|
|
|
|
|
|
|
-spec is_standalone_chat_state(stanza()) -> boolean().
|
|
|
|
is_standalone_chat_state(Stanza) ->
|
|
|
|
case unwrap_carbon(Stanza) of
|
2016-09-13 11:30:05 +02:00
|
|
|
#message{body = [], subject = [], sub_els = Els} ->
|
2016-11-08 13:15:19 +01:00
|
|
|
IgnoreNS = [?NS_CHATSTATES, ?NS_DELAY, ?NS_EVENT],
|
2016-07-18 14:01:32 +02:00
|
|
|
Stripped = [El || El <- Els,
|
|
|
|
not lists:member(xmpp:get_ns(El), IgnoreNS)],
|
|
|
|
Stripped == [];
|
|
|
|
_ ->
|
|
|
|
false
|
|
|
|
end.
|
|
|
|
|
2016-07-25 12:50:30 +02:00
|
|
|
-spec get_xdata_values(binary(), xdata()) -> [binary()].
|
|
|
|
get_xdata_values(Var, #xdata{fields = Fields}) ->
|
|
|
|
case lists:keyfind(Var, #xdata_field.var, Fields) of
|
|
|
|
#xdata_field{values = Vals} -> Vals;
|
|
|
|
false -> []
|
|
|
|
end.
|
|
|
|
|
|
|
|
-spec has_xdata_var(binary(), xdata()) -> boolean().
|
|
|
|
has_xdata_var(Var, #xdata{fields = Fields}) ->
|
|
|
|
lists:keymember(Var, #xdata_field.var, Fields).
|
|
|
|
|
2016-07-26 08:52:29 +02:00
|
|
|
-spec make_adhoc_response(adhoc_command(), adhoc_command()) -> adhoc_command().
|
|
|
|
make_adhoc_response(#adhoc_command{lang = Lang, node = Node, sid = SID},
|
|
|
|
Command) ->
|
2016-11-12 11:27:15 +01:00
|
|
|
make_adhoc_response(
|
|
|
|
Command#adhoc_command{lang = Lang, node = Node, sid = SID}).
|
2016-07-26 08:52:29 +02:00
|
|
|
|
|
|
|
-spec make_adhoc_response(adhoc_command()) -> adhoc_command().
|
2016-11-12 11:27:15 +01:00
|
|
|
make_adhoc_response(#adhoc_command{sid = <<"">>,
|
|
|
|
status = Status,
|
|
|
|
actions = Actions} = Command) ->
|
2016-10-07 09:31:03 +02:00
|
|
|
SID = encode_timestamp(p1_time_compat:timestamp()),
|
2016-11-12 11:27:15 +01:00
|
|
|
NewActions = if Actions == undefined, Status /= completed ->
|
|
|
|
#adhoc_actions{execute = complete, complete = true};
|
|
|
|
true ->
|
|
|
|
undefined
|
|
|
|
end,
|
|
|
|
Command#adhoc_command{sid = SID, actions = NewActions};
|
2016-07-26 08:52:29 +02:00
|
|
|
make_adhoc_response(Command) ->
|
|
|
|
Command.
|
|
|
|
|
2016-10-07 09:31:03 +02:00
|
|
|
-spec decode_timestamp(binary()) -> erlang:timestamp().
|
|
|
|
decode_timestamp(S) ->
|
|
|
|
try try_decode_timestamp(S)
|
|
|
|
catch _:_ -> erlang:error({bad_timestamp, S})
|
|
|
|
end.
|
|
|
|
|
|
|
|
-spec encode_timestamp(erlang:timestamp()) -> binary().
|
|
|
|
encode_timestamp({MegaSecs, Secs, MicroSecs}) ->
|
|
|
|
{{Year, Month, Day}, {Hour, Minute, Second}} =
|
|
|
|
calendar:now_to_universal_time({MegaSecs, Secs, MicroSecs}),
|
|
|
|
Fraction = if MicroSecs > 0 ->
|
|
|
|
io_lib:format(".~6..0B", [MicroSecs]);
|
|
|
|
true ->
|
|
|
|
""
|
|
|
|
end,
|
|
|
|
list_to_binary(io_lib:format("~4..0B-~2..0B-~2..0BT"
|
|
|
|
"~2..0B:~2..0B:~2..0B~sZ",
|
|
|
|
[Year, Month, Day, Hour, Minute, Second,
|
|
|
|
Fraction])).
|
|
|
|
|
2016-07-18 14:01:32 +02:00
|
|
|
%%%===================================================================
|
|
|
|
%%% Internal functions
|
|
|
|
%%%===================================================================
|
2016-10-07 09:31:03 +02:00
|
|
|
try_decode_timestamp(<<Y:4/binary, $-, Mo:2/binary, $-, D:2/binary, $T,
|
|
|
|
H:2/binary, $:, Mi:2/binary, $:, S:2/binary, T/binary>>) ->
|
|
|
|
Date = {to_integer(Y, 1970, 9999), to_integer(Mo, 1, 12), to_integer(D, 1, 31)},
|
|
|
|
Time = {to_integer(H, 0, 23), to_integer(Mi, 0, 59), to_integer(S, 0, 59)},
|
|
|
|
{MS, {TZH, TZM}} = try_decode_fraction(T),
|
|
|
|
Seconds = calendar:datetime_to_gregorian_seconds({Date, Time}) -
|
|
|
|
calendar:datetime_to_gregorian_seconds({{1970,1,1}, {0,0,0}}) -
|
|
|
|
TZH * 60 * 60 - TZM * 60,
|
|
|
|
{Seconds div 1000000, Seconds rem 1000000, MS};
|
|
|
|
try_decode_timestamp(<<Y:4/binary, Mo:2/binary, D:2/binary, $T,
|
|
|
|
H:2/binary, $:, Mi:2/binary, $:, S:2/binary>>) ->
|
|
|
|
try_decode_timestamp(<<Y:4/binary, $-, Mo:2/binary, $-, D:2/binary, $T,
|
|
|
|
H:2/binary, $:, Mi:2/binary, $:, S:2/binary, $Z>>).
|
|
|
|
|
|
|
|
try_decode_fraction(<<$., T/binary>>) ->
|
2016-11-13 08:44:53 +01:00
|
|
|
{match, [V]} = re:run(T, <<"^[0-9]+">>, [{capture, [0], list}]),
|
|
|
|
Size = length(V),
|
|
|
|
<<_:Size/binary, TZD/binary>> = T,
|
|
|
|
{list_to_integer(string:left(V, 6, $0)),
|
2016-10-07 09:31:03 +02:00
|
|
|
try_decode_tzd(TZD)};
|
|
|
|
try_decode_fraction(TZD) ->
|
|
|
|
{0, try_decode_tzd(TZD)}.
|
|
|
|
|
|
|
|
try_decode_tzd(<<$Z>>) ->
|
|
|
|
{0, 0};
|
|
|
|
try_decode_tzd(<<$-, H:2/binary, $:, M:2/binary>>) ->
|
|
|
|
{-1 * to_integer(H, 0, 12), to_integer(M, 0, 59)};
|
|
|
|
try_decode_tzd(<<$+, H:2/binary, $:, M:2/binary>>) ->
|
|
|
|
{to_integer(H, 0, 12), to_integer(M, 0, 59)}.
|
|
|
|
|
|
|
|
to_integer(S, Min, Max) ->
|
|
|
|
case binary_to_integer(S) of
|
|
|
|
I when I >= Min, I =< Max ->
|
|
|
|
I
|
|
|
|
end.
|