1162

Committed 20 Oct 2025 08:53AM UTC coverage: 33.87% (+0.1%) from 33.752%

Build # 1162

Build Type

push

github

Committed by

badlop

Commit Message

CI: Fix updating ubuntu packages

Run Details

15494 of 45746 relevant lines covered (33.87%)

1082.17 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

53.29

/src/nodetree_tree_sql.erl

%%%----------------------------------------------------------------------
%%% File    : nodetree_tree_sql.erl
%%% Author  : Christophe Romain <christophe.romain@process-one.net>
%%% Purpose : Standard node tree plugin with ODBC backend
%%% Created :  1 Dec 2007 by Christophe Romain <christophe.romain@process-one.net>
%%%
%%%
%%% ejabberd, Copyright (C) 2002-2025   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.
%%%
%%%----------------------------------------------------------------------

%%% @doc The module <strong>{@module}</strong> is the default PubSub node tree plugin.
%%% <p>It is used as a default for all unknown PubSub node type.  It can serve
%%% as a developer basis and reference to build its own custom pubsub node tree
%%% types.</p>
%%% <p>PubSub node tree plugins are using the {@link gen_nodetree} behaviour.</p>
%%% <p><strong>The API isn't stabilized yet</strong>. The pubsub plugin
%%% development is still a work in progress. However, the system is already
%%% usable and useful as is. Please, send us comments, feedback and
%%% improvements.</p>

-module(nodetree_tree_sql).
-behaviour(gen_pubsub_nodetree).
-author('christophe.romain@process-one.net').


-include("pubsub.hrl").
-include_lib("xmpp/include/xmpp.hrl").
-include("ejabberd_sql_pt.hrl").
-include("translate.hrl").

-export([init/3, terminate/2, options/0, set_node/1,
    get_node/3, get_node/2, get_node/1, get_nodes/2,
    get_nodes/1, get_all_nodes/1,
    get_parentnodes/3, get_parentnodes_tree/3,
    get_subnodes/3, get_subnodes_tree/3, create_node/6,
    delete_node/2]).

-export([raw_to_node/2]).

init(_Host, _ServerHost, _Opts) ->
    ok.

terminate(_Host, _ServerHost) ->
    ok.

options() ->
    [{sql, true} | nodetree_tree:options()].

set_node(Record) when is_record(Record, pubsub_node) ->
    {Host, Node} = Record#pubsub_node.nodeid,
    Parent = case Record#pubsub_node.parents of
        [] -> <<>>;
        [First | _] -> First
    end,
    Type = Record#pubsub_node.type,
    H = node_flat_sql:encode_host(Host),
    Nidx = case nodeidx(Host, Node) of
        {result, OldNidx} ->
            catch
            ejabberd_sql:sql_query_t(
              ?SQL("delete from pubsub_node_option "
                   "where nodeid=%(OldNidx)d")),
            catch
            ejabberd_sql:sql_query_t(
              ?SQL("update pubsub_node set"
                   " host=%(H)s, node=%(Node)s,"
                   " parent=%(Parent)s, plugin=%(Type)s "
                   "where nodeid=%(OldNidx)d")),
            OldNidx;
        {error, not_found} ->
            catch
            ejabberd_sql:sql_query_t(
              ?SQL("insert into pubsub_node(host, node, parent, plugin) "
                   "values(%(H)s, %(Node)s, %(Parent)s, %(Type)s)")),
            case nodeidx(Host, Node) of
                {result, NewNidx} -> NewNidx;
                {error, not_found} -> none;  % this should not happen
                {error, _} -> db_error
            end;
        {error, _} ->
            db_error
    end,
    case Nidx of
        db_error ->
            {error, xmpp:err_internal_server_error(?T("Database failure"), ejabberd_option:language())};
        none ->
            Txt = ?T("Node index not found"),
            {error, xmpp:err_internal_server_error(Txt, ejabberd_option:language())};
        _ ->
            lists:foreach(fun ({Key, Value}) ->
                        SKey = iolist_to_binary(atom_to_list(Key)),
                        SValue = misc:term_to_expr(Value),
                        catch
                        ejabberd_sql:sql_query_t(
                          ?SQL("insert into pubsub_node_option(nodeid, name, val) "
                               "values (%(Nidx)d, %(SKey)s, %(SValue)s)"))
                end,
                Record#pubsub_node.options),
            {result, Nidx}
    end.

get_node(Host, Node, _From) ->
    get_node(Host, Node).

get_node(Host, Node) ->
    H = node_flat_sql:encode_host(Host),
    case catch
        ejabberd_sql:sql_query_t(
          ?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d from pubsub_node "
               "where host=%(H)s and node=%(Node)s"))
    of
        {selected, [RItem]} ->
            raw_to_node(Host, RItem);
        {'EXIT', _Reason} ->
            {error, xmpp:err_internal_server_error(?T("Database failure"), ejabberd_option:language())};
        _ ->
            {error, xmpp:err_item_not_found(?T("Node not found"), ejabberd_option:language())}
    end.

