reign_view 0.2.1

use super::consts::DOCTYPE;
use super::{
    tag_name_regex, Comment, Doctype, Element, Error, Parse, ParseStream, Text, Tokenize,
    ViewFields,
};
use proc_macro2::TokenStream;

#[derive(Debug)]
pub enum Node {
    Element(Element),
    Comment(Comment),
    Text(Text),
    Doctype(Doctype),
}

impl Parse for Node {
    fn parse(input: &mut ParseStream) -> Result<Self, Error> {
        if input.cursor == 0 {
            input.skip_spaces()?;
        }

        if input.peek("<!--") {
            Ok(Node::Comment(input.parse()?))
        } else if input.is_match(DOCTYPE) {
            Ok(Node::Doctype(input.parse()?))
        } else if input.is_match(&tag_name_regex()) {
            Ok(Node::Element(input.parse()?))
        } else {
            let text: Text = input.parse()?;

            if text.content.is_empty() {
                return Err(input.error("unable to continue parsing"));
            }

            Ok(Node::Text(text))
        }
    }
}

impl Tokenize for Node {
    fn tokenize(&self, tokens: &mut TokenStream, idents: &mut ViewFields, scopes: &ViewFields) {
        match self {
            Node::Element(e) => e.tokenize(tokens, idents, scopes),
            Node::Comment(c) => c.tokenize(tokens, idents, scopes),
            Node::Text(t) => t.tokenize(tokens, idents, scopes),
            Node::Doctype(d) => d.tokenize(tokens, idents, scopes),
        };
    }
}