--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_storage_muconference_readonly/mod_storage_muconference_readonly.lua Thu Jun 30 05:22:12 2016 +0100
@@ -0,0 +1,155 @@
+
+-- luacheck: ignore 212/self
+
+local sql = require "util.sql";
+local xml_parse = require "util.xml".parse;
+local resolve_relative_path = require "util.paths".resolve_relative_path;
+local stanza_preserialize = require "util.stanza".preserialize;
+
+local unpack = unpack
+local function iterator(result)
+ return function(result_)
+ local row = result_();
+ if row ~= nil then
+ return unpack(row);
+ end
+ end, result, nil;
+end
+
+local default_params = { driver = "MySQL" };
+
+local engine;
+
+local host = module.host;
+local room, store;
+
+local function get_best_affiliation(a, b)
+ if a == 'owner' or b == 'owner' then
+ return 'owner';
+ elseif a == 'administrator' or b == 'administrator' then
+ return 'administrator';
+ elseif a == 'outcast' or b == 'outcast' then
+ return 'outcast';
+ elseif a == 'member' or b == 'member' then
+ return 'member';
+ end
+ assert(false);
+end
+
+local function keyval_store_get()
+ if store == "muc" then
+ local room_jid = room.."@"..host;
+ local result;
+ for row in engine:select("SELECT `name`,`desc`,`topic`,`public`,`secret` FROM `rooms` WHERE `jid`=? LIMIT 1", room_jid or "") do result = row end
+ local name = result[1];
+ local desc = result[2];
+ local subject = result[3];
+ local public = result[4];
+ local hidden = public == 0 and true or nil;
+ local secret = result[5];
+ if secret == '' then secret = nil end
+ local affiliations = {};
+ for row in engine:select("SELECT `jid_user`,`affil` FROM `rooms_lists` WHERE `jid_room`=?", room_jid or "") do
+ local jid_user = row[1];
+ local affil = row[2];
+ -- mu-conference has a bug where full JIDs get stored…
+ local bare_jid = jid_user:gsub('/.*', '');
+ local old_affil = affiliations[bare_jid];
+ -- mu-conference has a bug where it can record multiple affiliations…
+ if old_affil ~= nil and old_affil ~= affil then
+ affil = get_best_affiliation(old_affil, affil);
+ end
+ -- terminology is clearly “admin”, not “administrator”.
+ if affil == 'administrator' then
+ affil = 'admin';
+ end
+ affiliations[bare_jid] = affil;
+ end
+ return {
+ jid = room_jid,
+ _data = {
+ persistent = true,
+ name = name,
+ subject = subject,
+ password = secret,
+ hidden = hidden,
+ },
+ _affiliations = affiliations,
+ };
+ end
+end
+
+--- Key/value store API (default store type)
+
+local keyval_store = {};
+keyval_store.__index = keyval_store;
+function keyval_store:get(roomname)
+ room, store = roomname, self.store;
+ local ok, result = engine:transaction(keyval_store_get);
+ if not ok then
+ module:log("error", "Unable to read from database %s store for %s: %s", store, roomname or "<host>", result);
+ return nil, result;
+ end
+ return result;
+end
+
+function keyval_store:users()
+ local host_length = host:len() + 1;
+ local ok, result = engine:transaction(function()
+ return engine:select("SELECT SUBSTRING_INDEX(jid, '@', 1) FROM `rooms`");
+ end);
+ if not ok then return ok, result end
+ return iterator(result);
+end
+
+local stores = {
+ keyval = keyval_store;
+};
+
+--- Implement storage driver API
+
+-- FIXME: Some of these operations need to operate on the archive store(s) too
+
+local driver = {};
+
+function driver:open(store, typ)
+ local store_mt = stores[typ or "keyval"];
+ if store_mt then
+ return setmetatable({ store = store }, store_mt);
+ end
+ return nil, "unsupported-store";
+end
+
+function driver:stores(roomname)
+ local query = "SELECT 'config'";
+ if roomname == true or not roomname then
+ roomname = "";
+ end
+ local ok, result = engine:transaction(function()
+ return engine:select(query, host, roomname);
+ end);
+ if not ok then return ok, result end
+ return iterator(result);
+end
+
+--- Initialization
+
+
+local function normalize_params(params)
+ assert(params.driver and params.database, "Configuration error: Both the SQL driver and the database need to be specified");
+ return params;
+end
+
+function module.load()
+ if prosody.prosodyctl then return; end
+ local engines = module:shared("/*/sql/connections");
+ local params = normalize_params(module:get_option("sql", default_params));
+ engine = engines[sql.db2uri(params)];
+ if not engine then
+ module:log("debug", "Creating new engine");
+ engine = sql:create_engine(params);
+ engines[sql.db2uri(params)] = engine;
+ end
+
+ module:provides("storage", driver);
+end