use std::fmt::Display; use std::rc::Rc; use anyhow::{Error, anyhow}; use common::config_dir; use gpui::{App, AppContext, Context, Entity, Global, Subscription, Task, Window}; use nostr_sdk::prelude::*; use serde::{Deserialize, Serialize}; use smallvec::{SmallVec, smallvec}; use theme::{Theme, ThemeFamily, ThemeMode}; pub fn init(window: &mut Window, cx: &mut App) { AppSettings::set_global(cx.new(|cx| AppSettings::new(window, cx)), cx) } macro_rules! setting_accessors { ($(pub $field:ident: $type:ty),* $(,)?) => { impl AppSettings { $( paste::paste! { pub fn [](cx: &App) -> $type { Self::global(cx).read(cx).inner.read(cx).$field.clone() } pub fn [](value: $type, cx: &mut App) { Self::global(cx).update(cx, |this, cx| { this.inner.update(cx, |inner, cx| { inner.$field = value; cx.notify(); }); }); } } )* } }; } setting_accessors! { pub theme: Option, pub theme_mode: ThemeMode, pub hide_avatar: bool, pub screening: bool, pub nip4e: bool, pub auth_mode: AuthMode, pub trusted_relays: Vec, pub file_server: Url, } /// Authentication mode #[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq, PartialOrd, Ord)] pub enum AuthMode { #[default] Auto, Manual, } impl Display for AuthMode { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { match self { AuthMode::Auto => write!(f, "Auto"), AuthMode::Manual => write!(f, "Ask every time"), } } } /// Signer kind #[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq, PartialOrd, Ord)] pub enum SignerKind { Auto, Encryption, #[default] User, } impl SignerKind { pub fn auto(&self) -> bool { matches!(self, SignerKind::Auto) } pub fn user(&self) -> bool { matches!(self, SignerKind::User) } pub fn encryption(&self) -> bool { matches!(self, SignerKind::Encryption) } } /// Room configuration #[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq, PartialOrd, Ord)] pub struct RoomConfig { backup: bool, signer_kind: SignerKind, } impl RoomConfig { pub fn new() -> Self { Self { backup: true, signer_kind: SignerKind::default(), } } /// Get backup config pub fn backup(&self) -> bool { self.backup } /// Set backup config pub fn toggle_backup(&mut self) { self.backup = !self.backup; } /// Get signer kind config pub fn signer_kind(&self) -> &SignerKind { &self.signer_kind } /// Set signer kind config pub fn set_signer_kind(&mut self, kind: &SignerKind) { self.signer_kind = kind.to_owned(); } } /// Settings #[derive(Debug, Clone, Serialize, Deserialize)] pub struct Settings { /// Theme pub theme: Option, /// Theme mode pub theme_mode: ThemeMode, /// Hide user avatars pub hide_avatar: bool, /// Enable screening for unknown chat requests pub screening: bool, /// Enable decoupling encryption key pub nip4e: bool, /// Authentication mode pub auth_mode: AuthMode, /// Trusted relays; Coop will automatically authenticate with these relays pub trusted_relays: Vec, /// Server for blossom media attachments pub file_server: Url, } impl Default for Settings { fn default() -> Self { Self { theme: None, theme_mode: ThemeMode::default(), hide_avatar: false, screening: true, nip4e: false, auth_mode: AuthMode::default(), trusted_relays: vec![], file_server: Url::parse("https://blossom.band/").unwrap(), } } } impl AsRef for Settings { fn as_ref(&self) -> &Settings { self } } struct GlobalAppSettings(Entity); impl Global for GlobalAppSettings {} /// Application settings pub struct AppSettings { /// Settings inner: Entity, /// Event subscriptions _subscriptions: SmallVec<[Subscription; 2]>, } impl AppSettings { /// Retrieve the global settings instance pub fn global(cx: &App) -> Entity { cx.global::().0.clone() } /// Set the global settings instance fn set_global(state: Entity, cx: &mut App) { cx.set_global(GlobalAppSettings(state)); } fn new(window: &mut Window, cx: &mut Context) -> Self { let inner = cx.new(|_| Settings::default()); let mut subscriptions = smallvec![]; subscriptions.push( // Observe and automatically save settings on changes cx.observe(&inner, |this, _inner, cx| { this.save(cx); }), ); // Run at the end of current cycle cx.defer_in(window, |this, window, cx| { this.load(window, cx); }); Self { inner, _subscriptions: subscriptions, } } /// Update settings fn set_settings(&mut self, settings: Settings, cx: &mut Context) { self.inner.update(cx, |this, cx| { *this = settings; cx.notify(); }); } /// Load settings fn load(&mut self, window: &mut Window, cx: &mut Context) { let task: Task> = cx.background_spawn(async move { let path = config_dir().join(".settings"); if let Ok(content) = smol::fs::read_to_string(&path).await { Ok(serde_json::from_str(&content)?) } else { Err(anyhow!("Not found")) } }); cx.spawn_in(window, async move |this, cx| { let settings = task.await.unwrap_or(Settings::default()); // Update settings this.update_in(cx, |this, window, cx| { this.set_settings(settings, cx); this.apply_theme(window, cx); }) .ok(); }) .detach(); } /// Save settings pub fn save(&mut self, cx: &mut Context) { let settings = self.inner.read(cx); if let Ok(content) = serde_json::to_string(&settings) { cx.background_spawn(async move { let path = config_dir().join(".settings"); // Write settings to file smol::fs::write(&path, content).await.ok(); }) .detach(); } } /// Set theme pub fn set_theme(&mut self, theme: T, window: &mut Window, cx: &mut Context) where T: Into, { // Update settings self.inner.update(cx, |this, cx| { this.theme = Some(theme.into()); cx.notify(); }); // Apply the new theme self.apply_theme(window, cx); } /// Reset theme pub fn reset_theme(&mut self, window: &mut Window, cx: &mut Context) { self.inner.update(cx, |this, cx| { this.theme = None; cx.notify(); }); self.apply_theme(window, cx); } /// Apply theme pub fn apply_theme(&mut self, window: &mut Window, cx: &mut Context) { if let Some(name) = self.inner.read(cx).theme.as_ref() { let mode = self.inner.read(cx).theme_mode; if let Ok(new_theme) = ThemeFamily::from_assets(name) { Theme::apply_theme(Rc::new(new_theme), Some(window), cx); Theme::change(mode, Some(window), cx); } else { log::info!("Failed to load theme: {name}"); } } else { Theme::apply_theme(Rc::new(ThemeFamily::default()), Some(window), cx); } } /// Check if decoupling encryption key is enabled pub fn is_nip4e_enabled(&self, cx: &App) -> bool { self.inner.read(cx).nip4e } /// Check if the given relay is already authenticated pub fn trusted_relay(&self, url: &RelayUrl, cx: &App) -> bool { self.inner .read(cx) .trusted_relays .iter() .any(|relay| relay == url.as_str_without_trailing_slash()) } /// Add a relay to the trusted list pub fn add_trusted_relay(&mut self, url: &RelayUrl, cx: &mut Context) { self.inner.update(cx, |this, cx| { if !this .trusted_relays .iter() .any(|relay| relay == url.as_str_without_trailing_slash()) { this.trusted_relays .push(url.as_str_without_trailing_slash().to_string()); cx.notify(); } }); } }