natter/src/intermediate/ast_node.rs

use super::IBabelCall;
use super::ICenterBlock;
use super::IClock;
use super::ICommentBlock;
use super::IDiarySexp;
use super::IDrawer;
use super::IDynamicBlock;
use super::IExampleBlock;
use super::IExportBlock;
use super::IFixedWidthArea;
use super::IFootnoteDefinition;
use super::IHeading;
use super::IHorizontalRule;
use super::ILatexEnvironment;
use super::IParagraph;
use super::IPlainList;
use super::IPlanning;
use super::IPropertyDrawer;
use super::IQuoteBlock;
use super::ISection;
use super::ISpecialBlock;
use super::ISrcBlock;
use super::ITable;
use super::ITarget;
use super::ITimestamp;
use super::IUnderline;
use super::IVerbatim;
use super::IVerseBlock;
use super::IntermediateContext;
use super::angle_link::IAngleLink;
use super::bold::IBold;
use super::citation::ICitation;
use super::citation_reference::ICitationReference;
use super::code::ICode;
use super::comment::IComment;
use super::entity::IEntity;
use super::export_snippet::IExportSnippet;
use super::footnote_reference::IFootnoteReference;
use super::inline_babel_call::IInlineBabelCall;
use super::inline_source_block::IInlineSourceBlock;
use super::italic::IItalic;
use super::keyword::IKeyword;
use super::latex_fragment::ILatexFragment;
use super::line_break::ILineBreak;
use super::org_macro::IOrgMacro;
use super::plain_link::IPlainLink;
use super::plain_text::IPlainText;
use super::radio_link::IRadioLink;
use super::radio_target::IRadioTarget;
use super::regular_link::IRegularLink;
use super::statistics_cookie::IStatisticsCookie;
use super::strike_through::IStrikeThrough;
use super::subscript::ISubscript;
use super::superscript::ISuperscript;
use crate::error::CustomError;
use futures::future::{BoxFuture, FutureExt};

#[derive(Debug, Clone)]
pub(crate) enum IAstNode {
    Heading(IHeading),
    Section(ISection),
    Paragraph(IParagraph),
    PlainList(IPlainList),
    CenterBlock(ICenterBlock),
    QuoteBlock(IQuoteBlock),
    SpecialBlock(ISpecialBlock),
    DynamicBlock(IDynamicBlock),
    FootnoteDefinition(IFootnoteDefinition),
    Comment(IComment),
    Drawer(IDrawer),
    PropertyDrawer(IPropertyDrawer),
    Table(ITable),
    VerseBlock(IVerseBlock),
    CommentBlock(ICommentBlock),
    ExampleBlock(IExampleBlock),
    ExportBlock(IExportBlock),
    SrcBlock(ISrcBlock),
    Clock(IClock),
    DiarySexp(IDiarySexp),
    Planning(IPlanning),
    FixedWidthArea(IFixedWidthArea),
    HorizontalRule(IHorizontalRule),
    Keyword(IKeyword),
    BabelCall(IBabelCall),
    LatexEnvironment(ILatexEnvironment),
    Bold(IBold),
    Italic(IItalic),
    Underline(IUnderline),
    StrikeThrough(IStrikeThrough),
    Code(ICode),
    Verbatim(IVerbatim),
    PlainText(IPlainText),
    RegularLink(IRegularLink),
    RadioLink(IRadioLink),
    RadioTarget(IRadioTarget),
    PlainLink(IPlainLink),
    AngleLink(IAngleLink),
    OrgMacro(IOrgMacro),
    Entity(IEntity),
    LatexFragment(ILatexFragment),
    ExportSnippet(IExportSnippet),
    FootnoteReference(IFootnoteReference),
    Citation(ICitation),
    CitationReference(ICitationReference),
    InlineBabelCall(IInlineBabelCall),
    InlineSourceBlock(IInlineSourceBlock),
    LineBreak(ILineBreak),
    Target(ITarget),
    StatisticsCookie(IStatisticsCookie),
    Subscript(ISubscript),
    Superscript(ISuperscript),
    Timestamp(ITimestamp),
}

pub(crate) trait IntoIAstNode<'parse> {
    fn as_ast_node<'orig>(
        &'orig self,
        intermediate_context: IntermediateContext<'orig, 'parse>,
    ) -> BoxFuture<'orig, Result<IAstNode, CustomError>>;
}

impl<'parse> IntoIAstNode<'parse> for organic::types::DocumentElement<'parse> {
    fn as_ast_node<'orig>(
        &'orig self,
        intermediate_context: IntermediateContext<'orig, 'parse>,
    ) -> BoxFuture<'orig, Result<IAstNode, CustomError>> {
        async move {
            match self {
                organic::types::DocumentElement::Heading(inner) => Ok(IAstNode::Heading(
                    IHeading::new(intermediate_context, inner).await?,
                )),
                organic::types::DocumentElement::Section(inner) => Ok(IAstNode::Section(
                    ISection::new(intermediate_context, inner).await?,
                )),
            }
        }
        .boxed()
    }
}

