From d8e8ecaacd94bc6746ffa47a7854c6d0d047ced2 Mon Sep 17 00:00:00 2001
From: Alexey Melnichuk <mimir@newmail.ru>
Date: Mon, 25 Aug 2014 16:31:41 +0500
Subject: [PATCH] Add. Easy wrapper example

---
 examples/easy.obj.lua | 168 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 168 insertions(+)
 create mode 100644 examples/easy.obj.lua

diff --git a/examples/easy.obj.lua b/examples/easy.obj.lua
new file mode 100644
index 0000000..daddfbb
--- /dev/null
+++ b/examples/easy.obj.lua
@@ -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