From 5ea14b62950a6a9cbdc96c7354afddcf637d9d41 Mon Sep 17 00:00:00 2001 From: matthew Date: Sun, 24 Aug 2008 18:01:20 +0100 Subject: Added all the files to please hg :/ --- util/datamanager.lua | 79 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 79 insertions(+) create mode 100644 util/datamanager.lua (limited to 'util/datamanager.lua') 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 + -- cgit v1.2.3