mirror of
https://github.com/processone/ejabberd.git
synced 2024-06-14 22:00:16 +02:00
7bcbea2108
Since the main goal of jlib.erl is lost, all auxiliary functions are now moved to aux.erl, and the whole jlib.erl is now deprecated.
380 lines
12 KiB
Erlang
380 lines
12 KiB
Erlang
%%%-------------------------------------------------------------------
|
|
%%% File : mod_muc_sql.erl
|
|
%%% Author : Evgeny Khramtsov <ekhramtsov@process-one.net>
|
|
%%% Created : 13 Apr 2016 by Evgeny Khramtsov <ekhramtsov@process-one.net>
|
|
%%%
|
|
%%%
|
|
%%% ejabberd, Copyright (C) 2002-2017 ProcessOne
|
|
%%%
|
|
%%% 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.,
|
|
%%% 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
%%%
|
|
%%%----------------------------------------------------------------------
|
|
|
|
-module(mod_muc_sql).
|
|
|
|
-compile([{parse_transform, ejabberd_sql_pt}]).
|
|
|
|
-behaviour(mod_muc).
|
|
-behaviour(mod_muc_room).
|
|
|
|
%% API
|
|
-export([init/2, store_room/4, restore_room/3, forget_room/3,
|
|
can_use_nick/4, get_rooms/2, get_nick/3, set_nick/4,
|
|
import/3, export/1]).
|
|
-export([register_online_room/4, unregister_online_room/4, find_online_room/3,
|
|
get_online_rooms/3, count_online_rooms/2, rsm_supported/0,
|
|
register_online_user/4, unregister_online_user/4,
|
|
count_online_rooms_by_user/3, get_online_rooms_by_user/3]).
|
|
-export([set_affiliation/6, set_affiliations/4, get_affiliation/5,
|
|
get_affiliations/3, search_affiliation/4]).
|
|
|
|
-include("jid.hrl").
|
|
-include("mod_muc.hrl").
|
|
-include("logger.hrl").
|
|
-include("ejabberd_sql_pt.hrl").
|
|
|
|
%%%===================================================================
|
|
%%% API
|
|
%%%===================================================================
|
|
init(Host, Opts) ->
|
|
case gen_mod:ram_db_mod(Host, Opts, mod_muc) of
|
|
?MODULE ->
|
|
clean_tables(Host);
|
|
_ ->
|
|
ok
|
|
end.
|
|
|
|
store_room(LServer, Host, Name, Opts) ->
|
|
SOpts = aux:term_to_expr(Opts),
|
|
F = fun () ->
|
|
?SQL_UPSERT_T(
|
|
"muc_room",
|
|
["!name=%(Name)s",
|
|
"!host=%(Host)s",
|
|
"opts=%(SOpts)s"])
|
|
end,
|
|
ejabberd_sql:sql_transaction(LServer, F).
|
|
|
|
restore_room(LServer, Host, Name) ->
|
|
case catch ejabberd_sql:sql_query(
|
|
LServer,
|
|
?SQL("select @(opts)s from muc_room where name=%(Name)s"
|
|
" and host=%(Host)s")) of
|
|
{selected, [{Opts}]} ->
|
|
mod_muc:opts_to_binary(ejabberd_sql:decode_term(Opts));
|
|
_ ->
|
|
error
|
|
end.
|
|
|
|
forget_room(LServer, Host, Name) ->
|
|
F = fun () ->
|
|
ejabberd_sql:sql_query_t(
|
|
?SQL("delete from muc_room where name=%(Name)s"
|
|
" and host=%(Host)s"))
|
|
end,
|
|
ejabberd_sql:sql_transaction(LServer, F).
|
|
|
|
can_use_nick(LServer, Host, JID, Nick) ->
|
|
SJID = jid:encode(jid:tolower(jid:remove_resource(JID))),
|
|
case catch ejabberd_sql:sql_query(
|
|
LServer,
|
|
?SQL("select @(jid)s from muc_registered "
|
|
"where nick=%(Nick)s"
|
|
" and host=%(Host)s")) of
|
|
{selected, [{SJID1}]} -> SJID == SJID1;
|
|
_ -> true
|
|
end.
|
|
|
|
get_rooms(LServer, Host) ->
|
|
case catch ejabberd_sql:sql_query(
|
|
LServer,
|
|
?SQL("select @(name)s, @(opts)s from muc_room"
|
|
" where host=%(Host)s")) of
|
|
{selected, RoomOpts} ->
|
|
lists:map(
|
|
fun({Room, Opts}) ->
|
|
#muc_room{name_host = {Room, Host},
|
|
opts = mod_muc:opts_to_binary(
|
|
ejabberd_sql:decode_term(Opts))}
|
|
end, RoomOpts);
|
|
Err ->
|
|
?ERROR_MSG("failed to get rooms: ~p", [Err]),
|
|
[]
|
|
end.
|
|
|
|
get_nick(LServer, Host, From) ->
|
|
SJID = jid:encode(jid:tolower(jid:remove_resource(From))),
|
|
case catch ejabberd_sql:sql_query(
|
|
LServer,
|
|
?SQL("select @(nick)s from muc_registered where"
|
|
" jid=%(SJID)s and host=%(Host)s")) of
|
|
{selected, [{Nick}]} -> Nick;
|
|
_ -> error
|
|
end.
|
|
|
|
set_nick(LServer, Host, From, Nick) ->
|
|
JID = jid:encode(jid:tolower(jid:remove_resource(From))),
|
|
F = fun () ->
|
|
case Nick of
|
|
<<"">> ->
|
|
ejabberd_sql:sql_query_t(
|
|
?SQL("delete from muc_registered where"
|
|
" jid=%(JID)s and host=%(Host)s")),
|
|
ok;
|
|
_ ->
|
|
Allow = case ejabberd_sql:sql_query_t(
|
|
?SQL("select @(jid)s from muc_registered"
|
|
" where nick=%(Nick)s"
|
|
" and host=%(Host)s")) of
|
|
{selected, [{J}]} -> J == JID;
|
|
_ -> true
|
|
end,
|
|
if Allow ->
|
|
?SQL_UPSERT_T(
|
|
"muc_registered",
|
|
["!jid=%(JID)s",
|
|
"!host=%(Host)s",
|
|
"nick=%(Nick)s"]),
|
|
ok;
|
|
true ->
|
|
false
|
|
end
|
|
end
|
|
end,
|
|
ejabberd_sql:sql_transaction(LServer, F).
|
|
|
|
set_affiliation(_ServerHost, _Room, _Host, _JID, _Affiliation, _Reason) ->
|
|
{error, not_implemented}.
|
|
|
|
set_affiliations(_ServerHost, _Room, _Host, _Affiliations) ->
|
|
{error, not_implemented}.
|
|
|
|
get_affiliation(_ServerHost, _Room, _Host, _LUser, _LServer) ->
|
|
{error, not_implemented}.
|
|
|
|
get_affiliations(_ServerHost, _Room, _Host) ->
|
|
{error, not_implemented}.
|
|
|
|
search_affiliation(_ServerHost, _Room, _Host, _Affiliation) ->
|
|
{error, not_implemented}.
|
|
|
|
register_online_room(ServerHost, Room, Host, Pid) ->
|
|
PidS = enc_pid(Pid),
|
|
NodeS = erlang:atom_to_binary(node(Pid), latin1),
|
|
case ?SQL_UPSERT(ServerHost,
|
|
"muc_online_room",
|
|
["!name=%(Room)s",
|
|
"!host=%(Host)s",
|
|
"node=%(NodeS)s",
|
|
"pid=%(PidS)s"]) of
|
|
ok ->
|
|
ok;
|
|
Err ->
|
|
?ERROR_MSG("failed to update 'muc_online_room': ~p", [Err]),
|
|
Err
|
|
end.
|
|
|
|
unregister_online_room(ServerHost, Room, Host, Pid) ->
|
|
%% TODO: report errors
|
|
PidS = enc_pid(Pid),
|
|
NodeS = erlang:atom_to_binary(node(Pid), latin1),
|
|
ejabberd_sql:sql_query(
|
|
ServerHost,
|
|
?SQL("delete from muc_online_room where name=%(Room)s and "
|
|
"host=%(Host)s and node=%(NodeS)s and pid=%(PidS)s")).
|
|
|
|
find_online_room(ServerHost, Room, Host) ->
|
|
case ejabberd_sql:sql_query(
|
|
ServerHost,
|
|
?SQL("select @(pid)s, @(node)s from muc_online_room where "
|
|
"name=%(Room)s and host=%(Host)s")) of
|
|
{selected, [{PidS, NodeS}]} ->
|
|
try {ok, dec_pid(PidS, NodeS)}
|
|
catch _:{node_down, _} -> error
|
|
end;
|
|
{selected, []} ->
|
|
error;
|
|
Err ->
|
|
?ERROR_MSG("failed to select 'muc_online_room': ~p", [Err]),
|
|
error
|
|
end.
|
|
|
|
count_online_rooms(ServerHost, Host) ->
|
|
case ejabberd_sql:sql_query(
|
|
ServerHost,
|
|
?SQL("select @(count(*))d from muc_online_room "
|
|
"where host=%(Host)s")) of
|
|
{selected, [{Num}]} ->
|
|
Num;
|
|
Err ->
|
|
?ERROR_MSG("failed to select 'muc_online_room': ~p", [Err]),
|
|
0
|
|
end.
|
|
|
|
get_online_rooms(ServerHost, Host, _RSM) ->
|
|
case ejabberd_sql:sql_query(
|
|
ServerHost,
|
|
?SQL("select @(name)s, @(pid)s, @(node)s from muc_online_room "
|
|
"where host=%(Host)s")) of
|
|
{selected, Rows} ->
|
|
lists:flatmap(
|
|
fun({Room, PidS, NodeS}) ->
|
|
try [{Room, Host, dec_pid(PidS, NodeS)}]
|
|
catch _:{node_down, _} -> []
|
|
end
|
|
end, Rows);
|
|
Err ->
|
|
?ERROR_MSG("failed to select 'muc_online_room': ~p", [Err]),
|
|
[]
|
|
end.
|
|
|
|
rsm_supported() ->
|
|
false.
|
|
|
|
register_online_user(ServerHost, {U, S, R}, Room, Host) ->
|
|
NodeS = erlang:atom_to_binary(node(), latin1),
|
|
case ?SQL_UPSERT(ServerHost, "muc_online_users",
|
|
["!username=%(U)s",
|
|
"!server=%(S)s",
|
|
"!resource=%(R)s",
|
|
"!name=%(Room)s",
|
|
"!host=%(Host)s",
|
|
"node=%(NodeS)s"]) of
|
|
ok ->
|
|
ok;
|
|
Err ->
|
|
?ERROR_MSG("failed to update 'muc_online_users': ~p", [Err]),
|
|
Err
|
|
end.
|
|
|
|
unregister_online_user(ServerHost, {U, S, R}, Room, Host) ->
|
|
%% TODO: report errors
|
|
ejabberd_sql:sql_query(
|
|
ServerHost,
|
|
?SQL("delete from muc_online_users where username=%(U)s and "
|
|
"server=%(S)s and resource=%(R)s and name=%(Room)s and "
|
|
"host=%(Host)s")).
|
|
|
|
count_online_rooms_by_user(ServerHost, U, S) ->
|
|
case ejabberd_sql:sql_query(
|
|
ServerHost,
|
|
?SQL("select @(count(*))d from muc_online_users where "
|
|
"username=%(U)s and server=%(S)s")) of
|
|
{selected, [{Num}]} ->
|
|
Num;
|
|
Err ->
|
|
?ERROR_MSG("failed to select 'muc_online_users': ~p", [Err]),
|
|
0
|
|
end.
|
|
|
|
get_online_rooms_by_user(ServerHost, U, S) ->
|
|
case ejabberd_sql:sql_query(
|
|
ServerHost,
|
|
?SQL("select @(name)s, @(host)s from muc_online_users where "
|
|
"username=%(U)s and server=%(S)s")) of
|
|
{selected, Rows} ->
|
|
Rows;
|
|
Err ->
|
|
?ERROR_MSG("failed to select 'muc_online_users': ~p", [Err]),
|
|
[]
|
|
end.
|
|
|
|
export(_Server) ->
|
|
[{muc_room,
|
|
fun(Host, #muc_room{name_host = {Name, RoomHost}, opts = Opts}) ->
|
|
case str:suffix(Host, RoomHost) of
|
|
true ->
|
|
SOpts = aux:term_to_expr(Opts),
|
|
[?SQL("delete from muc_room where name=%(Name)s"
|
|
" and host=%(RoomHost)s;"),
|
|
?SQL("insert into muc_room(name, host, opts) "
|
|
"values ("
|
|
"%(Name)s, %(RoomHost)s, %(SOpts)s);")];
|
|
false ->
|
|
[]
|
|
end
|
|
end},
|
|
{muc_registered,
|
|
fun(Host, #muc_registered{us_host = {{U, S}, RoomHost},
|
|
nick = Nick}) ->
|
|
case str:suffix(Host, RoomHost) of
|
|
true ->
|
|
SJID = jid:encode(jid:make(U, S)),
|
|
[?SQL("delete from muc_registered where"
|
|
" jid=%(SJID)s and host=%(RoomHost)s;"),
|
|
?SQL("insert into muc_registered(jid, host, "
|
|
"nick) values ("
|
|
"%(SJID)s, %(RoomHost)s, %(Nick)s);")];
|
|
false ->
|
|
[]
|
|
end
|
|
end}].
|
|
|
|
import(_, _, _) ->
|
|
ok.
|
|
|
|
%%%===================================================================
|
|
%%% Internal functions
|
|
%%%===================================================================
|
|
-spec enc_pid(pid()) -> binary().
|
|
enc_pid(Pid) ->
|
|
list_to_binary(erlang:pid_to_list(Pid)).
|
|
|
|
-spec dec_pid(binary(), binary()) -> pid().
|
|
dec_pid(PidBin, NodeBin) ->
|
|
PidStr = binary_to_list(PidBin),
|
|
Pid = erlang:list_to_pid(PidStr),
|
|
case erlang:binary_to_atom(NodeBin, latin1) of
|
|
Node when Node == node() ->
|
|
Pid;
|
|
Node ->
|
|
try set_node_id(PidStr, NodeBin)
|
|
catch _:badarg ->
|
|
erlang:error({node_down, Node})
|
|
end
|
|
end.
|
|
|
|
-spec set_node_id(string(), binary()) -> pid().
|
|
set_node_id(PidStr, NodeBin) ->
|
|
ExtPidStr = erlang:pid_to_list(
|
|
binary_to_term(
|
|
<<131,103,100,(size(NodeBin)):16,NodeBin/binary,0:72>>)),
|
|
[H|_] = string:tokens(ExtPidStr, "."),
|
|
[_|T] = string:tokens(PidStr, "."),
|
|
erlang:list_to_pid(string:join([H|T], ".")).
|
|
|
|
clean_tables(ServerHost) ->
|
|
NodeS = erlang:atom_to_binary(node(), latin1),
|
|
?INFO_MSG("Cleaning SQL muc_online_room table...", []),
|
|
case ejabberd_sql:sql_query(
|
|
ServerHost,
|
|
?SQL("delete from muc_online_room where node=%(NodeS)s")) of
|
|
{updated, _} ->
|
|
ok;
|
|
Err1 ->
|
|
?ERROR_MSG("failed to clean 'muc_online_room' table: ~p", [Err1]),
|
|
Err1
|
|
end,
|
|
?INFO_MSG("Cleaning SQL muc_online_users table...", []),
|
|
case ejabberd_sql:sql_query(
|
|
ServerHost,
|
|
?SQL("delete from muc_online_users where node=%(NodeS)s")) of
|
|
{updated, _} ->
|
|
ok;
|
|
Err2 ->
|
|
?ERROR_MSG("failed to clean 'muc_online_users' table: ~p", [Err2]),
|
|
Err2
|
|
end.
|