darklua_core/nodes/statements/
do_statement.rs

1use crate::nodes::{Block, Token};
2
3#[derive(Clone, Debug, PartialEq, Eq)]
4pub struct DoTokens {
5    pub r#do: Token,
6    pub end: Token,
7}
8
9impl DoTokens {
10    super::impl_token_fns!(target = [r#do, end]);
11}
12
13#[derive(Clone, Debug, Default, PartialEq, Eq)]
14pub struct DoStatement {
15    block: Block,
16    tokens: Option<DoTokens>,
17}
18
19impl DoStatement {
20    pub fn new(block: Block) -> Self {
21        Self {
22            block,
23            tokens: None,
24        }
25    }
26
27    #[inline]
28    pub fn get_block(&self) -> &Block {
29        &self.block
30    }
31
32    #[inline]
33    pub fn mutate_block(&mut self) -> &mut Block {
34        &mut self.block
35    }
36
37    pub fn with_tokens(mut self, tokens: DoTokens) -> Self {
38        self.tokens = Some(tokens);
39        self
40    }
41
42    #[inline]
43    pub fn set_tokens(&mut self, tokens: DoTokens) {
44        self.tokens = Some(tokens);
45    }
46
47    #[inline]
48    pub fn get_tokens(&self) -> Option<&DoTokens> {
49        self.tokens.as_ref()
50    }
51
52    #[inline]
53    pub fn mutate_tokens(&mut self) -> Option<&mut DoTokens> {
54        self.tokens.as_mut()
55    }
56
57    super::impl_token_fns!(iter = [tokens]);
58}