dtd-parser 0.1.0-alpha3

use std::{fmt::Display, ops::Deref};

use nom::{
    branch::alt,
    bytes::complete::tag,
    character::complete::{multispace0, multispace1, space0},
    combinator::{map, opt, value},
    multi::{many0, many1},
    sequence::{preceded, terminated, tuple},
};
use nom_tracable::tracable_parser;

use super::{name, MixedPCDATA, Name, Repeatable, Result, Span};

/// 元素是 XML 以及 HTML 文档的主要构建模块。
///
/// HTML 元素的例子是 "body" 和 "table"。
/// XML 元素的例子是 "note" 和 "message" 。
/// 元素可包含文本、其他元素或者是空的。空的 HTML 元素的例子是 "hr"、"br" 以及 "img"。
#[derive(Debug, Display)]
#[display(fmt = "<!ELEMENT {} {}>", name, category)]
pub struct ElementDecl {
    name: Name,
    category: ElementCategory,
}

impl ElementDecl {
    pub fn name(&self) -> &str {
        self.name.deref()
    }

    pub fn category(&self) -> &ElementCategory {
        &self.category
    }
}

#[derive(Debug, Display)]
pub enum ElementCategory {
    #[display(fmt = "EMPTY")]
    Empty,
    #[display(fmt = "#PCDATA")]
    PCDATA,
    #[display(fmt = "#CDATA")]
    CDATA,
    #[display(fmt = "ANY")]
    Any,
    #[display(fmt = "{}", "_0")]
    Mixed(Repeatable<MixedPCDATA>),
    #[display(fmt = "{}", "_0")]
    Children(Repeatable<Child>),
}

#[tracable_parser]
fn parenthesis_open(i: Span) -> Result<Span> {
    tag("(")(i)
}

#[tracable_parser]
fn parenthesis_close(i: Span) -> Result<Span> {
    tag(")")(i)
}

/// Mixed	   ::=   	'(' S? '#PCDATA' (S? '|' S? Name)* S? ')*'
/// 			| '(' S? '#PCDATA' S? ')'
#[tracable_parser]
fn mixed(i: Span) -> Result<ElementCategory> {
    alt((
        map(mixed_many_names, |x| {
            ElementCategory::Mixed(Repeatable::ZeroOrManyTimes(MixedPCDATA(x)))
        }),
        map(mixed_no_names, |_| {
            ElementCategory::Mixed(Repeatable::Once(MixedPCDATA(Vec::new())))
        }),
    ))(i)
}

#[tracable_parser]
fn mixed_no_names(i: Span) -> Result<()> {
    value(
        (),
        tuple((
            tag("("),
            multispace0,
            tag("#PCDATA"),
            multispace0,
            tag(")"),
            space0,
        )),
    )(i)
}

#[tracable_parser]
fn mixed_many_names(i: Span) -> Result<Vec<Name>> {
    map(
        tuple((
            tuple((tag("("), multispace0, tag("#PCDATA"), multispace0)),
            many0(preceded(tuple((multispace0, tag("|"), multispace0)), name)),
            tuple((multispace0, tag(")*"))),
        )),
        |(_, x, _)| x,
    )(i)
}

#[derive(AsRef, AsMut, Debug, Deref, DerefMut, Display, IntoIterator)]
#[display(
    fmt = "({})",
    "_0.iter().map(|v|v.to_string()).collect::<Vec<_>>().join(\", \")"
)]
pub struct Seq<T>(Vec<T>)
where
    T: Display;

#[derive(AsRef, AsMut, Debug, Deref, DerefMut, Display, IntoIterator)]
#[display(
    fmt = "({})",
    "_0.iter().map(|v|v.to_string()).collect::<Vec<_>>().join(\" | \")"
)]
pub struct Choices<T>(Vec<T>)
where
    T: Display;

#[derive(Debug, Display)]
pub enum Child {
    #[display(fmt = "{}", "_0")]
    Name(Name),
    #[display(fmt = "{}", "_0")]
    Seq(Seq<Repeatable<Child>>),
    #[display(fmt = "{}", "_0")]
    Choices(Choices<Repeatable<Child>>),
}