get_node(Nidx) ->
    case catch
        ejabberd_sql:sql_query_t(
          ?SQL("select @(host)s, @(node)s, @(parent)s, @(plugin)s from pubsub_node "
               "where nodeid=%(Nidx)d"))
    of
        {selected, [{Host, Node, Parent, Type}]} ->
            raw_to_node(Host, {Node, Parent, Type, Nidx});
        {'EXIT', _Reason} ->
            {error, xmpp:err_internal_server_error(?T("Database failure"), ejabberd_option:language())};
        _ ->
            {error, xmpp:err_item_not_found(?T("Node not found"), ejabberd_option:language())}
    end.

get_nodes(Host) ->
    get_nodes(Host, infinity).

get_nodes(Host, Limit) ->
    H = node_flat_sql:encode_host(Host),
    Query = fun(mssql, _) when is_integer(Limit), Limit>=0 ->
                    ejabberd_sql:sql_query_t(
                      ?SQL("select top %(Limit)d @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "
                           "from pubsub_node where host=%(H)s"));
               (_, _) when is_integer(Limit), Limit>=0 ->
                    ejabberd_sql:sql_query_t(
                      ?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "
                           "from pubsub_node where host=%(H)s limit %(Limit)d"));
               (_, _) ->
                    ejabberd_sql:sql_query_t(
                      ?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "
                           "from pubsub_node where host=%(H)s"))
            end,
    case ejabberd_sql:sql_query_t(Query) of
        {selected, RItems} ->
            [raw_to_node(Host, Item) || Item <- RItems];
        _ ->
            []
    end.

get_all_nodes({_U, _S, _R} = JID) ->
    SubKey = jid:tolower(JID),
    GenKey = jid:remove_resource(SubKey),
    EncKey = node_flat_sql:encode_jid(GenKey),
    Pattern = <<(node_flat_sql:encode_jid_like(GenKey))/binary, "/%">>,
    case ejabberd_sql:sql_query_t(
           ?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "
                "from pubsub_node where host=%(EncKey)s "
                "or host like %(Pattern)s %ESCAPE")) of
        {selected, RItems} ->
            [raw_to_node(GenKey, Item) || Item <- RItems];
        _ ->
            []
    end;
get_all_nodes(Host) ->
    Pattern1 = <<"%@", Host/binary>>,
    Pattern2 = <<"%@", Host/binary, "/%">>,
    case ejabberd_sql:sql_query_t(
           ?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "
                "from pubsub_node where host=%(Host)s "
                "or host like %(Pattern1)s "
                "or host like %(Pattern2)s %ESCAPE")) of
        {selected, RItems} ->
            [raw_to_node(Host, Item) || Item <- RItems];
        _ ->
            []
    end.

get_parentnodes(Host, Node, _From) ->
    case get_node(Host, Node) of
        Record when is_record(Record, pubsub_node) ->
            Record#pubsub_node.parents;
        _ ->
            []
    end.

get_parentnodes_tree(Host, Node, _From) ->
    get_parentnodes_tree(Host, Node, 0, []).
get_parentnodes_tree(Host, Node, Level, Acc) ->
    case get_node(Host, Node) of
        Record when is_record(Record, pubsub_node) ->
            Tree = [{Level, [Record]}|Acc],
            case Record#pubsub_node.parents of
                [Parent] -> get_parentnodes_tree(Host, Parent, Level+1, Tree);
                _ -> Tree
            end;
        _ ->
            Acc
    end.

