unidok-parser 0.2.0

use aho_corasick::AhoCorasick;
use unidok_repr::ast::segments::ImageAst;

use super::links::{ParseLinkTargetReference, ParseLinkTargetUrl};
use super::Segments;
use crate::parsing_mode::ParsingMode;
use crate::{Context, Indents, Input, Parse};

pub(crate) struct ParseImage<'a> {
    pub ind: Indents<'a>,
    pub mode: Option<ParsingMode>,
    pub ac: &'a AhoCorasick,
}

impl Parse for ParseImage<'_> {
    type Output = ImageAst;

    fn parse(&mut self, input: &mut Input) -> Option<Self::Output> {
        if let Some(img) =
            input.parse(ParseFullImage { ind: self.ind, ac: self.ac, mode: self.mode })
        {
            Some(img)
        } else {
            let mut input = input.start();
            input.parse('!')?;
            let target = input.parse(ParseLinkTargetReference)?;
            input.apply();
            Some(ImageAst { alt: None, target })
        }
    }
}

pub(crate) struct ParseFullImage<'a> {
    ind: Indents<'a>,
    mode: Option<ParsingMode>,
    ac: &'a AhoCorasick,
}

impl Parse for ParseFullImage<'_> {
    type Output = ImageAst;

    fn parse(&mut self, input: &mut Input) -> Option<Self::Output> {
        let mut input = input.start();

        input.parse("![")?;
        let alt = input
            .parse(Segments::parser(
                self.ind,
                Context::LinkOrImg,
                self.mode.unwrap_or_else(ParsingMode::new_all),
                self.ac,
            ))?
            .into_segments_no_underline_zero()?;
        input.parse(']')?;

        let target =
            input.parse(ParseLinkTargetUrl).or_else(|| input.parse(ParseLinkTargetReference))?;

        input.apply();
        Some(ImageAst { alt: Some(alt), target })
    }
}