Mountain/ApplicationState/State/FeatureState/Webviews/
WebviewState.rs1use std::{
34 collections::HashMap,
35 sync::{Arc, Mutex as StandardMutex},
36};
37
38use crate::{ApplicationState::DTO::WebviewStateDTO::WebviewStateDTO, dev_log};
39
40#[derive(Clone)]
42pub struct WebviewState {
43 pub ActiveWebviews:Arc<StandardMutex<HashMap<String, WebviewStateDTO>>>,
45}
46
47impl Default for WebviewState {
48 fn default() -> Self {
49 dev_log!("extensions", "[WebviewState] Initializing default webview state...");
50
51 Self { ActiveWebviews:Arc::new(StandardMutex::new(HashMap::new())) }
52 }
53}
54
55impl WebviewState {
56 pub fn GetAll(&self) -> HashMap<String, WebviewStateDTO> {
58 self.ActiveWebviews.lock().ok().map(|guard| guard.clone()).unwrap_or_default()
59 }
60
61 pub fn Get(&self, id:&str) -> Option<WebviewStateDTO> {
63 self.ActiveWebviews.lock().ok().and_then(|guard| guard.get(id).cloned())
64 }
65
66 pub fn AddOrUpdate(&self, id:String, webview:WebviewStateDTO) {
68 if let Ok(mut guard) = self.ActiveWebviews.lock() {
69 guard.insert(id, webview);
70
71 dev_log!("extensions", "[WebviewState] Webview added/updated");
72 }
73 }
74
75 pub fn Remove(&self, id:&str) {
77 if let Ok(mut guard) = self.ActiveWebviews.lock() {
78 guard.remove(id);
79
80 dev_log!("extensions", "[WebviewState] Webview removed: {}", id);
81 }
82 }
83
84 pub fn Clear(&self) {
86 if let Ok(mut guard) = self.ActiveWebviews.lock() {
87 guard.clear();
88
89 dev_log!("extensions", "[WebviewState] All webviews cleared");
90 }
91 }
92
93 pub fn Count(&self) -> usize { self.ActiveWebviews.lock().ok().map(|guard| guard.len()).unwrap_or(0) }
95
96 pub fn Contains(&self, id:&str) -> bool {
98 self.ActiveWebviews
99 .lock()
100 .ok()
101 .map(|guard| guard.contains_key(id))
102 .unwrap_or(false)
103 }
104}