get_subnodes(Host, Node, Limit) ->
    H = node_flat_sql:encode_host(Host),
    Query = fun(mssql, _) when is_integer(Limit), Limit>=0 ->
                    ejabberd_sql:sql_query_t(
                      ?SQL("select top %(Limit)d @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "
                           "from pubsub_node where host=%(H)s and parent=%(Node)s"));
               (_, _) when is_integer(Limit), Limit>=0 ->
                    ejabberd_sql:sql_query_t(
                      ?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "
                           "from pubsub_node where host=%(H)s and parent=%(Node)s "
                           "limit %(Limit)d"));
               (_, _) ->
                    ejabberd_sql:sql_query_t(
                      ?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "
                           "from pubsub_node where host=%(H)s and parent=%(Node)s"))
            end,
    case ejabberd_sql:sql_query_t(Query) of
        {selected, RItems} ->
            [raw_to_node(Host, Item) || Item <- RItems];
        _ ->
            []
    end.

get_subnodes_tree(Host, Node, _From) ->
    get_subnodes_tree(Host, Node).

get_subnodes_tree(Host, Node) ->
    case get_node(Host, Node) of
        {error, _} ->
            [];
        Rec ->
            Type = Rec#pubsub_node.type,
            H = node_flat_sql:encode_host(Host),
            N = <<(ejabberd_sql:escape_like_arg(Node))/binary, "/%">>,
            Sub = case catch
                ejabberd_sql:sql_query_t(
                ?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d from pubsub_node "
                     "where host=%(H)s and plugin=%(Type)s and"
                     " (parent=%(Node)s or parent like %(N)s %ESCAPE)"))
            of
                {selected, RItems} ->
                    [raw_to_node(Host, Item) || Item <- RItems];
                _ ->
                    []
            end,
            [Rec|Sub]
    end.

create_node(Host, Node, Type, Owner, Options, Parents) ->
    BJID = jid:tolower(jid:remove_resource(Owner)),
    case nodeidx(Host, Node) of
        {error, not_found} ->
            ParentExists = case Host of
                {_U, _S, _R} ->
                    %% This is special case for PEP handling
                    %% PEP does not uses hierarchy
                    true;
                _ ->
                    case Parents of
                        [] ->
                            true;
                        [Parent | _] ->
                            case nodeidx(Host, Parent) of
                                {result, PNode} ->
                                    case nodeowners(PNode) of
                                        [{<<>>, Host, <<>>}] -> true;
                                        Owners -> lists:member(BJID, Owners)
                                    end;
                                _ ->
                                    false
                            end;
                        _ ->
                            false
                    end
            end,
            case ParentExists of
                true ->
                    case set_node(#pubsub_node{nodeid = {Host, Node},
                                parents = Parents, type = Type,
                                options = Options})
                    of
                        {result, Nidx} -> {ok, Nidx};
                        Other -> Other
                    end;
                false ->
                    {error, xmpp:err_forbidden()}
            end;
        {result, _} ->
            {error, xmpp:err_conflict(?T("Node already exists"), ejabberd_option:language())};
        {error, db_fail} ->
            {error, xmpp:err_internal_server_error(?T("Database failure"), ejabberd_option:language())}
    end.

delete_node(Host, Node) ->
    lists:map(
        fun(Rec) ->
            Nidx = Rec#pubsub_node.id,
            catch ejabberd_sql:sql_query_t(
                    ?SQL("delete from pubsub_node where nodeid=%(Nidx)d")),
            Rec
        end, get_subnodes_tree(Host, Node)).

%% helpers
raw_to_node(Host, [Node, Parent, Type, Nidx]) ->
    raw_to_node(Host, {Node, Parent, Type, binary_to_integer(Nidx)});
raw_to_node(Host, {Node, Parent, Type, Nidx}) ->
    Options = case catch
        ejabberd_sql:sql_query_t(
          ?SQL("select @(name)s, @(val)s from pubsub_node_option "
               "where nodeid=%(Nidx)d"))
    of
        {selected, ROptions} ->
            DbOpts = lists:map(
                fun({<<"max_items">>, <<"infinity">>}) ->
                       {max_items, max};
                   ({Key, Value}) ->
                       RKey = misc:binary_to_atom(Key),
                       Tokens = element(2, erl_scan:string(binary_to_list(<<Value/binary, ".">>))),
                       RValue = element(2, erl_parse:parse_term(Tokens)),
                       {RKey, RValue}
                end,
                ROptions),
            Module = misc:binary_to_atom(<<"node_", Type/binary, "_sql">>),
            StdOpts = Module:options(),
            lists:foldl(fun ({Key, Value}, Acc) ->
                        lists:keystore(Key, 1, Acc, {Key, Value})
                end,
                StdOpts, DbOpts);
        _ ->
            []
    end,
    Parents = case Parent of
        <<>> -> [];
        _ -> [Parent]
    end,
    #pubsub_node{nodeid = {Host, Node}, id = Nidx,
        parents = Parents, type = Type, options = Options}.

