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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

//! # Documentation
//!
//! This crate provides a simple interface for fetching oEmbed data from known providers.
//!
//! ## Example
//! ```rust
//! use oembed_rs::{find_provider, fetch, ConsumerRequest};
//!
//! async fn example() {
//!     let url = "https://twitter.com/user/status/1000000000000000000";
//!     let (_, endpoint) = find_provider(url).expect("unknown provider");
//!
//!     let response = fetch(
//!        &endpoint.url,
//!        ConsumerRequest {
//!            url,
//!            max_width: Some(1000),
//!            max_height: Some(500),
//!            ..ConsumerRequest::default()
//!        },
//!     )
//!     .await
//!     .expect("failed to fetch oembed data");
//! }
//! ```

use lazy_static::lazy_static;

mod error;
mod request;
mod spec;

pub use error::Error;
pub use request::{fetch, ConsumerRequest};
pub use spec::*;

lazy_static! {
    static ref PROVIDERS: Vec<Provider> =
        serde_json::from_slice(include_bytes!("../providers.json"))
            .expect("failed to load providers");
}

/// Find the oEmbed provider and endpoint based on the URL
pub fn find_provider(url: &str) -> Option<(&Provider, &Endpoint)> {
    PROVIDERS.iter().find_map(|p| {
        p.endpoints
            .iter()
            .find(|e| e.schemes.iter().any(|s| matches_scheme(s, url)))
            .map(|e| (p, e))
    })
}

/// Checks if the URL matches the scheme
pub fn matches_scheme(scheme: &str, mut url: &str) -> bool {
    let mut fragments = scheme.split('*').filter(|f| !f.is_empty()).peekable();

    while let Some(fragment) = fragments.next() {
        if !url.starts_with(fragment) {
            return false;
        }

        url = &url[fragment.len()..];

        if let Some(fragment) = fragments.peek() {
            let Some(idx) = url.find(fragment) else {
                return false;
            };

            url = &url[idx..];
        }
    }

    scheme.ends_with('*') || url.is_empty()
}

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

    #[test]
    fn test_twitter_provider() {
        let url = "https://twitter.com/user/status/1640004220000000000?s=20";
        let (provider, endpoint) = find_provider(url).unwrap();

        assert_eq!(provider.provider_name, "Twitter");
        assert_eq!(endpoint.url, "https://publish.twitter.com/oembed");
    }

    #[test]
    fn test_youtube_provider() {
        let url = "https://youtu.be/rAn0MId";
        let (provider, endpoint) = find_provider(url).unwrap();

        assert_eq!(provider.provider_name, "YouTube");
        assert_eq!(endpoint.url, "https://www.youtube.com/oembed");
    }

    #[test]
    fn test_youtube_provider_with_query() {
        let url = "https://www.youtube.com/watch?v=rAn0MId";
        let (provider, endpoint) = find_provider(url).unwrap();

        assert_eq!(provider.provider_name, "YouTube");
        assert_eq!(endpoint.url, "https://www.youtube.com/oembed");
    }

    #[test]
    fn test_invalid() {
        let url = "https://twitter.nl/user/status/1640004220000000000?s=20";
        assert!(find_provider(url).is_none());

        let url = "https://www.youtube.com/watcx?v=test";
        assert!(find_provider(url).is_none());
    }
}