natter/src/intermediate/object.rs

use crate::error::CustomError;

use super::angle_link::IAngleLink;
use super::bold::IBold;
use super::citation::ICitation;
use super::citation_reference::ICitationReference;
use super::code::ICode;
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::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::registry::Registry;
use super::regular_link::IRegularLink;
use super::statistics_cookie::IStatisticsCookie;
use super::strike_through::IStrikeThrough;
use super::subscript::ISubscript;
use super::superscript::ISuperscript;
use super::timestamp::ITimestamp;
use super::underline::IUnderline;
use super::verbatim::IVerbatim;
use super::ITarget;
use futures::future::{BoxFuture, FutureExt};

#[derive(Debug)]
pub(crate) enum IObject {
    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),
}

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