oak_bash/parser/
element_type.rs

1use crate::lexer::BashTokenType;
2use oak_core::{ElementType, UniversalElementRole};
3use serde::Serialize;
4
5/// Represents all possible element kinds in the Bash shell scripting language.
6#[derive(Debug, Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize)]
7pub enum BashElementType {
8    /// A wrapper for tokens
9    Token(BashTokenType),
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 while loop
19    WhileStatement,
20    /// A function definition
21    FunctionDefinition,
22    /// Error node for syntax errors
23    Error,
24}
25
26impl From<BashTokenType> for BashElementType {
27    fn from(token: BashTokenType) -> Self {
28        Self::Token(token)
29    }
30}
31
32impl ElementType for BashElementType {
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::FunctionDefinition => UniversalElementRole::Definition,
47            Self::CommandStatement | Self::IfStatement | Self::ForStatement | Self::WhileStatement => UniversalElementRole::Statement,
48            Self::Error => UniversalElementRole::Error,
49            _ => UniversalElementRole::None,
50        }
51    }
52}