xmpp-rs/xmpp/src/message/receive/group_chat.rs

// Copyright (c) 2023 xmpp-rs contributors.
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

use tokio_xmpp::connect::ServerConnector;
use tokio_xmpp::{parsers::message::Message, Jid};

use crate::message::MucMessageId;
use crate::parsers::ns::SID;
use crate::parsers::stanza_id::StanzaId;
use crate::{delay::StanzaTimeInfo, Agent, Event};

pub async fn handle_message_group_chat<C: ServerConnector>(
    agent: &mut Agent<C>,
    events: &mut Vec<Event>,
    from: Jid,
    message: &Message,
    time_info: StanzaTimeInfo,
) {
    let langs: Vec<&str> = agent.lang.iter().map(String::as_str).collect();

    let id = extract_muc_message_id(message);

    if let Some((_lang, subject)) = message.get_best_subject(langs.clone()) {
        events.push(Event::RoomSubject(
            id.clone(),
            from.to_bare(),
            from.resource().map(|x| x.to_string()),
            subject.0.clone(),
            time_info.clone(),
        ));
    }

    if let Some((_lang, body)) = message.get_best_body(langs) {
        let event = match from.clone() {
            Jid::Full(full) => Event::RoomMessage(
                id,
                from.to_bare(),
                full.resource().to_string(),
                body.clone(),
                time_info,
            ),
            Jid::Bare(bare) => {
                Event::ServiceMessage(message.id.clone(), bare, body.clone(), time_info)
            }
        };
        events.push(event)
    }
}

/// Extract the sender-assigned and room-assigned IDs from a message.
pub fn extract_muc_message_id(message: &Message) -> MucMessageId {
    // Find a stanza-id element, if any.
    let stanza_id = message
        .payloads
        .iter()
        .find(|payload| payload.is("stanza-id", SID))
        .map(|elem| {
            StanzaId::try_from(elem.clone())
                .expect("Failed to parse stanza-id")
                .id
        });

    let id = MucMessageId {
        sender_assigned_id: message.id.clone(),
        room_assigned_id: stanza_id,
    };
    id
}