mistralrs_server_core/
util.rs1use image::DynamicImage;
4use tokio::{
5 fs::{self, File},
6 io::AsyncReadExt,
7};
8
9pub 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 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 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 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 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 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 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 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}