impl<'parse> IntoIAstNode<'parse> for organic::types::Element<'parse> {
    fn as_ast_node<'orig>(
        &'orig self,
        intermediate_context: IntermediateContext<'orig, 'parse>,
    ) -> BoxFuture<'orig, Result<IAstNode, CustomError>> {
        async move {
            match self {
                organic::types::Element::Paragraph(inner) => Ok(IAstNode::Paragraph(
                    IParagraph::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::PlainList(inner) => Ok(IAstNode::PlainList(
                    IPlainList::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::CenterBlock(inner) => Ok(IAstNode::CenterBlock(
                    ICenterBlock::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::QuoteBlock(inner) => Ok(IAstNode::QuoteBlock(
                    IQuoteBlock::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::SpecialBlock(inner) => Ok(IAstNode::SpecialBlock(
                    ISpecialBlock::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::DynamicBlock(inner) => Ok(IAstNode::DynamicBlock(
                    IDynamicBlock::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::FootnoteDefinition(inner) => {
                    Ok(IAstNode::FootnoteDefinition(
                        IFootnoteDefinition::new(intermediate_context, inner).await?,
                    ))
                }
                organic::types::Element::Comment(inner) => Ok(IAstNode::Comment(
                    IComment::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::Drawer(inner) => Ok(IAstNode::Drawer(
                    IDrawer::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::PropertyDrawer(inner) => Ok(IAstNode::PropertyDrawer(
                    IPropertyDrawer::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::Table(inner) => Ok(IAstNode::Table(
                    ITable::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::VerseBlock(inner) => Ok(IAstNode::VerseBlock(
                    IVerseBlock::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::CommentBlock(inner) => Ok(IAstNode::CommentBlock(
                    ICommentBlock::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::ExampleBlock(inner) => Ok(IAstNode::ExampleBlock(
                    IExampleBlock::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::ExportBlock(inner) => Ok(IAstNode::ExportBlock(
                    IExportBlock::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::SrcBlock(inner) => Ok(IAstNode::SrcBlock(
                    ISrcBlock::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::Clock(inner) => Ok(IAstNode::Clock(
                    IClock::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::DiarySexp(inner) => Ok(IAstNode::DiarySexp(
                    IDiarySexp::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::Planning(inner) => Ok(IAstNode::Planning(
                    IPlanning::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::FixedWidthArea(inner) => Ok(IAstNode::FixedWidthArea(
                    IFixedWidthArea::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::HorizontalRule(inner) => Ok(IAstNode::HorizontalRule(
                    IHorizontalRule::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::Keyword(inner) => Ok(IAstNode::Keyword(
                    IKeyword::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::BabelCall(inner) => Ok(IAstNode::BabelCall(
                    IBabelCall::new(intermediate_context, inner).await?,
                )),
                organic::types::Element::LatexEnvironment(inner) => Ok(IAstNode::LatexEnvironment(
                    ILatexEnvironment::new(intermediate_context, inner).await?,
                )),
            }
        }
        .boxed()
    }
}

impl<'parse> IntoIAstNode<'parse> for organic::types::Object<'parse> {
    fn as_ast_node<'orig>(
        &'orig self,
        intermediate_context: IntermediateContext<'orig, 'parse>,
    ) -> BoxFuture<'orig, Result<IAstNode, CustomError>> {
        async move {
            match self {
                organic::types::Object::Bold(inner) => Ok(IAstNode::Bold(
                    IBold::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::Italic(inner) => Ok(IAstNode::Italic(
                    IItalic::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::Underline(inner) => Ok(IAstNode::Underline(
                    IUnderline::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::StrikeThrough(inner) => Ok(IAstNode::StrikeThrough(
                    IStrikeThrough::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::Code(inner) => Ok(IAstNode::Code(
                    ICode::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::Verbatim(inner) => Ok(IAstNode::Verbatim(
                    IVerbatim::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::PlainText(inner) => Ok(IAstNode::PlainText(
                    IPlainText::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::RegularLink(inner) => Ok(IAstNode::RegularLink(
                    IRegularLink::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::RadioLink(inner) => Ok(IAstNode::RadioLink(
                    IRadioLink::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::RadioTarget(inner) => Ok(IAstNode::RadioTarget(
                    IRadioTarget::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::PlainLink(inner) => Ok(IAstNode::PlainLink(
                    IPlainLink::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::AngleLink(inner) => Ok(IAstNode::AngleLink(
                    IAngleLink::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::OrgMacro(inner) => Ok(IAstNode::OrgMacro(
                    IOrgMacro::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::Entity(inner) => Ok(IAstNode::Entity(
                    IEntity::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::LatexFragment(inner) => Ok(IAstNode::LatexFragment(
                    ILatexFragment::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::ExportSnippet(inner) => Ok(IAstNode::ExportSnippet(
                    IExportSnippet::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::FootnoteReference(inner) => {
                    Ok(IAstNode::FootnoteReference(
                        IFootnoteReference::new(intermediate_context, inner).await?,
                    ))
                }
                organic::types::Object::Citation(inner) => Ok(IAstNode::Citation(
                    ICitation::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::CitationReference(inner) => {
                    Ok(IAstNode::CitationReference(
                        ICitationReference::new(intermediate_context, inner).await?,
                    ))
                }
                organic::types::Object::InlineBabelCall(inner) => Ok(IAstNode::InlineBabelCall(
                    IInlineBabelCall::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::InlineSourceBlock(inner) => {
                    Ok(IAstNode::InlineSourceBlock(
                        IInlineSourceBlock::new(intermediate_context, inner).await?,
                    ))
                }
                organic::types::Object::LineBreak(inner) => Ok(IAstNode::LineBreak(
                    ILineBreak::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::Target(inner) => Ok(IAstNode::Target(
                    ITarget::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::StatisticsCookie(inner) => Ok(IAstNode::StatisticsCookie(
                    IStatisticsCookie::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::Subscript(inner) => Ok(IAstNode::Subscript(
                    ISubscript::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::Superscript(inner) => Ok(IAstNode::Superscript(
                    ISuperscript::new(intermediate_context, inner).await?,
                )),
                organic::types::Object::Timestamp(inner) => Ok(IAstNode::Timestamp(
                    ITimestamp::new(intermediate_context, inner).await?,
                )),
            }
        }
        .boxed()
    }
}