prosody-modules: mod_carbons/mod

annotate mod_carbons/mod_carbons.lua @ 558:66de25ffc8d9

mod_mam: Implement archiving preferences.

author	Kim Alvefur <zash@zash.se>
date	Sun, 15 Jan 2012 03:58:54 +0100
parents	77f350635f4f
children	ce2798a1bc56

rev	line source
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 f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	6 local st = require "util.stanza";
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	7 local jid_bare = require "util.jid".bare;
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	8 local jid_split = require "util.jid".split;
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	9 local xmlns_carbons = "urn:xmpp:carbons:1";
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	10 local xmlns_forward = "urn:xmpp:forward:0";
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	11 local host_sessions = hosts[module.host].sessions;
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	12
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	13 module:hook("iq/self/"..xmlns_carbons..":enable", function(event)
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	14 local origin, stanza = event.origin, event.stanza;
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	15 if stanza.attr.type == "set" then
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	16 module:log("debug", "%s enabled carbons", origin.full_jid);
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	17 origin.want_carbons = true;
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	18 origin.send(st.reply(stanza));
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	19 return true
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	20 end
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	21 end);
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	22
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	23 module:hook("iq/self/"..xmlns_carbons..":disable", function(event)
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	24 local origin, stanza = event.origin, event.stanza;
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	25 if stanza.attr.type == "set" then
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	26 module:log("debug", "%s disabled carbons", origin.full_jid);
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	27 origin.want_carbons = nil;
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	28 origin.send(st.reply(stanza));
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	29 return true
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	30 end
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	31 end);
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	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 f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	34 local origin, stanza = event.origin, event.stanza;
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	35 local orig_type = stanza.attr.type;
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	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 f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	38
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	39 if not (orig_type == nil
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	40 or orig_type == "normal"
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	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 f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	43 end
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	44
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	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 = {};
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	47 module:log("debug", "origin (%s) type: %s", tostring(origin), origin.type)
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 f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	49 bare_jid = (origin.username.."@"..origin.host)
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	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
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	56 no_carbon_to[resource] = true;
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	57 else
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	58 local top_resources = user_sessions.top_resources;
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	59 for _, session in ipairs(top_resources) do
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	60 no_carbon_to[session.resource] = true;
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	61 end
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	62 end
462 f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	63 end
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	64
480 0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	65 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	66 stanza:maptags(function(tag)
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	67 return tag.attr.xmlns == xmlns_carbons
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	68 and tag.name == "private" and tag or nil;
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	69 end);
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	70 return
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	71 end
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	72
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	73 if not stanza:get_child("forwarded", xmlns_forward) then
462 f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	74 user_sessions = user_sessions and user_sessions.sessions;
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	75 for resource, session in pairs(user_sessions) do
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	76 local full_jid = bare_jid .. "/" .. resource;
480 0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	77 if session.want_carbons then
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	78 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	79 local msg = st.clone(stanza);
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	80 msg.attr.xmlns = msg.attr.xmlns or "jabber:client";
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	81 local fwd = st.message{
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	82 from = bare_jid,
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	83 to = full_jid,
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	84 type = orig_type,
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	85 }
545 77f350635f4f mod_carbons: Update to Carbons v0.6. Kim Alvefur <zash@zash.se> parents: 521 diff changeset	86 :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	87 :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	88 :add_child(msg);
480 0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	89 core_post_stanza(origin, fwd);
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	90 end
462 f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	91 end
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	92 end
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	93 end
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	94 end
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	95
480 0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	96 local function c2s_message_handler(event)
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	97 return message_handler(event, true)
462 f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	98 end
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	99
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	100 -- Stanzas sent by local clients
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	101 module:hook("pre-message/bare", c2s_message_handler, 1);
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	102 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	103 -- Stanzas to local clients
480 0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	104 module:hook("message/bare", message_handler, 1);
0cef5be669de mod_carbons: Consolidate message handlers Kim Alvefur <zash@zash.se> parents: 463 diff changeset	105 module:hook("message/full", message_handler, 1);
462 f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	106
f28a3f260fc2 mod_carbons: Initial commit. Kim Alvefur <zash@zash.se> parents: diff changeset	107 module:add_feature(xmlns_carbons);

Mercurial > prosody-modules

annotate mod_carbons/mod_carbons.lua @ 558:66de25ffc8d9