cosy 0.1.0

A parser for Cosense/Scrapbox markup syntax that produces a typed AST
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

use url::Url;

#[derive(Debug, PartialEq, Eq, Clone, Copy)]
pub enum UrlKind {
    Image,
    Other,
}

/// Parses `s` as a URL and classifies it. Returns the parsed [`Url`] together
/// with the inferred [`UrlKind`] so callers do not have to re-parse.
pub fn infer_url(s: &str) -> Option<(Url, UrlKind)> {
    if !s.contains("://") {
        return None;
    }
    let url = Url::parse(s).ok()?;
    let kind = if let Some(ext) = url
        .path_segments()
        .and_then(|mut segments| segments.next_back())
        .and_then(|name| name.split('.').next_back())
    {
        let mime = mime_guess::from_ext(ext).first_or_octet_stream();
        if mime.type_() == mime::IMAGE {
            UrlKind::Image
        } else {
            UrlKind::Other
        }
    } else {
        UrlKind::Other
    };
    Some((url, kind))
}

#[cfg(test)]
mod tests {
    use super::*;

    fn kind_of(s: &str) -> Option<UrlKind> {
        infer_url(s).map(|(_, k)| k)
    }

    #[test]
    fn test_infer_url_kind() {
        assert_eq!(
            kind_of("https://example.com/image.png"),
            Some(UrlKind::Image)
        );
        assert_eq!(
            kind_of("https://example.com/document.pdf"),
            Some(UrlKind::Other)
        );
        assert_eq!(kind_of("not a url"), None);
    }

    #[test]
    fn test_infer_url_kind_image_with_query() {
        // Query string should not affect image detection
        assert_eq!(
            kind_of("https://example.com/image.png?w=100"),
            Some(UrlKind::Image)
        );
    }

    #[test]
    fn test_infer_url_returns_parsed_url() {
        let (url, kind) = infer_url("https://example.com/photo.png").unwrap();
        assert_eq!(url.as_str(), "https://example.com/photo.png");
        assert_eq!(kind, UrlKind::Image);
    }
}