1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

use super::*;
use std::{cell::RefCell, rc::Rc};

pub struct ComponentHandle {
    state: Rc<RefCell<ComponentState>>,
}

impl ComponentHandle {
    pub(crate) fn new(state: Rc<RefCell<ComponentState>>) -> Self {
        Self { state }
    }

    /// Pauses component until [`Self::resume()`] is called.
    pub fn pause(&self) {
        *self.state.borrow_mut() = ComponentState::Paused;
    }

    /// Resumes component paused through [`Self::pause()`].
    pub fn resume(&self) {
        *self.state.borrow_mut() = ComponentState::Working;
    }

    /// Removes component, preventing it from running again.
    pub fn remove(self) {
        *self.state.borrow_mut() = ComponentState::Removed;
    }

    /// Returns component's state.
    pub fn state(&self) -> ComponentState {
        *self.state.borrow()
    }
}