state.rs

src-tauri/src/state.rs

1. use std::collections::HashMap;
2. use std::path::PathBuf;
3. use std::sync::{Mutex, MutexGuard, PoisonError};
4. use std::time::Instant;
6. use cortex_core::secrets::KeyCache;
7. use cortex_core::storage::Db;
8. use tokio_util::sync::CancellationToken;
10. pub struct AppState {
11. pub db: Mutex<Db>,
12. pub keys: KeyCache,
13. pub streams: Mutex<HashMap<String, CancellationToken>>,
14. pub data_dir: PathBuf,
15. /// When the quick window was last shown; blur right after showing is ignored.
16. pub quick_shown_at: Mutex<Option<Instant>>,
17. /// While a native dialog is open, losing focus must not hide the quick window.
18. pub dialog_open: Mutex<bool>,
19. pub tray_spend: Mutex<Option<tauri::menu::MenuItem<tauri::Wry>>>,
20. }
22. /// Lock even if a panic elsewhere poisoned the mutex, so one failed command
23. /// does not fail every later one until the app restarts.
24. pub fn lock<T>(m: &Mutex<T>) -> MutexGuard<'_, T> {
25. m.lock().unwrap_or_else(PoisonError::into_inner)
26. }
28. impl AppState {
29. pub fn db(&self) -> MutexGuard<'_, Db> {
30. lock(&self.db)
31. }
32. pub fn attachments_dir(&self) -> PathBuf {
33. self.data_dir.join("attachments")
34. }
35. }