use anyhow::{Context, Result};
use reqwest::Client;
use reqwest::multipart::{Form, Part};
use serde::Deserialize;
use super::parse_json;
use crate::image::get_image_ext;
pub const API_URL: &str = "https://beeimg.com/api/upload/file/json/";
#[derive(Deserialize)]
struct Response {
files: Files,
}
#[derive(Deserialize)]
struct Files {
url: String,
}
pub async fn upload(client: &Client, data: Vec<u8>, url: &str) -> Result<String> {
let ext = get_image_ext(&data)?;
let name = format!("img.{}", ext.extensions_str()[0]);
let content_type = ext.to_mime_type();
let form = Form::new().part(
"file",
Part::bytes(data)
.file_name(name)
.mime_str(content_type)
.context("invalid mime type")?,
);
let resp = client
.post(url)
.multipart(form)
.send()
.await
.context("failed to send request to beeimg")?;
let resp: Response = parse_json(resp, "beeimg").await?;
Ok(format!("https:{}", resp.files.url))
}
#[cfg(test)]
#[expect(clippy::unwrap_used)]
mod tests {
use wiremock::matchers::method;
use wiremock::{Mock, MockServer, ResponseTemplate};
use super::*;
#[tokio::test]
async fn test_upload_success() {
let mock_server = MockServer::start().await;
Mock::given(method("POST"))
.respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
"files": {"url": "//beeimg.com/images/x80784797021.png"}
})))
.mount(&mock_server)
.await;
let png = crate::image::create_test_png();
let client = Client::new();
let url = upload(&client, png, &mock_server.uri()).await.unwrap();
assert_eq!(url, "https://beeimg.com/images/x80784797021.png");
}
}