2011-11-29 14:13:13 +01:00
|
|
|
--- mod_pubsub.erl 2011-11-29 14:10:41.000000000 +0100
|
|
|
|
+++ mod_pubsub_odbc.erl 2011-11-29 14:12:01.000000000 +0100
|
2009-10-13 18:43:02 +02:00
|
|
|
@@ -42,7 +42,7 @@
|
|
|
|
%%% 6.2.3.1, 6.2.3.5, and 6.3. For information on subscription leases see
|
|
|
|
%%% XEP-0060 section 12.18.
|
2009-08-07 10:26:47 +02:00
|
|
|
|
|
|
|
--module(mod_pubsub).
|
|
|
|
+-module(mod_pubsub_odbc).
|
|
|
|
-author('christophe.romain@process-one.net').
|
2009-09-24 21:28:43 +02:00
|
|
|
-version('1.13-0').
|
2009-08-07 10:26:47 +02:00
|
|
|
|
2009-10-13 18:43:02 +02:00
|
|
|
@@ -54,9 +54,9 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
-include("jlib.hrl").
|
|
|
|
-include("pubsub.hrl").
|
|
|
|
|
|
|
|
--define(STDTREE, "tree").
|
|
|
|
--define(STDNODE, "flat").
|
|
|
|
--define(PEPNODE, "pep").
|
|
|
|
+-define(STDTREE, "tree_odbc").
|
|
|
|
+-define(STDNODE, "flat_odbc").
|
|
|
|
+-define(PEPNODE, "pep_odbc").
|
|
|
|
|
|
|
|
%% exports for hooks
|
|
|
|
-export([presence_probe/3,
|
2010-08-02 17:07:23 +02:00
|
|
|
@@ -103,7 +103,7 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
string_to_affiliation/1,
|
|
|
|
extended_error/2,
|
2009-08-25 16:13:26 +02:00
|
|
|
extended_error/3,
|
2009-08-27 10:57:45 +02:00
|
|
|
- rename_default_nodeplugin/0
|
|
|
|
+ escape/1
|
2009-08-07 10:26:47 +02:00
|
|
|
]).
|
|
|
|
|
2009-08-27 10:57:45 +02:00
|
|
|
%% API and gen_server callbacks
|
2010-08-02 17:07:23 +02:00
|
|
|
@@ -122,7 +122,7 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
-export([send_loop/1
|
|
|
|
]).
|
|
|
|
|
|
|
|
--define(PROCNAME, ejabberd_mod_pubsub).
|
|
|
|
+-define(PROCNAME, ejabberd_mod_pubsub_odbc).
|
2010-01-12 16:14:47 +01:00
|
|
|
-define(LOOPNAME, ejabberd_mod_pubsub_loop).
|
2009-08-07 10:26:47 +02:00
|
|
|
-define(PLUGIN_PREFIX, "node_").
|
|
|
|
-define(TREE_PREFIX, "nodetree_").
|
2010-08-02 17:07:23 +02:00
|
|
|
@@ -217,8 +217,6 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
ok
|
|
|
|
end,
|
|
|
|
ejabberd_router:register_route(Host),
|
|
|
|
- update_node_database(Host, ServerHost),
|
|
|
|
- update_state_database(Host, ServerHost),
|
2011-11-29 14:13:13 +01:00
|
|
|
put(server_host, ServerHost), % not clean, but needed to plug hooks at any location
|
2009-09-25 15:12:29 +02:00
|
|
|
init_nodes(Host, ServerHost, NodeTree, Plugins),
|
2009-08-07 10:26:47 +02:00
|
|
|
State = #state{host = Host,
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -283,207 +281,14 @@
|
2009-10-20 17:03:07 +02:00
|
|
|
|
|
|
|
init_nodes(Host, ServerHost, _NodeTree, Plugins) ->
|
|
|
|
%% TODO, this call should be done plugin side
|
|
|
|
- case lists:member("hometree", Plugins) of
|
|
|
|
+ case lists:member("hometree_odbc", Plugins) of
|
|
|
|
true ->
|
|
|
|
- create_node(Host, ServerHost, string_to_node("/home"), service_jid(Host), "hometree"),
|
|
|
|
- create_node(Host, ServerHost, string_to_node("/home/"++ServerHost), service_jid(Host), "hometree");
|
|
|
|
+ create_node(Host, ServerHost, string_to_node("/home"), service_jid(Host), "hometree_odbc"),
|
|
|
|
+ create_node(Host, ServerHost, string_to_node("/home/"++ServerHost), service_jid(Host), "hometree_odbc");
|
|
|
|
false ->
|
2009-09-25 15:12:29 +02:00
|
|
|
ok
|
|
|
|
end.
|
2009-08-25 16:13:26 +02:00
|
|
|
|
2009-08-27 10:57:45 +02:00
|
|
|
-update_node_database(Host, ServerHost) ->
|
|
|
|
- mnesia:del_table_index(pubsub_node, type),
|
|
|
|
- mnesia:del_table_index(pubsub_node, parentid),
|
|
|
|
- case catch mnesia:table_info(pubsub_node, attributes) of
|
|
|
|
- [host_node, host_parent, info] ->
|
|
|
|
- ?INFO_MSG("upgrade node pubsub tables",[]),
|
|
|
|
- F = fun() ->
|
|
|
|
- {Result, LastIdx} = lists:foldl(
|
|
|
|
- fun({pubsub_node, NodeId, ParentId, {nodeinfo, Items, Options, Entities}}, {RecList, NodeIdx}) ->
|
|
|
|
- ItemsList =
|
|
|
|
- lists:foldl(
|
|
|
|
- fun({item, IID, Publisher, Payload}, Acc) ->
|
|
|
|
- C = {unknown, Publisher},
|
|
|
|
- M = {now(), Publisher},
|
|
|
|
- mnesia:write(
|
|
|
|
- #pubsub_item{itemid = {IID, NodeIdx},
|
|
|
|
- creation = C,
|
|
|
|
- modification = M,
|
|
|
|
- payload = Payload}),
|
|
|
|
- [{Publisher, IID} | Acc]
|
|
|
|
- end, [], Items),
|
|
|
|
- Owners =
|
|
|
|
- dict:fold(
|
|
|
|
- fun(JID, {entity, Aff, Sub}, Acc) ->
|
|
|
|
- UsrItems =
|
|
|
|
- lists:foldl(
|
|
|
|
- fun({P, I}, IAcc) ->
|
|
|
|
- case P of
|
|
|
|
- JID -> [I | IAcc];
|
|
|
|
- _ -> IAcc
|
|
|
|
- end
|
|
|
|
- end, [], ItemsList),
|
|
|
|
- mnesia:write({pubsub_state,
|
|
|
|
- {JID, NodeIdx},
|
|
|
|
- UsrItems,
|
|
|
|
- Aff,
|
|
|
|
- Sub}),
|
|
|
|
- case Aff of
|
|
|
|
- owner -> [JID | Acc];
|
|
|
|
- _ -> Acc
|
|
|
|
- end
|
|
|
|
- end, [], Entities),
|
|
|
|
- mnesia:delete({pubsub_node, NodeId}),
|
|
|
|
- {[#pubsub_node{nodeid = NodeId,
|
|
|
|
- id = NodeIdx,
|
|
|
|
- parents = [element(2, ParentId)],
|
|
|
|
- owners = Owners,
|
|
|
|
- options = Options} |
|
|
|
|
- RecList], NodeIdx + 1}
|
|
|
|
- end, {[], 1},
|
|
|
|
- mnesia:match_object(
|
|
|
|
- {pubsub_node, {Host, '_'}, '_', '_'})),
|
|
|
|
- mnesia:write(#pubsub_index{index = node, last = LastIdx, free = []}),
|
|
|
|
- Result
|
|
|
|
- end,
|
|
|
|
- {atomic, NewRecords} = mnesia:transaction(F),
|
|
|
|
- {atomic, ok} = mnesia:delete_table(pubsub_node),
|
|
|
|
- {atomic, ok} = mnesia:create_table(pubsub_node,
|
|
|
|
- [{disc_copies, [node()]},
|
|
|
|
- {attributes, record_info(fields, pubsub_node)}]),
|
|
|
|
- FNew = fun() -> lists:foreach(fun(Record) ->
|
|
|
|
- mnesia:write(Record)
|
|
|
|
- end, NewRecords)
|
|
|
|
- end,
|
|
|
|
- case mnesia:transaction(FNew) of
|
|
|
|
- {atomic, Result} ->
|
|
|
|
- ?INFO_MSG("Pubsub node tables updated correctly: ~p", [Result]);
|
|
|
|
- {aborted, Reason} ->
|
|
|
|
- ?ERROR_MSG("Problem updating Pubsub node tables:~n~p", [Reason])
|
|
|
|
- end;
|
|
|
|
- [nodeid, parentid, type, owners, options] ->
|
|
|
|
- F = fun({pubsub_node, NodeId, {_, Parent}, Type, Owners, Options}) ->
|
|
|
|
- #pubsub_node{
|
|
|
|
- nodeid = NodeId,
|
|
|
|
- id = 0,
|
|
|
|
- parents = [Parent],
|
|
|
|
- type = Type,
|
|
|
|
- owners = Owners,
|
|
|
|
- options = Options}
|
|
|
|
- end,
|
|
|
|
- mnesia:transform_table(pubsub_node, F, [nodeid, id, parents, type, owners, options]),
|
|
|
|
- FNew = fun() ->
|
|
|
|
- LastIdx = lists:foldl(fun(#pubsub_node{nodeid = NodeId} = PubsubNode, NodeIdx) ->
|
|
|
|
- mnesia:write(PubsubNode#pubsub_node{id = NodeIdx}),
|
|
|
|
- lists:foreach(fun(#pubsub_state{stateid = StateId} = State) ->
|
|
|
|
- {JID, _} = StateId,
|
|
|
|
- mnesia:delete({pubsub_state, StateId}),
|
|
|
|
- mnesia:write(State#pubsub_state{stateid = {JID, NodeIdx}})
|
|
|
|
- end, mnesia:match_object(#pubsub_state{stateid = {'_', NodeId}, _ = '_'})),
|
|
|
|
- lists:foreach(fun(#pubsub_item{itemid = ItemId} = Item) ->
|
|
|
|
- {IID, _} = ItemId,
|
|
|
|
- {M1, M2} = Item#pubsub_item.modification,
|
|
|
|
- {C1, C2} = Item#pubsub_item.creation,
|
|
|
|
- mnesia:delete({pubsub_item, ItemId}),
|
|
|
|
- mnesia:write(Item#pubsub_item{itemid = {IID, NodeIdx},
|
|
|
|
- modification = {M2, M1},
|
|
|
|
- creation = {C2, C1}})
|
|
|
|
- end, mnesia:match_object(#pubsub_item{itemid = {'_', NodeId}, _ = '_'})),
|
|
|
|
- NodeIdx + 1
|
|
|
|
- end, 1, mnesia:match_object(
|
|
|
|
- {pubsub_node, {Host, '_'}, '_', '_', '_', '_', '_'})
|
|
|
|
- ++ mnesia:match_object(
|
|
|
|
- {pubsub_node, {{'_', ServerHost, '_'}, '_'}, '_', '_', '_', '_', '_'})),
|
|
|
|
- mnesia:write(#pubsub_index{index = node, last = LastIdx, free = []})
|
|
|
|
- end,
|
|
|
|
- case mnesia:transaction(FNew) of
|
|
|
|
- {atomic, Result} ->
|
|
|
|
- rename_default_nodeplugin(),
|
|
|
|
- ?INFO_MSG("Pubsub node tables updated correctly: ~p", [Result]);
|
|
|
|
- {aborted, Reason} ->
|
|
|
|
- ?ERROR_MSG("Problem updating Pubsub node tables:~n~p", [Reason])
|
|
|
|
- end;
|
|
|
|
- [nodeid, id, parent, type, owners, options] ->
|
|
|
|
- F = fun({pubsub_node, NodeId, Id, Parent, Type, Owners, Options}) ->
|
|
|
|
- #pubsub_node{
|
|
|
|
- nodeid = NodeId,
|
|
|
|
- id = Id,
|
|
|
|
- parents = [Parent],
|
|
|
|
- type = Type,
|
|
|
|
- owners = Owners,
|
|
|
|
- options = Options}
|
|
|
|
- end,
|
|
|
|
- mnesia:transform_table(pubsub_node, F, [nodeid, id, parents, type, owners, options]),
|
|
|
|
- rename_default_nodeplugin();
|
|
|
|
- _ ->
|
|
|
|
- ok
|
2009-10-20 17:03:07 +02:00
|
|
|
- end,
|
|
|
|
- mnesia:transaction(fun() ->
|
|
|
|
- case catch mnesia:first(pubsub_node) of
|
|
|
|
- {_, L} when is_list(L) ->
|
|
|
|
- lists:foreach(
|
|
|
|
- fun({H, N}) when is_list(N) ->
|
|
|
|
- [Node] = mnesia:read({pubsub_node, {H, N}}),
|
|
|
|
- Type = Node#pubsub_node.type,
|
|
|
|
- BN = element(2, node_call(Type, path_to_node, [N])),
|
|
|
|
- BP = case [element(2, node_call(Type, path_to_node, [P])) || P <- Node#pubsub_node.parents] of
|
|
|
|
- [<<>>] -> [];
|
|
|
|
- Parents -> Parents
|
|
|
|
- end,
|
|
|
|
- mnesia:write(Node#pubsub_node{nodeid={H, BN}, parents=BP}),
|
|
|
|
- mnesia:delete({pubsub_node, {H, N}});
|
|
|
|
- (_) ->
|
|
|
|
- ok
|
|
|
|
- end, mnesia:all_keys(pubsub_node));
|
|
|
|
- _ ->
|
|
|
|
- ok
|
|
|
|
- end
|
|
|
|
- end).
|
2009-08-27 10:57:45 +02:00
|
|
|
-
|
|
|
|
-rename_default_nodeplugin() ->
|
2009-08-25 16:13:26 +02:00
|
|
|
- lists:foreach(fun(Node) ->
|
|
|
|
- mnesia:dirty_write(Node#pubsub_node{type = "hometree"})
|
|
|
|
- end, mnesia:dirty_match_object(#pubsub_node{type = "default", _ = '_'})).
|
2009-08-27 10:57:45 +02:00
|
|
|
-
|
|
|
|
-update_state_database(_Host, _ServerHost) ->
|
|
|
|
- case catch mnesia:table_info(pubsub_state, attributes) of
|
|
|
|
- [stateid, items, affiliation, subscription] ->
|
|
|
|
- ?INFO_MSG("upgrade state pubsub tables", []),
|
|
|
|
- F = fun ({pubsub_state, {JID, NodeID}, Items, Aff, Sub}, Acc) ->
|
|
|
|
- Subs = case Sub of
|
|
|
|
- none ->
|
|
|
|
- [];
|
|
|
|
- _ ->
|
|
|
|
- {result, SubID} = pubsub_subscription:subscribe_node(JID, NodeID, []),
|
|
|
|
- [{Sub, SubID}]
|
|
|
|
- end,
|
|
|
|
- NewState = #pubsub_state{stateid = {JID, NodeID},
|
|
|
|
- items = Items,
|
|
|
|
- affiliation = Aff,
|
|
|
|
- subscriptions = Subs},
|
|
|
|
- [NewState | Acc]
|
|
|
|
- end,
|
|
|
|
- {atomic, NewRecs} = mnesia:transaction(fun mnesia:foldl/3,
|
|
|
|
- [F, [], pubsub_state]),
|
|
|
|
- {atomic, ok} = mnesia:delete_table(pubsub_state),
|
|
|
|
- {atomic, ok} = mnesia:create_table(pubsub_state,
|
|
|
|
- [{disc_copies, [node()]},
|
|
|
|
- {attributes, record_info(fields, pubsub_state)}]),
|
|
|
|
- FNew = fun () ->
|
|
|
|
- lists:foreach(fun mnesia:write/1, NewRecs)
|
|
|
|
- end,
|
|
|
|
- case mnesia:transaction(FNew) of
|
|
|
|
- {atomic, Result} ->
|
|
|
|
- ?INFO_MSG("Pubsub state tables updated correctly: ~p",
|
|
|
|
- [Result]);
|
|
|
|
- {aborted, Reason} ->
|
|
|
|
- ?ERROR_MSG("Problem updating Pubsub state tables:~n~p",
|
|
|
|
- [Reason])
|
|
|
|
- end;
|
|
|
|
- _ ->
|
|
|
|
- ok
|
|
|
|
- end.
|
|
|
|
-
|
2010-01-12 16:14:47 +01:00
|
|
|
send_loop(State) ->
|
|
|
|
receive
|
|
|
|
{presence, JID, Pid} ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -494,17 +299,15 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
%% for each node From is subscribed to
|
|
|
|
%% and if the node is so configured, send the last published item to From
|
|
|
|
lists:foreach(fun(PType) ->
|
|
|
|
- {result, Subscriptions} = node_action(Host, PType, get_entity_subscriptions, [Host, JID]),
|
|
|
|
+ Subscriptions = case catch node_action(Host, PType, get_entity_subscriptions_for_send_last, [Host, JID]) of
|
|
|
|
+ {result, S} -> S;
|
|
|
|
+ _ -> []
|
|
|
|
+ end,
|
|
|
|
lists:foreach(
|
|
|
|
fun({Node, subscribed, _, SubJID}) ->
|
|
|
|
if (SubJID == LJID) or (SubJID == BJID) ->
|
2009-08-28 02:06:28 +02:00
|
|
|
- #pubsub_node{nodeid = {H, N}, type = Type, id = NodeId, options = Options} = Node,
|
2009-08-07 10:26:47 +02:00
|
|
|
- case get_option(Options, send_last_published_item) of
|
|
|
|
- on_sub_and_presence ->
|
2009-09-24 21:28:43 +02:00
|
|
|
- send_items(H, N, NodeId, Type, LJID, last);
|
2009-08-07 10:26:47 +02:00
|
|
|
- _ ->
|
|
|
|
- ok
|
|
|
|
- end;
|
|
|
|
+ #pubsub_node{nodeid = {H, N}, type = Type, id = NodeId} = Node,
|
2009-09-24 21:28:43 +02:00
|
|
|
+ send_items(H, N, NodeId, Type, LJID, last);
|
2009-08-07 10:26:47 +02:00
|
|
|
true ->
|
|
|
|
% resource not concerned about that subscription
|
|
|
|
ok
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -623,7 +426,8 @@
|
2010-08-02 17:07:23 +02:00
|
|
|
disco_identity(_Host, <<>>, _From) ->
|
|
|
|
[{xmlelement, "identity", [{"category", "pubsub"}, {"type", "pep"}], []}];
|
|
|
|
disco_identity(Host, Node, From) ->
|
|
|
|
- Action = fun(#pubsub_node{id = Idx, type = Type, options = Options, owners = Owners}) ->
|
|
|
|
+ Action = fun(#pubsub_node{id = Idx, type = Type, options = Options}) ->
|
|
|
|
+ Owners = node_owners_call(Type, Idx),
|
|
|
|
case get_allowed_items_call(Host, Idx, From, Type, Options, Owners) of
|
|
|
|
{result, _} ->
|
|
|
|
{result, [{xmlelement, "identity", [{"category", "pubsub"}, {"type", "pep"}], []},
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -658,7 +462,8 @@
|
2010-08-02 17:07:23 +02:00
|
|
|
[?NS_PUBSUB
|
|
|
|
| [?NS_PUBSUB++"#"++Feature || Feature <- features("pep")]];
|
|
|
|
disco_features(Host, Node, From) ->
|
|
|
|
- Action = fun(#pubsub_node{id = Idx, type = Type, options = Options, owners = Owners}) ->
|
|
|
|
+ Action = fun(#pubsub_node{id = Idx, type = Type, options = Options}) ->
|
|
|
|
+ Owners = node_owners_call(Type, Idx),
|
|
|
|
case get_allowed_items_call(Host, Idx, From, Type, Options, Owners) of
|
|
|
|
{result, _} ->
|
|
|
|
{result, [?NS_PUBSUB
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -683,7 +488,8 @@
|
2010-08-02 17:07:23 +02:00
|
|
|
Acc.
|
|
|
|
|
|
|
|
disco_items(Host, <<>>, From) ->
|
|
|
|
- Action = fun(#pubsub_node{nodeid ={_, NodeID}, options = Options, type = Type, id = Idx, owners = Owners}, Acc) ->
|
|
|
|
+ Action = fun(#pubsub_node{nodeid ={_, NodeID}, options = Options, type = Type, id = Idx}, Acc) ->
|
|
|
|
+ Owners = node_owners_call(Type, Idx),
|
|
|
|
case get_allowed_items_call(Host, Idx, From, Type, Options, Owners) of
|
|
|
|
{result, _} ->
|
|
|
|
[{xmlelement, "item",
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -701,13 +507,14 @@
|
2010-08-02 17:07:23 +02:00
|
|
|
_ -> Acc
|
|
|
|
end
|
|
|
|
end,
|
|
|
|
- case transaction(Host, Action, sync_dirty) of
|
|
|
|
+ case transaction_on_nodes(Host, Action, sync_dirty) of
|
|
|
|
{result, Items} -> Items;
|
|
|
|
_ -> []
|
|
|
|
end;
|
|
|
|
|
|
|
|
disco_items(Host, Node, From) ->
|
|
|
|
- Action = fun(#pubsub_node{id = Idx, type = Type, options = Options, owners = Owners}) ->
|
|
|
|
+ Action = fun(#pubsub_node{id = Idx, type = Type, options = Options}) ->
|
|
|
|
+ Owners = node_owners_call(Type, Idx),
|
|
|
|
case get_allowed_items_call(Host, Idx, From, Type, Options, Owners) of
|
|
|
|
{result, Items} ->
|
|
|
|
{result, [{xmlelement, "item",
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -793,10 +600,10 @@
|
2010-01-12 16:14:47 +01:00
|
|
|
lists:foreach(fun(PType) ->
|
|
|
|
{result, Subscriptions} = node_action(Host, PType, get_entity_subscriptions, [Host, Entity]),
|
2009-10-13 18:43:02 +02:00
|
|
|
lists:foreach(fun
|
2010-01-12 16:14:47 +01:00
|
|
|
- ({#pubsub_node{options = Options, owners = Owners, id = NodeId}, subscribed, _, JID}) ->
|
|
|
|
+ ({#pubsub_node{options = Options, id = NodeId}, subscribed, _, JID}) ->
|
2009-10-13 18:43:02 +02:00
|
|
|
case get_option(Options, access_model) of
|
|
|
|
presence ->
|
|
|
|
- case lists:member(BJID, Owners) of
|
2010-01-12 16:14:47 +01:00
|
|
|
+ case lists:member(BJID, node_owners(Host, PType, NodeId)) of
|
2009-10-13 18:43:02 +02:00
|
|
|
true ->
|
2010-01-12 16:14:47 +01:00
|
|
|
node_action(Host, PType, unsubscribe_node, [NodeId, Entity, JID, all]);
|
2009-10-13 18:43:02 +02:00
|
|
|
false ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -964,7 +771,8 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
sub_el = SubEl} = IQ ->
|
|
|
|
{xmlelement, _, QAttrs, _} = SubEl,
|
|
|
|
Node = xml:get_attr_s("node", QAttrs),
|
|
|
|
- Res = case iq_disco_items(Host, Node, From) of
|
|
|
|
+ Rsm = jlib:rsm_decode(IQ),
|
|
|
|
+ Res = case iq_disco_items(Host, Node, From, Rsm) of
|
|
|
|
{result, IQRes} ->
|
|
|
|
jlib:iq_to_xml(
|
|
|
|
IQ#iq{type = result,
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1077,7 +885,7 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
[] ->
|
|
|
|
["leaf"]; %% No sub-nodes: it's a leaf node
|
|
|
|
_ ->
|
|
|
|
- case node_call(Type, get_items, [NodeId, From]) of
|
|
|
|
+ case node_call(Type, get_items, [NodeId, From, none]) of
|
|
|
|
{result, []} -> ["collection"];
|
|
|
|
{result, _} -> ["leaf", "collection"];
|
|
|
|
_ -> []
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1093,8 +901,9 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
[];
|
|
|
|
true ->
|
|
|
|
[{xmlelement, "feature", [{"var", ?NS_PUBSUB}], []} |
|
|
|
|
- lists:map(fun(T) ->
|
|
|
|
- {xmlelement, "feature", [{"var", ?NS_PUBSUB++"#"++T}], []}
|
|
|
|
+ lists:map(fun
|
|
|
|
+ ("rsm")-> {xmlelement, "feature", [{"var", ?NS_RSM}], []};
|
|
|
|
+ (T) -> {xmlelement, "feature", [{"var", ?NS_PUBSUB++"#"++T}], []}
|
|
|
|
end, features(Type))]
|
|
|
|
end,
|
|
|
|
%% TODO: add meta-data info (spec section 5.4)
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1123,8 +932,9 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
{xmlelement, "feature", [{"var", ?NS_PUBSUB}], []},
|
2009-11-04 20:54:15 +01:00
|
|
|
{xmlelement, "feature", [{"var", ?NS_COMMANDS}], []},
|
2009-08-07 10:26:47 +02:00
|
|
|
{xmlelement, "feature", [{"var", ?NS_VCARD}], []}] ++
|
|
|
|
- lists:map(fun(Feature) ->
|
|
|
|
- {xmlelement, "feature", [{"var", ?NS_PUBSUB++"#"++Feature}], []}
|
|
|
|
+ lists:map(fun
|
|
|
|
+ ("rsm")-> {xmlelement, "feature", [{"var", ?NS_RSM}], []};
|
|
|
|
+ (T) -> {xmlelement, "feature", [{"var", ?NS_PUBSUB++"#"++T}], []}
|
|
|
|
end, features(Host, Node))};
|
2009-11-04 20:54:15 +01:00
|
|
|
<<?NS_COMMANDS>> ->
|
|
|
|
command_disco_info(Host, Node, From);
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1134,7 +944,7 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
node_disco_info(Host, Node, From)
|
|
|
|
end.
|
|
|
|
|
|
|
|
-iq_disco_items(Host, [], From) ->
|
|
|
|
+iq_disco_items(Host, [], From, _RSM) ->
|
2009-10-27 15:13:18 +01:00
|
|
|
case tree_action(Host, get_subnodes, [Host, <<>>, From]) of
|
2010-03-05 11:53:26 +01:00
|
|
|
Nodes when is_list(Nodes) ->
|
|
|
|
{result, lists:map(
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1151,23 +961,24 @@
|
2010-03-05 11:53:26 +01:00
|
|
|
Other ->
|
|
|
|
Other
|
2009-10-27 15:13:18 +01:00
|
|
|
end;
|
2009-11-04 20:54:15 +01:00
|
|
|
-iq_disco_items(Host, ?NS_COMMANDS, _From) ->
|
|
|
|
+iq_disco_items(Host, ?NS_COMMANDS, _From, _RSM) ->
|
|
|
|
%% TODO: support localization of this string
|
|
|
|
CommandItems = [{xmlelement, "item", [{"jid", Host}, {"node", ?NS_PUBSUB_GET_PENDING}, {"name", "Get Pending"}], []}],
|
|
|
|
{result, CommandItems};
|
|
|
|
-iq_disco_items(_Host, ?NS_PUBSUB_GET_PENDING, _From) ->
|
|
|
|
+iq_disco_items(_Host, ?NS_PUBSUB_GET_PENDING, _From, _RSM) ->
|
|
|
|
CommandItems = [],
|
|
|
|
{result, CommandItems};
|
2009-08-07 10:26:47 +02:00
|
|
|
-iq_disco_items(Host, Item, From) ->
|
|
|
|
+iq_disco_items(Host, Item, From, RSM) ->
|
|
|
|
case string:tokens(Item, "!") of
|
|
|
|
[_SNode, _ItemID] ->
|
|
|
|
{result, []};
|
2010-08-02 17:07:23 +02:00
|
|
|
[SNode] ->
|
2009-10-20 17:03:07 +02:00
|
|
|
Node = string_to_node(SNode),
|
2010-08-02 17:07:23 +02:00
|
|
|
- Action = fun(#pubsub_node{id = Idx, type = Type, options = Options, owners = Owners}) ->
|
|
|
|
- NodeItems = case get_allowed_items_call(Host, Idx, From, Type, Options, Owners) of
|
|
|
|
+ Action = fun(#pubsub_node{id = Idx, type = Type, options = Options}) ->
|
|
|
|
+ Owners = node_owners_call(Type, Idx),
|
|
|
|
+ {NodeItems, RsmOut} = case get_allowed_items_call(Host, Idx, From, Type, Options, Owners, RSM) of
|
|
|
|
{result, R} -> R;
|
|
|
|
- _ -> []
|
|
|
|
+ _ -> {[], none}
|
|
|
|
end,
|
2009-08-07 10:26:47 +02:00
|
|
|
Nodes = lists:map(
|
2010-08-02 17:07:23 +02:00
|
|
|
fun(#pubsub_node{nodeid = {_, SubNode}, options = SubOptions}) ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1185,7 +996,7 @@
|
2010-03-05 11:53:26 +01:00
|
|
|
{result, Name} = node_call(Type, get_item_name, [Host, Node, RN]),
|
|
|
|
{xmlelement, "item", [{"jid", Host}, {"name", Name}], []}
|
|
|
|
end, NodeItems),
|
2009-08-07 10:26:47 +02:00
|
|
|
- {result, Nodes ++ Items}
|
|
|
|
+ {result, Nodes ++ Items ++ jlib:rsm_encode(RsmOut)}
|
|
|
|
end,
|
|
|
|
case transaction(Host, Node, Action, sync_dirty) of
|
|
|
|
{result, {_, Result}} -> {result, Result};
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1296,7 +1107,8 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
(_, Acc) ->
|
|
|
|
Acc
|
|
|
|
end, [], xml:remove_cdata(Els)),
|
|
|
|
- get_items(Host, Node, From, SubId, MaxItems, ItemIDs);
|
|
|
|
+ RSM = jlib:rsm_decode(SubEl),
|
|
|
|
+ get_items(Host, Node, From, SubId, MaxItems, ItemIDs, RSM);
|
|
|
|
{get, "subscriptions"} ->
|
|
|
|
get_subscriptions(Host, Node, From, Plugins);
|
|
|
|
{get, "affiliations"} ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1319,7 +1131,9 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
|
|
|
|
iq_pubsub_owner(Host, ServerHost, From, IQType, SubEl, Lang) ->
|
|
|
|
{xmlelement, _, _, SubEls} = SubEl,
|
|
|
|
- Action = xml:remove_cdata(SubEls),
|
2009-08-28 00:36:46 +02:00
|
|
|
+ Action = lists:filter(fun({xmlelement, "set", _, _}) -> false;
|
|
|
|
+ (_) -> true
|
|
|
|
+ end, xml:remove_cdata(SubEls)),
|
2009-08-07 10:26:47 +02:00
|
|
|
case Action of
|
|
|
|
[{xmlelement, Name, Attrs, Els}] ->
|
2009-10-20 17:03:07 +02:00
|
|
|
Node = string_to_node(xml:get_attr_s("node", Attrs)),
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1449,7 +1263,8 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
_ -> []
|
|
|
|
end
|
|
|
|
end,
|
|
|
|
- case transaction(fun () -> {result, lists:flatmap(Tr, Plugins)} end,
|
|
|
|
+ case transaction(Host,
|
|
|
|
+ fun () -> {result, lists:flatmap(Tr, Plugins)} end,
|
|
|
|
sync_dirty) of
|
|
|
|
{result, Res} -> Res;
|
|
|
|
Err -> Err
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1488,7 +1303,7 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
|
|
|
|
%%% authorization handling
|
|
|
|
|
|
|
|
-send_authorization_request(#pubsub_node{owners = Owners, nodeid = {Host, Node}}, Subscriber) ->
|
|
|
|
+send_authorization_request(#pubsub_node{nodeid = {Host, Node}, type = Type, id = NodeId}, Subscriber) ->
|
|
|
|
Lang = "en", %% TODO fix
|
|
|
|
Stanza = {xmlelement, "message",
|
|
|
|
[],
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1517,7 +1332,7 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
[{xmlelement, "value", [], [{xmlcdata, "false"}]}]}]}]},
|
|
|
|
lists:foreach(fun(Owner) ->
|
2009-12-04 17:31:25 +01:00
|
|
|
ejabberd_router:route(service_jid(Host), jlib:make_jid(Owner), Stanza)
|
2009-08-07 10:26:47 +02:00
|
|
|
- end, Owners).
|
|
|
|
+ end, node_owners(Host, Type, NodeId)).
|
|
|
|
|
|
|
|
find_authorization_response(Packet) ->
|
|
|
|
{xmlelement, _Name, _Attrs, Els} = Packet,
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1581,8 +1396,8 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
"true" -> true;
|
|
|
|
_ -> false
|
|
|
|
end,
|
|
|
|
- Action = fun(#pubsub_node{type = Type, owners = Owners, id = NodeId}) ->
|
|
|
|
- IsApprover = lists:member(jlib:jid_tolower(jlib:jid_remove_resource(From)), Owners),
|
|
|
|
+ Action = fun(#pubsub_node{type = Type, id = NodeId}) ->
|
|
|
|
+ IsApprover = lists:member(jlib:jid_tolower(jlib:jid_remove_resource(From)), node_owners_call(Type, NodeId)),
|
|
|
|
{result, Subscriptions} = node_call(Type, get_subscriptions, [NodeId, Subscriber]),
|
|
|
|
if
|
|
|
|
not IsApprover ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1781,7 +1596,7 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
Reply = [{xmlelement, "pubsub", [{"xmlns", ?NS_PUBSUB}],
|
|
|
|
[{xmlelement, "create", nodeAttr(Node),
|
|
|
|
[]}]}],
|
|
|
|
- case transaction(CreateNode, transaction) of
|
|
|
|
+ case transaction(Host, CreateNode, transaction) of
|
2010-05-28 13:34:05 +02:00
|
|
|
{result, {NodeId, SubsByDepth, {Result, broadcast}}} ->
|
|
|
|
broadcast_created_node(Host, Node, NodeId, Type, NodeOptions, SubsByDepth),
|
2011-11-29 14:13:13 +01:00
|
|
|
ejabberd_hooks:run(pubsub_create_node, ServerHost, [ServerHost, Host, Node, NodeId, NodeOptions]),
|
|
|
|
@@ -1898,7 +1713,7 @@
|
2009-08-27 23:38:23 +02:00
|
|
|
%%<li>The node does not exist.</li>
|
|
|
|
%%</ul>
|
|
|
|
subscribe_node(Host, Node, From, JID, Configuration) ->
|
2009-10-12 11:59:36 +02:00
|
|
|
- SubOpts = case pubsub_subscription:parse_options_xform(Configuration) of
|
|
|
|
+ SubOpts = case pubsub_subscription_odbc:parse_options_xform(Configuration) of
|
|
|
|
{result, GoodSubOpts} -> GoodSubOpts;
|
|
|
|
_ -> invalid
|
|
|
|
end,
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1906,7 +1721,7 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
error -> {"", "", ""};
|
|
|
|
J -> jlib:jid_tolower(J)
|
|
|
|
end,
|
2010-08-02 17:07:23 +02:00
|
|
|
- Action = fun(#pubsub_node{options = Options, owners = Owners, type = Type, id = NodeId}) ->
|
2009-08-07 10:26:47 +02:00
|
|
|
+ Action = fun(#pubsub_node{options = Options, type = Type, id = NodeId}) ->
|
|
|
|
Features = features(Type),
|
|
|
|
SubscribeFeature = lists:member("subscribe", Features),
|
|
|
|
OptionsFeature = lists:member("subscription-options", Features),
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -1915,6 +1730,7 @@
|
2010-08-02 17:07:23 +02:00
|
|
|
AccessModel = get_option(Options, access_model),
|
|
|
|
SendLast = get_option(Options, send_last_published_item),
|
|
|
|
AllowedGroups = get_option(Options, roster_groups_allowed, []),
|
|
|
|
+ Owners = node_owners_call(Type, NodeId),
|
|
|
|
{PresenceSubscription, RosterGroup} = get_presence_and_roster_permissions(Host, Subscriber, Owners, AccessModel, AllowedGroups),
|
2009-08-07 10:26:47 +02:00
|
|
|
if
|
2010-08-02 17:07:23 +02:00
|
|
|
not SubscribeFeature ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2036,12 +1852,9 @@
|
2011-08-31 16:11:01 +02:00
|
|
|
Features = features(Type),
|
|
|
|
PublishFeature = lists:member("publish", Features),
|
|
|
|
PublishModel = get_option(Options, publish_model),
|
|
|
|
+ MaxItems = max_items(Host, Options),
|
|
|
|
DeliverPayloads = get_option(Options, deliver_payloads),
|
|
|
|
PersistItems = get_option(Options, persist_items),
|
|
|
|
- MaxItems = case PersistItems of
|
|
|
|
- false -> 0;
|
|
|
|
- true -> max_items(Host, Options)
|
|
|
|
- end,
|
|
|
|
PayloadCount = payload_xmlelements(Payload),
|
|
|
|
PayloadSize = size(term_to_binary(Payload))-2, % size(term_to_binary([])) == 2
|
|
|
|
PayloadMaxSize = get_option(Options, max_payload_size),
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2092,7 +1905,7 @@
|
2011-08-31 16:11:01 +02:00
|
|
|
false ->
|
|
|
|
ok
|
|
|
|
end,
|
|
|
|
- set_cached_item(Host, NodeId, ItemId, Publisher, Payload),
|
|
|
|
+ set_cached_item(Host, NodeId, ItemId, Publisher, Payload),
|
|
|
|
case Result of
|
|
|
|
default -> {result, Reply};
|
|
|
|
_ -> {result, Result}
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2258,7 +2071,7 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
%% <p>The permission are not checked in this function.</p>
|
|
|
|
%% @todo We probably need to check that the user doing the query has the right
|
|
|
|
%% to read the items.
|
|
|
|
-get_items(Host, Node, From, SubId, SMaxItems, ItemIDs) ->
|
|
|
|
+get_items(Host, Node, From, SubId, SMaxItems, ItemIDs, RSM) ->
|
|
|
|
MaxItems =
|
|
|
|
if
|
2009-09-23 17:56:06 +02:00
|
|
|
SMaxItems == "" -> get_max_items_node(Host);
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2272,12 +2085,13 @@
|
2010-08-02 17:07:23 +02:00
|
|
|
{error, Error} ->
|
|
|
|
{error, Error};
|
|
|
|
_ ->
|
|
|
|
- Action = fun(#pubsub_node{options = Options, type = Type, id = NodeId, owners = Owners}) ->
|
|
|
|
+ Action = fun(#pubsub_node{options = Options, type = Type, id = NodeId}) ->
|
|
|
|
Features = features(Type),
|
|
|
|
RetreiveFeature = lists:member("retrieve-items", Features),
|
|
|
|
PersistentFeature = lists:member("persistent-items", Features),
|
|
|
|
AccessModel = get_option(Options, access_model),
|
|
|
|
AllowedGroups = get_option(Options, roster_groups_allowed, []),
|
|
|
|
+ Owners = node_owners_call(Type, NodeId),
|
|
|
|
{PresenceSubscription, RosterGroup} = get_presence_and_roster_permissions(Host, From, Owners, AccessModel, AllowedGroups),
|
|
|
|
if
|
|
|
|
not RetreiveFeature ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2290,11 +2104,11 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
node_call(Type, get_items,
|
|
|
|
[NodeId, From,
|
|
|
|
AccessModel, PresenceSubscription, RosterGroup,
|
|
|
|
- SubId])
|
|
|
|
+ SubId, RSM])
|
|
|
|
end
|
|
|
|
end,
|
|
|
|
case transaction(Host, Node, Action, sync_dirty) of
|
|
|
|
- {result, {_, Items}} ->
|
|
|
|
+ {result, {_, {Items, RSMOut}}} ->
|
|
|
|
SendItems = case ItemIDs of
|
|
|
|
[] ->
|
|
|
|
Items;
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2307,7 +2121,8 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
%% number of items sent to MaxItems:
|
|
|
|
{result, [{xmlelement, "pubsub", [{"xmlns", ?NS_PUBSUB}],
|
|
|
|
[{xmlelement, "items", nodeAttr(Node),
|
|
|
|
- itemsEls(lists:sublist(SendItems, MaxItems))}]}]};
|
|
|
|
+ itemsEls(lists:sublist(SendItems, MaxItems))}
|
|
|
|
+ | jlib:rsm_encode(RSMOut)]}]};
|
|
|
|
Error ->
|
|
|
|
Error
|
|
|
|
end
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2329,10 +2144,15 @@
|
2010-08-02 17:07:23 +02:00
|
|
|
Error -> Error
|
|
|
|
end.
|
|
|
|
get_allowed_items_call(Host, NodeIdx, From, Type, Options, Owners) ->
|
|
|
|
+ case get_allowed_items_call(Host, NodeIdx, From, Type, Options, Owners, none) of
|
|
|
|
+ {result, {I, _}} -> {result, I};
|
|
|
|
+ Error -> Error
|
|
|
|
+ end.
|
|
|
|
+get_allowed_items_call(Host, NodeIdx, From, Type, Options, Owners, RSM) ->
|
|
|
|
AccessModel = get_option(Options, access_model),
|
|
|
|
AllowedGroups = get_option(Options, roster_groups_allowed, []),
|
|
|
|
{PresenceSubscription, RosterGroup} = get_presence_and_roster_permissions(Host, From, Owners, AccessModel, AllowedGroups),
|
|
|
|
- node_call(Type, get_items, [NodeIdx, From, AccessModel, PresenceSubscription, RosterGroup, undefined]).
|
|
|
|
+ node_call(Type, get_items, [NodeIdx, From, AccessModel, PresenceSubscription, RosterGroup, undefined, RSM]).
|
|
|
|
|
|
|
|
|
|
|
|
%% @spec (Host, Node, NodeId, Type, LJID, Number) -> any()
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2344,31 +2164,29 @@
|
2011-08-31 16:11:01 +02:00
|
|
|
%% Number = last | integer()
|
2009-08-07 10:26:47 +02:00
|
|
|
%% @doc <p>Resend the items of a node to the user.</p>
|
|
|
|
%% @todo use cache-last-item feature
|
2011-08-31 16:11:01 +02:00
|
|
|
-send_items(Host, Node, NodeId, Type, {U,S,R} = LJID, last) ->
|
2009-08-07 10:26:47 +02:00
|
|
|
- case get_cached_item(Host, NodeId) of
|
2011-08-31 16:11:01 +02:00
|
|
|
+send_items(Host, Node, NodeId, Type, LJID, last) ->
|
2009-08-07 10:26:47 +02:00
|
|
|
+ Stanza = case get_cached_item(Host, NodeId) of
|
|
|
|
undefined ->
|
|
|
|
- send_items(Host, Node, NodeId, Type, LJID, 1);
|
|
|
|
+ % special ODBC optimization, works only with node_hometree_odbc, node_flat_odbc and node_pep_odbc
|
2009-09-23 11:27:12 +02:00
|
|
|
+ case node_action(Host, Type, get_last_items, [NodeId, LJID, 1]) of
|
|
|
|
+ {result, [LastItem]} ->
|
2010-05-28 11:56:09 +02:00
|
|
|
+ {ModifNow, ModifUSR} = LastItem#pubsub_item.modification,
|
2009-09-23 11:27:12 +02:00
|
|
|
+ event_stanza_with_delay(
|
|
|
|
+ [{xmlelement, "items", nodeAttr(Node),
|
2010-05-28 11:56:09 +02:00
|
|
|
+ itemsEls([LastItem])}], ModifNow, ModifUSR);
|
2009-09-23 11:27:12 +02:00
|
|
|
+ _ ->
|
|
|
|
+ event_stanza(
|
|
|
|
+ [{xmlelement, "items", nodeAttr(Node),
|
|
|
|
+ itemsEls([])}])
|
|
|
|
+ end;
|
2009-08-07 10:26:47 +02:00
|
|
|
LastItem ->
|
2010-05-28 11:56:09 +02:00
|
|
|
{ModifNow, ModifUSR} = LastItem#pubsub_item.modification,
|
2009-09-23 11:27:12 +02:00
|
|
|
- Stanza = event_stanza_with_delay(
|
|
|
|
+ event_stanza_with_delay(
|
2009-08-07 10:26:47 +02:00
|
|
|
[{xmlelement, "items", nodeAttr(Node),
|
2010-05-28 11:56:09 +02:00
|
|
|
- itemsEls([LastItem])}], ModifNow, ModifUSR),
|
2011-08-31 16:11:01 +02:00
|
|
|
- case is_tuple(Host) of
|
|
|
|
- false ->
|
|
|
|
- ejabberd_router:route(service_jid(Host), jlib:make_jid(LJID), Stanza);
|
|
|
|
- true ->
|
|
|
|
- case ejabberd_sm:get_session_pid(U,S,R) of
|
|
|
|
- C2SPid when is_pid(C2SPid) ->
|
|
|
|
- ejabberd_c2s:broadcast(C2SPid,
|
|
|
|
- {pep_message, binary_to_list(Node)++"+notify"},
|
|
|
|
- _Sender = service_jid(Host),
|
|
|
|
- Stanza);
|
|
|
|
- _ ->
|
|
|
|
- ok
|
|
|
|
- end
|
|
|
|
- end
|
2009-08-07 10:26:47 +02:00
|
|
|
- end;
|
2011-08-31 16:11:01 +02:00
|
|
|
-send_items(Host, Node, NodeId, Type, {U,S,R} = LJID, Number) ->
|
2010-05-28 11:56:09 +02:00
|
|
|
+ itemsEls([LastItem])}], ModifNow, ModifUSR)
|
2009-08-07 10:26:47 +02:00
|
|
|
+ end,
|
2009-12-04 17:31:25 +01:00
|
|
|
+ ejabberd_router:route(service_jid(Host), jlib:make_jid(LJID), Stanza);
|
2011-08-31 16:11:01 +02:00
|
|
|
+send_items(Host, Node, NodeId, Type, LJID, Number) ->
|
2009-08-07 10:26:47 +02:00
|
|
|
ToSend = case node_action(Host, Type, get_items, [NodeId, LJID]) of
|
|
|
|
{result, []} ->
|
2011-08-31 16:11:01 +02:00
|
|
|
[];
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2391,20 +2209,7 @@
|
2011-08-31 16:11:01 +02:00
|
|
|
[{xmlelement, "items", nodeAttr(Node),
|
|
|
|
itemsEls(ToSend)}])
|
|
|
|
end,
|
|
|
|
- case is_tuple(Host) of
|
|
|
|
- false ->
|
|
|
|
- ejabberd_router:route(service_jid(Host), jlib:make_jid(LJID), Stanza);
|
|
|
|
- true ->
|
|
|
|
- case ejabberd_sm:get_session_pid(U,S,R) of
|
|
|
|
- C2SPid when is_pid(C2SPid) ->
|
|
|
|
- ejabberd_c2s:broadcast(C2SPid,
|
|
|
|
- {pep_message, binary_to_list(Node)++"+notify"},
|
|
|
|
- _Sender = service_jid(Host),
|
|
|
|
- Stanza);
|
|
|
|
- _ ->
|
|
|
|
- ok
|
|
|
|
- end
|
|
|
|
- end.
|
|
|
|
+ ejabberd_router:route(service_jid(Host), jlib:make_jid(LJID), Stanza).
|
|
|
|
|
|
|
|
%% @spec (Host, JID, Plugins) -> {error, Reason} | {result, Response}
|
|
|
|
%% Host = host()
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2506,7 +2311,8 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
error ->
|
|
|
|
{error, ?ERR_BAD_REQUEST};
|
|
|
|
_ ->
|
|
|
|
- Action = fun(#pubsub_node{owners = Owners, type = Type, id = NodeId}=N) ->
|
|
|
|
+ Action = fun(#pubsub_node{type = Type, id = NodeId}) ->
|
2010-05-19 11:05:02 +02:00
|
|
|
+ Owners = node_owners_call(Type, NodeId),
|
|
|
|
case lists:member(Owner, Owners) of
|
2009-08-07 10:26:47 +02:00
|
|
|
true ->
|
2010-05-28 12:33:11 +02:00
|
|
|
OwnerJID = jlib:make_jid(Owner),
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2516,24 +2322,7 @@
|
2010-05-19 10:47:47 +02:00
|
|
|
end,
|
2009-08-07 10:26:47 +02:00
|
|
|
lists:foreach(
|
|
|
|
fun({JID, Affiliation}) ->
|
|
|
|
- node_call(Type, set_affiliation, [NodeId, JID, Affiliation]),
|
|
|
|
- case Affiliation of
|
|
|
|
- owner ->
|
|
|
|
- NewOwner = jlib:jid_tolower(jlib:jid_remove_resource(JID)),
|
|
|
|
- NewOwners = [NewOwner|Owners],
|
|
|
|
- tree_call(Host, set_node, [N#pubsub_node{owners = NewOwners}]);
|
|
|
|
- none ->
|
|
|
|
- OldOwner = jlib:jid_tolower(jlib:jid_remove_resource(JID)),
|
|
|
|
- case lists:member(OldOwner, Owners) of
|
|
|
|
- true ->
|
|
|
|
- NewOwners = Owners--[OldOwner],
|
|
|
|
- tree_call(Host, set_node, [N#pubsub_node{owners = NewOwners}]);
|
|
|
|
- _ ->
|
|
|
|
- ok
|
|
|
|
- end;
|
|
|
|
- _ ->
|
|
|
|
- ok
|
|
|
|
- end
|
|
|
|
+ node_call(Type, set_affiliation, [NodeId, JID, Affiliation])
|
2010-05-19 10:47:47 +02:00
|
|
|
end, FilteredEntities),
|
2009-08-07 10:26:47 +02:00
|
|
|
{result, []};
|
|
|
|
_ ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2586,11 +2375,11 @@
|
2009-08-27 23:38:23 +02:00
|
|
|
end.
|
|
|
|
|
2009-09-01 15:16:10 +02:00
|
|
|
read_sub(Subscriber, Node, NodeID, SubID, Lang) ->
|
2009-08-27 23:38:23 +02:00
|
|
|
- case pubsub_subscription:get_subscription(Subscriber, NodeID, SubID) of
|
|
|
|
+ case pubsub_subscription_odbc:get_subscription(Subscriber, NodeID, SubID) of
|
|
|
|
{error, notfound} ->
|
2009-10-12 11:59:36 +02:00
|
|
|
{error, extended_error(?ERR_NOT_ACCEPTABLE, "invalid-subid")};
|
2009-08-27 23:38:23 +02:00
|
|
|
{result, #pubsub_subscription{options = Options}} ->
|
2009-09-01 15:16:10 +02:00
|
|
|
- {result, XdataEl} = pubsub_subscription:get_options_xform(Lang, Options),
|
|
|
|
+ {result, XdataEl} = pubsub_subscription_odbc:get_options_xform(Lang, Options),
|
2009-10-20 17:03:07 +02:00
|
|
|
OptionsEl = {xmlelement, "options", [{"jid", jlib:jid_to_string(Subscriber)},
|
|
|
|
{"subid", SubID}|nodeAttr(Node)],
|
|
|
|
[XdataEl]},
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2616,7 +2405,7 @@
|
2009-08-27 23:38:23 +02:00
|
|
|
end.
|
|
|
|
|
2009-10-12 11:59:36 +02:00
|
|
|
set_options_helper(Configuration, JID, NodeID, SubID, Type) ->
|
|
|
|
- SubOpts = case pubsub_subscription:parse_options_xform(Configuration) of
|
|
|
|
+ SubOpts = case pubsub_subscription_odbc:parse_options_xform(Configuration) of
|
|
|
|
{result, GoodSubOpts} -> GoodSubOpts;
|
|
|
|
_ -> invalid
|
|
|
|
end,
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2645,7 +2434,7 @@
|
2009-10-12 11:59:36 +02:00
|
|
|
write_sub(_Subscriber, _NodeID, _SubID, invalid) ->
|
|
|
|
{error, extended_error(?ERR_BAD_REQUEST, "invalid-options")};
|
2009-08-27 23:38:23 +02:00
|
|
|
write_sub(Subscriber, NodeID, SubID, Options) ->
|
2009-08-28 00:14:48 +02:00
|
|
|
- case pubsub_subscription:set_subscription(Subscriber, NodeID, SubID, Options) of
|
|
|
|
+ case pubsub_subscription_odbc:set_subscription(Subscriber, NodeID, SubID, Options) of
|
2009-08-27 23:38:23 +02:00
|
|
|
{error, notfound} ->
|
2009-10-12 11:59:36 +02:00
|
|
|
{error, extended_error(?ERR_NOT_ACCEPTABLE, "invalid-subid")};
|
2009-08-28 00:14:48 +02:00
|
|
|
{result, _} ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -2813,8 +2602,8 @@
|
2009-08-25 18:35:43 +02:00
|
|
|
{"subscription", subscription_to_string(Sub)} | nodeAttr(Node)], []}]}]},
|
2009-12-04 17:31:25 +01:00
|
|
|
ejabberd_router:route(service_jid(Host), jlib:make_jid(JID), Stanza)
|
2009-08-25 18:35:43 +02:00
|
|
|
end,
|
2009-08-07 10:26:47 +02:00
|
|
|
- Action = fun(#pubsub_node{owners = Owners, type = Type, id = NodeId}) ->
|
2009-08-25 16:13:26 +02:00
|
|
|
- case lists:member(Owner, Owners) of
|
2009-08-07 10:26:47 +02:00
|
|
|
+ Action = fun(#pubsub_node{type = Type, id = NodeId}) ->
|
2009-08-25 16:13:26 +02:00
|
|
|
+ case lists:member(Owner, node_owners_call(Type, NodeId)) of
|
|
|
|
true ->
|
|
|
|
Result = lists:foldl(fun({JID, Subscription, SubId}, Acc) ->
|
|
|
|
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -3169,7 +2958,7 @@
|
2009-08-27 23:38:23 +02:00
|
|
|
{Depth, [{N, get_node_subs(N)} || N <- Nodes]}
|
|
|
|
end, tree_call(Host, get_parentnodes_tree, [Host, Node, service_jid(Host)]))}
|
|
|
|
end,
|
|
|
|
- case transaction(Action, sync_dirty) of
|
|
|
|
+ case transaction(Host, Action, sync_dirty) of
|
|
|
|
{result, CollSubs} -> CollSubs;
|
|
|
|
_ -> []
|
|
|
|
end.
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -3183,9 +2972,9 @@
|
2009-08-27 23:38:23 +02:00
|
|
|
|
|
|
|
get_options_for_subs(NodeID, Subs) ->
|
|
|
|
lists:foldl(fun({JID, subscribed, SubID}, Acc) ->
|
|
|
|
- case pubsub_subscription:read_subscription(JID, NodeID, SubID) of
|
|
|
|
+ case pubsub_subscription_odbc:get_subscription(JID, NodeID, SubID) of
|
|
|
|
{error, notfound} -> [{JID, SubID, []} | Acc];
|
|
|
|
- #pubsub_subscription{options = Options} -> [{JID, SubID, Options} | Acc];
|
|
|
|
+ {result, #pubsub_subscription{options = Options}} -> [{JID, SubID, Options} | Acc];
|
|
|
|
_ -> Acc
|
|
|
|
end;
|
|
|
|
(_, Acc) ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -3374,6 +3163,30 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
Result
|
|
|
|
end.
|
|
|
|
|
2009-08-27 20:15:12 +02:00
|
|
|
+%% @spec (Host, Type, NodeId) -> [ljid()]
|
2009-08-07 10:26:47 +02:00
|
|
|
+%% NodeId = pubsubNodeId()
|
|
|
|
+%% @doc <p>Return list of node owners.</p>
|
|
|
|
+node_owners(Host, Type, NodeId) ->
|
|
|
|
+ case node_action(Host, Type, get_node_affiliations, [NodeId]) of
|
|
|
|
+ {result, Affiliations} ->
|
|
|
|
+ lists:foldl(
|
|
|
|
+ fun({LJID, owner}, Acc) -> [LJID|Acc];
|
|
|
|
+ (_, Acc) -> Acc
|
|
|
|
+ end, [], Affiliations);
|
|
|
|
+ _ ->
|
|
|
|
+ []
|
|
|
|
+ end.
|
|
|
|
+node_owners_call(Type, NodeId) ->
|
|
|
|
+ case node_call(Type, get_node_affiliations, [NodeId]) of
|
|
|
|
+ {result, Affiliations} ->
|
|
|
|
+ lists:foldl(
|
|
|
|
+ fun({LJID, owner}, Acc) -> [LJID|Acc];
|
|
|
|
+ (_, Acc) -> Acc
|
|
|
|
+ end, [], Affiliations);
|
|
|
|
+ _ ->
|
|
|
|
+ []
|
|
|
|
+ end.
|
|
|
|
+
|
2009-08-21 13:54:22 +02:00
|
|
|
%% @spec (Host, Options) -> MaxItems
|
2009-08-07 10:26:47 +02:00
|
|
|
%% Host = host()
|
|
|
|
%% Options = [Option]
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -3770,7 +3583,13 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
tree_action(Host, Function, Args) ->
|
|
|
|
?DEBUG("tree_action ~p ~p ~p",[Host,Function,Args]),
|
|
|
|
Fun = fun() -> tree_call(Host, Function, Args) end,
|
|
|
|
- catch mnesia:sync_dirty(Fun).
|
|
|
|
+ case catch ejabberd_odbc:sql_bloc(odbc_conn(Host), Fun) of
|
|
|
|
+ {atomic, Result} ->
|
|
|
|
+ Result;
|
|
|
|
+ {aborted, Reason} ->
|
|
|
|
+ ?ERROR_MSG("transaction return internal error: ~p~n",[{aborted, Reason}]),
|
|
|
|
+ {error, ?ERR_INTERNAL_SERVER_ERROR}
|
|
|
|
+ end.
|
|
|
|
|
|
|
|
%% @doc <p>node plugin call.</p>
|
|
|
|
node_call(Type, Function, Args) ->
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -3790,13 +3609,13 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
|
|
|
|
node_action(Host, Type, Function, Args) ->
|
|
|
|
?DEBUG("node_action ~p ~p ~p ~p",[Host,Type,Function,Args]),
|
|
|
|
- transaction(fun() ->
|
|
|
|
+ transaction(Host, fun() ->
|
|
|
|
node_call(Type, Function, Args)
|
|
|
|
end, sync_dirty).
|
|
|
|
|
|
|
|
%% @doc <p>plugin transaction handling.</p>
|
|
|
|
transaction(Host, Node, Action, Trans) ->
|
|
|
|
- transaction(fun() ->
|
|
|
|
+ transaction(Host, fun() ->
|
|
|
|
case tree_call(Host, get_node, [Host, Node]) of
|
|
|
|
N when is_record(N, pubsub_node) ->
|
|
|
|
case Action(N) of
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -3808,13 +3627,19 @@
|
2010-08-02 17:07:23 +02:00
|
|
|
Error
|
2009-08-07 10:26:47 +02:00
|
|
|
end
|
|
|
|
end, Trans).
|
2010-08-02 17:07:23 +02:00
|
|
|
-transaction(Host, Action, Trans) ->
|
|
|
|
- transaction(fun() ->
|
|
|
|
+transaction_on_nodes(Host, Action, Trans) ->
|
|
|
|
+ transaction(Host, fun() ->
|
|
|
|
{result, lists:foldl(Action, [], tree_call(Host, get_nodes, [Host]))}
|
|
|
|
end, Trans).
|
2009-08-07 10:26:47 +02:00
|
|
|
|
|
|
|
-transaction(Fun, Trans) ->
|
|
|
|
- case catch mnesia:Trans(Fun) of
|
|
|
|
+transaction(Host, Fun, Trans) ->
|
|
|
|
+ transaction_retry(Host, Fun, Trans, 2).
|
|
|
|
+transaction_retry(Host, Fun, Trans, Count) ->
|
|
|
|
+ SqlFun = case Trans of
|
|
|
|
+ transaction -> sql_transaction;
|
|
|
|
+ _ -> sql_bloc
|
|
|
|
+ end,
|
|
|
|
+ case catch ejabberd_odbc:SqlFun(odbc_conn(Host), Fun) of
|
|
|
|
{result, Result} -> {result, Result};
|
|
|
|
{error, Error} -> {error, Error};
|
|
|
|
{atomic, {result, Result}} -> {result, Result};
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -3822,6 +3647,15 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
{aborted, Reason} ->
|
|
|
|
?ERROR_MSG("transaction return internal error: ~p~n", [{aborted, Reason}]),
|
|
|
|
{error, ?ERR_INTERNAL_SERVER_ERROR};
|
|
|
|
+ {'EXIT', {timeout, _} = Reason} ->
|
|
|
|
+ case Count of
|
|
|
|
+ 0 ->
|
|
|
|
+ ?ERROR_MSG("transaction return internal error: ~p~n", [{'EXIT', Reason}]),
|
|
|
|
+ {error, ?ERR_INTERNAL_SERVER_ERROR};
|
|
|
|
+ N ->
|
|
|
|
+ erlang:yield(),
|
|
|
|
+ transaction_retry(Host, Fun, Trans, N-1)
|
|
|
|
+ end;
|
|
|
|
{'EXIT', Reason} ->
|
|
|
|
?ERROR_MSG("transaction return internal error: ~p~n", [{'EXIT', Reason}]),
|
|
|
|
{error, ?ERR_INTERNAL_SERVER_ERROR};
|
2011-11-29 14:13:13 +01:00
|
|
|
@@ -3830,6 +3664,17 @@
|
2009-08-07 10:26:47 +02:00
|
|
|
{error, ?ERR_INTERNAL_SERVER_ERROR}
|
|
|
|
end.
|
|
|
|
|
|
|
|
+odbc_conn({_U, Host, _R})->
|
|
|
|
+ Host;
|
|
|
|
+odbc_conn(Host) ->
|
2009-12-04 17:08:31 +01:00
|
|
|
+ lists:dropwhile(fun(A) -> A/=$. end, Host) -- ".".
|
2009-08-07 10:26:47 +02:00
|
|
|
+
|
|
|
|
+%% escape value for database storage
|
|
|
|
+escape({_U, _H, _R}=JID)->
|
|
|
|
+ ejabberd_odbc:escape(jlib:jid_to_string(JID));
|
|
|
|
+escape(Value)->
|
|
|
|
+ ejabberd_odbc:escape(Value).
|
|
|
|
+
|
|
|
|
%%%% helpers
|
|
|
|
|
|
|
|
%% Add pubsub-specific error element
|