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                 return true; -- 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                         ok = module:fire_event('message/offline/handle', {
  52                                 username = node;
  53                                 origin = origin,
  54                                 stanza = stanza,
  55                         });
  56                 end
  57
  58                 if not ok then
  59                         origin.send(st.error_reply(stanza, "cancel", "service-unavailable"));
  60                 end
  61         end
  62         return true;
  63 end
  64
  65 module:hook("message/full", function(data)
  66         -- message to full JID received
  67         local origin, stanza = data.origin, data.stanza;
  68
  69         local session = full_sessions[stanza.attr.to];
  70         if session and session.send(stanza) then
  71                 return true;
  72         else -- resource not online
  73                 return process_to_bare(jid_bare(stanza.attr.to), origin, stanza);
  74         end
  75 end, -1);
  76
  77 module:hook("message/bare", function(data)
  78         -- message to bare JID received
  79         local origin, stanza = data.origin, data.stanza;
  80
  81         return process_to_bare(stanza.attr.to or (origin.username..'@'..origin.host), origin, stanza);
  82 end, -1);
  83
  84 module:add_feature("msgoffline");