Add. Easy wrapper example
parent
96fd2add7a
commit
d8e8ecaacd
|
@ -0,0 +1,168 @@
|
|||
--- High level wrapper around lcurl.easy
|
||||
--
|
||||
-- Usage:
|
||||
--
|
||||
-- local e = cURL.easy{
|
||||
-- url = 'http://example.com',
|
||||
-- write = io.stdout;
|
||||
-- header = function(msg) io.stderr:write(msg) end;
|
||||
-- }
|
||||
-- e:perform()
|
||||
--
|
||||
|
||||
local curl = require "lcurl.safe"
|
||||
|
||||
local easy = {} do
|
||||
|
||||
local LCURL_ERROR_EASY = 1
|
||||
local CURLE_UNKNOWN_OPTION = 48
|
||||
|
||||
local RENAME_OPT = {
|
||||
read = "readfunction";
|
||||
write = "writefunction";
|
||||
header = "headerfunction";
|
||||
}
|
||||
|
||||
local postfields_storage = setmetatable({}, {__mode = "k"})
|
||||
|
||||
easy.__index = function(self, k)
|
||||
if k:sub(1, 4) == 'set_' then
|
||||
local h = self:handle()
|
||||
local name = k:sub(5)
|
||||
if h['setopt_' .. name] then
|
||||
easy[k] = function(self, value) return self:set(name, value) end
|
||||
end
|
||||
elseif k:sub(1, 4) == 'get_' then
|
||||
local h = self:handle()
|
||||
local name = k:sub(5)
|
||||
if h['getinfo_' .. name] then
|
||||
easy[k] = function(self) return self:get(name) end
|
||||
end
|
||||
end
|
||||
return easy[k]
|
||||
end
|
||||
|
||||
local function setopt(handle, k, v, ...)
|
||||
k = RENAME_OPT[k] or k
|
||||
local fn = handle["setopt_" .. k]
|
||||
if not fn then
|
||||
return nil, curl.error(LCURL_ERROR_EASY, CURLE_UNKNOWN_OPTION)
|
||||
end
|
||||
local ok, err = fn(handle, v, ...)
|
||||
if not ok then return nil, err end
|
||||
return true
|
||||
end
|
||||
|
||||
function easy:new(opt)
|
||||
local handle, err = curl.easy()
|
||||
if not handle then return nil, err end
|
||||
|
||||
if opt then for k, v in pairs(opt) do
|
||||
local ok, err = setopt(handle, k, v)
|
||||
if not ok then
|
||||
handle:close()
|
||||
return nil, err
|
||||
end
|
||||
end end
|
||||
|
||||
local o = setmetatable({
|
||||
_handle = handle;
|
||||
}, self)
|
||||
|
||||
return o
|
||||
end
|
||||
|
||||
function easy:handle()
|
||||
return self._handle
|
||||
end
|
||||
|
||||
function easy:set(k, v, ...)
|
||||
local handle = self:handle()
|
||||
|
||||
if type(k) == "table" then
|
||||
assert(v == nil)
|
||||
for k, v in pairs(k) do
|
||||
local ok, err = setopt(handle, k, v)
|
||||
if not ok then return nil, err, k end
|
||||
end
|
||||
return self
|
||||
end
|
||||
|
||||
assert(type(k) == 'string')
|
||||
local ok, err = setopt(handle, k, v, ...)
|
||||
if not ok then return nil, err end
|
||||
return self
|
||||
end
|
||||
|
||||
function easy:perform(opt)
|
||||
local e = self:handle()
|
||||
|
||||
local oerror = opt.error or function(err) return nil, err end
|
||||
|
||||
if opt.read then
|
||||
local ok, err = e:setopt_readfunction(opt.read)
|
||||
if not ok then return oerror(err) end
|
||||
end
|
||||
|
||||
if opt.write then
|
||||
local ok, err = e:setopt_writefunction(opt.write)
|
||||
if not ok then return oerror(err) end
|
||||
end
|
||||
|
||||
if opt.header then
|
||||
local ok, err = e:setopt_headerfunction(opt.header)
|
||||
if not ok then return oerror(err) end
|
||||
end
|
||||
|
||||
local ok, err = e:perform()
|
||||
if ok then return oerror(err) end
|
||||
|
||||
return self
|
||||
end
|
||||
|
||||
function easy:post(data)
|
||||
local post = curl.httppost()
|
||||
local ok, err
|
||||
for k, v in data do
|
||||
if type(v) == "string" then
|
||||
ok, err = post:add_content(k, v)
|
||||
else
|
||||
assert(type(v) == "table")
|
||||
if v.data then
|
||||
ok, err = post:add_buffer(k, v.file, v.data, v.type, v.headers)
|
||||
else
|
||||
ok, err = post:add_file(k, v.file, v.data, v.type, v.filename, v.headers)
|
||||
end
|
||||
end
|
||||
if not ok then break end
|
||||
end
|
||||
if not ok then
|
||||
post:free()
|
||||
return nil, err
|
||||
end
|
||||
|
||||
local e = self:handle()
|
||||
ok, err = e:setopt_httppost(post)
|
||||
if not ok then
|
||||
post:free()
|
||||
return nil, err
|
||||
end
|
||||
|
||||
return self
|
||||
end
|
||||
|
||||
function easy:get(k, ...)
|
||||
assert(type(k) == "string")
|
||||
local handle = self:handle()
|
||||
local fn = handle["getinfo_" .. k]
|
||||
if not fn then return nil, "Unknown info: " .. k end
|
||||
return fn(handle, ...)
|
||||
end
|
||||
|
||||
end
|
||||
|
||||
local cURL = {}
|
||||
|
||||
cURL.easy = function(...) return easy:new(...) end
|
||||
|
||||
return cURL
|
Loading…
Reference in New Issue