Skip to main content

oak_cmd/parser/
element_type.rs

1use crate::lexer::token_type::CmdTokenType;
2use oak_core::UniversalElementRole;
3
4/// Represents all possible element kinds in the Windows Command (CMD) scripting language.
5#[derive(Debug, Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
6#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
7pub enum CmdElementType {
8    /// A wrapper for tokens
9    Token(CmdTokenType),
10    /// Root node representing the entire source file
11    Root,
12    /// A single command or a pipeline
13    CommandStatement,
14    /// An if statement
15    IfStatement,
16    /// A for loop
17    ForStatement,
18    /// A set statement
19    SetStatement,
20    /// A label definition
21    LabelDefinition,
22    /// Error node for syntax errors
23    Error,
24}
25
26impl From<CmdTokenType> for CmdElementType {
27    fn from(token: CmdTokenType) -> Self {
28        Self::Token(token)
29    }
30}
31
32impl oak_core::ElementType for CmdElementType {
33    type Role = UniversalElementRole;
34
35    fn is_root(&self) -> bool {
36        matches!(self, Self::Root)
37    }
38
39    fn is_error(&self) -> bool {
40        matches!(self, Self::Error)
41    }
42
43    fn role(&self) -> Self::Role {
44        match self {
45            Self::Root => UniversalElementRole::Root,
46            Self::LabelDefinition => UniversalElementRole::Definition,
47            Self::CommandStatement | Self::IfStatement | Self::ForStatement | Self::SetStatement => UniversalElementRole::Statement,
48            Self::Error => UniversalElementRole::Error,
49            _ => UniversalElementRole::None,
50        }
51    }
52}