use alloc::sync::Arc;
use std::collections::HashMap;
use std::path::{Path, PathBuf};
#[cfg(feature = "escape-hatch")]
use tokio::io;
use tokio::sync::RwLock;
use crate::{
Config, Error, Event, RoomNick, event_loop,
jid::{BareJid, FullJid, Jid},
message, muc,
parsers::disco::DiscoInfoResult,
upload,
};
use tokio_xmpp::Client as TokioXmppClient;
#[cfg(feature = "escape-hatch")]
use tokio_xmpp::{Stanza, stanzastream::StanzaToken};
pub struct Agent {
pub(crate) client: TokioXmppClient,
config: Arc<RwLock<Config>>,
pub(crate) disco: DiscoInfoResult,
pub(crate) uploads: Vec<(String, Jid, PathBuf)>,
pub(crate) awaiting_disco_bookmarks_type: bool,
pub(crate) rooms_joined: HashMap<BareJid, RoomNick>,
pub(crate) rooms_joining: HashMap<BareJid, RoomNick>,
pub(crate) rooms_leaving: HashMap<BareJid, RoomNick>,
}
impl Agent {
pub fn new(client: TokioXmppClient, config: Config, disco: DiscoInfoResult) -> Agent {
Agent {
client,
config: Arc::new(RwLock::new(config)),
disco,
uploads: Vec::new(),
awaiting_disco_bookmarks_type: false,
rooms_joined: HashMap::new(),
rooms_joining: HashMap::new(),
rooms_leaving: HashMap::new(),
}
}
pub async fn set_config(&mut self, config: Config) {
let mut c = self.config.write().await;
*c = config;
}
pub async fn get_config(&self) -> Config {
self.config.read().await.clone()
}
pub async fn disconnect(self) -> Result<(), Error> {
self.client.send_end().await
}
#[cfg(feature = "escape-hatch")]
pub async fn send_stanza<S: Into<Stanza>>(&mut self, st: S) -> Result<StanzaToken, io::Error> {
self.client.send_stanza(st.into()).await
}
pub async fn join_room<'a>(&mut self, settings: muc::room::JoinRoomSettings<'a>) {
muc::room::join_room(self, settings).await
}
pub async fn leave_room<'a>(&mut self, settings: muc::room::LeaveRoomSettings<'a>) {
muc::room::leave_room(self, settings).await
}
pub async fn send_raw_message<'a>(&mut self, settings: message::send::RawMessageSettings<'a>) {
message::send::send_raw_message(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
}
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
}
pub fn bound_jid(&self) -> Option<&FullJid> {
self.client.bound_jid()
}
}