aboutsummaryrefslogtreecommitdiffstats
path: root/util/datamanager.lua
diff options
context:
space:
mode:
authormatthew <matthew@silver>2008-08-24 18:01:20 +0100
committermatthew <matthew@silver>2008-08-24 18:01:20 +0100
commit5ea14b62950a6a9cbdc96c7354afddcf637d9d41 (patch)
tree1285f7cc14c92306dab94b104648dd1747812069 /util/datamanager.lua
downloadprosody-5ea14b62950a6a9cbdc96c7354afddcf637d9d41.tar.gz
prosody-5ea14b62950a6a9cbdc96c7354afddcf637d9d41.zip
Added all the files to please hg :/
Diffstat (limited to 'util/datamanager.lua')
-rw-r--r--util/datamanager.lua79
1 files changed, 79 insertions, 0 deletions
diff --git a/util/datamanager.lua b/util/datamanager.lua
new file mode 100644
index 00000000..be63673e
--- /dev/null
+++ b/util/datamanager.lua
@@ -0,0 +1,79 @@
+local format = string.format;
+local setmetatable, type = setmetatable, type;
+local pairs = pairs;
+local char = string.char;
+local loadfile, setfenv, pcall = loadfile, setfenv, pcall;
+local log = log;
+local io_open = io.open;
+
+module "datamanager"
+
+
+---- utils -----
+local encode, decode;
+
+local log = function (type, msg) return log(type, "datamanager", msg); end
+
+do
+ local urlcodes = setmetatable({}, { __index = function (t, k) t[k] = char(tonumber("0x"..k)); return t[k]; end });
+
+ decode = function (s)
+ return s and (s:gsub("+", " "):gsub("%%([a-fA-F0-9][a-fA-F0-9])", urlcodes));
+ end
+
+ encode = function (s)
+ return s and (s:gsub("%W", function (c) return format("%%%x", c:byte()); end));
+ end
+end
+
+local function basicSerialize (o)
+ if type(o) == "number" or type(o) == "boolean" then
+ return tostring(o)
+ else -- assume it is a string
+ return string.format("%q", tostring(o))
+ end
+end
+
+
+local function simplesave (f, o)
+ if type(o) == "number" then
+ f:write(o)
+ elseif type(o) == "string" then
+ f:write(format("%q", o))
+ elseif type(o) == "table" then
+ f:write("{\n")
+ for k,v in pairs(o) do
+ f:write(" [", format("%q", k), "] = ")
+ simplesave(f, v)
+ f:write(",\n")
+ end
+ f:write("}\n")
+ else
+ error("cannot serialize a " .. type(o))
+ end
+ end
+
+------- API -------------
+
+function getpath(username, host, datastore)
+ return format("data/%s/%s/%s.dat", encode(host), datastore, encode(username));
+end
+
+function load(username, host, datastore)
+ local data, ret = loadfile(getpath(username, host, datastore));
+ if not data then log("warn", "Failed to load "..datastore.." storage ('"..ret.."') for user: "..username.."@"..host); return nil; end
+ setfenv(data, {});
+ local success, ret = pcall(data);
+ if not success then log("error", "Unable to load "..datastore.." storage ('"..ret.."') for user: "..username.."@"..host); return nil; end
+ return ret;
+end
+
+function store(username, host, datastore, data)
+ local f, msg = io_open(getpath(username, host, datastore), "w+");
+ if not f then log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..username.."@"..host); return nil; end
+ f:write("return ");
+ simplesave(f, data);
+ f:close();
+ return true;
+end
+