use crate::lexer::token_type::CobolTokenType;
use oak_core::{ElementType, UniversalElementRole};
#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub enum CobolElementType {
Eof,
Whitespace,
Comment,
Identifier,
String,
Number,
Identification,
Division,
Root,
DivisionElement,
Paragraph,
}
impl ElementType for CobolElementType {
type Role = UniversalElementRole;
fn role(&self) -> Self::Role {
match self {
Self::Root => UniversalElementRole::Root,
_ => UniversalElementRole::None,
}
}
}
impl From<CobolTokenType> for CobolElementType {
fn from(token: CobolTokenType) -> Self {
match token {
CobolTokenType::Eof => Self::Eof,
CobolTokenType::Whitespace => Self::Whitespace,
CobolTokenType::Comment => Self::Comment,
CobolTokenType::Identifier => Self::Identifier,
CobolTokenType::String => Self::String,
CobolTokenType::Number => Self::Number,
CobolTokenType::Identification => Self::Identification,
CobolTokenType::Division => Self::Division,
}
}
}