mistralrs_server_core/
util.rs

1//! ## General utilities.
2
3use image::DynamicImage;
4use tokio::{
5    fs::{self, File},
6    io::AsyncReadExt,
7};
8
9/// Parses and loads an image from a URL, file path, or data URL.
10///
11/// This function accepts various input formats and attempts to parse them in order:
12/// 1. First tries to parse as a complete URL (http/https/file/data schemes)
13/// 2. If that fails, checks if it's a local file path and converts to file URL
14/// 3. Finally falls back to treating it as a malformed URL and returns an error
15///
16/// ### Arguments
17///
18/// * `url_unparsed` - A string that can be:
19///   - An HTTP/HTTPS URL (e.g., "<https://example.com/image.png>")
20///   - A file path (e.g., "/path/to/image.jpg" or "image.png")
21///   - A data URL with base64 encoded image (e.g., "data:image/png;base64,...")
22///   - A file URL (e.g., "file:///path/to/image.jpg")
23///
24/// ### Examples
25///
26/// ```ignore
27/// use mistralrs_server_core::util::parse_image_url;
28///
29/// // Load from HTTP URL
30/// let image = parse_image_url("https://example.com/photo.jpg").await?;
31///
32/// // Load from local file path
33/// let image = parse_image_url("./assets/logo.png").await?;
34///
35/// // Load from data URL
36/// let image = parse_image_url("data:image/png;base64,iVBORw0KGgoAAAANS...").await?;
37///
38/// // Load from file URL
39/// let image = parse_image_url("file:///home/user/picture.jpg").await?;
40/// ```
41pub async fn parse_image_url(url_unparsed: &str) -> Result<DynamicImage, anyhow::Error> {
42    let url = if let Ok(url) = url::Url::parse(url_unparsed) {
43        url
44    } else if File::open(url_unparsed).await.is_ok() {
45        url::Url::from_file_path(std::path::absolute(url_unparsed)?)
46            .map_err(|_| anyhow::anyhow!("Could not parse file path: {}", url_unparsed))?
47    } else {
48        url::Url::parse(url_unparsed)
49            .map_err(|_| anyhow::anyhow!("Could not parse as base64 data: {}", url_unparsed))?
50    };
51
52    let bytes = if url.scheme() == "http" || url.scheme() == "https" {
53        // Read from http
54        match reqwest::get(url.clone()).await {
55            Ok(http_resp) => http_resp.bytes().await?.to_vec(),
56            Err(e) => anyhow::bail!(e),
57        }
58    } else if url.scheme() == "file" {
59        let path = url
60            .to_file_path()
61            .map_err(|_| anyhow::anyhow!("Could not parse file path: {}", url))?;
62
63        if let Ok(mut f) = File::open(&path).await {
64            // Read from local file
65            let metadata = fs::metadata(&path).await?;
66            let mut buffer = vec![0; metadata.len() as usize];
67            f.read_exact(&mut buffer).await?;
68            buffer
69        } else {
70            anyhow::bail!("Could not open file at path: {}", url);
71        }
72    } else if url.scheme() == "data" {
73        // Decode with base64
74        let data_url = data_url::DataUrl::process(url.as_str())?;
75        data_url.decode_to_vec()?.0
76    } else {
77        anyhow::bail!("Unsupported URL scheme: {}", url.scheme());
78    };
79
80    Ok(image::load_from_memory(&bytes)?)
81}
82
83#[cfg(test)]
84mod tests {
85    use image::GenericImageView;
86
87    use super::*;
88
89    #[tokio::test]
90    async fn test_parse_image_url() {
91        // from URL
92        let url = "https://www.rust-lang.org/logos/rust-logo-32x32.png";
93        let image = parse_image_url(url).await.unwrap();
94        assert_eq!(image.dimensions(), (32, 32));
95
96        let url = "http://www.rust-lang.org/logos/rust-logo-32x32.png";
97        let image = parse_image_url(url).await.unwrap();
98        assert_eq!(image.dimensions(), (32, 32));
99
100        // from file path
101        let url = "resources/rust-logo-32x32.png";
102        let image = parse_image_url(url).await.unwrap();
103        assert_eq!(image.dimensions(), (32, 32));
104
105        // URL must be an absolute path
106        let absolute_path = std::path::absolute(url).unwrap();
107        let url = format!("file://{}", absolute_path.as_os_str().to_str().unwrap());
108        let image = parse_image_url(&url).await.unwrap();
109        assert_eq!(image.dimensions(), (32, 32));
110
111        // from base64 encoded image (rust-logo-32x32.png)
112        let url = "
113        iVBORw0KGgoAAAANSUhEUgAAACAAAAAgCAYAAABzenr0AAAHhElEQVR4AZXVA5Aky9bA8f/JzKrq
114        npleX9u2bdu2bdu2bdv29z1e2zZm7k5PT3dXZeZ56I6J3o03sbG/iDLOSQuT6fptF11OREYDj4uR
115        i9UmAAdHa9ZH6QP+n8kg1+26HJMU44rG+4OjL3YqCv+693HOwcHiTJeYY2NUch/PLI3sOdZY82lU
116        Xbynp3yzEXMH8CCTINfuujzDEXQlVN9sju8/uFHPTy2KWLVpWsl9ZGQCvY2AF0ulu0RTBRHIi1AV
117        iZU0sSd0dWWXZKVsUeAVhiFX7roCwzGDA9rXV6uaqH/YcmnmPEQGg4IYIoLAYRHRABcaQIGuNMVa
118        IS98tZnnjOxJK4AwDDlzs0XoNGUmlWDsPr/98ucLIerrPVlCI8KAWMAQAYWXo8rKipyuMDewuaAv
119        g6wMgEa6M0dX6ugdqOPQxSs96WqlcukqoEoHuWiHZelki3yF/vHVV0OhdCUJfzZyQlYiiPlR4RxV
120        bgKqAbNthDto2Q64U6ACbAicKzCtAON6Uqr1HAk5XYlZEXiNDnLaBgvQxqiSzPdLX70PNT9U/pN9
121        0xNdSjT2UoXjJ84+x6ygwMQ/bSdyOnCgamSqSpmBepOY53OliXHAh7TJsesuCMBMU/XM/+dvve/9
122        PhgYl2X8Xi8IWZkobAg8xuQjx24L3KEamaY7oX/8IDZ6ukZkCwDvA8gpGy1EG9Vq44fRpXTa3oZv
123        BVeIQERQQBFUQQGE4frWj+3hdyxQtei2oHe4UDB1KvxWL34EpqPNLjzdWKYZXVqpr3fgdDV2QSJZ
124        A4M3loC0gqu0ggsgrXMQhlEBlgR2Au6OyF+AWby4hbvU4xVtRF2x7OQ7a+QbOWKN+Rjp4lF/NOLZ
125        o0sZvw96MIJPM6IYVEFFAFrnTEhF6CSqdHgaWEeEzQXuc9EzlYv8VPdkwtHAOS4P8Fsw52A40Mc4
126        rRp5ICKzR2WhCC8hsrgqFaWlXRPfCfJtRIxCVGQWYFoAERCU9rY2AKqXAO/7qHFA7YIi4ccczgFw
127        U490G/7WV7/KZdm0/YVHxBwcka2jyEKI7K3KdQorarvqI4aIXAWcRQcv5ixBjgZFVBEUg2KJiyFM
128        i+p2EpWB3L+UJHbamPsfMo37uFoj/8RjKqkRmiqAfKcioPKjwoCCjWKIGALtBERmi5glFHGAgswC
129        ur4AoEO1YDReAvITaNVIfInEVWOzoJwaBqGSwCeuVucTMebUIsTzBCHCD9G63dH4tCh4m5qIELAE
130        ERRDRHbT/24AAhMch5rgqSDm4AIARpS1uZ3k4SqLEsUCnFoX84nLupPLaoN+/8xZ6kUBYr82wT9N
131        W7AlghgChnbwdlMICCgCgMLQmaiAcDAdqtJ1R0+ie4VQrNCFosh5TpjJSe6fVGWnqFrx/2Nwe7G0
132        233oqNIxL9D5iSIIiCLwenu8VwEy9ad4cTNL9AQMXqlaa/7uxqt7SiQcVCvijQGDUR1Nh4jRdvt3
133        BJdjgLPbISsKVwHbgSBA66gVgY2A252GSlQ90eRNECEP4MUd5AO3u5Els2Gtrjd2p5a81iSYZB7v
134        ssUKl6UBm0dkRECI0k4Ag8LsiiyhkAHvANsDGwIVBQRoH/cW+CiKORBtt70oYi1i/I2p8LsL8BLW
135        3FHLwwZZYkcMBlDM68rQsEMZCk4EFNkN2A0EhTOB44D3gWWYgC4HvB4RsI5gLJlEGj72q5pHrY1f
136        mmpuqiD7RB+lK7UYUWIMRCxDHU4mCA5IR/tT0PIcbQoTvBMR1BfEEEjTlIZXKaVm32DcByYYR0Y4
137        0ahWvIIRxWiEULSDT9zZBGUChpZrAIZLQsWAS4gKZXzFKCcaBWMUSNypwNq1PL7dlbZe0qgovK7I
138        i4q8oPCywl//szG08TrwZscquF773kvACwovgbwOhugjXaljoFl8Z4ysHdFTI4qLKOO9q46o8Jei
139        VswWFMoOGj6iToyKfKKwIMgTAmcJyvB48j9bRM4DlqaVwPr4BiUTEAzdJowyxvwlQhXARQSAclc2
140        a+H101rD10ZWulbsq4GE5qKOdOoc+5kaORM4i0lQpAIcDnyIcjC+USEGxpSgVq9+4JKskZg4a3v0
141        IPussxSdTGNw2jrJD7Zcpmg2iaKr9LrRqMpLWLsE8DrDQ1UXA15HZDppDq5p0Zum6TbUBmq4LJsO
142        +JEOro6j0xQjyrMzWNCs1/4Y210a21+El0blvdU+NwZCeFGNuQGRe4APgCotFWA+YtwK1d3QiAb/
143        j9EujBmXKL9U69UscRUncfaJE5Dd112G4RT1hmZpQuYM3+UJRYBoE8QYMA40gmrrKIKGCNaSaMHU
144        iQfvaeYBQBiG7LTKEgxndI/przbii001lR7HqnVXphmU3EdCFHLjEI1ojKQWyonQI4pGT72Rv2OE
145        r7qtrAaMYBiy1xpLMClSTk/Nm/6EZtAHUms3y1BKzvCHZESEMVqnXkRyHwjIAxbdLEnsqcBJTILz
146        xjApU46pnBe8fwF4pb+/8Ywv/DK9zbCKsfWXUBhf+A1dOX00S+xfgc3L3dmKWSn7iklDjthxbSaH
147        c7YCVIAfi6JYn5bHjTHTGmurQJXJ8C/um928G9zK4gAAAABJRU5ErkJggg==
148        ";
149
150        let url = format!("data:image/png;base64,{}", url);
151        let image = parse_image_url(&url).await.unwrap();
152        assert_eq!(image.dimensions(), (32, 32));
153    }
154}