nodeidx(Host, Node) ->
    H = node_flat_sql:encode_host(Host),
    case catch
        ejabberd_sql:sql_query_t(
          ?SQL("select @(nodeid)d from pubsub_node "
               "where host=%(H)s and node=%(Node)s"))
    of
        {selected, [{Nidx}]} ->
            {result, Nidx};
        {'EXIT', _Reason} ->
            {error, db_fail};
        _ ->
            {error, not_found}
    end.

nodeowners(Nidx) ->
    {result, Res} = node_flat_sql:get_node_affiliations(Nidx),
    [LJID || {LJID, Aff} <- Res, Aff =:= owner].

1	%%%----------------------------------------------------------------------
2	%%% File : nodetree_tree_sql.erl
3	%%% Author : Christophe Romain <christophe.romain@process-one.net>
4	%%% Purpose : Standard node tree plugin with ODBC backend
5	%%% Created : 1 Dec 2007 by Christophe Romain <christophe.romain@process-one.net>
6	%%%
7	%%%
8	%%% ejabberd, Copyright (C) 2002-2025 ProcessOne
9	%%%
10	%%% This program is free software; you can redistribute it and/or
11	%%% modify it under the terms of the GNU General Public License as
12	%%% published by the Free Software Foundation; either version 2 of the
13	%%% License, or (at your option) any later version.
14	%%%
15	%%% This program is distributed in the hope that it will be useful,
16	%%% but WITHOUT ANY WARRANTY; without even the implied warranty of
17	%%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18	%%% General Public License for more details.
19	%%%
20	%%% You should have received a copy of the GNU General Public License along
21	%%% with this program; if not, write to the Free Software Foundation, Inc.,
22	%%% 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
23	%%%
24	%%%----------------------------------------------------------------------
25
26	%%% @doc The module <strong>{@module}</strong> is the default PubSub node tree plugin.
27	%%% <p>It is used as a default for all unknown PubSub node type. It can serve
28	%%% as a developer basis and reference to build its own custom pubsub node tree
29	%%% types.</p>
30	%%% <p>PubSub node tree plugins are using the {@link gen_nodetree} behaviour.</p>
31	%%% <p><strong>The API isn't stabilized yet</strong>. The pubsub plugin
32	%%% development is still a work in progress. However, the system is already
33	%%% usable and useful as is. Please, send us comments, feedback and
34	%%% improvements.</p>
35
36	-module(nodetree_tree_sql).
37	-behaviour(gen_pubsub_nodetree).
38	-author('christophe.romain@process-one.net').
39
40
41	-include("pubsub.hrl").
42	-include_lib("xmpp/include/xmpp.hrl").
43	-include("ejabberd_sql_pt.hrl").
44	-include("translate.hrl").
45
46	-export([init/3, terminate/2, options/0, set_node/1,
47	get_node/3, get_node/2, get_node/1, get_nodes/2,
48	get_nodes/1, get_all_nodes/1,
49	get_parentnodes/3, get_parentnodes_tree/3,
50	get_subnodes/3, get_subnodes_tree/3, create_node/6,
51	delete_node/2]).
52
53	-export([raw_to_node/2]).
54
55	init(_Host, _ServerHost, _Opts) ->
56	ok.	6✔
57
58	terminate(_Host, _ServerHost) ->
59	ok.	6✔
60
61	options() ->
62	[{sql, true} \| nodetree_tree:options()].	×
63
64	set_node(Record) when is_record(Record, pubsub_node) ->
65	{Host, Node} = Record#pubsub_node.nodeid,	216✔
66	Parent = case Record#pubsub_node.parents of	216✔
67	[] -> <<>>;	216✔
68	[First \| _] -> First	×
69	end,
70	Type = Record#pubsub_node.type,	216✔
71	H = node_flat_sql:encode_host(Host),	216✔
72	Nidx = case nodeidx(Host, Node) of	216✔
73	{result, OldNidx} ->
74	catch	24✔
75	ejabberd_sql:sql_query_t(
76	?SQL("delete from pubsub_node_option "	36✔
77	"where nodeid=%(OldNidx)d")),
78	catch	24✔
79	ejabberd_sql:sql_query_t(
80	?SQL("update pubsub_node set"	36✔
81	" host=%(H)s, node=%(Node)s,"
82	" parent=%(Parent)s, plugin=%(Type)s "
83	"where nodeid=%(OldNidx)d")),
84	OldNidx;	24✔
85	{error, not_found} ->
86	catch	192✔
87	ejabberd_sql:sql_query_t(
88	?SQL("insert into pubsub_node(host, node, parent, plugin) "	260✔
89	"values(%(H)s, %(Node)s, %(Parent)s, %(Type)s)")),
90	case nodeidx(Host, Node) of	192✔
91	{result, NewNidx} -> NewNidx;	192✔
92	{error, not_found} -> none; % this should not happen	×
93	{error, _} -> db_error	×
94	end;
95	{error, _} ->
96	db_error	×
97	end,
98	case Nidx of	216✔
99	db_error ->
100	{error, xmpp:err_internal_server_error(?T("Database failure"), ejabberd_option:language())};	×
101	none ->
102	Txt = ?T("Node index not found"),	×
103	{error, xmpp:err_internal_server_error(Txt, ejabberd_option:language())};	×
104	_ ->
105	lists:foreach(fun ({Key, Value}) ->	216✔
106	SKey = iolist_to_binary(atom_to_list(Key)),	4,176✔
107	SValue = misc:term_to_expr(Value),	4,176✔
108	catch	4,176✔
109	ejabberd_sql:sql_query_t(
110	?SQL("insert into pubsub_node_option(nodeid, name, val) "	5,572✔
111	"values (%(Nidx)d, %(SKey)s, %(SValue)s)"))
112	end,
113	Record#pubsub_node.options),
114	{result, Nidx}	216✔
115	end.
116
117	get_node(Host, Node, _From) ->
118	get_node(Host, Node).	×
119
120	get_node(Host, Node) ->
121	H = node_flat_sql:encode_host(Host),	2,016✔
122	case catch	2,016✔
123	ejabberd_sql:sql_query_t(
124	?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d from pubsub_node "	2,692✔
125	"where host=%(H)s and node=%(Node)s"))
126	of
127	{selected, [RItem]} ->
128	raw_to_node(Host, RItem);	1,932✔
129	{'EXIT', _Reason} ->
130	{error, xmpp:err_internal_server_error(?T("Database failure"), ejabberd_option:language())};	×
131	_ ->
132	{error, xmpp:err_item_not_found(?T("Node not found"), ejabberd_option:language())}	84✔
133	end.
134
135	get_node(Nidx) ->
136	case catch	×
137	ejabberd_sql:sql_query_t(
138	?SQL("select @(host)s, @(node)s, @(parent)s, @(plugin)s from pubsub_node "	×
139	"where nodeid=%(Nidx)d"))
140	of
141	{selected, [{Host, Node, Parent, Type}]} ->
142	raw_to_node(Host, {Node, Parent, Type, Nidx});	×
143	{'EXIT', _Reason} ->
144	{error, xmpp:err_internal_server_error(?T("Database failure"), ejabberd_option:language())};	×
145	_ ->
146	{error, xmpp:err_item_not_found(?T("Node not found"), ejabberd_option:language())}	×
147	end.
148
149	get_nodes(Host) ->
150	get_nodes(Host, infinity).	×
151
152	get_nodes(Host, Limit) ->
153	H = node_flat_sql:encode_host(Host),	300✔
154	Query = fun(mssql, _) when is_integer(Limit), Limit>=0 ->	300✔
155	ejabberd_sql:sql_query_t(	×
156	?SQL("select top %(Limit)d @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "	×
157	"from pubsub_node where host=%(H)s"));
158	(_, _) when is_integer(Limit), Limit>=0 ->
159	ejabberd_sql:sql_query_t(	×
160	?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "	×
161	"from pubsub_node where host=%(H)s limit %(Limit)d"));
162	(_, _) ->
163	ejabberd_sql:sql_query_t(	300✔
164	?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "	404✔
165	"from pubsub_node where host=%(H)s"))
166	end,
167	case ejabberd_sql:sql_query_t(Query) of	300✔
168	{selected, RItems} ->
169	[raw_to_node(Host, Item) \|\| Item <- RItems];	300✔
170	_ ->
171	[]	×
172	end.
173
174	get_all_nodes({_U, _S, _R} = JID) ->
175	SubKey = jid:tolower(JID),	×
176	GenKey = jid:remove_resource(SubKey),	×
177	EncKey = node_flat_sql:encode_jid(GenKey),	×
178	Pattern = <<(node_flat_sql:encode_jid_like(GenKey))/binary, "/%">>,	×
179	case ejabberd_sql:sql_query_t(	×
180	?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "	×
181	"from pubsub_node where host=%(EncKey)s "
182	"or host like %(Pattern)s %ESCAPE")) of
183	{selected, RItems} ->
184	[raw_to_node(GenKey, Item) \|\| Item <- RItems];	×
185	_ ->
186	[]	×
187	end;
188	get_all_nodes(Host) ->
189	Pattern1 = <<"%@", Host/binary>>,	×
190	Pattern2 = <<"%@", Host/binary, "/%">>,	×
191	case ejabberd_sql:sql_query_t(	×
192	?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "	×
193	"from pubsub_node where host=%(Host)s "
194	"or host like %(Pattern1)s "
195	"or host like %(Pattern2)s %ESCAPE")) of
196	{selected, RItems} ->
197	[raw_to_node(Host, Item) \|\| Item <- RItems];	×
198	_ ->
199	[]	×
200	end.
201
202	get_parentnodes(Host, Node, _From) ->
203	case get_node(Host, Node) of	×
204	Record when is_record(Record, pubsub_node) ->
205	Record#pubsub_node.parents;	×
206	_ ->
207	[]	×
208	end.
209
210	get_parentnodes_tree(Host, Node, _From) ->
211	get_parentnodes_tree(Host, Node, 0, []).	600✔
212	get_parentnodes_tree(Host, Node, Level, Acc) ->
213	case get_node(Host, Node) of	600✔
214	Record when is_record(Record, pubsub_node) ->
215	Tree = [{Level, [Record]}\|Acc],	600✔
216	case Record#pubsub_node.parents of	600✔
217	[Parent] -> get_parentnodes_tree(Host, Parent, Level+1, Tree);	×
218	_ -> Tree	600✔
219	end;
220	_ ->
221	Acc	×
222	end.
223
224	get_subnodes(Host, Node, Limit) ->
225	H = node_flat_sql:encode_host(Host),	×
226	Query = fun(mssql, _) when is_integer(Limit), Limit>=0 ->	×
227	ejabberd_sql:sql_query_t(	×
228	?SQL("select top %(Limit)d @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "	×
229	"from pubsub_node where host=%(H)s and parent=%(Node)s"));
230	(_, _) when is_integer(Limit), Limit>=0 ->
231	ejabberd_sql:sql_query_t(	×
232	?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "	×
233	"from pubsub_node where host=%(H)s and parent=%(Node)s "
234	"limit %(Limit)d"));
235	(_, _) ->
236	ejabberd_sql:sql_query_t(	×
237	?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d "	×
238	"from pubsub_node where host=%(H)s and parent=%(Node)s"))
239	end,
240	case ejabberd_sql:sql_query_t(Query) of	×
241	{selected, RItems} ->
242	[raw_to_node(Host, Item) \|\| Item <- RItems];	×
243	_ ->
244	[]	×
245	end.
246
247	get_subnodes_tree(Host, Node, _From) ->
248	get_subnodes_tree(Host, Node).	×
249
250	get_subnodes_tree(Host, Node) ->
251	case get_node(Host, Node) of	192✔
252	{error, _} ->
253	[];	×
254	Rec ->
255	Type = Rec#pubsub_node.type,	192✔
256	H = node_flat_sql:encode_host(Host),	192✔
257	N = <<(ejabberd_sql:escape_like_arg(Node))/binary, "/%">>,	192✔
258	Sub = case catch	192✔
259	ejabberd_sql:sql_query_t(
260	?SQL("select @(node)s, @(parent)s, @(plugin)s, @(nodeid)d from pubsub_node "	260✔
261	"where host=%(H)s and plugin=%(Type)s and"
262	" (parent=%(Node)s or parent like %(N)s %ESCAPE)"))
263	of
264	{selected, RItems} ->
265	[raw_to_node(Host, Item) \|\| Item <- RItems];	192✔
266	_ ->
267	[]	×
268	end,
269	[Rec\|Sub]	192✔
270	end.
271
272	create_node(Host, Node, Type, Owner, Options, Parents) ->
273	BJID = jid:tolower(jid:remove_resource(Owner)),	192✔
274	case nodeidx(Host, Node) of	192✔
275	{error, not_found} ->
276	ParentExists = case Host of	192✔
277	{_U, _S, _R} ->
278	%% This is special case for PEP handling
279	%% PEP does not uses hierarchy
280	true;	12✔
281	_ ->
282	case Parents of	180✔
283	[] ->
284	true;	180✔
285	[Parent \| _] ->
286	case nodeidx(Host, Parent) of	×
287	{result, PNode} ->
288	case nodeowners(PNode) of	×
289	[{<<>>, Host, <<>>}] -> true;	×
290	Owners -> lists:member(BJID, Owners)	×
291	end;
292	_ ->
293	false	×
294	end;
295	_ ->
296	false	×
297	end
298	end,
299	case ParentExists of	192✔
300	true ->
301	case set_node(#pubsub_node{nodeid = {Host, Node},	192✔
302	parents = Parents, type = Type,
303	options = Options})
304	of
305	{result, Nidx} -> {ok, Nidx};	192✔
306	Other -> Other	×
307	end;
308	false ->
309	{error, xmpp:err_forbidden()}	×
310	end;
311	{result, _} ->
312	{error, xmpp:err_conflict(?T("Node already exists"), ejabberd_option:language())};	×
313	{error, db_fail} ->
314	{error, xmpp:err_internal_server_error(?T("Database failure"), ejabberd_option:language())}	×
315	end.
316
317	delete_node(Host, Node) ->
318	lists:map(	192✔
319	fun(Rec) ->
320	Nidx = Rec#pubsub_node.id,	192✔
321	catch ejabberd_sql:sql_query_t(	192✔
322	?SQL("delete from pubsub_node where nodeid=%(Nidx)d")),	260✔
323	Rec	192✔
324	end, get_subnodes_tree(Host, Node)).
325
326	%% helpers
327	raw_to_node(Host, [Node, Parent, Type, Nidx]) ->
328	raw_to_node(Host, {Node, Parent, Type, binary_to_integer(Nidx)});	×
329	raw_to_node(Host, {Node, Parent, Type, Nidx}) ->
330	Options = case catch	2,076✔
331	ejabberd_sql:sql_query_t(
332	?SQL("select @(name)s, @(val)s from pubsub_node_option "	2,772✔
333	"where nodeid=%(Nidx)d"))
334	of
335	{selected, ROptions} ->
336	DbOpts = lists:map(	2,076✔
337	fun({<<"max_items">>, <<"infinity">>}) ->
338	{max_items, max};	×
339	({Key, Value}) ->
340	RKey = misc:binary_to_atom(Key),	40,350✔
341	Tokens = element(2, erl_scan:string(binary_to_list(<<Value/binary, ".">>))),	40,350✔
342	RValue = element(2, erl_parse:parse_term(Tokens)),	40,350✔
343	{RKey, RValue}	40,350✔
344	end,
345	ROptions),
346	Module = misc:binary_to_atom(<<"node_", Type/binary, "_sql">>),	2,076✔
347	StdOpts = Module:options(),	2,076✔
348	lists:foldl(fun ({Key, Value}, Acc) ->	2,076✔
349	lists:keystore(Key, 1, Acc, {Key, Value})	40,350✔
350	end,
351	StdOpts, DbOpts);
352	_ ->
353	[]	×
354	end,
355	Parents = case Parent of	2,076✔
356	<<>> -> [];	2,076✔
357	_ -> [Parent]	×
358	end,
359	#pubsub_node{nodeid = {Host, Node}, id = Nidx,	2,076✔
360	parents = Parents, type = Type, options = Options}.
361
362	nodeidx(Host, Node) ->
363	H = node_flat_sql:encode_host(Host),	600✔
364	case catch	600✔
365	ejabberd_sql:sql_query_t(
366	?SQL("select @(nodeid)d from pubsub_node "	804✔
367	"where host=%(H)s and node=%(Node)s"))
368	of
369	{selected, [{Nidx}]} ->
370	{result, Nidx};	216✔
371	{'EXIT', _Reason} ->
372	{error, db_fail};	×
373	_ ->
374	{error, not_found}	384✔
375	end.
376
377	nodeowners(Nidx) ->
378	{result, Res} = node_flat_sql:get_node_affiliations(Nidx),	×
379	[LJID \|\| {LJID, Aff} <- Res, Aff =:= owner].	×

processone / ejabberd / 1162

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous