Packages

ejabberd

16.4.1
26.4.0 26.3.0 26.2.0 26.1.0 25.10.0 25.8.0 25.7.0 25.4.0 25.3.0 24.12.0 24.10.0 24.7.0 24.6.0 24.2.6 23.10.0 23.4.0 23.1.0 22.10.0 22.5.0 21.12.0 21.7.0 21.4.0 21.1.0 20.12.0 20.7.0 20.4.0 20.3.0 20.2.0 20.1.0 19.9.1 19.9.0 19.8.0 19.5.0 19.2.0 18.12.1 18.12.0 18.6.0 18.4.0 18.3.0 18.1.0 17.11.0 17.9.0 17.6.0 17.3.0 17.1.0 16.12.0-beta1 16.9.0 16.8.0 16.6.2 16.6.1 16.6.0 16.4.1 16.4.0 16.3.0 16.2.0 16.1.0-beta1

Robust, Ubiquitous and Massively Scalable Messaging Platform (XMPP, MQTT, SIP Server)

HexDocs

Current section

Files

Jump to
Readme
56 Versions
18 Dependencies
Dependants
1 Advisory
Files
Activity
ejabberd src mod_irc_sql.erl
Raw

src/mod_irc_sql.erl 

%%%-------------------------------------------------------------------

%%% @author Evgeny Khramtsov <ekhramtsov@process-one.net>

%%% @copyright (C) 2016, Evgeny Khramtsov

%%% @doc

%%%

%%% @end

%%% Created : 14 Apr 2016 by Evgeny Khramtsov <ekhramtsov@process-one.net>

%%%-------------------------------------------------------------------

-module(mod_irc_sql).



-behaviour(mod_irc).



%% API

-export([init/2, get_data/3, set_data/4, import/1, import/2, export/1]).



-include("jlib.hrl").

-include("mod_irc.hrl").



%%%===================================================================

%%% API

%%%===================================================================

init(_Host, _Opts) ->

    ok.



get_data(LServer, Host, From) ->

    LJID = jid:tolower(jid:remove_resource(From)),

    SJID = ejabberd_sql:escape(jid:to_string(LJID)),

    SHost = ejabberd_sql:escape(Host),

    case catch ejabberd_sql:sql_query(

		 LServer,

		 [<<"select data from irc_custom where jid='">>,

		  SJID, <<"' and host='">>, SHost,

		  <<"';">>]) of

	{selected, [<<"data">>], [[SData]]} ->

	    mod_irc:data_to_binary(From, ejabberd_sql:decode_term(SData));

	{'EXIT', _} -> error;

	{selected, _, _} -> empty

    end.



set_data(LServer, Host, From, Data) ->

    LJID = jid:tolower(jid:remove_resource(From)),

    SJID = ejabberd_sql:escape(jid:to_string(LJID)),

    SHost = ejabberd_sql:escape(Host),

    SData = ejabberd_sql:encode_term(Data),

    F = fun () ->

		sql_queries:update_t(<<"irc_custom">>,

				      [<<"jid">>, <<"host">>, <<"data">>],

				      [SJID, SHost, SData],

				      [<<"jid='">>, SJID, <<"' and host='">>,

				       SHost, <<"'">>]),

		ok

	end,

    ejabberd_sql:sql_transaction(LServer, F).



export(_Server) ->

    [{irc_custom,

      fun(Host, #irc_custom{us_host = {{U, S}, IRCHost},

                            data = Data}) ->

              case str:suffix(Host, IRCHost) of

                  true ->

                      SJID = ejabberd_sql:escape(

                               jid:to_string(

                                 jid:make(U, S, <<"">>))),

                      SIRCHost = ejabberd_sql:escape(IRCHost),

                      SData = ejabberd_sql:encode_term(Data),

                      [[<<"delete from irc_custom where jid='">>, SJID,

                        <<"' and host='">>, SIRCHost, <<"';">>],

                       [<<"insert into irc_custom(jid, host, "

                          "data) values ('">>,

                        SJID, <<"', '">>, SIRCHost, <<"', '">>, SData,

                        <<"');">>]];

                  false ->

                      []

              end

      end}].



import(_LServer) ->

    [{<<"select jid, host, data from irc_custom;">>,

      fun([SJID, IRCHost, SData]) ->

              #jid{luser = U, lserver = S} = jid:from_string(SJID),

              Data = ejabberd_sql:decode_term(SData),

              #irc_custom{us_host = {{U, S}, IRCHost},

                          data = Data}

      end}].



import(_, _) ->

    pass.



%%%===================================================================

%%% Internal functions

%%%===================================================================