Software /
code /
prosody-modules
File
mod_bookmarks/mod_bookmarks.lua @ 3238:786ba175f2e5
mod_bookmarks: Require forgotten util.jid module.
author | Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> |
---|---|
date | Sat, 18 Aug 2018 18:08:47 +0100 |
parent | 3235:bd8e94ff726b |
child | 3253:cf8ad9fd7f15 |
line wrap: on
line source
local st = require "util.stanza"; local jid_split = require "util.jid".split; local mod_pep = module:depends "pep"; local private_storage = module:open_store("private", "map"); module:hook("account-disco-info", function (event) event.reply:tag("feature", { var = "urn:xmpp:bookmarks-conversion:0" }):up(); end); local function on_retrieve_private_xml(event) local stanza, session = event.stanza, event.origin; local query = stanza:get_child("query", "jabber:iq:private"); if query == nil then return; end local bookmarks = query:get_child("storage", "storage:bookmarks"); if bookmarks == nil then return; end module:log("debug", "Getting private bookmarks: %s", bookmarks); local username = session.username; local service = mod_pep.get_pep_service(username); local ok, id, item = service:get_last_item("storage:bookmarks", session.full_jid); if not ok then module:log("error", "Failed to retrieve PEP bookmarks of %s: %s", username, id); session.send(st.error_reply(stanza, "cancel", "internal-server-error", "Failed to retrive bookmarks from PEP")); return; end if not id or not item then module:log("debug", "Got no PEP bookmarks item for %s, returning empty private bookmarks", username); session.send(st.reply(stanza):add_child(query)); return end module:log("debug", "Got item %s: %s", id, item); local content = item.tags[1]; module:log("debug", "Sending back private for %s: %s", username, content); session.send(st.reply(stanza):query("jabber:iq:private"):add_child(content)); return true; end function publish_to_pep(jid, bookmarks) local service = mod_pep.get_pep_service(jid_split(jid)); local item = st.stanza("item", { xmlns = "http://jabber.org/protocol/pubsub", id = "current" }) :add_child(bookmarks); local options = { ["persist_items"] = true; ["access_model"] = "whitelist"; }; return service:publish("storage:bookmarks", jid, "current", item, options); end -- Synchronise Private XML to PEP. local function on_publish_private_xml(event) local stanza, session = event.stanza, event.origin; local query = stanza:get_child("query", "jabber:iq:private"); if query == nil then return; end local bookmarks = query:get_child("storage", "storage:bookmarks"); if bookmarks == nil then return; end module:log("debug", "Private bookmarks set by client, publishing to pep"); local ok, err = publish_to_pep(session.full_jid, bookmarks); if not ok then module:log("error", "Failed to publish to PEP bookmarks for %s: %s", session.username, err); session.send(st.error_reply(stanza, "cancel", "internal-server-error", "Failed to store bookmarks to PEP")); return; end session.send(st.reply(stanza)); return true; end local function on_resource_bind(event) local session = event.session; local username = session.username; local data, err = private_storage:get(username, "storage:storage:bookmarks"); if not data then module:log("debug", "No existing Private XML bookmarks for %s, migration already done: %s", username, err); local service = mod_pep.get_pep_service(username); local ok, id = service:get_last_item("storage:bookmarks", session.full_jid); if not ok or not id then module:log("debug", "Additionally, no PEP bookmarks were existing for %s", username); module:fire_event("bookmarks/empty", { session = session }); end return; end local bookmarks = st.deserialize(data); module:log("debug", "Got private bookmarks of %s: %s", username, bookmarks); module:log("debug", "Going to store PEP item for %s", username); local ok, err = publish_to_pep(session.full_jid, bookmarks); if not ok then module:log("error", "Failed to store bookmarks to PEP for %s, aborting migration: %s", username, err); return; end module:log("debug", "Stored bookmarks to PEP for %s", username); local ok, err = private_storage:set(username, "storage:storage:bookmarks", nil); if not ok then module:log("error", "Failed to remove private bookmarks of %s: %s", username, err); return; end module:log("debug", "Removed private bookmarks of %s, migration done!", username); end local function on_item_published(event) if event.node == "storage:bookmarks" then module:fire_event("bookmarks/updated", event); end end module:hook("iq-get/bare/jabber:iq:private:query", on_retrieve_private_xml); module:hook("iq-set/bare/jabber:iq:private:query", on_publish_private_xml); module:hook("resource-bind", on_resource_bind); module:hook("item-published", on_item_published);