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>
96 lines
2.8 KiB
Rust
96 lines
2.8 KiB
Rust
use super::{DeleteFileData, FileHost, FileHostingError, UploadFileData};
|
|
use async_trait::async_trait;
|
|
use bytes::Bytes;
|
|
use reqwest::Response;
|
|
use serde::Deserialize;
|
|
use sha2::Digest;
|
|
|
|
mod authorization;
|
|
mod delete;
|
|
mod upload;
|
|
|
|
pub struct BackblazeHost {
|
|
upload_url_data: authorization::UploadUrlData,
|
|
authorization_data: authorization::AuthorizationData,
|
|
}
|
|
|
|
impl BackblazeHost {
|
|
pub async fn new(key_id: &str, key: &str, bucket_id: &str) -> Self {
|
|
let authorization_data = authorization::authorize_account(key_id, key).await.unwrap();
|
|
let upload_url_data = authorization::get_upload_url(&authorization_data, bucket_id)
|
|
.await
|
|
.unwrap();
|
|
|
|
BackblazeHost {
|
|
upload_url_data,
|
|
authorization_data,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl FileHost for BackblazeHost {
|
|
async fn upload_file(
|
|
&self,
|
|
content_type: &str,
|
|
file_name: &str,
|
|
file_bytes: Bytes,
|
|
) -> Result<UploadFileData, FileHostingError> {
|
|
let content_sha512 = format!("{:x}", sha2::Sha512::digest(&file_bytes));
|
|
|
|
let upload_data =
|
|
upload::upload_file(&self.upload_url_data, content_type, file_name, file_bytes).await?;
|
|
Ok(UploadFileData {
|
|
file_id: upload_data.file_id,
|
|
file_name: upload_data.file_name,
|
|
content_length: upload_data.content_length,
|
|
content_sha512,
|
|
content_sha1: upload_data.content_sha1,
|
|
content_md5: upload_data.content_md5,
|
|
content_type: upload_data.content_type,
|
|
upload_timestamp: upload_data.upload_timestamp,
|
|
})
|
|
}
|
|
|
|
/*
|
|
async fn upload_file_streaming(
|
|
&self,
|
|
content_type: &str,
|
|
file_name: &str,
|
|
stream: reqwest::Body
|
|
) -> Result<UploadFileData, FileHostingError> {
|
|
use futures::stream::StreamExt;
|
|
|
|
let mut data = Vec::new();
|
|
while let Some(chunk) = stream.next().await {
|
|
data.extend_from_slice(&chunk.map_err(|e| FileHostingError::Other(e))?);
|
|
}
|
|
self.upload_file(content_type, file_name, data).await
|
|
}
|
|
*/
|
|
|
|
async fn delete_file_version(
|
|
&self,
|
|
file_id: &str,
|
|
file_name: &str,
|
|
) -> Result<DeleteFileData, FileHostingError> {
|
|
let delete_data =
|
|
delete::delete_file_version(&self.authorization_data, file_id, file_name).await?;
|
|
Ok(DeleteFileData {
|
|
file_id: delete_data.file_id,
|
|
file_name: delete_data.file_name,
|
|
})
|
|
}
|
|
}
|
|
|
|
pub async fn process_response<T>(response: Response) -> Result<T, FileHostingError>
|
|
where
|
|
T: for<'de> Deserialize<'de>,
|
|
{
|
|
if response.status().is_success() {
|
|
Ok(response.json().await?)
|
|
} else {
|
|
Err(FileHostingError::BackblazeError(response.json().await?))
|
|
}
|
|
}
|