MUC: Treat explicit join presence as join
[prosody.git] / plugins / muc / muc.lib.lua
index d084d9d534c189a519c71d9131a6a5becc28b468..71f199dd48206d1d4dce2ab350b54f88478faec3 100644 (file)
@@ -384,34 +384,90 @@ module:hook("muc-occupant-pre-join", function(event)
        end
 end, -10);
 
+function room_mt:handle_first_presence(origin, stanza)
+       local real_jid = stanza.attr.from;
+       local dest_jid = stanza.attr.to;
+       local bare_jid = jid_bare(real_jid);
+       if module:fire_event("muc-room-pre-create", {
+                       room = self;
+                       origin = origin;
+                       stanza = stanza;
+               }) then return true; end
+       local is_first_dest_session = true;
+       local dest_occupant = self:new_occupant(bare_jid, dest_jid);
+
+       -- TODO Handle this case sensibly
+       if not stanza:get_child("x", "http://jabber.org/protocol/muc") then
+               module:log("debug", "Room creation without <x>, possibly desynced");
+       end
+
+       if module:fire_event("muc-occupant-pre-join", {
+               room = self;
+               origin = origin;
+               stanza = stanza;
+               is_first_session = is_first_dest_session;
+               is_new_room = true;
+               occupant = dest_occupant;
+       }) then return true; end
+
+       dest_occupant:set_session(real_jid, stanza);
+       local dest_x = st.stanza("x", {xmlns = "http://jabber.org/protocol/muc#user";});
+       dest_x:tag("status", {code = "201"}):up();
+       if self:get_whois() == "anyone" then
+               dest_x:tag("status", {code = "100"}):up();
+       end
+       self:save_occupant(dest_occupant);
+
+       self:publicise_occupant_status(dest_occupant, dest_x);
+
+       module:fire_event("muc-occupant-joined", {
+               room = self;
+               nick = dest_occupant.nick;
+               occupant = dest_occupant;
+               stanza = stanza;
+               origin = origin;
+       });
+       module:fire_event("muc-occupant-session-new", {
+               room = self;
+               nick = dest_occupant.nick;
+               occupant = dest_occupant;
+               stanza = stanza;
+               origin = origin;
+               jid = real_jid;
+       });
+       module:fire_event("muc-room-created", {
+               room = self;
+               creator = dest_occupant;
+               stanza = stanza;
+               origin = origin;
+       });
+       return true;
+end
+
 function room_mt:handle_normal_presence(origin, stanza)
        local type = stanza.attr.type;
+       local muc_x = stanza:get_child("x", "http://jabber.org/protocol/muc");
        local real_jid = stanza.attr.from;
        local bare_jid = jid_bare(real_jid);
-       local orig_occupant, dest_occupant;
-       local is_new_room = next(self._affiliations) == nil;
-       if is_new_room then
-               if type == "unavailable" then return true; end -- Unavailable from someone not in the room
-               if module:fire_event("muc-room-pre-create", {
-                               room = self;
-                               origin = origin;
-                               stanza = stanza;
-                       }) then return true; end
-       else
-               orig_occupant = self:get_occupant_by_real_jid(real_jid);
-               if type == "unavailable" and orig_occupant == nil then return true; end -- Unavailable from someone not in the room
-       end
+       local orig_occupant = self:get_occupant_by_real_jid(real_jid);
+       if type == "unavailable" and orig_occupant == nil then return true; end -- Unavailable from someone not in the room
        local is_first_dest_session;
+       local dest_occupant;
        if type == "unavailable" then -- luacheck: ignore 542
                -- FIXME Why the empty if branch?
                -- dest_occupant = nil
-       elseif orig_occupant and orig_occupant.nick == stanza.attr.to then -- Just a presence update
+       elseif orig_occupant and not muc_x and orig_occupant.nick == stanza.attr.to then -- Just a presence update
                log("debug", "presence update for %s from session %s", orig_occupant.nick, real_jid);
                dest_occupant = orig_occupant;
        else
                local dest_jid = stanza.attr.to;
                dest_occupant = self:get_occupant_by_nick(dest_jid);
-               if dest_occupant == nil then
+               if muc_x then
+                       dest_occupant = self:new_occupant(bare_jid, dest_jid);
+                       if dest_occupant == nil then
+                               is_first_dest_session = true;
+                       end
+               elseif dest_occupant == nil then
                        log("debug", "no occupant found for %s; creating new occupant object for %s", dest_jid, real_jid);
                        is_first_dest_session = true;
                        dest_occupant = self:new_occupant(bare_jid, dest_jid);
@@ -427,7 +483,6 @@ function room_mt:handle_normal_presence(origin, stanza)
        end
 
        -- TODO Handle these cases sensibly
-       local muc_x = stanza:get_child("x", "http://jabber.org/protocol/muc");
        if orig_occupant == nil and not muc_x then
                module:log("debug", "Join without <x>, possibly desynced");
        elseif orig_occupant ~= nil and muc_x then
@@ -443,7 +498,6 @@ function room_mt:handle_normal_presence(origin, stanza)
        };
        if orig_occupant == nil then
                event_name = "muc-occupant-pre-join";
