You've already forked pages
forked from didirus/AstralRinth
13187de97d
* typos :help_me: * (part 1/?) massive cleanup to make the code more Rust-ic and cut down heap allocations. * (part 2/?) massive cleanup to make the code more Rust-ic and cut down heap allocations. * (part 3/?) cut down some pretty major heap allocations here - more Bytes and BytesMuts, less Vec<u8>s also I don't really understand why you need to `to_vec` when you don't really use it again afterwards * (part 4/?) deduplicate error handling in backblaze logic * (part 5/?) fixes, cleanups, refactors, and reformatting * (part 6/?) cleanups and refactors * remove loads of `as_str` in types that already are `Display` * Revert "remove loads of `as_str` in types that already are `Display`" This reverts commit 4f974310cfb167ceba03001d81388db4f0fbb509. * reformat and move routes util to the util module * use streams * Run prepare + formatting issues Co-authored-by: Jai A <jaiagr+gpg@pm.me> Co-authored-by: Geometrically <18202329+Geometrically@users.noreply.github.com>
72 lines
2.1 KiB
Rust
72 lines
2.1 KiB
Rust
use crate::file_hosting::FileHostingError;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
#[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, Clone)]
|
|
#[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: &str,
|
|
application_key: &str,
|
|
) -> Result<AuthorizationData, FileHostingError> {
|
|
let combined_key = format!("{}:{}", key_id, application_key);
|
|
let formatted_key = format!("Basic {}", base64::encode(combined_key));
|
|
|
|
let response = 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?;
|
|
|
|
super::process_response(response).await
|
|
}
|
|
|
|
pub async fn get_upload_url(
|
|
authorization_data: &AuthorizationData,
|
|
bucket_id: &str,
|
|
) -> Result<UploadUrlData, FileHostingError> {
|
|
let response = 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(
|
|
serde_json::json!({
|
|
"bucketId": bucket_id,
|
|
})
|
|
.to_string(),
|
|
)
|
|
.send()
|
|
.await?;
|
|
|
|
super::process_response(response).await
|
|
}
|