plugins/mod_message.lua

   1
   2 local full_sessions = full_sessions;
   3 local bare_sessions = bare_sessions;
   4
   5 local st = require "util.stanza";
   6 local jid_bare = require "util.jid".bare;
   7 local jid_split = require "util.jid".split;
   8 local user_exists = require "core.usermanager".user_exists;
   9 local offlinemanager = require "core.offlinemanager";
  10 local t_insert = table.insert;
  11
  12 local function select_top_resources(user)
  13         local priority = 0;
  14         local recipients = {};
  15         for _, session in pairs(user.sessions) do -- find resource with greatest priority
  16                 if session.presence then
  17                         -- TODO check active privacy list for session
  18                         local p = session.priority;
  19                         if p > priority then
  20                                 priority = p;
  21                                 recipients = {session};
  22                         elseif p == priority then
  23                                 t_insert(recipients, session);
  24                         end
  25                 end
  26         end
  27         return recipients;
  28 end
  29
  30 local function process_to_bare(bare, origin, stanza)
  31         local user = bare_sessions[bare];
  32
  33         local t = stanza.attr.type;
  34         if t == "error" then
  35                 -- discard
  36         elseif t == "groupchat" then
  37                 origin.send(st.error_reply(stanza, "cancel", "service-unavailable"));
  38         elseif t == "headline" then
  39                 if user then
  40                         for _, session in pairs(user.sessions) do
  41                                 if session.presence and session.priority >= 0 then
  42                                         session.send(stanza);
  43                                 end
  44                         end
  45                 end  -- current policy is to discard headlines if no recipient is available
  46         else -- chat or normal message
  47                 if user then -- some resources are connected
  48                         local recipients = select_top_resources(user);
  49                         if #recipients > 0 then
  50                                 for i=1,#recipients do
  51                                         recipients[i].send(stanza);
  52                                 end
  53                                 return true;
  54                         end
  55                 end
  56                 -- no resources are online
  57                 local node, host = jid_split(bare);
  58                 if user_exists(node, host) then
  59                         -- TODO apply the default privacy list
  60                         offlinemanager.store(node, host, stanza);
  61                 else
  62                         origin.send(st.error_reply(stanza, "cancel", "service-unavailable"));
  63                 end
  64         end
  65         return true;
  66 end
  67
  68 module:hook("message/full", function(data)
  69         -- message to full JID recieved
  70         local origin, stanza = data.origin, data.stanza;
  71
  72         local session = full_sessions[stanza.attr.to];
  73         if session then
  74                 -- TODO fire post processing event
  75                 session.send(stanza);
  76                 return true;
  77         else -- resource not online
  78                 return process_to_bare(jid_bare(stanza.attr.to), origin, stanza);
  79         end
  80 end);
  81
  82 module:hook("message/bare", function(data)
  83         -- message to bare JID recieved
  84         local origin, stanza = data.origin, data.stanza;
  85
  86         return process_to_bare(stanza.attr.to or (origin.username..'@'..origin.host), origin, stanza);
  87 end);