Software /
code /
prosody
Annotate
util/datamanager.lua @ 91:6d66eb6b24cb
Fixed: util.stanza.deserialize now handles nil stanzas
author | Waqas Hussain <waqas20@gmail.com> |
---|---|
date | Thu, 09 Oct 2008 22:19:35 +0500 |
parent | 88:023320150c65 |
child | 117:8e5c5e6a3240 |
rev | line source |
---|---|
0 | 1 local format = string.format; |
2 local setmetatable, type = setmetatable, type; | |
3 local pairs = pairs; | |
4 local char = string.char; | |
5 local loadfile, setfenv, pcall = loadfile, setfenv, pcall; | |
6 local log = log; | |
7 local io_open = io.open; | |
88 | 8 local tostring = tostring; |
0 | 9 |
10 module "datamanager" | |
11 | |
12 | |
13 ---- utils ----- | |
14 local encode, decode; | |
15 | |
16 local log = function (type, msg) return log(type, "datamanager", msg); end | |
17 | |
18 do | |
19 local urlcodes = setmetatable({}, { __index = function (t, k) t[k] = char(tonumber("0x"..k)); return t[k]; end }); | |
20 | |
21 decode = function (s) | |
22 return s and (s:gsub("+", " "):gsub("%%([a-fA-F0-9][a-fA-F0-9])", urlcodes)); | |
23 end | |
24 | |
25 encode = function (s) | |
26 return s and (s:gsub("%W", function (c) return format("%%%x", c:byte()); end)); | |
27 end | |
28 end | |
29 | |
30 local function basicSerialize (o) | |
31 if type(o) == "number" or type(o) == "boolean" then | |
32 return tostring(o) | |
33 else -- assume it is a string | |
88 | 34 return format("%q", tostring(o)) |
0 | 35 end |
36 end | |
37 | |
38 | |
39 local function simplesave (f, o) | |
40 if type(o) == "number" then | |
41 f:write(o) | |
42 elseif type(o) == "string" then | |
43 f:write(format("%q", o)) | |
44 elseif type(o) == "table" then | |
45 f:write("{\n") | |
46 for k,v in pairs(o) do | |
82
cbf387f29d56
Fix for saving tables with non-string keys
Matthew Wild <mwild1@gmail.com>
parents:
0
diff
changeset
|
47 f:write(" [", basicSerialize(k), "] = ") |
0 | 48 simplesave(f, v) |
49 f:write(",\n") | |
50 end | |
51 f:write("}\n") | |
52 else | |
53 error("cannot serialize a " .. type(o)) | |
54 end | |
55 end | |
56 | |
57 ------- API ------------- | |
58 | |
59 function getpath(username, host, datastore) | |
84
d0a0bac6815e
Added: Datastore support for hosts and global data in addition to users
Waqas Hussain <waqas20@gmail.com>
parents:
0
diff
changeset
|
60 if username then |
d0a0bac6815e
Added: Datastore support for hosts and global data in addition to users
Waqas Hussain <waqas20@gmail.com>
parents:
0
diff
changeset
|
61 return format("data/%s/%s/%s.dat", encode(host), datastore, encode(username)); |
d0a0bac6815e
Added: Datastore support for hosts and global data in addition to users
Waqas Hussain <waqas20@gmail.com>
parents:
0
diff
changeset
|
62 elseif host then |
d0a0bac6815e
Added: Datastore support for hosts and global data in addition to users
Waqas Hussain <waqas20@gmail.com>
parents:
0
diff
changeset
|
63 return format("data/%s/%s.dat", encode(host), datastore); |
d0a0bac6815e
Added: Datastore support for hosts and global data in addition to users
Waqas Hussain <waqas20@gmail.com>
parents:
0
diff
changeset
|
64 else |
d0a0bac6815e
Added: Datastore support for hosts and global data in addition to users
Waqas Hussain <waqas20@gmail.com>
parents:
0
diff
changeset
|
65 return format("data/%s.dat", datastore); |
d0a0bac6815e
Added: Datastore support for hosts and global data in addition to users
Waqas Hussain <waqas20@gmail.com>
parents:
0
diff
changeset
|
66 end |
0 | 67 end |
68 | |
69 function load(username, host, datastore) | |
70 local data, ret = loadfile(getpath(username, host, datastore)); | |
71 if not data then log("warn", "Failed to load "..datastore.." storage ('"..ret.."') for user: "..username.."@"..host); return nil; end | |
72 setfenv(data, {}); | |
73 local success, ret = pcall(data); | |
74 if not success then log("error", "Unable to load "..datastore.." storage ('"..ret.."') for user: "..username.."@"..host); return nil; end | |
75 return ret; | |
76 end | |
77 | |
78 function store(username, host, datastore, data) | |
79 local f, msg = io_open(getpath(username, host, datastore), "w+"); | |
80 if not f then log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..username.."@"..host); return nil; end | |
81 f:write("return "); | |
82 simplesave(f, data); | |
83 f:close(); | |
84 return true; | |
85 end | |
86 |