///   	children	   ::=   	(choice | seq) ('?' | '*' | '+')?
#[tracable_parser]
fn children(i: Span) -> Result<ElementCategory> {
    map(
        tuple((
            alt((map(choices, Child::Choices), map(sequence, Child::Seq))),
            opt(alt((tag("?"), tag("*"), tag("+")))),
        )),
        |(child, repeatable)| {
            let repeatable = if let Some(repeatable) = repeatable {
                match *repeatable {
                    "?" => |child| Repeatable::AtMostOnce(child),
                    "*" => |child| Repeatable::ZeroOrManyTimes(child),
                    "+" => |child| Repeatable::AtLeastOnce(child),
                    _ => unreachable!(),
                }
            } else {
                |child| Repeatable::Once(child)
            };
            ElementCategory::Children(repeatable(child))
        },
    )(i)
}

///   	seq	   ::=   	'(' S? cp ( S? ',' S? cp )* S? ')'	[VC: Proper Group/PE Nesting]
#[tracable_parser]
fn sequence(i: Span) -> Result<Seq<Repeatable<Child>>> {
    map(
        tuple((
            preceded(tuple((parenthesis_open, multispace0)), cuple),
            terminated(
                many0(preceded(tuple((multispace0, tag(","), multispace0)), cuple)),
                tuple((multispace0, parenthesis_close)),
            ),
        )),
        |(cp, many)| Seq(std::iter::once(cp).chain(many.into_iter()).collect()),
    )(i)
}

///   	choice	   ::=   	'(' S? cp ( S? '|' S? cp )+ S? ')'	[VC: Proper Group/PE Nesting]
#[tracable_parser]
fn choices(i: Span) -> Result<Choices<Repeatable<Child>>> {
    map(
        tuple((
            parenthesis_open,
            multispace0,
            cuple,
            many1(tuple((multispace0, tag("|"), multispace0, cuple))),
            multispace0,
            parenthesis_close,
        )),
        |(_, _, cp, many, _, _)| {
            Choices(
                std::iter::once(cp)
                    .chain(many.into_iter().map(|(_, _, _, x)| x))
                    .collect(),
            )
        },
    )(i)
}

///   	cp	   ::=   	(Name | choice | seq) ('?' | '*' | '+')?
#[tracable_parser]
fn cuple(i: Span) -> Result<Repeatable<Child>> {
    map(
        tuple((
            alt((
                map(name, Child::Name),
                map(choices, Child::Choices),
                map(sequence, Child::Seq),
            )),
            opt(alt((tag("?"), tag("*"), tag("+")))),
        )),
        |(child, repeatable)| {
            let repeatable = if let Some(repeatable) = repeatable {
                match *repeatable {
                    "?" => |child| Repeatable::AtMostOnce(child),
                    "*" => |child| Repeatable::ZeroOrManyTimes(child),
                    "+" => |child| Repeatable::AtLeastOnce(child),
                    _ => unreachable!(),
                }
            } else {
                |child| Repeatable::Once(child)
            };
            repeatable(child)
        },
    )(i)
}

#[tracable_parser]
fn empty(i: Span) -> Result<ElementCategory> {
    map(tag("EMPTY"), |_| ElementCategory::Empty)(i)
}

#[tracable_parser]
fn any(i: Span) -> Result<ElementCategory> {
    map(tag("ANY"), |_| ElementCategory::Any)(i)
}

/// <!ELEMENT 元素名称 类别>
#[tracable_parser]
pub(super) fn element_decl(i: Span) -> Result<ElementDecl> {
    map(
        tuple((
            tag("<!ELEMENT"),
            multispace1,
            name,
            multispace1,
            alt((empty, any, mixed, children)),
            multispace0,
            tag(">"),
        )),
        |(_, _, n, _, c, _, _)| ElementDecl {
            name: n,
            category: c,
        },
    )(i)
}

