xmpp-rs/xmpp/src/agent.rs

82 lines
2.9 KiB
Rust

// 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 std::collections::HashMap;
use std::path::{Path, PathBuf};
use std::sync::Arc;
use tokio::sync::RwLock;
pub use tokio_xmpp::parsers;
use tokio_xmpp::parsers::disco::DiscoInfoResult;
pub use tokio_xmpp::{
jid::{BareJid, FullJid, Jid, ResourcePart},
minidom::Element,
Client as TokioXmppClient,
};
use crate::{event_loop, message, muc, upload, Error, Event};
pub struct Agent {
pub(crate) client: TokioXmppClient,
pub(crate) default_nick: Arc<RwLock<ResourcePart>>,
pub(crate) lang: Arc<Vec<String>>,
pub(crate) disco: DiscoInfoResult,
pub(crate) node: String,
pub(crate) uploads: Vec<(String, Jid, PathBuf)>,
pub(crate) awaiting_disco_bookmarks_type: bool,
// Mapping of room->nick
pub(crate) rooms_joined: HashMap<BareJid, ResourcePart>,
pub(crate) rooms_joining: HashMap<BareJid, ResourcePart>,
pub(crate) rooms_leaving: HashMap<BareJid, ResourcePart>,
}
impl Agent {
pub async fn disconnect(self) -> Result<(), Error> {
self.client.send_end().await
}
pub async fn join_room<'a>(&mut self, settings: muc::room::JoinRoomSettings<'a>) {
muc::room::join_room(self, settings).await
}
/// Request to leave a chatroom.
///
/// If successful, an [Event::RoomLeft] event will be produced. This method does not remove the room
/// from bookmarks nor remove the autojoin flag. See [muc::room::leave_room] for more information.
pub async fn leave_room<'a>(&mut self, settings: muc::room::LeaveRoomSettings<'a>) {
muc::room::leave_room(self, settings).await
}
pub async fn send_message<'a>(&mut self, settings: message::send::MessageSettings<'a>) {
message::send::send_message(self, settings).await
}
pub async fn send_room_message<'a>(&mut self, settings: muc::room::RoomMessageSettings<'a>) {
muc::room::send_room_message(self, settings).await
}
pub async fn send_room_private_message<'a>(
&mut self,
settings: muc::private_message::RoomPrivateMessageSettings<'a>,
) {
muc::private_message::send_room_private_message(self, settings).await
}
/// Wait for new events, or Error::Disconnected when connection is closed and will not reconnect.
pub async fn wait_for_events(&mut self) -> Vec<Event> {
event_loop::wait_for_events(self).await
}
pub async fn upload_file_with(&mut self, service: &str, path: &Path) {
upload::send::upload_file_with(self, service, path).await
}
/// Get the bound jid of the client.
///
/// If the client is not connected, this will be None.
pub fn bound_jid(&self) -> Option<&Jid> {
self.client.bound_jid()
}
}