|
204
|
1
|
|
|
2 local http = require "socket.http";
|
|
|
3 local url = require "socket.url";
|
|
|
4
|
|
|
5 local couchapi = module:require("couchdb/couchapi");
|
|
|
6 local json = module:require("couchdb/json");
|
|
|
7
|
|
|
8 local couchdb_url = assert(module:get_option("couchdb_url"), "Option couchdb_url not specified");
|
|
|
9 local db = couchapi.db(couchdb_url);
|
|
|
10
|
|
|
11 local function couchdb_get(key)
|
|
|
12 local a,b = db:doc(key):get()
|
|
|
13 print(json.encode(a));
|
|
|
14 if b == 404 then return nil; end
|
|
|
15 if b == 200 then b = nil; end
|
|
|
16 return a.payload,b;
|
|
|
17 end
|
|
|
18
|
|
|
19 local function couchdb_put(key, value)
|
|
|
20 local a,b = db:doc(key):get();
|
|
|
21 return db:doc(key):put({ payload = value, _rev = a and a._rev });
|
|
|
22 end
|
|
|
23
|
|
|
24 local st = require "util.stanza";
|
|
|
25
|
|
|
26 local handlers = {};
|
|
|
27
|
|
|
28 handlers.accounts = {
|
|
|
29 get = function(self, user)
|
|
|
30 return couchdb_get(self.host.."/"..user.."/account");
|
|
|
31 end;
|
|
|
32 set = function(self, user, data)
|
|
|
33 return couchdb_put(self.host.."/"..user.."/account", data);
|
|
|
34 end;
|
|
|
35 };
|
|
|
36 handlers.vcard = {
|
|
|
37 get = function(self, user)
|
|
|
38 return couchdb_get(self.host.."/"..user.."/vcard");
|
|
|
39 end;
|
|
|
40 set = function(self, user, data)
|
|
|
41 return couchdb_put(self.host.."/"..user.."/vcard", data);
|
|
|
42 end;
|
|
|
43 };
|
|
|
44 handlers.private = {
|
|
|
45 get = function(self, user)
|
|
|
46 return couchdb_get(self.host.."/"..user.."/private");
|
|
|
47 end;
|
|
|
48 set = function(self, user, data)
|
|
|
49 return couchdb_put(self.host.."/"..user.."/private", data);
|
|
|
50 end;
|
|
|
51 };
|
|
|
52 handlers.roster = {
|
|
|
53 get = function(self, user)
|
|
|
54 return couchdb_get(self.host.."/"..user.."/roster");
|
|
|
55 end;
|
|
|
56 set = function(self, user, data)
|
|
|
57 return couchdb_put(self.host.."/"..user.."/roster", data);
|
|
|
58 end;
|
|
|
59 };
|
|
|
60
|
|
|
61 -----------------------------
|
|
|
62 local driver = {};
|
|
|
63 driver.__index = driver;
|
|
|
64
|
|
|
65 function driver:open(host, datastore, typ)
|
|
|
66 local cache_key = host.." "..datastore;
|
|
|
67 if self.ds_cache[cache_key] then return self.ds_cache[cache_key]; end
|
|
|
68 local instance = setmetatable({}, self);
|
|
|
69 instance.host = host;
|
|
|
70 instance.datastore = datastore;
|
|
|
71 local handler = handlers[datastore];
|
|
|
72 if not handler then return nil; end
|
|
|
73 for key,val in pairs(handler) do
|
|
|
74 instance[key] = val;
|
|
|
75 end
|
|
|
76 if instance.init then instance:init(); end
|
|
|
77 self.ds_cache[cache_key] = instance;
|
|
|
78 return instance;
|
|
|
79 end
|
|
|
80
|
|
|
81 -----------------------------
|
|
|
82 local _M = {};
|
|
|
83
|
|
|
84 function _M.new()
|
|
|
85 local instance = setmetatable({}, driver);
|
|
|
86 instance.__index = instance;
|
|
|
87 instance.ds_cache = {};
|
|
|
88 return instance;
|
|
|
89 end
|
|
|
90
|
|
|
91 return _M;
|
