Mercurial > prosody-modules
changeset 2223:c3ad652cb71f
mod_storage_ejabberdsql_readonly: Initial commit
author | Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> |
---|---|
date | Thu, 30 Jun 2016 05:27:02 +0100 |
parents | 51596d73157e |
children | 20accfa5b6d6 |
files | mod_storage_ejabberdsql_readonly/README.markdown mod_storage_ejabberdsql_readonly/mod_storage_ejabberdsql_readonly.lua |
diffstat | 2 files changed, 226 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_storage_ejabberdsql_readonly/README.markdown Thu Jun 30 05:27:02 2016 +0100 @@ -0,0 +1,39 @@ +--- +labels: +- 'Type-Storage' +- 'Stage-Alpha' +summary: Ejabberd SQL Read-only Storage Module +... + +Introduction +============ + +This is a storage backend using Ejabberd’s SQL backend. It depends on +[LuaDBI](https://prosody.im/doc/depends#luadbi) + +This module only works in read-only, and was made to be used by +[mod\_migrate](https://modules.prosody.im/mod_migrate.html) to migrate +from Ejabberd’s SQL backend. + +Configuration +============= + +Copy the module to the prosody modules/plugins directory. + +In Prosody's configuration file, set: + + storage = "ejabberdsql_readonly" + +EjabberdSQL options are the same as the +[https://prosody.im/doc/modules/mod\_storage\_sql](SQL ones): + +Compatibility +============= + + ------- --------------------------- + trunk Works + ------- --------------------------- + 0.10 Untested, but should work + ------- --------------------------- + 0.9 Does not work + ------- ---------------------------
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_storage_ejabberdsql_readonly/mod_storage_ejabberdsql_readonly.lua Thu Jun 30 05:27:02 2016 +0100 @@ -0,0 +1,187 @@ + +-- 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 = "SQLite3" }; + +local engine; + +local host = module.host; +local user, store; + +local function keyval_store_get() + if store == "accounts" then + --for row in engine:select("SELECT `password`,`created_at` FROM `users` WHERE `username`=?", user or "") do + local result; + for row in engine:select("SELECT `password` FROM `users` WHERE `username`=? LIMIT 1", user or "") do result = row end + local password = result[1]; + --local created_at = result[2]; + return { password = password }; + + elseif store == "roster" then + local roster = {}; + local pending = nil; + --for row in engine:select("SELECT `jid`,`nick`,`subscription`,`ask`,`askmessage`,`server`,`subscribe`,`type`,`created_at` FROM `rosterusers` WHERE `username`=?", user or "") do + for row in engine:select("SELECT `jid`,`nick`,`subscription`,`ask` FROM `rosterusers` WHERE `username`=?", user or "") do + local contact = row[1]; + local name = row[2]; + if name == "" then name = nil; end + local subscription = row[3]; + if subscription == "N" then + subscription = "none" + elseif subscription == "B" then + subscription = "both" + elseif subscription == "F" then + subscription = "from" + elseif subscription == "T" then + subscription = "to" + else error("Unknown subscription type: "..subscription) end; + local ask = row[4]; + if ask == "N" then + ask = nil; + elseif ask == "O" then + ask = "subscribe"; + elseif ask == "I" then + if pending == nil then pending = {} end; + pending[contact] = true; + ask = nil; + elseif ask == "B" then + if pending == nil then pending = {} end; + pending[contact] = true; + ask = "subscribe"; + else error("Unknown ask type: "..ask); end + + --local askmessage = row[5]; + --local server = row[6]; + --local subscribe = row[7]; + --local type = row[8]; + --local created_at = row[9]; + + local groups = {}; + for row in engine:select("SELECT `grp` FROM `rostergroups` WHERE `username`=? AND `jid`=?", user or "", contact) do + local group = row[1]; + groups[group] = true; + end + + roster[contact] = { name = name, ask = ask, subscription = subscription, groups = groups }; + end + return roster; + + elseif store == "vcard" then + local result = nil; + for row in engine:select("SELECT `vcard` FROM `vcard` WHERE `username`=? LIMIT 1", user or "") do result = row end + if not result then + return nil; + end + local data, err = xml_parse(result[1]); + if data then + return stanza_preserialize(data); + end + + elseif store == "private" then + local private = nil; + local result; + for row in engine:select("SELECT `namespace`,`data` FROM `private_storage` WHERE `username`=?", user or "") do + if private == nil then private = {} end; + local namespace = row[1]; + local data, err = xml_parse(row[2]); + if data then + private[namespace] = stanza_preserialize(data); + end + end + return private; + end +end + +--- Key/value store API (default store type) + +local keyval_store = {}; +keyval_store.__index = keyval_store; +function keyval_store:get(username) + user, store = username, 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, username or "<host>", result); + return nil, result; + end + return result; +end + +function keyval_store:users() + local ok, result = engine:transaction(function() + return engine:select("SELECT `username` FROM `users`"); + 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(username) + local query = "SELECT 'accounts', 'roster', 'vcard', 'private'"; + if username == true or not username then + username = ""; + end + local ok, result = engine:transaction(function() + return engine:select(query, host, username); + end); + if not ok then return ok, result end + return iterator(result); +end + +--- Initialization + + +local function normalize_params(params) + if params.driver == "SQLite3" then + if params.database ~= ":memory:" then + params.database = resolve_relative_path(prosody.paths.data or ".", params.database or "prosody.sqlite"); + end + end + 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