mod auth; mod app; mod components; mod page; mod config; mod actors; mod activities; mod objects; mod timeline; mod getters; pub use app::App; pub use config::Config; pub use auth::Auth; pub mod prelude; pub const URL_BASE: &str = "https://dev.upub.social"; pub const URL_PREFIX: &str = "/web"; pub const CONTACT: &str = "abuse@alemi.dev"; pub const URL_SENSITIVE: &str = "https://cdn.alemi.dev/social/nsfw.png"; pub const FALLBACK_IMAGE_URL: &str = "https://cdn.alemi.dev/social/gradient.png"; pub const NAME: &str = "μ"; pub const DEFAULT_COLOR: &str = "#BF616A"; pub const VERSION: &str = env!("CARGO_PKG_VERSION"); use std::{ops::Deref, sync::Arc}; use uriproxy::UriClass; pub type Object = Arc; pub mod cache { use super::DashmapCache; lazy_static::lazy_static! { pub static ref OBJECTS: DashmapCache = DashmapCache::default(); pub static ref WEBFINGER: DashmapCache = DashmapCache::default(); } } #[derive(Debug)] pub enum LookupStatus { Resolving, Found(T), NotFound, } impl LookupStatus { fn inner(&self) -> Option<&T> { if let Self::Found(x) = self { return Some(x); } None } } pub trait Cache { type Item; fn lookup(&self, key: &str) -> Option>>; fn store(&self, key: &str, value: Self::Item) -> Option; fn get(&self, key: &str) -> Option where Self::Item : Clone { Some(self.lookup(key)?.deref().inner()?.clone()) } fn get_or(&self, key: &str, or: Self::Item) -> Self::Item where Self::Item : Clone { self.get(key).unwrap_or(or) } fn get_or_default(&self, key: &str) -> Self::Item where Self::Item : Clone + Default { self.get(key).unwrap_or_default() } } #[derive(Default, Clone)] pub struct DashmapCache(Arc>>); impl Cache for DashmapCache { type Item = T; fn lookup(&self, key: &str) -> Option>> { self.0.get(key) } fn store(&self, key: &str, value: Self::Item) -> Option { self.0.insert(key.to_string(), LookupStatus::Found(value)) .and_then(|x| if let LookupStatus::Found(x) = x { Some(x) } else { None } ) } } // TODO would be cool unifying a bit the fetch code too impl DashmapCache { pub async fn fetch(&self, k: &str, kind: UriClass) -> reqwest::Result { match self.get(k) { Some(x) => Ok(x), None => { let obj = reqwest::get(Uri::api(kind, k, true)) .await? .json::() .await?; self.store(k, Arc::new(obj)); Ok(self.get(k).expect("not found in cache after insertion")) } } } } impl DashmapCache { pub async fn blocking_resolve(&self, user: &str, domain: &str) -> Option { if let Some(x) = self.resource(user, domain) { return Some(x); } self.fetch(user, domain).await; self.resource(user, domain) } pub fn resolve(&self, user: &str, domain: &str) -> Option { if let Some(x) = self.resource(user, domain) { return Some(x); } let (_self, user, domain) = (self.clone(), user.to_string(), domain.to_string()); leptos::spawn_local(async move { _self.fetch(&user, &domain).await }); None } fn resource(&self, user: &str, domain: &str) -> Option { let query = format!("{user}@{domain}"); self.get(&query) } async fn fetch(&self, user: &str, domain: &str) { let query = format!("{user}@{domain}"); self.0.insert(query.to_string(), LookupStatus::Resolving); match reqwest::get(format!("{URL_BASE}/.well-known/webfinger?resource=acct:{query}")).await { Ok(res) => match res.error_for_status() { Ok(res) => match res.json::().await { Ok(doc) => { if let Some(uid) = doc.links.into_iter().find(|x| x.rel == "self").and_then(|x| x.href) { self.0.insert(query, LookupStatus::Found(uid)); } else { self.0.insert(query, LookupStatus::NotFound); } }, Err(e) => { tracing::error!("invalid webfinger response: {e:?}"); self.0.remove(&query); }, }, Err(e) => { tracing::error!("could not resolve webfinbger: {e:?}"); self.0.insert(query, LookupStatus::NotFound); }, }, Err(e) => { tracing::error!("failed accessing webfinger server: {e:?}"); self.0.remove(&query); }, } } } use leptos_router::Params; // TODO can i remove this? #[derive(Clone, leptos::Params, PartialEq)] pub struct IdParam { id: Option, } pub struct Http; impl Http { pub async fn request( method: reqwest::Method, url: &str, data: Option<&T>, auth: Auth, ) -> reqwest::Result { use leptos::SignalGetUntracked; let mut req = reqwest::Client::new() .request(method, url); if let Some(auth) = auth.token.get_untracked().filter(|x| !x.is_empty()) { req = req.header("Authorization", format!("Bearer {}", auth)); } if let Some(data) = data { req = req.json(data); } req.send().await } pub async fn fetch(url: &str, token: Auth) -> reqwest::Result { Self::request::<()>(reqwest::Method::GET, url, None, token) .await? .error_for_status()? .json::() .await } pub async fn post(url: &str, data: &T, token: Auth) -> reqwest::Result<()> { Self::request(reqwest::Method::POST, url, Some(data), token) .await? .error_for_status()?; Ok(()) } } pub struct Uri; impl Uri { pub fn full(kind: UriClass, id: &str) -> String { uriproxy::uri(URL_BASE, kind, id) } pub fn pretty(url: &str, len: usize) -> String { let bare = url.replace("https://", ""); if bare.len() < len { bare } else { format!("{}..", bare.get(..len).unwrap_or_default()) } //.replace('/', "\u{200B}/\u{200B}") } pub fn short(url: &str) -> String { if url.starts_with(URL_BASE) || url.starts_with('/') { uriproxy::decompose(url) } else if url.starts_with("https://") || url.starts_with("http://") { uriproxy::compact(url) } else { url.to_string() } } /// convert url id to valid frontend view id: /// /// accepts: /// pub fn web(kind: UriClass, url: &str) -> String { let kind = kind.as_ref(); format!("/web/{kind}/{}", Self::short(url)) } /// convert url id to valid backend api id /// /// accepts: /// pub fn api(kind: UriClass, url: &str, fetch: bool) -> String { let kind = kind.as_ref(); format!("{URL_BASE}/{kind}/{}{}", Self::short(url), if fetch { "?fetch=true" } else { "" }) } }