Skip to main content

oak_bash/parser/
element_type.rs

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