2007-12-24 14:57:53 +01:00
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
%%% File : odbc_queries.erl
|
|
|
|
%%% Author : Mickael Remond <mremond@process-one.net>
|
|
|
|
%%% Purpose : ODBC queries dependind on back-end
|
|
|
|
%%% Created : by Mickael Remond <mremond@process-one.net>
|
|
|
|
%%%
|
|
|
|
%%%
|
2008-07-31 13:18:49 +02:00
|
|
|
%%% ejabberd, Copyright (C) 2002-2008 ProcessOne
|
2007-12-24 14:57:53 +01:00
|
|
|
%%%
|
|
|
|
%%% 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.
|
2008-02-15 16:39:30 +01:00
|
|
|
%%%
|
2007-12-24 14:57:53 +01:00
|
|
|
%%% 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., 59 Temple Place, Suite 330, Boston, MA
|
|
|
|
%%% 02111-1307 USA
|
|
|
|
%%%
|
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
|
2006-09-03 17:15:46 +02:00
|
|
|
-module(odbc_queries).
|
2007-12-24 14:57:53 +01:00
|
|
|
-author("mremond@process-one.net").
|
2006-09-03 17:15:46 +02:00
|
|
|
|
|
|
|
-export([get_db_type/0,
|
|
|
|
sql_transaction/2,
|
|
|
|
get_last/2,
|
|
|
|
set_last_t/4,
|
|
|
|
del_last/2,
|
|
|
|
get_password/2,
|
|
|
|
set_password_t/3,
|
|
|
|
add_user/3,
|
|
|
|
del_user/2,
|
|
|
|
del_user_return_password/3,
|
|
|
|
list_users/1,
|
2007-11-03 18:35:23 +01:00
|
|
|
list_users/2,
|
2007-05-12 20:09:38 +02:00
|
|
|
users_number/1,
|
2007-11-03 18:35:23 +01:00
|
|
|
users_number/2,
|
2006-09-03 17:15:46 +02:00
|
|
|
add_spool_sql/2,
|
|
|
|
add_spool/2,
|
|
|
|
get_and_del_spool_msg_t/2,
|
|
|
|
del_spool_msg/2,
|
|
|
|
get_roster/2,
|
|
|
|
get_roster_jid_groups/2,
|
|
|
|
get_roster_groups/3,
|
|
|
|
del_user_roster_t/2,
|
|
|
|
get_roster_by_jid/3,
|
|
|
|
get_rostergroup_by_jid/3,
|
|
|
|
del_roster/3,
|
|
|
|
del_roster_sql/2,
|
|
|
|
update_roster/5,
|
|
|
|
update_roster_sql/4,
|
|
|
|
roster_subscribe/4,
|
|
|
|
get_subscription/3,
|
2006-11-05 16:51:26 +01:00
|
|
|
set_private_data/4,
|
2006-11-20 14:20:47 +01:00
|
|
|
set_private_data_sql/3,
|
2006-11-05 16:51:26 +01:00
|
|
|
get_private_data/3,
|
|
|
|
del_user_private_storage/2,
|
2007-08-28 16:36:39 +02:00
|
|
|
escape/1,
|
|
|
|
count_records_where/3]).
|
2006-09-03 17:15:46 +02:00
|
|
|
|
2007-11-30 19:06:26 +01:00
|
|
|
%% We have only two compile time options for db queries:
|
2006-09-03 17:15:46 +02:00
|
|
|
%-define(generic, true).
|
|
|
|
%-define(mssql, true).
|
2007-12-01 15:25:31 +01:00
|
|
|
-ifndef(mssql).
|
|
|
|
-undef(generic).
|
|
|
|
-define(generic, true).
|
|
|
|
-endif.
|
2006-09-03 17:15:46 +02:00
|
|
|
|
|
|
|
%% -----------------
|
|
|
|
%% Generic queries
|
|
|
|
-ifdef(generic).
|
|
|
|
|
|
|
|
get_db_type() ->
|
|
|
|
generic.
|
|
|
|
|
2006-11-20 14:20:47 +01:00
|
|
|
%% F can be either a fun or a list of queries
|
|
|
|
%% TODO: We should probably move the list of queries transaction wrapper from the ejabberd_odbc module
|
|
|
|
%% to this one (odbc_queries)
|
2006-09-03 17:15:46 +02:00
|
|
|
sql_transaction(LServer, F) ->
|
|
|
|
ejabberd_odbc:sql_transaction(LServer, F).
|
|
|
|
|
|
|
|
get_last(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["select seconds, state from last "
|
|
|
|
"where username='", Username, "'"]).
|
|
|
|
|
|
|
|
set_last_t(LServer, Username, Seconds, State) ->
|
|
|
|
%% MREMOND: I think this should be turn into a non transactional behaviour
|
|
|
|
ejabberd_odbc:sql_transaction(
|
|
|
|
LServer,
|
|
|
|
[["delete from last where username='", Username, "';"],
|
|
|
|
["insert into last(username, seconds, state) "
|
|
|
|
"values ('", Username, "', '", Seconds, "', '", State, "');"]]).
|
|
|
|
|
|
|
|
del_last(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["delete from last where username='", Username, "'"]).
|
|
|
|
|
|
|
|
get_password(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["select password from users "
|
|
|
|
"where username='", Username, "';"]).
|
|
|
|
|
|
|
|
set_password_t(LServer, Username, Pass) ->
|
|
|
|
ejabberd_odbc:sql_transaction(
|
|
|
|
LServer,
|
|
|
|
[["delete from users where username='", Username ,"';"],
|
|
|
|
["insert into users(username, password) "
|
|
|
|
"values ('", Username, "', '", Pass, "');"]]).
|
|
|
|
|
|
|
|
add_user(LServer, Username, Pass) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["insert into users(username, password) "
|
|
|
|
"values ('", Username, "', '", Pass, "');"]).
|
|
|
|
|
|
|
|
del_user(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["delete from users where username='", Username ,"';"]).
|
|
|
|
|
|
|
|
del_user_return_password(_LServer, Username, Pass) ->
|
2008-05-31 19:09:13 +02:00
|
|
|
P = ejabberd_odbc:sql_query_t(
|
|
|
|
["select password from users where username='",
|
|
|
|
Username, "';"]),
|
2006-09-03 17:15:46 +02:00
|
|
|
ejabberd_odbc:sql_query_t(["delete from users "
|
|
|
|
"where username='", Username,
|
2008-05-31 19:09:13 +02:00
|
|
|
"' and password='", Pass, "';"]),
|
|
|
|
P.
|
2006-09-03 17:15:46 +02:00
|
|
|
|
|
|
|
list_users(LServer) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
"select username from users").
|
|
|
|
|
2007-11-03 18:35:23 +01:00
|
|
|
list_users(LServer, [{from, Start}, {to, End}]) when is_integer(Start) and
|
|
|
|
is_integer(End) ->
|
2008-02-15 16:39:30 +01:00
|
|
|
list_users(LServer, [{limit, End-Start+1}, {offset, Start-1}]);
|
2007-11-03 18:35:23 +01:00
|
|
|
list_users(LServer, [{prefix, Prefix}, {from, Start}, {to, End}]) when is_list(Prefix) and
|
|
|
|
is_integer(Start) and
|
|
|
|
is_integer(End) ->
|
2008-02-15 16:39:30 +01:00
|
|
|
list_users(LServer, [{prefix, Prefix}, {limit, End-Start+1}, {offset, Start-1}]);
|
2007-11-03 18:35:23 +01:00
|
|
|
|
|
|
|
list_users(LServer, [{limit, Limit}, {offset, Offset}]) when is_integer(Limit) and
|
|
|
|
is_integer(Offset) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
io_lib:format(
|
|
|
|
"select username from users " ++
|
|
|
|
"order by username " ++
|
|
|
|
"limit ~w offset ~w", [Limit, Offset]));
|
|
|
|
list_users(LServer, [{prefix, Prefix},
|
|
|
|
{limit, Limit},
|
|
|
|
{offset, Offset}]) when is_list(Prefix) and
|
|
|
|
is_integer(Limit) and
|
|
|
|
is_integer(Offset) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
io_lib:format("select username from users " ++
|
|
|
|
"where username like '~s%' " ++
|
|
|
|
"order by username " ++
|
|
|
|
"limit ~w offset ~w ", [Prefix, Limit, Offset])).
|
|
|
|
|
2007-05-12 20:09:38 +02:00
|
|
|
users_number(LServer) ->
|
2007-12-20 18:12:45 +01:00
|
|
|
case element(1, ejabberd_config:get_local_option({odbc_server, LServer})) of
|
|
|
|
mysql ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
"select table_rows from information_schema.tables where table_name='users'");
|
|
|
|
pgsql ->
|
|
|
|
case ejabberd_config:get_local_option({pgsql_users_number_estimate, LServer}) of
|
2007-05-12 20:09:38 +02:00
|
|
|
true ->
|
|
|
|
ejabberd_odbc:sql_query(
|
2007-12-20 18:12:45 +01:00
|
|
|
LServer,
|
|
|
|
"select reltuples from pg_class where oid = 'users'::regclass::oid");
|
2007-05-12 20:09:38 +02:00
|
|
|
_ ->
|
|
|
|
ejabberd_odbc:sql_query(
|
2007-12-20 18:12:45 +01:00
|
|
|
LServer,
|
|
|
|
"select count(*) from users")
|
|
|
|
end;
|
|
|
|
_ ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
"select count(*) from users")
|
2007-05-12 20:09:38 +02:00
|
|
|
end.
|
|
|
|
|
2007-11-03 18:35:23 +01:00
|
|
|
users_number(LServer, [{prefix, Prefix}]) when is_list(Prefix) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
io_lib:fwrite("select count(*) from users " ++
|
|
|
|
%% Warning: Escape prefix at higher level to prevent SQL
|
2008-02-15 16:39:30 +01:00
|
|
|
%% injection.
|
2007-11-03 18:35:23 +01:00
|
|
|
"where username like '~s%'", [Prefix]));
|
|
|
|
users_number(LServer, []) ->
|
|
|
|
users_number(LServer).
|
|
|
|
|
|
|
|
|
2006-09-03 17:15:46 +02:00
|
|
|
add_spool_sql(Username, XML) ->
|
|
|
|
["insert into spool(username, xml) "
|
|
|
|
"values ('", Username, "', '",
|
|
|
|
XML,
|
|
|
|
"');"].
|
|
|
|
|
|
|
|
add_spool(LServer, Queries) ->
|
|
|
|
ejabberd_odbc:sql_transaction(
|
|
|
|
LServer, Queries).
|
|
|
|
|
|
|
|
get_and_del_spool_msg_t(LServer, Username) ->
|
|
|
|
F = fun() ->
|
|
|
|
Result = ejabberd_odbc:sql_query_t(
|
|
|
|
["select username, xml from spool where username='", Username, "'"
|
|
|
|
" order by seq;"]),
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["delete from spool where username='", Username, "';"]),
|
|
|
|
Result
|
|
|
|
end,
|
|
|
|
ejabberd_odbc:sql_transaction(LServer,F).
|
|
|
|
|
|
|
|
del_spool_msg(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["delete from spool where username='", Username, "';"]).
|
|
|
|
|
|
|
|
get_roster(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["select username, jid, nick, subscription, ask, "
|
|
|
|
"askmessage, server, subscribe, type from rosterusers "
|
|
|
|
"where username='", Username, "'"]).
|
|
|
|
|
|
|
|
get_roster_jid_groups(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["select jid, grp from rostergroups "
|
|
|
|
"where username='", Username, "'"]).
|
|
|
|
|
|
|
|
get_roster_groups(_LServer, Username, SJID) ->
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["select grp from rostergroups "
|
|
|
|
"where username='", Username, "' "
|
|
|
|
"and jid='", SJID, "';"]).
|
|
|
|
|
|
|
|
del_user_roster_t(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_transaction(
|
|
|
|
LServer,
|
|
|
|
fun() ->
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["delete from rosterusers "
|
|
|
|
" where username='", Username, "';"]),
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["delete from rostergroups "
|
|
|
|
" where username='", Username, "';"])
|
|
|
|
end).
|
|
|
|
|
|
|
|
get_roster_by_jid(_LServer, Username, SJID) ->
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["select username, jid, nick, subscription, "
|
|
|
|
"ask, askmessage, server, subscribe, type from rosterusers "
|
|
|
|
"where username='", Username, "' "
|
|
|
|
"and jid='", SJID, "';"]).
|
|
|
|
|
|
|
|
get_rostergroup_by_jid(LServer, Username, SJID) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["select grp from rostergroups "
|
|
|
|
"where username='", Username, "' "
|
|
|
|
"and jid='", SJID, "'"]).
|
|
|
|
|
|
|
|
del_roster(_LServer, Username, SJID) ->
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["delete from rosterusers "
|
|
|
|
" where username='", Username, "' "
|
|
|
|
" and jid='", SJID, "';"]),
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["delete from rostergroups "
|
|
|
|
" where username='", Username, "' "
|
|
|
|
" and jid='", SJID, "';"]).
|
|
|
|
|
|
|
|
del_roster_sql(Username, SJID) ->
|
|
|
|
[["delete from rosterusers "
|
|
|
|
" where username='", Username, "' "
|
|
|
|
" and jid='", SJID, "';"],
|
|
|
|
["delete from rostergroups "
|
|
|
|
" where username='", Username, "' "
|
|
|
|
" and jid='", SJID, "';"]].
|
|
|
|
|
|
|
|
update_roster(_LServer, Username, SJID, ItemVals, ItemGroups) ->
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["delete from rosterusers "
|
|
|
|
" where username='", Username, "' "
|
|
|
|
" and jid='", SJID, "';"]),
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["insert into rosterusers("
|
|
|
|
" username, jid, nick, "
|
|
|
|
" subscription, ask, askmessage, "
|
|
|
|
" server, subscribe, type) "
|
|
|
|
" values (", ItemVals, ");"]),
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["delete from rostergroups "
|
|
|
|
" where username='", Username, "' "
|
|
|
|
" and jid='", SJID, "';"]),
|
|
|
|
lists:foreach(fun(ItemGroup) ->
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["insert into rostergroups("
|
|
|
|
" username, jid, grp) "
|
|
|
|
" values (", ItemGroup, ");"])
|
|
|
|
end,
|
|
|
|
ItemGroups).
|
|
|
|
|
|
|
|
update_roster_sql(Username, SJID, ItemVals, ItemGroups) ->
|
|
|
|
[["delete from rosterusers "
|
|
|
|
" where username='", Username, "' "
|
|
|
|
" and jid='", SJID, "';"],
|
|
|
|
["insert into rosterusers("
|
|
|
|
" username, jid, nick, "
|
2008-10-03 17:49:24 +02:00
|
|
|
" subscription, ask, askmessage, "
|
2006-09-03 17:15:46 +02:00
|
|
|
" server, subscribe, type) "
|
|
|
|
" values (", ItemVals, ");"],
|
|
|
|
["delete from rostergroups "
|
|
|
|
" where username='", Username, "' "
|
2008-10-03 17:47:30 +02:00
|
|
|
" and jid='", SJID, "';"]] ++
|
2006-09-03 17:15:46 +02:00
|
|
|
[["insert into rostergroups("
|
|
|
|
" username, jid, grp) "
|
|
|
|
" values (", ItemGroup, ");"] ||
|
2008-10-03 17:47:30 +02:00
|
|
|
ItemGroup <- ItemGroups].
|
2006-09-03 17:15:46 +02:00
|
|
|
|
|
|
|
roster_subscribe(_LServer, Username, SJID, ItemVals) ->
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["delete from rosterusers "
|
|
|
|
" where username='", Username, "' "
|
|
|
|
" and jid='", SJID, "';"]),
|
|
|
|
ejabberd_odbc:sql_query_t(
|
|
|
|
["insert into rosterusers("
|
|
|
|
" username, jid, nick, "
|
|
|
|
" subscription, ask, askmessage, "
|
|
|
|
" server, subscribe, type) "
|
|
|
|
" values (", ItemVals, ");"]).
|
|
|
|
|
|
|
|
get_subscription(LServer, Username, SJID) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["select subscription from rosterusers "
|
|
|
|
"where username='", Username, "' "
|
|
|
|
"and jid='", SJID, "'"]).
|
|
|
|
|
2006-11-05 16:51:26 +01:00
|
|
|
set_private_data(_LServer, Username, LXMLNS, SData) ->
|
2006-11-20 14:20:47 +01:00
|
|
|
lists:foreach(fun(Query) ->
|
|
|
|
ejabberd_odbc:sql_query_t(Query)
|
|
|
|
end,
|
|
|
|
set_private_data_sql(Username, LXMLNS, SData)).
|
2008-02-15 16:39:30 +01:00
|
|
|
|
2006-11-20 14:20:47 +01:00
|
|
|
set_private_data_sql(Username, LXMLNS, SData) ->
|
|
|
|
[["delete from private_storage "
|
2006-11-05 16:51:26 +01:00
|
|
|
"where username='", Username, "' and "
|
2006-11-20 14:20:47 +01:00
|
|
|
"namespace='", LXMLNS, "';"],
|
2006-11-05 16:51:26 +01:00
|
|
|
["insert into private_storage(username, namespace, data) "
|
|
|
|
"values ('", Username, "', '", LXMLNS, "', "
|
2008-02-15 16:39:30 +01:00
|
|
|
"'", SData, "');"]].
|
|
|
|
|
2006-11-05 16:51:26 +01:00
|
|
|
get_private_data(LServer, Username, LXMLNS) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["select data from private_storage "
|
|
|
|
"where username='", Username, "' and "
|
|
|
|
"namespace='", LXMLNS, "';"]).
|
|
|
|
|
|
|
|
del_user_private_storage(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_transaction(
|
|
|
|
LServer,
|
|
|
|
["delete from private_storage where username='", Username, "';"]).
|
|
|
|
|
2006-09-03 17:15:46 +02:00
|
|
|
%% Characters to escape
|
|
|
|
escape($\0) -> "\\0";
|
|
|
|
escape($\n) -> "\\n";
|
|
|
|
escape($\t) -> "\\t";
|
|
|
|
escape($\b) -> "\\b";
|
|
|
|
escape($\r) -> "\\r";
|
|
|
|
escape($') -> "\\'";
|
|
|
|
escape($") -> "\\\"";
|
|
|
|
escape($\\) -> "\\\\";
|
|
|
|
escape(C) -> C.
|
|
|
|
|
2007-08-28 16:36:39 +02:00
|
|
|
%% Count number of records in a table given a where clause
|
|
|
|
count_records_where(LServer, Table, WhereClause) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["select count(*) from ", Table, " ", WhereClause, ";"]).
|
2006-09-03 17:15:46 +02:00
|
|
|
-endif.
|
|
|
|
|
|
|
|
%% -----------------
|
|
|
|
%% MSSQL queries
|
|
|
|
-ifdef(mssql).
|
|
|
|
|
|
|
|
get_db_type() ->
|
|
|
|
mssql.
|
|
|
|
|
2006-11-20 14:20:47 +01:00
|
|
|
%% Queries can be either a fun or a list of queries
|
2007-12-17 15:10:26 +01:00
|
|
|
sql_transaction(LServer, Queries) when is_list(Queries) ->
|
2006-11-20 14:20:47 +01:00
|
|
|
%% SQL transaction based on a list of queries
|
2008-02-15 16:39:30 +01:00
|
|
|
%% This function automatically
|
2006-11-20 14:20:47 +01:00
|
|
|
F = fun() ->
|
|
|
|
lists:foreach(fun(Query) ->
|
2007-12-17 15:10:26 +01:00
|
|
|
ejabberd_odbc:sql_query(LServer, Query)
|
|
|
|
end, Queries)
|
2006-11-20 14:20:47 +01:00
|
|
|
end,
|
|
|
|
{atomic, catch F()};
|
|
|
|
sql_transaction(_LServer, FQueries) ->
|
|
|
|
{atomic, catch FQueries()}.
|
2006-09-03 17:15:46 +02:00
|
|
|
|
|
|
|
get_last(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.get_last '", Username, "'"]).
|
|
|
|
|
|
|
|
set_last_t(LServer, Username, Seconds, State) ->
|
|
|
|
Result = ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.set_last '", Username, "', '", Seconds,
|
|
|
|
"', '", State, "'"]),
|
|
|
|
{atomic, Result}.
|
|
|
|
|
|
|
|
del_last(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
2008-02-15 16:39:30 +01:00
|
|
|
LServer,
|
2006-09-03 17:15:46 +02:00
|
|
|
["EXECUTE dbo.del_last '", Username, "'"]).
|
|
|
|
|
|
|
|
get_password(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.get_password '", Username, "'"]).
|
|
|
|
|
|
|
|
set_password_t(LServer, Username, Pass) ->
|
|
|
|
Result = ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.set_password '", Username, "', '", Pass, "'"]),
|
|
|
|
{atomic, Result}.
|
|
|
|
|
|
|
|
add_user(LServer, Username, Pass) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.add_user '", Username, "', '", Pass, "'"]).
|
|
|
|
|
|
|
|
del_user(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.del_user '", Username ,"'"]).
|
|
|
|
|
|
|
|
del_user_return_password(LServer, Username, Pass) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.del_user_return_password '", Username, "'"]),
|
|
|
|
Pass.
|
|
|
|
|
|
|
|
list_users(LServer) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
"EXECUTE dbo.list_users").
|
|
|
|
|
2007-12-11 18:43:49 +01:00
|
|
|
list_users(LServer, _) ->
|
|
|
|
% scope listing not supported
|
|
|
|
list_users(LServer).
|
2008-02-15 16:39:30 +01:00
|
|
|
|
2007-06-18 12:22:54 +02:00
|
|
|
users_number(LServer) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
"select count(*) from users with (nolock)").
|
|
|
|
|
2007-12-11 18:43:49 +01:00
|
|
|
users_number(LServer, _) ->
|
|
|
|
% scope listing not supported
|
|
|
|
users_number(LServer).
|
|
|
|
|
2006-09-03 17:15:46 +02:00
|
|
|
add_spool_sql(Username, XML) ->
|
|
|
|
["EXECUTE dbo.add_spool '", Username, "' , '",XML,"'"].
|
|
|
|
|
|
|
|
add_spool(LServer, Queries) ->
|
|
|
|
lists:foreach(fun(Query) ->
|
|
|
|
ejabberd_odbc:sql_query(LServer, Query)
|
|
|
|
end,
|
|
|
|
Queries).
|
|
|
|
|
|
|
|
get_and_del_spool_msg_t(LServer, Username) ->
|
|
|
|
[Result] = case ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.get_and_del_spool_msg '", Username, "'"]) of
|
|
|
|
Rs when list(Rs) ->
|
|
|
|
lists:filter(fun({selected, _Header, _Row}) ->
|
|
|
|
true;
|
|
|
|
({updated, _N}) ->
|
|
|
|
false
|
|
|
|
end,
|
|
|
|
Rs);
|
|
|
|
Rs -> [Rs]
|
|
|
|
end,
|
|
|
|
{atomic, Result}.
|
|
|
|
|
|
|
|
del_spool_msg(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.del_spool_msg '", Username, "'"]).
|
|
|
|
|
|
|
|
get_roster(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.get_roster '", Username, "'"]).
|
|
|
|
|
|
|
|
get_roster_jid_groups(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.get_roster_jid_groups '", Username, "'"]).
|
|
|
|
|
|
|
|
get_roster_groups(LServer, Username, SJID) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.get_roster_groups '", Username, "' , '", SJID, "'"]).
|
2008-02-15 16:39:30 +01:00
|
|
|
|
2006-09-03 17:15:46 +02:00
|
|
|
del_user_roster_t(LServer, Username) ->
|
|
|
|
Result = ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.del_user_roster '", Username, "'"]),
|
|
|
|
{atomic, Result}.
|
|
|
|
|
|
|
|
get_roster_by_jid(LServer, Username, SJID) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.get_roster_by_jid '", Username, "' , '", SJID, "'"]).
|
|
|
|
|
|
|
|
get_rostergroup_by_jid(LServer, Username, SJID) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.get_rostergroup_by_jid '", Username, "' , '", SJID, "'"]).
|
|
|
|
|
|
|
|
del_roster(LServer, Username, SJID) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.del_roster '", Username, "', '", SJID, "'"]).
|
|
|
|
|
|
|
|
del_roster_sql(Username, SJID) ->
|
|
|
|
["EXECUTE dbo.del_roster '", Username, "', '", SJID, "'"].
|
|
|
|
|
|
|
|
update_roster(LServer, Username, SJID, ItemVals, ItemGroups) ->
|
|
|
|
Query1 = ["EXECUTE dbo.del_roster '", Username, "', '", SJID, "' "],
|
|
|
|
ejabberd_odbc:sql_query(LServer, lists:flatten(Query1)),
|
|
|
|
Query2 = ["EXECUTE dbo.add_roster_user ", ItemVals],
|
|
|
|
ejabberd_odbc:sql_query(LServer, lists:flatten(Query2)),
|
|
|
|
Query3 = ["EXECUTE dbo.del_roster_groups '", Username, "', '", SJID, "' "],
|
|
|
|
ejabberd_odbc:sql_query(LServer, lists:flatten(Query3)),
|
|
|
|
lists:foreach(fun(ItemGroup) ->
|
|
|
|
Query = ["EXECUTE dbo.add_roster_group ",
|
|
|
|
ItemGroup],
|
|
|
|
ejabberd_odbc:sql_query(LServer,
|
|
|
|
lists:flatten(Query))
|
|
|
|
end,
|
|
|
|
ItemGroups).
|
|
|
|
|
|
|
|
update_roster_sql(Username, SJID, ItemVals, ItemGroups) ->
|
|
|
|
["BEGIN TRANSACTION ",
|
|
|
|
"EXECUTE dbo.del_roster_groups '", Username, "','", SJID, "' ",
|
|
|
|
"EXECUTE dbo.add_roster_user ", ItemVals, " "] ++
|
|
|
|
[lists:flatten("EXECUTE dbo.add_roster_group ", ItemGroup, " ")
|
|
|
|
|| ItemGroup <- ItemGroups] ++
|
|
|
|
["COMMIT"].
|
|
|
|
|
|
|
|
roster_subscribe(LServer, _Username, _SJID, ItemVals) ->
|
|
|
|
catch ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.add_roster_user ", ItemVals]).
|
|
|
|
|
|
|
|
get_subscription(LServer, Username, SJID) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.get_subscription '", Username, "' , '", SJID, "'"]).
|
2006-11-20 14:20:47 +01:00
|
|
|
|
2007-12-11 18:43:49 +01:00
|
|
|
set_private_data(LServer, Username, LXMLNS, SData) ->
|
2006-11-05 16:51:26 +01:00
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
2006-11-20 14:20:47 +01:00
|
|
|
set_private_data_sql(Username, LXMLNS, SData)).
|
|
|
|
|
|
|
|
set_private_data_sql(Username, LXMLNS, SData) ->
|
|
|
|
["EXECUTE dbo.set_private_data '", Username, "' , '", LXMLNS, "' , '", SData, "'"].
|
2008-02-15 16:39:30 +01:00
|
|
|
|
2006-11-05 16:51:26 +01:00
|
|
|
get_private_data(LServer, Username, LXMLNS) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.get_private_data '", Username, "' , '", LXMLNS, "'"]).
|
|
|
|
|
|
|
|
del_user_private_storage(LServer, Username) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["EXECUTE dbo.del_user_storage '", Username, "'"]).
|
2006-09-03 17:15:46 +02:00
|
|
|
|
|
|
|
%% Characters to escape
|
|
|
|
escape($\0) -> "\\0";
|
|
|
|
escape($\t) -> "\\t";
|
|
|
|
escape($\b) -> "\\b";
|
|
|
|
escape($\r) -> "\\r";
|
|
|
|
escape($') -> "\''";
|
|
|
|
escape($") -> "\\\"";
|
|
|
|
escape(C) -> C.
|
2007-08-28 16:36:39 +02:00
|
|
|
|
|
|
|
%% Count number of records in a table given a where clause
|
|
|
|
count_records_where(LServer, Table, WhereClause) ->
|
|
|
|
ejabberd_odbc:sql_query(
|
|
|
|
LServer,
|
|
|
|
["select count(*) from ", Table, " ", WhereClause, " with (nolock)"]).
|
2006-09-03 17:15:46 +02:00
|
|
|
-endif.
|