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