2 -- Copyright (C) 2008 Matthew Wild
3 -- Copyright (C) 2008 Waqas Hussain
5 -- This program is free software; you can redistribute it and/or
6 -- modify it under the terms of the GNU General Public License
7 -- as published by the Free Software Foundation; either version 2
8 -- of the License, or (at your option) any later version.
10 -- This program is distributed in the hope that it will be useful,
11 -- but WITHOUT ANY WARRANTY; without even the implied warranty of
12 -- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 -- GNU General Public License for more details.
15 -- You should have received a copy of the GNU General Public License
16 -- along with this program; if not, write to the Free Software
17 -- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
21 local format = string.format;
22 local setmetatable, type = setmetatable, type;
23 local pairs, ipairs = pairs, ipairs;
24 local char = string.char;
25 local loadfile, setfenv, pcall = loadfile, setfenv, pcall;
26 local log = require "util.logger".init("datamanager");
27 local io_open = io.open;
28 local os_remove = os.remove;
29 local tostring, tonumber = tostring, tonumber;
32 local t_insert = table.insert;
34 local indent = function(f, i)
40 local data_path = "data";
49 local urlcodes = setmetatable({}, { __index = function (t, k) t[k] = char(tonumber("0x"..k)); return t[k]; end });
52 return s and (s:gsub("+", " "):gsub("%%([a-fA-F0-9][a-fA-F0-9])", urlcodes));
56 return s and (s:gsub("%W", function (c) return format("%%%x", c:byte()); end));
60 local function basicSerialize (o)
61 if type(o) == "number" or type(o) == "boolean" then
63 else -- assume it is a string -- FIXME make sure it's a string. throw an error otherwise.
64 return (format("%q", tostring(o)):gsub("\\\n", "\\n"));
69 local function simplesave (f, o, ind)
70 if type(o) == "number" then
72 elseif type(o) == "string" then
73 f:write((format("%q", o):gsub("\\\n", "\\n")))
74 elseif type(o) == "table" then
76 for k,v in pairs(o) do
78 f:write("[", basicSerialize(k), "] = ")
79 simplesave(f, v, ind+1)
84 elseif type(o) == "boolean" then
85 f:write(o and "true" or "false");
87 error("cannot serialize a " .. type(o))
91 ------- API -------------
93 function set_data_path(path)
97 function getpath(username, host, datastore, ext)
100 return format("%s/%s/%s/%s.%s", data_path, encode(host), datastore, encode(username), ext);
102 return format("%s/%s/%s.%s", data_path, encode(host), datastore, ext);
104 return format("%s/%s.%s", data_path, datastore, ext);
108 function load(username, host, datastore)
109 local data, ret = loadfile(getpath(username, host, datastore));
111 log("warn", "Failed to load "..datastore.." storage ('"..ret.."') for user: "..(username or "nil").."@"..(host or "nil"));
115 local success, ret = pcall(data);
117 log("error", "Unable to load "..datastore.." storage ('"..ret.."') for user: "..(username or "nil").."@"..(host or "nil"));
123 function store(username, host, datastore, data)
127 -- save the datastore
128 local f, msg = io_open(getpath(username, host, datastore), "w+");
130 log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..(username or "nil").."@"..(host or "nil"));
134 simplesave(f, data, 1);
136 if not next(data) then -- try to delete empty datastore
137 os_remove(getpath(username, host, datastore));
139 -- we write data even when we are deleting because lua doesn't have a
140 -- platform independent way of checking for non-exisitng files
144 function list_append(username, host, datastore, data)
145 if not data then return; end
146 -- save the datastore
147 local f, msg = io_open(getpath(username, host, datastore, "list"), "a+");
149 log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..(username or "nil").."@"..(host or "nil"));
153 simplesave(f, data, 1);
159 function list_store(username, host, datastore, data)
163 -- save the datastore
164 local f, msg = io_open(getpath(username, host, datastore, "list"), "w+");
166 log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..(username or "nil").."@"..(host or "nil"));
169 for _, d in ipairs(data) do
175 if not next(data) then -- try to delete empty datastore
176 os_remove(getpath(username, host, datastore, "list"));
178 -- we write data even when we are deleting because lua doesn't have a
179 -- platform independent way of checking for non-exisitng files
183 function list_load(username, host, datastore)
184 local data, ret = loadfile(getpath(username, host, datastore, "list"));
186 log("warn", "Failed to load "..datastore.." storage ('"..ret.."') for user: "..(username or "nil").."@"..(host or "nil"));
190 setfenv(data, {item = function(i) t_insert(items, i); end});
191 local success, ret = pcall(data);
193 log("error", "Unable to load "..datastore.." storage ('"..ret.."') for user: "..(username or "nil").."@"..(host or "nil"));