openai_dive 1.4.3

use serde_json::Value;

use crate::v1::{
    api::Client,
    error::APIError,
    helpers::format_response,
    resources::upload::{
        AddPartParameters, CompleteUploadParameters, CreateUploadParameters, Upload, UploadPart,
    },
};

pub struct Uploads<'a> {
    pub client: &'a Client,
}

impl Client {
    /// Allows you to upload large files in multiple parts.
    pub fn uploads(&self) -> Uploads<'_> {
        Uploads { client: self }
    }
}

impl Uploads<'_> {
    /// Creates an intermediate Upload object that you can add Parts to.
    pub async fn create(&self, parameters: CreateUploadParameters) -> Result<Upload, APIError> {
        let response = self.client.post("/uploads", &parameters, None).await?;

        let response: Upload = format_response(response.data)?;

        Ok(response)
    }

    /// Adds a Part to an Upload object. A Part represents a chunk of bytes from the file you are trying to upload.
    pub async fn add_part(
        &self,
        id: &str,
        parameters: AddPartParameters,
    ) -> Result<UploadPart, APIError> {
        let mut form = reqwest::multipart::Form::new();

        let file_part = parameters.data.into_part().await?;
        form = form.part("data", file_part);

        let response = self
            .client
            .post_with_form(&format!("/uploads/{id}/parts"), form)
            .await?;

        let response: UploadPart = format_response(response)?;

        Ok(response)
    }

    /// Completes the Upload.
    pub async fn complete(
        &self,
        id: &str,
        parameters: CompleteUploadParameters,
    ) -> Result<Upload, APIError> {
        let response = self
            .client
            .post(&format!("/uploads/{id}/complete"), &parameters, None)
            .await?;

        let response: Upload = format_response(response.data)?;

        Ok(response)
    }

    /// Cancels the Upload.
    pub async fn cancel(&self, id: &str) -> Result<Upload, APIError> {
        let response = self
            .client
            .post(&format!("/uploads/{id}/cancel"), &Value::Null, None)
            .await?;

        let response: Upload = format_response(response.data)?;

        Ok(response)
    }
}