clix/raw.lua

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
local verse = require "verse";
local envload = require "prosody.util.envload".envload;
local xml = require "prosody.util.xml";
short_opts.i = "interactive";
short_opts.e = "echo";

return function (opts, args)
	if opts.short_help then
		print("Send/receive raw XML to/from the server");
		return;
	end
	
	local send_xml;
	if opts.stdin then
		send_xml = io.read("*a");
	end
	
	local function on_connect(conn)
		_G.conn = conn
		local print = print;
		local function stprint(stanza)
			if stanza.attr.to == conn.jid then
				stanza.attr.to = nil;
			end
			if opts.pretty and stanza.indent then
				return print(stanza:indent(1, "  "):pretty_print());
			end
			return print(stanza);
		end
		conn:hook("stanza", stprint, 1)
		if opts.interactive then
			local stdin = {
				getfd = function () return 0; end;
				dirty = function (self) return false; end;
				settimeout = function () end;
				send = function (_, d) return #d, 0; end;
				close = function () end;
				receive = function (_, patt)
					local data = io.stdin:read(patt);
				if data == nil then
					conn:close();
				end
					if opts.echo then
						io.write(data, patt == "*l" and "\n" or "");
					end
					return data;
				end
			};
			local stwrap;
			do
				local f_mt, r_mt = {}, {};
				function f_mt:__call(...) if ... and type(...) == "string" then return self{ to=... } end return self._f(...) end
				function f_mt:__index(k) return setmetatable({_st = self._f{ type = k }}, r_mt); end
				function r_mt:__call(to) self._st.attr.to = to; return self._st end
				function stwrap(f) return setmetatable({_f=f},f_mt) end
			end
			local env = setmetatable({}, { __index = {
				s = verse.stanza,
				m = stwrap(verse.message),
				p = stwrap(verse.presence),
				iq = stwrap(verse.iq),
				r = function ()
					return verse.stanza("r", { xmlns = "urn:xmpp:sm:3" });
				end,
				ping = function(host)
					return verse.iq{ type="get", to=host}:tag("ping", {xmlns="urn:xmpp:ping"});
				end,
				version = function(host)
					return verse.iq{ type="get", to=host}:query"jabber:iq:version";
				end,
				vcard = function(who)
					return verse.iq{ type="get", to=who}:tag("vCard",{xmlns="vcard-temp"});
				end,
				pubsub = function(...)
					return conn.pubsub(...);
				end,
				disco = function (to, node)
					return verse.iq{ type="get", to=to }:tag("query", { xmlns="http://jabber.org/protocol/disco#info", node = node });
				end,
				items = function (to, node)
					return verse.iq{ type="get", to=to }:tag("query", { xmlns="http://jabber.org/protocol/disco#items", node = node });
				end,
				join = function (roomnick)
					return verse.presence({ to=roomnick }):tag("x", { xmlns="http://jabber.org/protocol/muc" });
				end,
			}});
			local function on_incoming(stdin, data)
				if not data then
					conn:close();
					return
				end
				if data:sub(1,1) ~= "<" then
					local sandboxed = true;
					if data:sub(1,1) == ">" then
						sandboxed = false;
						data = data:sub(2);
					else
						data = "return "..data;
					end
					local chunk, err = envload(data, "@stdin", sandboxed and env or _G);
					if not chunk then
						conn:error(err);
						return;
					end
					data = "";
					local ok, ret = pcall(chunk);
					if ok then
						data = ret
					else
						conn:error(ret);
						return;
					end
				end
				if data then
					conn:send(data);
				end
				if not opts.interactive then
					conn:close();
				end
			end
			stdin = verse.server.wrapclient(stdin, "stdin", 0, {
				onincoming = on_incoming, ondisconnect = function () conn:close() end, onreadtimeout = function () return true; end
				}, "*l");
		else
			if not send_xml then
				send_xml = table.concat(args, " ");
			end
			send_xml = assert(xml.parse(send_xml));
			if send_xml.name == "iq" then
				conn:send_iq(send_xml, function ()
					conn:close();
				end);
			else
				conn:send(send_xml);
				conn:close();
			end
		end

	end
	local plugins = {};
	for i=#args,1,-1 do
		if args[i]:sub(1,1) == "+" then
			table.insert(plugins, table.remove(args, i):sub(2))
		end
	end
	return clix_connect(opts, on_connect, plugins);
end