AstralRinth/src/file_hosting/authorization.rs

use crate::file_hosting::FileHostingError;
use base64::encode;
use serde::{Deserialize, Serialize};
use serde_json::json;

#[derive(Serialize, Deserialize, Debug, Clone)]
#[serde(rename_all = "camelCase")]
pub struct AuthorizationPermissions {
    bucket_id: Option<String>,
    bucket_name: Option<String>,
    capabilities: Vec<String>,
    name_prefix: Option<String>,
}

#[derive(Serialize, Deserialize, Debug, Clone)]
#[serde(rename_all = "camelCase")]
pub struct AuthorizationData {
    pub absolute_minimum_part_size: i32,
    pub account_id: String,
    pub allowed: AuthorizationPermissions,
    pub api_url: String,
    pub authorization_token: String,
    pub download_url: String,
    pub recommended_part_size: i32,
}

#[derive(Serialize, Deserialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct UploadUrlData {
    pub bucket_id: String,
    pub upload_url: String,
    pub authorization_token: String,
}

pub async fn authorize_account(
    key_id: String,
    application_key: String,
) -> Result<AuthorizationData, FileHostingError> {
    let combined_key = format!("{}:{}", key_id, application_key);
    let formatted_key = format!("Basic {}", encode(combined_key));

    Ok(reqwest::Client::new()
        .get("https://api.backblazeb2.com/b2api/v2/b2_authorize_account")
        .header(reqwest::header::CONTENT_TYPE, "application/json")
        .header(reqwest::header::AUTHORIZATION, formatted_key)
        .send()
        .await?
        .json()
        .await?)
}

pub async fn get_upload_url(
    authorization_data: AuthorizationData,
    bucket_id: String,
) -> Result<UploadUrlData, FileHostingError> {
    Ok(reqwest::Client::new()
        .post(&format!("{}/b2api/v2/b2_get_upload_url", authorization_data.api_url).to_string())
        .header(reqwest::header::CONTENT_TYPE, "application/json")
        .header(
            reqwest::header::AUTHORIZATION,
            authorization_data.authorization_token,
        )
        .body(
            json!({
                "bucketId": bucket_id,
            })
            .to_string(),
        )
        .send()
        .await?
        .json()
        .await?)
}