// <!ELEMENT 元素名称 (元素内容)>
// 空元素通过类别关键词EMPTY进行声明：
// <!ELEMENT 元素名称 EMPTY>
// 只有 PCDATA 的元素通过圆括号中的 #PCDATA 进行声明：
// <!ELEMENT 元素名称 (#PCDATA)>
// 通过类别关键词 ANY 声明的元素，可包含任何可解析数据的组合：
// <!ELEMENT 元素名称 ANY>
// 带有一个或多个子元素的元素通过圆括号中的子元素名进行声明：
// <!ELEMENT 元素名称 (子元素名称 1)>
// 或者
// <!ELEMENT 元素名称 (子元素名称 1,子元素名称 2,.....)>
// 声明只出现一次的元素
// <!ELEMENT 元素名称 (子元素名称)>
// 例子：
// <!ELEMENT note (message)>
// 上面的例子声明了：message 子元素必须出现一次，并且必须只在 "note" 元素中出现一次。
//
// 声明最少出现一次的元素
// <!ELEMENT 元素名称 (子元素名称+)>
// 例子：
// <!ELEMENT note (message+)>
// 上面的例子中的加号声明了：message 子元素必须在 "note" 元素内出现至少一次。
//
// 声明出现零次或多次的元素
// <!ELEMENT 元素名称 (子元素名称*)>
// 例子：
// <!ELEMENT note (message*)>
// 上面的例子中的星号声明了：子元素 message 可在 "note" 元素内出现零次或多次。
//
// 声明出现零次或一次的元素
// <!ELEMENT 元素名称 (子元素名称?)>
// 例子：
// <!ELEMENT note (message?)>
// 上面的例子中的问号声明了：子元素 message 可在 "note" 元素内出现零次或一次。
//
// 声明“非.../既...”类型的内容
// 例子：
// <!ELEMENT note (to,from,header,(message|body))>
// 上面的例子声明了："note" 元素必须包含 "to" 元素、"from" 元素、"header" 元素，以及非 "message" 元素既 "body" 元素。
//
// 声明混合型的内容
// 例子：
// <!ELEMENT note (#PCDATA|to|from|header|message)*>
// 上面的例子声明了："note" 元素可包含出现零次或多次的 PCDATA、"to"、"from"、"header" 或者 "message"。

#[cfg(test)]
mod tests {

    use super::{children, element_decl, mixed};
    use crate::assert_ok;
    use crate::span;

    use nom::Finish;

    #[test]
    fn test_element_decl() {
        let input = span("<!ELEMENT b (#PCDATA)>");
        let el = element_decl(input).finish();
        assert_ok!(input, el);
        let input = span("<!ELEMENT p (#PCDATA|a|ul|b|i|em)*>");
        let el = element_decl(input).finish();
        assert_ok!(input, el);
        let el = element_decl(span(
            "<!ELEMENT p (#PCDATA | %font; | %phrase; | %special; | %form;)* >",
        ))
        .finish();
        assert_ok!(input, el);
    }

    #[test]
    fn test_element_decl_multiline() {
        let input = span(
            r#"<!ELEMENT generated  (#PCDATA |   emphasis | strong | literal | math
    | reference | footnote_reference | citation_reference
    | substitution_reference | title_reference
    | abbreviation | acronym | subscript | superscript
    | inline | problematic | generated
    | target | image | raw
        )* >"#,
        );
        let el = element_decl(input).finish();
        assert_ok!(input, el);
    }

    #[test]
    fn test_element_decl_multiline2() {
        let el = element_decl(span(
            r#"<!ELEMENT document
    ( (title, subtitle?)?,
      meta?,
      decoration?,
      (docinfo, transition?)?,
       ( ( (  paragraph | compound | container | literal_block | doctest_block
    | math_block | line_block | block_quote
    | table | figure | image | footnote | citation | rubric
    | bullet_list | enumerated_list | definition_list | field_list
    | option_list
    | attention | caution | danger | error | hint | important | note
    | tip | warning | admonition
    | reference | target | substitution_definition | comment | pending
    | system_message | raw
         | topic | sidebar)+, transition? )*,
       ( (  section
        ), (transition?, (  section
        ) )* )? ) )>"#,
        ));
        assert!(el.is_ok(), "{:?}", el.as_ref().unwrap_err());
    }

    #[test]
    fn test_child() {
        let el = children(span("(title, subtitle?)?"));
        assert!(el.is_ok(), "{:?}", el.as_ref().unwrap_err());
    }

    #[test]
    fn test_mixed_simple() {
        let input = span("<!ELEMENT option_string (#PCDATA)>");
        let el = element_decl(input).finish();
        assert_ok!(input, el);
    }
}