plugins/adhoc/mod_adhoc.lua

1
-- Copyright (C) 2009 Thilo Cestonaro
2
-- Copyright (C) 2009-2010 Florian Zeitz
3
--
4
-- This file is MIT/X11 licensed. Please see the
5
-- COPYING file in the source package for more information.
6
--
7
8
local st = require "util.stanza";
9
local is_admin = require "core.usermanager".is_admin;
10
local adhoc_handle_cmd = module:require "adhoc".handle_cmd;
11
local xmlns_cmd = "http://jabber.org/protocol/commands";
12
local xmlns_disco = "http://jabber.org/protocol/disco";
13
local commands = {};
14
15
module:add_feature(xmlns_cmd);
16
17
module:hook("iq/host/"..xmlns_disco.."#info:query", function (event)
18
	local origin, stanza = event.origin, event.stanza;
19
	local node = stanza.tags[1].attr.node;
20
	if stanza.attr.type == "get" and node then
21
		if commands[node] then
22
			local privileged = is_admin(stanza.attr.from, stanza.attr.to);
23
			if (commands[node].permission == "admin" and privileged)
24
			    or (commands[node].permission == "user") then
25
				reply = st.reply(stanza);
26
				reply:tag("query", { xmlns = xmlns_disco.."#info",
27
				    node = node });
28
				reply:tag("identity", { name = commands[node].name,
29
				    category = "automation", type = "command-node" }):up();
30
				reply:tag("feature", { var = xmlns_cmd }):up();
31
				reply:tag("feature", { var = "jabber:x:data" }):up();
32
			else
33
				reply = st.error_reply(stanza, "auth", "forbidden", "This item is not available to you");
34
			end
35
			origin.send(reply);
36
			return true;
37
		elseif node == xmlns_cmd then
38
			reply = st.reply(stanza);
39
			reply:tag("query", { xmlns = xmlns_disco.."#info",
40
			    node = node });
41
			reply:tag("identity", { name = "Ad-Hoc Commands",
42
			    category = "automation", type = "command-list" }):up();
43
			origin.send(reply);
44
			return true;
45
46
		end
47
	end
48
end);
49
50
module:hook("iq/host/"..xmlns_disco.."#items:query", function (event)
51
	local origin, stanza = event.origin, event.stanza;
52
	if stanza.attr.type == "get" and stanza.tags[1].attr.node
53
	    and stanza.tags[1].attr.node == xmlns_cmd then
54
		local privileged = is_admin(stanza.attr.from, stanza.attr.to);
55
		reply = st.reply(stanza);
56
		reply:tag("query", { xmlns = xmlns_disco.."#items",
57
		    node = xmlns_cmd });
58
		for node, command in pairs(commands) do
59
			if (command.permission == "admin" and privileged)
60
			    or (command.permission == "user") then
61
				reply:tag("item", { name = command.name,
62
				    node = node, jid = module:get_host() });
63
				reply:up();
64
			end
65
		end
66
		origin.send(reply);
67
		return true;
68
	end
69
end, 500);
70
71
module:hook("iq/host/"..xmlns_cmd..":command", function (event)
72
	local origin, stanza = event.origin, event.stanza;
73
	if stanza.attr.type == "set" then
74
		local node = stanza.tags[1].attr.node
75
		if commands[node] then
76
			local privileged = is_admin(stanza.attr.from, stanza.attr.to);
77
			if commands[node].permission == "admin"
78
			    and not privileged then
79
				origin.send(st.error_reply(stanza, "auth", "forbidden", "You don't have permission to execute this command"):up()
80
				    :add_child(commands[node]:cmdtag("canceled")
81
					:tag("note", {type="error"}):text("You don't have permission to execute this command")));
82
				return true
83
			end
84
			-- User has permission now execute the command
85
			return adhoc_handle_cmd(commands[node], origin, stanza);
86
		end
87
	end
88
end, 500);
89
90
local function handle_item_added(item)
91
	commands[item.node] = item;
92
end
93
94
module:hook("item-added/adhoc", function (event)
95
	return handle_item_added(event.item);
96
end, 500);
97
98
module:hook("item-removed/adhoc", function (event)
99
	commands[event.item.node] = nil;
100
end, 500);
101
102
-- Pick up any items that are already added
103
for _, item in ipairs(module:get_host_items("adhoc")) do
104
	handle_item_added(item);
105
end