use proc_macro2::Span;
use std::fmt;
use syn::{punctuated::Punctuated, spanned::Spanned, token::Colon, Expr, ExprPath, Ident, Lit};
use crate::punctuation::Dash;
#[derive(Debug)]
pub struct Node {
pub name: Option<NodeName>,
pub node_type: NodeType,
pub value: Option<Expr>,
pub attributes: Vec<Node>,
pub children: Vec<Node>,
}
impl Node {
pub fn name_as_string(&self) -> Option<String> {
match self.name.as_ref() {
Some(NodeName::Path(expr)) => Some(
expr.path
.segments
.iter()
.map(|segment| segment.ident.to_string())
.collect::<Vec<String>>()
.join("::"),
),
Some(NodeName::Dash(name)) => Some(
name.iter()
.map(|ident| ident.to_string())
.collect::<Vec<String>>()
.join("-"),
),
Some(NodeName::Colon(name)) => Some(
name.iter()
.map(|ident| ident.to_string())
.collect::<Vec<String>>()
.join(":"),
),
None => None,
}
}
pub fn name_span(&self) -> Option<Span> {
match self.name.as_ref() {
Some(NodeName::Path(name)) => Some(name.span()),
Some(NodeName::Dash(name)) => Some(name.span()),
Some(NodeName::Colon(name)) => Some(name.span()),
None => None,
}
}
pub fn value_as_string(&self) -> Option<String> {
match self.value.as_ref() {
Some(Expr::Lit(expr)) => match &expr.lit {
Lit::Str(lit_str) => Some(lit_str.value()),
_ => None,
},
_ => None,
}
}
}
#[derive(Debug, Clone, PartialEq)]
pub enum NodeType {
Element,
Attribute,
Text,
Block,
}
impl fmt::Display for NodeType {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(
f,
"{}",
match self {
Self::Element => "NodeType::Element",
Self::Attribute => "NodeType::Attribute",
Self::Text => "NodeType::Text",
Self::Block => "NodeType::Block",
}
)
}
}
#[derive(Debug)]
pub enum NodeName {
Path(ExprPath),
Dash(Punctuated<Ident, Dash>),
Colon(Punctuated<Ident, Colon>),
}
impl PartialEq for NodeName {
fn eq(&self, other: &NodeName) -> bool {
match self {
Self::Path(this) => match other {
Self::Path(other) => this == other,
_ => false,
},
Self::Dash(this) => match other {
Self::Dash(other) => this == other,
_ => false,
},
Self::Colon(this) => match other {
Self::Colon(other) => this == other,
_ => false,
},
}
}
}