2 -- Copyright (C) 2008-2010 Matthew Wild
3 -- Copyright (C) 2008-2010 Waqas Hussain
5 -- This project is MIT/X11 licensed. Please see the
6 -- COPYING file in the source package for more information.
10 local socket = require "socket"
11 local mime = require "mime"
12 local url = require "socket.url"
13 local httpstream_new = require "util.httpstream".new;
15 local server = require "net.server"
17 local connlisteners_get = require "net.connlisteners".get;
18 local listener = connlisteners_get("httpclient") or error("No httpclient listener!");
20 local t_insert, t_concat = table.insert, table.concat;
21 local pairs, ipairs = pairs, ipairs;
22 local tonumber, tostring, xpcall, select, debug_traceback, char, format =
23 tonumber, tostring, xpcall, select, debug.traceback, string.char, string.format;
25 local log = require "util.logger".init("http");
29 function urlencode(s) return s and (s:gsub("%W", function (c) return format("%%%02x", c:byte()); end)); end
30 function urldecode(s) return s and (s:gsub("%%(%x%x)", function (c) return char(tonumber(c,16)); end)); end
32 local function _formencodepart(s)
33 return s and (s:gsub("%W", function (c)
35 return format("%%%02x", c:byte());
41 function formencode(form)
43 for _, field in ipairs(form) do
44 t_insert(result, _formencodepart(field.name).."=".._formencodepart(field.value));
46 return t_concat(result, "&");
49 local function request_reader(request, data, startpos)
50 if not request.parser then
51 local function success_cb(r)
52 if request.callback then
53 for k,v in pairs(r) do request[k] = v; end
54 request.callback(r.body, r.code, request);
55 request.callback = nil;
57 destroy_request(request);
59 local function error_cb(r)
60 if request.callback then
61 request.callback(r or "connection-closed", 0, request);
62 request.callback = nil;
64 destroy_request(request);
66 local function options_cb()
69 request.parser = httpstream_new(success_cb, error_cb, "client", options_cb);
71 request.parser:feed(data);
74 local function handleerr(err) log("error", "Traceback[http]: %s: %s", tostring(err), debug_traceback()); end
75 function request(u, ex, callback)
76 local req = url.parse(u);
78 if not (req and req.host) then
79 callback(nil, 0, req);
80 return nil, "invalid-url";
87 local custom_headers, body;
88 local default_headers = { ["Host"] = req.host, ["User-Agent"] = "Prosody XMPP Server" }
92 default_headers["Authorization"] = "Basic "..mime.b64(req.userinfo);
96 custom_headers = ex.headers;
97 req.onlystatus = ex.onlystatus;
100 req.method = "POST ";
101 default_headers["Content-Length"] = tostring(#body);
102 default_headers["Content-Type"] = "application/x-www-form-urlencoded";
104 if ex.method then req.method = ex.method; end
107 req.handler, req.conn = server.wrapclient(socket.tcp(), req.host, req.port or 80, listener, "*a");
108 req.write = function (...) return req.handler:write(...); end
109 req.conn:settimeout(0);
110 local ok, err = req.conn:connect(req.host, req.port or 80);
111 if not ok and err ~= "timeout" then
112 callback(nil, 0, req);
116 local request_line = { req.method or "GET", " ", req.path, " HTTP/1.1\r\n" };
119 t_insert(request_line, 4, "?");
120 t_insert(request_line, 5, req.query);
123 req.write(t_concat(request_line));
124 local t = { [2] = ": ", [4] = "\r\n" };
125 if custom_headers then
126 for k, v in pairs(custom_headers) do
128 req.write(t_concat(t));
129 default_headers[k] = nil;
133 for k, v in pairs(default_headers) do
135 req.write(t_concat(t));
136 default_headers[k] = nil;
144 req.callback = function (content, code, request) log("debug", "Calling callback, status %s", code or "---"); return select(2, xpcall(function () return callback(content, code, request) end, handleerr)); end
145 req.reader = request_reader;
146 req.state = "status";
148 listener.register_request(req.handler, req);
153 function destroy_request(request)
156 request.handler:close()
157 listener.ondisconnect(request.handler, "closed");
161 _M.urlencode = urlencode;