%%%------------------------------------------------------------------- %%% File : mod_muc_sql.erl %%% Author : Evgeny Khramtsov %%% Created : 13 Apr 2016 by Evgeny Khramtsov %%% %%% %%% 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/3, unregister_online_room/3, find_online_room/2, get_online_rooms/2, count_online_rooms/1, rsm_supported/0, register_online_user/3, unregister_online_user/3, count_online_rooms_by_user/2, get_online_rooms_by_user/2]). -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) -> ok. store_room(LServer, Host, Name, Opts) -> SOpts = jlib: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:to_string(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:to_string(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:to_string(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(_, _, _) -> erlang:error(not_implemented). unregister_online_room(_, _, _) -> erlang:error(not_implemented). find_online_room(_, _) -> erlang:error(not_implemented). count_online_rooms(_) -> erlang:error(not_implemented). get_online_rooms(_, _) -> erlang:error(not_implemented). rsm_supported() -> false. register_online_user(_, _, _) -> erlang:error(not_implemented). unregister_online_user(_, _, _) -> erlang:error(not_implemented). count_online_rooms_by_user(_, _) -> erlang:error(not_implemented). get_online_rooms_by_user(_, _) -> erlang:error(not_implemented). export(_Server) -> [{muc_room, fun(Host, #muc_room{name_host = {Name, RoomHost}, opts = Opts}) -> case str:suffix(Host, RoomHost) of true -> SOpts = jlib: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:to_string(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 %%%===================================================================