Mercurial > prosody-modules
diff mod_privacy/mod_privacy.lua @ 8:10502594a49b
adds mod_privacy; lists creating, editing and deletion working.
author | Thilo Cestonaro <thilo@cestona.ro> |
---|---|
date | Fri, 25 Sep 2009 16:56:50 +0200 |
parents | |
children | 7d70faba234c |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_privacy/mod_privacy.lua Fri Sep 25 16:56:50 2009 +0200 @@ -0,0 +1,214 @@ +-- Prosody IM +-- Copyright (C) 2008-2009 Matthew Wild +-- Copyright (C) 2008-2009 Waqas Hussain +-- Copyright (C) 2009 Thilo Cestonaro +-- +-- This project is MIT/X11 licensed. Please see the +-- COPYING file in the source package for more information. +-- + + +local prosody = prosody; +local helpers = require "util/helpers"; +local st = require "util.stanza"; +local datamanager = require "util.datamanager"; +local privacy_lists = nil +local bare_sessions = bare_sessions; + + +function findNamedList (name) + local ret = nil + if privacy_lists.lists == nil then return nil; end + + for i=1, #privacy_lists.lists do + if privacy_lists.lists[i].name == name then + ret = i; + break; + end + end + return ret; +end + +function declineList (origin, stanza, which) + module:log("info", "User requests to decline the use of privacy list: %s", which); + privacy_lists[which] = nil; + origin.send(st.reply(stanza)); + return true; +end + +function activateList (origin, stanza, which, name) + module:log("info", "User requests to change the privacy list: %s, to be list named %s", which, name); + local ret = false; + local idx = findNamedList(name); + + if privacy_lists[which] == nil then + privacy_lists[which] = ""; + end + + if privacy_lists[which] ~= name and idx ~= nil then + privacy_lists[which] = name; + origin.send(st.reply(stanza)); + ret = true; + end + return ret; +end + +function deleteList (origin, stanza, name) + module:log("info", "User requests to delete privacy list: %s", name); + local ret = false; + local idx = findNamedList(name); + + if idx ~= nil then + table.remove(privacy_lists.lists, idx); + origin.send(st.reply(stanza)); + ret = true; + end + return ret; +end + +function createOrReplaceList (origin, stanza, name, entries) + module:log("info", "User requests to create / replace list named %s, item count: %d", name, #entries); + local ret = true; + local idx = findNamedList(name); + local bare_jid = origin.username.."@"..origin.host; + + if privacy_lists.lists == nil then + privacy_lists.lists = {}; + end + + if idx == nil then + idx = #privacy_lists.lists + 1; + end + + local list = {}; + list.name = name; + list.items = {}; + + for _,item in ipairs(entries) do + tmp = {}; + tmp["type"] = item.attr.type; + tmp["value"] = item.attr.value; + tmp["action"] = item.attr.action; + tmp["order"] = item.attr.order; + tmp["presence-in"] = false; + tmp["presence-out"] = false; + tmp["message"] = false; + tmp["iq"] = false; + + if #item.tags > 0 then + for _,tag in ipairs(item.tags) do + tmp[tag.name] = true; + end + end + list.items[#list.items + 1] = tmp; + end + + privacy_lists.lists[idx] = list; + origin.send(st.reply(stanza)); + if bare_sessions[bare_jid] ~= nil then + iq = st.iq ( { type = "set", id="push1" } ); + iq:tag ("query", { xmlns = "jabber:iq:privacy" } ); + iq:tag ("list", { name = list.name } ):up(); + iq:up(); + for resource, session in pairs(bare_sessions[bare_jid].sessions) do + iq.attr.to = bare_jid.."/"..resource + session.send(iq); + end + end + return true; +end + +function getList(origin, stanza, name) + module:log("info", "User requests list named: %s", name or "nil"); + local ret = true; + local idx = findNamedList(name); + local reply = st.reply(stanza); + reply = reply:tag("query", {xmlns="jabber:iq:privacy"}); + + if idx == nil then + reply:tag("active", {name=privacy_lists.active or ""}):up(); + reply:tag("default", {name=privacy_lists.default or ""}):up(); + if privacy_lists.lists then + for _,list in ipairs(privacy_lists.lists) do + reply:tag("list", {name=list.name}):up(); + end + end + else + list = privacy_lists.lists[idx]; + reply = reply:tag("list", {name=list.name}); + for _,item in ipairs(list.items) do + reply:tag("item", {type=item.type, value=item.value, action=item.action, order=item.order}); + if item["message"] then reply:tag("message"):up(); end + if item["iq"] then reply:tag("iq"):up(); end + if item["presence-in"] then reply:tag("presence-in"):up(); end + if item["presence-out"] then reply:tag("presence-out"):up(); end + reply:up(); + end + end + + origin.send(reply); + return ret; +end + +-- "[tagname]/[target-type]/[payload-namespace]:[payload-tagname]" +module:hook("iq/bare/jabber:iq:privacy:query", function(data) + local origin, stanza = data.origin, data.stanza; + + if stanza.attr.to == nil then -- only service requests to own bare JID + local query = stanza.tags[1]; -- the query element + local valid = false; + privacy_lists = datamanager.load(origin.username, origin.host, "privacy") or {}; + + if stanza.attr.type == "set" then + if #query.tags >= 1 then + for _,tag in ipairs(query.tags) do + if tag.name == "active" or tag.name == "default" then + if tag.attr.name == nil then -- Client declines the use of active / default list + valid = declineList(origin, stanza, tag.name); + else -- Client requests change of active / default list + valid = activateList(origin, stanza, tag.name, tag.attr.name); + end + elseif tag.name == "list" and tag.attr.name then -- Client adds / edits a privacy list + if #tag.tags == 0 then -- Client removes a privacy list + valid = deleteList(origin, stanza, tag.attr.name); + else -- Client edits a privacy list + valid = createOrReplaceList(origin, stanza, tag.attr.name, tag.tags) + end + end + end + end + elseif stanza.attr.type == "get" then + local name = nil; + if #query.tags >= 1 then + for _,tag in ipairs(query.tags) do + if tag.name == "list" then -- Client requests a privacy list from server + name = tag.attr.name; + break; + end + end + end + valid = getList(origin, stanza, name); + end + + if valid == false then + origin.send(st.error_reply(stanza, "modify", "bad-request")); + else + datamanager.store(origin.username, origin.host, "privacy", privacy_lists); + end + return true; + end + return false; +end, 500); + +function checkIfNeedToBeBlocked(e) + return false; +end + +module:hook("pre-message/full", checkIfNeedToBeBlocked, 500); +module:hook("pre-iq/bare", checkIfNeedToBeBlocked, 500); +module:hook("pre-presence/bare", checkIfNeedToBeBlocked, 500); + +-- helpers.log_events(hosts["albastru.de"].events, "albastru.de"); +-- helpers.log_events(prosody.events, "*"); + +module:log("info", "mod_privacy loaded ...");