Skip to main content

oak_c/parser/
element_type.rs

1use crate::lexer::CTokenType;
2use oak_core::{ElementType, UniversalElementRole};
3use serde::{Deserialize, Serialize};
4
5/// Represents all possible element kinds in the C programming language.
6#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
7pub enum CElementType {
8    /// A wrapper for tokens
9    Token(CTokenType),
10    /// Root node representing the entire source file
11    Root,
12    /// Function definition
13    FunctionDefinition,
14    /// Parameter list
15    ParameterList,
16    /// Compound statement (block)
17    CompoundStatement,
18    /// Expression statement
19    ExpressionStatement,
20    /// Declaration statement
21    DeclarationStatement,
22    /// If statement
23    IfStatement,
24    /// While statement
25    WhileStatement,
26    /// For statement
27    ForStatement,
28    /// Return statement
29    ReturnStatement,
30    /// Error element
31    Error,
32}
33
34impl From<CTokenType> for CElementType {
35    fn from(token: CTokenType) -> Self {
36        Self::Token(token)
37    }
38}
39
40impl ElementType for CElementType {
41    type Role = UniversalElementRole;
42
43    fn is_root(&self) -> bool {
44        matches!(self, Self::Root)
45    }
46
47    fn is_error(&self) -> bool {
48        matches!(self, Self::Error)
49    }
50
51    fn role(&self) -> Self::Role {
52        match self {
53            Self::Root => UniversalElementRole::Root,
54            Self::FunctionDefinition => UniversalElementRole::Definition,
55            Self::CompoundStatement | Self::ExpressionStatement | Self::DeclarationStatement | Self::IfStatement | Self::WhileStatement | Self::ForStatement | Self::ReturnStatement => UniversalElementRole::Statement,
56            Self::Error => UniversalElementRole::Error,
57            _ => UniversalElementRole::None,
58        }
59    }
60}