74 lines
2.2 KiB
Rust
74 lines
2.2 KiB
Rust
|
use aws_config::{BehaviorVersion, Region};
|
||
|
|
use aws_sdk_s3::{config::Credentials, Client, Config};
|
||
|
|
use std::env;
|
||
|
|
|
||
|
|
#[derive(Debug)]
|
||
|
|
pub struct S3ClientConfig {
|
||
|
|
pub access_key: String,
|
||
|
|
secret_key: String,
|
||
|
|
endpoint: String,
|
||
|
|
pub bucket: String,
|
||
|
|
region: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl S3ClientConfig {
|
||
|
|
pub fn from_env() -> Result<Self, Box<dyn std::error::Error>> {
|
||
|
|
Ok(S3ClientConfig {
|
||
|
|
access_key: env::var("LINODE_ACCESS_KEY")
|
||
|
|
.map_err(|_| "LINODE_ACCESS_KEY environment variable not set")?,
|
||
|
|
secret_key: env::var("LINODE_SECRET_KEY")
|
||
|
|
.map_err(|_| "LINODE_SECRET_KEY environment variable not set")?,
|
||
|
|
endpoint: env::var("LINODE_ENDPOINT")
|
||
|
|
.unwrap_or_else(|_| "us-ord-1.linodeobjects.com".to_string()),
|
||
|
|
bucket: env::var("LINODE_BUCKET")
|
||
|
|
.map_err(|_| "LINODE_BUCKET environment variable not set")?,
|
||
|
|
region: env::var("LINODE_REGION").unwrap_or_else(|_| "us-ord".to_string()),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn create_s3_client(
|
||
|
|
config: &S3ClientConfig,
|
||
|
|
) -> Result<Client, Box<dyn std::error::Error>> {
|
||
|
|
let credentials = Credentials::new(
|
||
|
|
&config.access_key,
|
||
|
|
&config.secret_key,
|
||
|
|
None,
|
||
|
|
None,
|
||
|
|
"linode-object-storage",
|
||
|
|
);
|
||
|
|
|
||
|
|
let s3_config = Config::builder()
|
||
|
|
.behavior_version(BehaviorVersion::latest())
|
||
|
|
.region(Region::new(config.region.clone()))
|
||
|
|
.endpoint_url(format!("https://{}", config.endpoint))
|
||
|
|
.credentials_provider(credentials)
|
||
|
|
.force_path_style(false)
|
||
|
|
.build();
|
||
|
|
|
||
|
|
Ok(Client::from_conf(s3_config))
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn upload(
|
||
|
|
client: &Client,
|
||
|
|
bucket: &str,
|
||
|
|
key: &str,
|
||
|
|
content: &str,
|
||
|
|
) -> Result<(), Box<dyn std::error::Error>> {
|
||
|
|
println!("Uploading to Linode Object Storage...");
|
||
|
|
println!("Bucket: {}", bucket);
|
||
|
|
|
||
|
|
let put_object_req = client
|
||
|
|
.put_object()
|
||
|
|
.bucket(bucket)
|
||
|
|
.key(key)
|
||
|
|
.body(content.as_bytes().to_vec().into())
|
||
|
|
.acl(aws_sdk_s3::types::ObjectCannedAcl::PublicRead)
|
||
|
|
.content_type("application/rss+xml")
|
||
|
|
.send()
|
||
|
|
.await?;
|
||
|
|
|
||
|
|
println!("Upload successful! ETag: {:?}", put_object_req.e_tag());
|
||
|
|
Ok(())
|
||
|
|
}
|