1 

     2 local setmetatable = setmetatable;

     3 local pcall = pcall;

     4 local type = type;

     5 local t_concat = table.concat;

     6 local print = print;

     7 

     8 local socket_url = require "socket.url";

     9 local http = require "socket.http";

    10 local ltn12 = require "ltn12";

    11 

    12 --local json = require "json";

    13 local json = module:require("couchdb/json");

    14 

    15 --module("couchdb")

    16 local _M = {};

    17 

    18 local function urlcat(url, path)

    19 	return url:gsub("/*$", "").."/"..path:gsub("^/*", "");

    20 end

    21 

    22 local doc_mt = {};

    23 doc_mt.__index = doc_mt;

    24 

    25 function doc_mt:get()

    26 	return self.db:get(socket_url.escape(self.id));

    27 end

    28 function doc_mt:put(val)

    29 	return self.db:put(socket_url.escape(self.id), val);

    30 end

    31 function doc_mt:__tostring()

    32 	return "couchdb.doc("..self.url..")";

    33 end

    34 

    35 

    36 local db_mt = {};

    37 db_mt.__index = db_mt;

    38 

    39 function db_mt:__tostring()

    40 	return "couchdb.db("..self.url..")";

    41 end

    42 function db_mt:doc(id)

    43 	local url = urlcat(self.url, socket_url.escape(id));

    44 	return setmetatable({ url = url, db = self, id = id }, doc_mt);

    45 end

    46 function db_mt:get(id)

    47 	local url = urlcat(self.url, id);

    48 	local a,b = http.request(url);

    49 	local r,x = pcall(json.decode, a);

    50 	if r then a = x; end

    51 	return a,b;

    52 end

    53 function db_mt:put(id, value)

    54 	local url = urlcat(self.url, id);

    55 	if type(value) == "table" then

    56 		value = json.encode(value);

    57 	elseif value ~= nil and type(value) ~= "string" then

    58 		return nil, "Invalid type";

    59 	end

    60 	local t = {};

    61 	local a,b = http.request {

    62 		url = url,

    63 		sink = ltn12.sink.table(t),

    64 		source = ltn12.source.string(value),

    65 		method = "PUT",

    66 		headers = {

    67 			["Content-Length"] = #value,

    68 			["Content-Type"] = "application/json"

    69 		}

    70 	};

    71 	a = t_concat(t);

    72 	local r,x = pcall(json.decode, a);

    73 	if r then a = x; end

    74 	return a,b;

    75 end

    76 

    77 

    78 local server_mt = {};

    79 server_mt.__index = server_mt;

    80 

    81 function server_mt:db(name)

    82 	local url = urlcat(self.url, socket_url.escape(name));

    83 	return setmetatable({ url = url }, db_mt);

    84 end

    85 function server_mt:__tostring()

    86 	return "couchdb.server("..self.url..")";

    87 end

    88 

    89 

    90 function _M.server(url)

    91 	return setmetatable({ url = url }, server_mt);

    92 end

    93 function _M.db(url)

    94 	return setmetatable({ url = url }, db_mt);

    95 end

    96 

    97 return _M;