clap_handler/
context.rs

1use std::any::Any;
2
3#[derive(Default)]
4pub struct Context {
5    #[cfg(not(feature = "async"))]
6    inner: Vec<Option<Box<dyn Any>>>,
7    #[cfg(feature = "async")]
8    inner: Vec<Option<Box<dyn Any + Send>>>,
9}
10
11impl Context {
12    #[cfg(not(feature = "async"))]
13    pub fn insert<T>(&mut self, param: T)
14        where T: 'static {
15        self.inner.push(Some(Box::new(param)));
16    }
17
18    #[cfg(feature = "async")]
19    pub fn insert<T>(&mut self, param: T)
20        where T: 'static + Send {
21        self.inner.push(Some(Box::new(param)));
22    }
23
24    pub fn get<T>(&self) -> Option<&T>
25        where T: 'static {
26        // iterate from end to start
27        for item in self.inner.iter().rev() {
28            if let Some(item) = item {
29                if let Some(data) = item.downcast_ref::<T>() {
30                    return Some(data);
31                }
32            }
33        }
34        None
35    }
36
37    pub fn get_mut<T>(&mut self) -> Option<&mut T>
38        where T: 'static {
39        // iterate from end to start
40        for item in self.inner.iter_mut().rev() {
41            if let Some(item) = item {
42                if let Some(data) = item.downcast_mut::<T>() {
43                    return Some(data);
44                }
45            }
46        }
47        None
48    }
49
50    pub fn take<T>(&mut self) -> Option<Box<T>>
51        where T: 'static {
52        // iterate from end to start
53        for item in self.inner.iter_mut().rev() {
54            if let Some(inner) = item {
55                if inner.is::<T>() {
56                    let data = item.take().unwrap();
57                    return Some(data.downcast().unwrap());
58                }
59            }
60        }
61        None
62    }
63}