37 lines
1023 B
Rust
37 lines
1023 B
Rust
pub mod encrypted;
|
|
pub mod unencrypted;
|
|
|
|
// use async_trait::async_trait;
|
|
|
|
use crate::stanza::stream::StreamFeature;
|
|
use crate::JabberError;
|
|
use crate::Result;
|
|
|
|
pub enum JabberClientType<'j> {
|
|
Encrypted(encrypted::JabberClient<'j>),
|
|
Unencrypted(unencrypted::JabberClient<'j>),
|
|
}
|
|
|
|
impl<'j> JabberClientType<'j> {
|
|
pub async fn ensure_tls(self) -> Result<encrypted::JabberClient<'j>> {
|
|
match self {
|
|
Self::Encrypted(c) => Ok(c),
|
|
Self::Unencrypted(mut c) => {
|
|
let features = c.get_features().await?;
|
|
if features.contains(&StreamFeature::StartTls) {
|
|
Ok(c.starttls().await?)
|
|
} else {
|
|
Err(JabberError::StartTlsUnavailable)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// TODO: jabber client trait over both client types
|
|
// #[async_trait]
|
|
// pub trait JabberTrait {
|
|
// async fn start_stream(&mut self) -> Result<()>;
|
|
// async fn get_features(&self) -> Result<Vec<StreamFeatures>>;
|
|
// }
|