cosy 0.1.0

use super::super::node::parse_nodes;
use crate::CosyParserExtension;
use crate::ast::Link;
use crate::ast::Node;
use crate::url::{UrlKind, infer_url};
use winnow::Result as PResult;

enum BracketToken<'a> {
    Image(::url::Url),
    Link(::url::Url),
    Plain(&'a str),
}

impl<'a> BracketToken<'a> {
    fn of(s: &'a str) -> Self {
        match infer_url(s) {
            Some((url, UrlKind::Image)) => Self::Image(url),
            Some((url, UrlKind::Other)) => Self::Link(url),
            None => Self::Plain(s),
        }
    }
}

pub(super) fn parse_links_and_pages<'s, 'i, E>(
    extension: &'s E,
) -> impl FnMut(&mut &'i str) -> PResult<Node<E::Output>> + 's
where
    E: CosyParserExtension,
{
    move |input: &mut &'i str| {
        let content = *input;
        *input = ""; // Always consumes the whole bracket content.

        // Single-token content.
        let Some((first_token, rest)) = content.split_once(' ') else {
            return Ok(match BracketToken::of(content) {
                BracketToken::Image(url) => Node::Image(url),
                BracketToken::Link(url) => Node::Link(Link::Url(url)),
                BracketToken::Plain(s) => Node::Link(Link::Page(s.to_string())),
            });
        };
        // `split_once` succeeded, so `rsplit_once` must too.
        let (start, last_token) = content.rsplit_once(' ').expect("content contains ' '");

        let with_label = |href: ::url::Url, text: &str| -> PResult<Node<E::Output>> {
            let mut label_input = text;
            let label = parse_nodes(&mut label_input, extension)?;
            Ok(Node::Link(Link::WithLabel { href, label }))
        };

        // Only the first and last tokens can be URLs; classify each end and
        // dispatch on the pair.
        match (BracketToken::of(first_token), BracketToken::of(last_token)) {
            // LinkedImage — both ends are URLs and at least one is an image.
            // (Only reachable for 2-token inputs, since 3+ tokens means the
            // "label side" string contains spaces and can't be a URL.)
            (BracketToken::Image(href), BracketToken::Image(src)) => {
                Ok(Node::LinkedImage { src, href })
            }
            (BracketToken::Image(src), BracketToken::Link(href)) => {
                Ok(Node::LinkedImage { src, href })
            }
            (BracketToken::Link(href), BracketToken::Image(src)) => {
                Ok(Node::LinkedImage { src, href })
            }

            // URL anchor on the left → label is everything that follows.
            (BracketToken::Link(href), _) => with_label(href, rest),

            // URL anchor on the right → label is everything that precedes.
            (_, BracketToken::Link(href) | BracketToken::Image(href)) => with_label(href, start),

            // (Image, Plain) and (Plain, Plain) — no usable URL pair.
            _ => Ok(Node::Link(Link::Page(content.to_string()))),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::super::parse_bracket;
    use crate::ast::{Link, Node};
    use winnow::Parser;

    fn parse(input: &str) -> Node<()> {
        let mut s = input;
        parse_bracket(&()).parse_next(&mut s).unwrap()
    }

    // LinkedImage tests
    const IMAGE_URL: &str = "https://example.com/photo.png";
    const IMAGE_URL_2: &str = "https://example.com/other.jpg";
    const LINK_URL: &str = "https://example.com/page";

    #[test]
    fn test_linked_image_img_then_link() {
        // [image_url link_url] → LinkedImage { src: image_url, href: link_url }
        let node = parse(&format!("[{IMAGE_URL} {LINK_URL}]"));
        assert_eq!(
            node,
            Node::LinkedImage {
                src: ::url::Url::parse(IMAGE_URL).unwrap(),
                href: ::url::Url::parse(LINK_URL).unwrap(),
            }
        );
    }

    #[test]
    fn test_linked_image_link_then_img() {
        // [link_url image_url] → LinkedImage { src: image_url, href: link_url }
        let node = parse(&format!("[{LINK_URL} {IMAGE_URL}]"));
        assert_eq!(
            node,
            Node::LinkedImage {
                src: ::url::Url::parse(IMAGE_URL).unwrap(),
                href: ::url::Url::parse(LINK_URL).unwrap(),
            }
        );
    }

    #[test]
    fn test_linked_image_two_images() {
        // [img1 img2] → LinkedImage { src: img2, href: img1 }
        let node = parse(&format!("[{IMAGE_URL} {IMAGE_URL_2}]"));
        assert_eq!(
            node,
            Node::LinkedImage {
                src: ::url::Url::parse(IMAGE_URL_2).unwrap(),
                href: ::url::Url::parse(IMAGE_URL).unwrap(),
            }
        );
    }

    // Regression tests for non-image URL cases
    #[test]
    fn test_with_label_url_then_single_word() {
        // [url label] → WithLabel { href: url, label: [Text("label")] }
        let node = parse(&format!("[{LINK_URL} label]"));
        assert_eq!(
            node,
            Node::Link(Link::WithLabel {
                href: ::url::Url::parse(LINK_URL).unwrap(),
                label: vec![Node::Text("label".to_string())],
            })
        );
    }

    #[test]
    fn test_with_label_single_word_then_url() {
        // [label url] → WithLabel { href: url, label: [Text("label")] }
        let node = parse(&format!("[label {LINK_URL}]"));
        assert_eq!(
            node,
            Node::Link(Link::WithLabel {
                href: ::url::Url::parse(LINK_URL).unwrap(),
                label: vec![Node::Text("label".to_string())],
            })
        );
    }

    #[test]
    fn test_with_label_url_then_multi_word() {
        // [url some text] → WithLabel { href: url, label: [Text("some text")] }
        let node = parse(&format!("[{LINK_URL} some text]"));
        assert_eq!(
            node,
            Node::Link(Link::WithLabel {
                href: ::url::Url::parse(LINK_URL).unwrap(),
                label: vec![Node::Text("some text".to_string())],
            })
        );
    }

    #[test]
    fn test_with_label_multi_word_then_url() {
        // [some text url] → WithLabel { href: url, label: [Text("some text")] }
        let node = parse(&format!("[some text {LINK_URL}]"));
        assert_eq!(
            node,
            Node::Link(Link::WithLabel {
                href: ::url::Url::parse(LINK_URL).unwrap(),
                label: vec![Node::Text("some text".to_string())],
            })
        );
    }

    #[test]
    fn test_with_label_url_then_multiple_urls() {
        // [url1 url2 url3] → WithLabel { href: url1, label: [Text("url2 url3")] }
        let url2 = "https://example.com/other";
        let url3 = "https://example.com/another";
        let node = parse(&format!("[{LINK_URL} {url2} {url3}]"));
        assert_eq!(
            node,
            Node::Link(Link::WithLabel {
                href: ::url::Url::parse(LINK_URL).unwrap(),
                label: vec![Node::Text(format!("{url2} {url3}"))],
            })
        );
    }

    #[test]
    fn test_with_label_two_non_image_urls() {
        // [url1 url2] → WithLabel { href: url1, label: [Text(url2)] }
        // parse_nodes on a bare URL (no brackets) produces Text, not Link::Url
        let url2 = "https://example.com/other";
        let node = parse(&format!("[{LINK_URL} {url2}]"));
        assert_eq!(
            node,
            Node::Link(Link::WithLabel {
                href: ::url::Url::parse(LINK_URL).unwrap(),
                label: vec![Node::Text(url2.to_string())],
            })
        );
    }

    #[test]
    fn test_page_with_spaces() {
        // [text1 text2] → Page("text1 text2")
        let node = parse("[hello world]");
        assert_eq!(node, Node::Link(Link::Page("hello world".to_string())));
    }

    #[test]
    fn test_page_with_multiple_spaces() {
        // [text1 text2 text3] → Page("text1 text2 text3")
        let node = parse("[hello beautiful world]");
        assert_eq!(
            node,
            Node::Link(Link::Page("hello beautiful world".to_string()))
        );
    }
}