use std::{path::Path, str::FromStr, sync::Arc, time::Duration}; use filamento::{Client, db::Db, files::FileStore}; use jid::JID; use tokio::io::{self, AsyncReadExt}; use tracing::info; #[derive(Clone, Debug)] pub struct Files; impl FileStore for Files { type Err = Arc; async fn is_stored(&self, name: &str) -> Result { tracing::debug!("checking if {} is stored", name); let res = tokio::fs::try_exists(format!("files/{}", name)) .await .map_err(|err| Arc::new(err)); tracing::debug!("file check res: {:?}", res); res } async fn store(&self, name: &str, data: &[u8]) -> Result<(), Self::Err> { tracing::debug!("storing {} is stored", name); let res = tokio::fs::write(format!("files/{}", name), data) .await .map_err(|err| Arc::new(err)); tracing::debug!("file store res: {:?}", res); res } async fn delete(&self, name: &str) -> Result<(), Self::Err> { tracing::debug!("deleting {}", name); let res = tokio::fs::remove_file(format!("files/{}", name)) .await .map_err(|err| Arc::new(err)); tracing::debug!("file delete res: {:?}", res); res } } #[tokio::main] async fn main() { tracing_subscriber::fmt::init(); let db = Db::create_connect_and_migrate(Path::new("./filamento.db")) .await .unwrap(); let (client, mut recv) = Client::new( "test@blos.sm/testing2".try_into().unwrap(), "slayed".to_string(), db, Files, ); tokio::spawn(async move { while let Some(msg) = recv.recv().await { info!("{:#?}", msg) } }); client.connect().await.unwrap(); tokio::time::sleep(Duration::from_secs(5)).await; info!("changing nick"); client .change_nick(Some("britney".to_string())) .await .unwrap(); let mut profile_pic = tokio::fs::File::open("files/britney_starbies.jpg") .await .unwrap(); let mut data = Vec::new(); profile_pic.read_to_end(&mut data).await.unwrap(); client.change_avatar(Some(data)).await.unwrap(); info!("sending message"); client .send_message( JID::from_str("cel@blos.sm").unwrap(), filamento::chat::Body { body: "hallo!!!".to_string(), }, ) .await .unwrap(); info!("sent message"); client .send_message( JID::from_str("cel@blos.sm").unwrap(), filamento::chat::Body { body: "hallo 2".to_string(), }, ) .await .unwrap(); tokio::time::sleep(Duration::from_secs(15)).await; // info!("sending disco query"); // let info = client.disco_info(None, None).await.unwrap(); // info!("got disco result: {:#?}", info); // let items = client.disco_items(None, None).await.unwrap(); // info!("got disco result: {:#?}", items); // let info = client // .disco_info(Some("blos.sm".parse().unwrap()), None) // .await // .unwrap(); // info!("got disco result: {:#?}", info); // let items = client // .disco_items(Some("blos.sm".parse().unwrap()), None) // .await // .unwrap(); // info!("got disco result: {:#?}", items); // let info = client // .disco_info(Some("pubsub.blos.sm".parse().unwrap()), None) // .await // .unwrap(); // info!("got disco result: {:#?}", info); // let items = client // .disco_items(Some("pubsub.blos.sm".parse().unwrap()), None) // .await // .unwrap(); // info!("got disco result: {:#?}", items); }