tripo-api 0.1.4

Unofficial async Rust client for the Tripo 3D Generation API
Documentation 
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

//! `/upload` multipart helper.

use std::path::Path;

use serde::Deserialize;

use crate::client::Client;
use crate::error::{Error, Result};
use crate::types::UploadedFile;

#[derive(Deserialize)]
struct ImageTokenBody {
    image_token: uuid::Uuid,
}

impl Client {
    /// Upload a local file and return a token usable as `ImageInput::FileToken`.
    #[tracing::instrument(skip(self), fields(path = %path.as_ref().display()))]
    pub async fn upload_file(&self, path: impl AsRef<Path>) -> Result<UploadedFile> {
        let path = path.as_ref();
        let file_name = path
            .file_name()
            .and_then(|s| s.to_str())
            .ok_or_else(|| {
                Error::Io(std::io::Error::new(
                    std::io::ErrorKind::InvalidInput,
                    "non-utf8 filename",
                ))
            })?
            .to_string();
        let bytes = tokio::fs::read(path).await?;
        let part = reqwest::multipart::Part::bytes(bytes).file_name(file_name);
        let form = reqwest::multipart::Form::new().part("file", part);

        let url = self.url(&["upload"]);
        let resp = self
            .http
            .post(url)
            .headers(self.region_headers())
            .multipart(form)
            .send()
            .await?;
        let status = resp.status();
        let body = resp.bytes().await?;
        if !status.is_success() {
            return Err(crate::envelope::map_http_error(status, &body));
        }
        let env: crate::envelope::Envelope<ImageTokenBody> = serde_json::from_slice(&body)?;
        let data = env.into_result()?;
        Ok(UploadedFile {
            file_token: data.image_token,
        })
    }
}