author | Kim Alvefur <zash@zash.se> |
Sun, 12 Feb 2012 16:35:54 +0100 | |
changeset 606 | a9249614b0fe |
parent 605 | 5175f6d33470 |
child 617 | 3ca933c9d8ff |
permissions | -rw-r--r-- |
521
6d0d2673f95e
mod_carbons: Add MIT license statement.
Kim Alvefur <zash@zash.se>
parents:
520
diff
changeset
|
1 |
-- XEP-0280: Message Carbons implementation for Prosody |
6d0d2673f95e
mod_carbons: Add MIT license statement.
Kim Alvefur <zash@zash.se>
parents:
520
diff
changeset
|
2 |
-- Copyright (C) 2011 Kim Alvefur |
6d0d2673f95e
mod_carbons: Add MIT license statement.
Kim Alvefur <zash@zash.se>
parents:
520
diff
changeset
|
3 |
-- |
6d0d2673f95e
mod_carbons: Add MIT license statement.
Kim Alvefur <zash@zash.se>
parents:
520
diff
changeset
|
4 |
-- This file is MIT/X11 licensed. |
6d0d2673f95e
mod_carbons: Add MIT license statement.
Kim Alvefur <zash@zash.se>
parents:
520
diff
changeset
|
5 |
|
462 | 6 |
local st = require "util.stanza"; |
7 |
local jid_bare = require "util.jid".bare; |
|
8 |
local jid_split = require "util.jid".split; |
|
9 |
local xmlns_carbons = "urn:xmpp:carbons:1"; |
|
10 |
local xmlns_forward = "urn:xmpp:forward:0"; |
|
11 |
local host_sessions = hosts[module.host].sessions; |
|
12 |
||
13 |
module:hook("iq/self/"..xmlns_carbons..":enable", function(event) |
|
14 |
local origin, stanza = event.origin, event.stanza; |
|
15 |
if stanza.attr.type == "set" then |
|
16 |
module:log("debug", "%s enabled carbons", origin.full_jid); |
|
17 |
origin.want_carbons = true; |
|
18 |
origin.send(st.reply(stanza)); |
|
19 |
return true |
|
20 |
end |
|
21 |
end); |
|
22 |
||
23 |
module:hook("iq/self/"..xmlns_carbons..":disable", function(event) |
|
24 |
local origin, stanza = event.origin, event.stanza; |
|
25 |
if stanza.attr.type == "set" then |
|
26 |
module:log("debug", "%s disabled carbons", origin.full_jid); |
|
27 |
origin.want_carbons = nil; |
|
28 |
origin.send(st.reply(stanza)); |
|
29 |
return true |
|
30 |
end |
|
31 |
end); |
|
32 |
||
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
33 |
local function message_handler(event, c2s) |
462 | 34 |
local origin, stanza = event.origin, event.stanza; |
35 |
local orig_type = stanza.attr.type; |
|
36 |
local orig_to = stanza.attr.to; |
|
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
37 |
local orig_from = stanza.attr.from; |
462 | 38 |
|
39 |
if not (orig_type == nil |
|
40 |
or orig_type == "normal" |
|
41 |
or orig_type == "chat") then |
|
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
42 |
return -- No carbons for messages of type error or headline |
462 | 43 |
end |
44 |
||
45 |
local bare_jid, user_sessions; |
|
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
46 |
local no_carbon_to = {}; |
605
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
47 |
module:log("debug", "Message from %s to %s", orig_from, orig_to); |
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
48 |
if c2s then -- Stanza sent by a local client |
462 | 49 |
bare_jid = (origin.username.."@"..origin.host) |
50 |
user_sessions = host_sessions[origin.username]; |
|
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
51 |
else -- Stanza about to be delivered to a local client |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
52 |
local username, hostname, resource = jid_split(orig_to); |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
53 |
bare_jid = jid_bare(orig_to); |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
54 |
user_sessions = host_sessions[username]; |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
55 |
if resource then |
605
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
56 |
module:log("debug", "Message was to resource %s, it will not get carbon", resource); |
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
57 |
no_carbon_to[resource] = true; |
585
ce2798a1bc56
mod_carbons: Don't try to send carbons for entirely offline users.
Kim Alvefur <zash@zash.se>
parents:
545
diff
changeset
|
58 |
elseif user_sessions then |
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
59 |
local top_resources = user_sessions.top_resources; |
585
ce2798a1bc56
mod_carbons: Don't try to send carbons for entirely offline users.
Kim Alvefur <zash@zash.se>
parents:
545
diff
changeset
|
60 |
if top_resources then |
ce2798a1bc56
mod_carbons: Don't try to send carbons for entirely offline users.
Kim Alvefur <zash@zash.se>
parents:
545
diff
changeset
|
61 |
for _, session in ipairs(top_resources) do |
605
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
62 |
module:log("debug", "Not sending carbons to top resource %s", session.resource); |
585
ce2798a1bc56
mod_carbons: Don't try to send carbons for entirely offline users.
Kim Alvefur <zash@zash.se>
parents:
545
diff
changeset
|
63 |
no_carbon_to[session.resource] = true; |
ce2798a1bc56
mod_carbons: Don't try to send carbons for entirely offline users.
Kim Alvefur <zash@zash.se>
parents:
545
diff
changeset
|
64 |
end |
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
65 |
end |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
66 |
end |
462 | 67 |
end |
68 |
||
585
ce2798a1bc56
mod_carbons: Don't try to send carbons for entirely offline users.
Kim Alvefur <zash@zash.se>
parents:
545
diff
changeset
|
69 |
if not user_sessions then |
605
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
70 |
module:log("debug", "Skip carbons for offline user"); |
585
ce2798a1bc56
mod_carbons: Don't try to send carbons for entirely offline users.
Kim Alvefur <zash@zash.se>
parents:
545
diff
changeset
|
71 |
return -- No use in sending carbons to an offline user |
ce2798a1bc56
mod_carbons: Don't try to send carbons for entirely offline users.
Kim Alvefur <zash@zash.se>
parents:
545
diff
changeset
|
72 |
end |
ce2798a1bc56
mod_carbons: Don't try to send carbons for entirely offline users.
Kim Alvefur <zash@zash.se>
parents:
545
diff
changeset
|
73 |
|
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
74 |
if not c2s and stanza:get_child("private", xmlns_carbons) then |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
75 |
stanza:maptags(function(tag) |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
76 |
return tag.attr.xmlns == xmlns_carbons |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
77 |
and tag.name == "private" and tag or nil; |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
78 |
end); |
605
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
79 |
module:log("debug", "Message tagged private, ignoring"); |
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
80 |
return |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
81 |
end |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
82 |
|
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
83 |
if not stanza:get_child("forwarded", xmlns_forward) then |
462 | 84 |
user_sessions = user_sessions and user_sessions.sessions; |
85 |
for resource, session in pairs(user_sessions) do |
|
86 |
local full_jid = bare_jid .. "/" .. resource; |
|
605
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
87 |
module:log("debug", "%s wants carbons: %s", session.full_jid, tostring(session.want_carbons)); |
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
88 |
if session.want_carbons then |
605
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
89 |
if c2s then |
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
90 |
module:log("debug", "Session is origin: %s", tostring(session == origin)); |
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
91 |
else |
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
92 |
module:log("debug", "Session is in ignore list: %s", tostring(no_carbon_to[resource])); |
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
93 |
end |
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
94 |
if (c2s and session ~= origin) or (not c2s and not no_carbon_to[resource]) then |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
95 |
local msg = st.clone(stanza); |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
96 |
msg.attr.xmlns = msg.attr.xmlns or "jabber:client"; |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
97 |
local fwd = st.message{ |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
98 |
from = bare_jid, |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
99 |
to = full_jid, |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
100 |
type = orig_type, |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
101 |
} |
545
77f350635f4f
mod_carbons: Update to Carbons v0.6.
Kim Alvefur <zash@zash.se>
parents:
521
diff
changeset
|
102 |
:tag(c2s and "sent" or "received", { xmlns = xmlns_carbons }):up() |
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
103 |
:tag("forwarded", { xmlns = xmlns_forward }) |
510
59e80326f2b3
mod_carbons: Fix a typo and unindent a line.
Kim Alvefur <zash@zash.se>
parents:
480
diff
changeset
|
104 |
:add_child(msg); |
605
5175f6d33470
mod_carbons: Add more debug logging
Kim Alvefur <zash@zash.se>
parents:
585
diff
changeset
|
105 |
module:log("debug", "Sending carbon"); |
606
a9249614b0fe
mod_carbons: Use session.send instead of passing carbon through all the routing
Kim Alvefur <zash@zash.se>
parents:
605
diff
changeset
|
106 |
session.send(fwd); |
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
107 |
end |
462 | 108 |
end |
109 |
end |
|
110 |
end |
|
111 |
end |
|
112 |
||
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
113 |
local function c2s_message_handler(event) |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
114 |
return message_handler(event, true) |
462 | 115 |
end |
116 |
||
117 |
-- Stanzas sent by local clients |
|
118 |
module:hook("pre-message/bare", c2s_message_handler, 1); |
|
119 |
module:hook("pre-message/full", c2s_message_handler, 1); |
|
510
59e80326f2b3
mod_carbons: Fix a typo and unindent a line.
Kim Alvefur <zash@zash.se>
parents:
480
diff
changeset
|
120 |
-- Stanzas to local clients |
480
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
121 |
module:hook("message/bare", message_handler, 1); |
0cef5be669de
mod_carbons: Consolidate message handlers
Kim Alvefur <zash@zash.se>
parents:
463
diff
changeset
|
122 |
module:hook("message/full", message_handler, 1); |
462 | 123 |
|
124 |
module:add_feature(xmlns_carbons); |