imgup 4.0.1

Upload images via APIs
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

use anyhow::{Context, Result};
use base64::Engine as _;
use reqwest::Client;
use tracing::debug;

use super::response_text;

pub const API_URL: &str = "https://api.postimage.org/1/upload";

/// Upload image bytes to postimages.org.
///
/// Requires API key.
pub async fn upload(client: &Client, data: Vec<u8>, url: &str, key: &str) -> Result<String> {
    let image_b64 = base64::engine::general_purpose::STANDARD.encode(&data);
    // o/m are tokens required by the API.
    let resp = client
        .post(url)
        .form(&[
            ("key", key),
            ("image", image_b64.as_str()),
            ("name", "img"),
            ("type", "jpg"),
            ("version", "1.0.1"),
            ("o", "2b819584285c102318568238c7d4a4c7"),
            ("m", "fb733cccce28e7db3ff9f17d7ccff3d1"),
        ])
        .send()
        .await
        .context("failed to send request to postimages")?;

    let body = response_text(resp, "postimages").await?;
    extract_hotlink(&body)
}

/// Extract the direct image URL from the XML response.
fn extract_hotlink(xml: &str) -> Result<String> {
    xml.split_once("<hotlink>")
        .and_then(|(_, rest)| rest.split_once("</hotlink>"))
        .map(|(url, _)| url.to_owned())
        .inspect(|url| debug!("postimages hotlink: {url}"))
        .ok_or_else(|| {
            debug!("Response text:\n{xml}");
            anyhow::anyhow!("hotlink not found in postimages response")
        })
}

#[cfg(test)]
#[expect(clippy::unwrap_used)]
mod tests {
    use wiremock::matchers::method;
    use wiremock::{Mock, MockServer, ResponseTemplate};

    use super::*;

    const XML_RESPONSE: &str = r#"<?xml version="1.0" encoding="utf-8"?>
<data success="1" status="200">
  <links>
    <hotlink>https://i.postimg.cc/abc123/image.jpg</hotlink>
  </links>
</data>"#;

    #[tokio::test]
    async fn test_upload_success() {
        let mock_server = MockServer::start().await;

        Mock::given(method("POST"))
            .respond_with(ResponseTemplate::new(200).set_body_string(XML_RESPONSE))
            .mount(&mock_server)
            .await;

        let client = Client::new();
        let url = upload(&client, vec![1, 2, 3], &mock_server.uri(), "test_key")
            .await
            .unwrap();
        assert_eq!(url, "https://i.postimg.cc/abc123/image.jpg");
    }

    #[test]
    fn test_extract_hotlink() {
        let url = extract_hotlink(XML_RESPONSE).unwrap();
        assert_eq!(url, "https://i.postimg.cc/abc123/image.jpg");
    }
}