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 dev_log!("extensions", "[TreeViewState] Tree view added/updated");
71 }
72 }
73
74 pub fn Remove(&self, id:&str) {
76 if let Ok(mut guard) = self.ActiveTreeViews.lock() {
77 guard.remove(id);
78 dev_log!("extensions", "[TreeViewState] Tree view removed: {}", id);
79 }
80 }
81
82 pub fn Clear(&self) {
84 if let Ok(mut guard) = self.ActiveTreeViews.lock() {
85 guard.clear();
86 dev_log!("extensions", "[TreeViewState] All tree views cleared");
87 }
88 }
89
90 pub fn Count(&self) -> usize { self.ActiveTreeViews.lock().ok().map(|guard| guard.len()).unwrap_or(0) }
92
93 pub fn Contains(&self, id:&str) -> bool {
95 self.ActiveTreeViews
96 .lock()
97 .ok()
98 .map(|guard| guard.contains_key(id))
99 .unwrap_or(false)
100 }
101}