Software /
code /
prosody-modules
File
mod_storage_ejabberdsql_readonly/mod_storage_ejabberdsql_readonly.lua @ 4989:b74d592df9e2
mod_http_muc_log: Remove dead code
This might be something left over since a different variant where the
loop went like `for n = i-1, i-100, -1 do ... end` i.e. it went trough a
fixed number of items instead of all the page until the current message.
Then it would have needed something to stop going over the end, but
since the checks are simple it shouldn't be much of a problem looping
over even a very busy day.
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Tue, 16 Aug 2022 01:27:59 +0200 |
parent | 2244:e0663dcd934d |
child | 5707:7c264a2cb970 |
line wrap: on
line source
-- 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 assert(namespace == data.attr.xmlns, "Wrong namespace in private data!"); local key = data.name..":"..data.attr.xmlns; private[key] = 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