-               event.is_new_room = is_new_room;
                event.occupant = dest_occupant;
        elseif dest_occupant == nil then
                event_name = "muc-occupant-pre-leave";
@@ -482,10 +536,10 @@ function room_mt:handle_normal_presence(origin, stanza)
                        if not is_first_dest_session then -- User is swapping into another pre-existing session
                                log("debug", "session %s is swapping into multisession %s, showing it leave.", real_jid, dest_occupant.nick);
                                -- Show the other session leaving
-                               local x = st.stanza("x", {xmlns = "http://jabber.org/protocol/muc#user";})
-                                       :tag("status"):text("you are joining pre-existing session " .. dest_nick):up();
+                               local x = st.stanza("x", {xmlns = "http://jabber.org/protocol/muc#user";});
                                add_item(x, self:get_affiliation(bare_jid), "none");
                                local pr = st.presence{from = dest_occupant.nick, to = real_jid, type = "unavailable"}
+                                       :tag("status"):text("you are joining pre-existing session " .. dest_nick):up()
                                        :add_child(x);
                                self:route_stanza(pr);
                        end
@@ -523,9 +577,6 @@ function room_mt:handle_normal_presence(origin, stanza)
        if dest_occupant ~= nil then
                dest_occupant:set_session(real_jid, stanza);
                local dest_x = st.stanza("x", {xmlns = "http://jabber.org/protocol/muc#user";});
-               if is_new_room then
-                       dest_x:tag("status", {code = "201"}):up();
-               end
                if orig_occupant == nil and self:get_whois() == "anyone" then
                        dest_x:tag("status", {code = "100"}):up();
                end
@@ -723,7 +774,7 @@ function room_mt:process_form(origin, stanza)
                end
                event.field, event.value = nil, nil;
 
-               self:save();
+               self:save(true);
                origin.send(st.reply(stanza));
 
                if next(event.status_codes) then
@@ -806,7 +857,7 @@ function room_mt:handle_admin_query_set_command(origin, stanza)
        else
                success, errtype, err = nil, "cancel", "bad-request";
        end
-       self:save();
+       self:save(true);
        if not success then
                origin.send(st.error_reply(stanza, errtype, err));
        else
@@ -1173,7 +1224,7 @@ function room_mt:set_affiliation(actor, jid, affiliation, reason)
                end
        end
 
-       self:save();
+       self:save(true);
 
        module:fire_event("muc-set-affiliation", {
                room = self;
@@ -1258,7 +1309,7 @@ function _M.new_room(jid, config)
 end
 
 function room_mt:freeze(live)
-       local frozen = {
+       local frozen, state = {
                _jid = self.jid;
                _data = self._data;
        };
@@ -1266,21 +1317,27 @@ function room_mt:freeze(live)
                frozen[user] = affiliation;
        end
        if live then
+               state = {};
                for nick, occupant in self:each_occupant() do
-                       frozen[nick] = {
+                       state[nick] = {
                                bare_jid = occupant.bare_jid;
                                role = occupant.role;
                                jid = occupant.jid;
                        }
                        for jid, presence in occupant:each_session() do
-                               frozen[jid] = st.preserialize(presence);
+                               state[jid] = st.preserialize(presence);
                        end
                end
+               local history = self._history;
+               if history then
+                       state._last_message = st.preserialize(history[#history].stanza);
+                       state._last_message_at = history[#history].timestamp;
+               end
        end
-       return frozen;
+       return frozen, state;
 end
 
-function _M.restore_room(frozen)
+function _M.restore_room(frozen, state)
        -- COMPAT
        if frozen.jid and frozen._affiliations then
                local room = _M.new_room(frozen.jid, frozen._data);
@@ -1291,16 +1348,27 @@ function _M.restore_room(frozen)
        local room_jid = frozen._jid;
        local room = _M.new_room(room_jid, frozen._data);
 
+       if state and state._last_message and state._last_message_at then
+               room._history = {
+                       { stanza = st.deserialize(state._last_message),
+                         timestamp = state._last_message_at, },
+               };
+       end
+
        local occupants = {};
        local occupant_sessions = {};
        local room_name, room_host = jid_split(room_jid);
        for jid, data in pairs(frozen) do
+               local node, host, resource = jid_split(jid);
+               if host:sub(1,1) ~= "_" and not resource and type(data) == "string" then
+                       -- bare jid: affiliation
+                       room._affiliations[jid] = data;
+               end
+       end
+       for jid, data in pairs(state or frozen) do
                local node, host, resource = jid_split(jid);
                if node or host:sub(1,1) ~= "_" then
-                       if not resource and type(data) == "string" then
-                               -- bare jid: affiliation
-                               room._affiliations[jid] = data;
-                       elseif host == room_host and node == room_name and resource then
+                       if host == room_host and node == room_name and resource and type(data) == "table" then
                                -- full room jid: bare real jid and role
                                local bare_jid = data.bare_jid;
                                local   occupant = occupant_lib.new(bare_jid, jid);
@@ -1314,7 +1382,7 @@ function _M.restore_room(frozen)
                                        end
                                end
                                occupant_sessions[bare_jid] = nil;
-                       else
+                       elseif type(data) == "table" and data.name then
                                -- full user jid: presence
                                local presence = st.deserialize(data);
                                local bare_jid = jid_bare(jid);