plugins/mod_message.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2010 Matthew Wild
   3 -- Copyright (C) 2008-2010 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9
  10 local full_sessions = prosody.full_sessions;
  11 local bare_sessions = prosody.bare_sessions;
  12
  13 local st = require "util.stanza";
  14 local jid_bare = require "util.jid".bare;
  15 local jid_split = require "util.jid".split;
  16 local user_exists = require "core.usermanager".user_exists;
  17
  18 local function process_to_bare(bare, origin, stanza)
  19         local user = bare_sessions[bare];
  20
  21         local t = stanza.attr.type;
  22         if t == "error" then
  23                 -- discard
  24         elseif t == "groupchat" then
  25                 origin.send(st.error_reply(stanza, "cancel", "service-unavailable"));
  26         elseif t == "headline" then
  27                 if user and stanza.attr.to == bare then
  28                         for _, session in pairs(user.sessions) do
  29                                 if session.presence and session.priority >= 0 then
  30                                         session.send(stanza);
  31                                 end
  32                         end
  33                 end  -- current policy is to discard headlines if no recipient is available
  34         else -- chat or normal message
  35                 if user then -- some resources are connected
  36                         local recipients = user.top_resources;
  37                         if recipients then
  38                                 local sent;
  39                                 for i=1,#recipients do
  40                                         sent = recipients[i].send(stanza) or sent;
  41                                 end
  42                                 if sent then
  43                                         return true;
  44                                 end
  45                         end
  46                 end
  47                 -- no resources are online
  48                 local node, host = jid_split(bare);
  49                 local ok
  50                 if user_exists(node, host) then
  51                         -- TODO apply the default privacy list
  52
  53                         ok = module:fire_event('message/offline/handle', {
  54                             origin = origin,
  55                             stanza = stanza,
  56                         });
  57                 end
  58
  59                 if not ok then
  60                         origin.send(st.error_reply(stanza, "cancel", "service-unavailable"));
  61                 end
  62         end
  63         return true;
  64 end
  65
  66 module:hook("message/full", function(data)
  67         -- message to full JID recieved
  68         local origin, stanza = data.origin, data.stanza;
  69
  70         local session = full_sessions[stanza.attr.to];
  71         if session and session.send(stanza) then
  72                 return true;
  73         else -- resource not online
  74                 return process_to_bare(jid_bare(stanza.attr.to), origin, stanza);
  75         end
  76 end);
  77
  78 module:hook("message/bare", function(data)
  79         -- message to bare JID recieved
  80         local origin, stanza = data.origin, data.stanza;
  81
  82         return process_to_bare(stanza.attr.to or (origin.username..'@'..origin.host), origin, stanza);
  83 end);
  84
  85 module:add_feature("msgoffline");