spacebar/account
6
0
Fork 0
Code Issues 1 Pull requests 7 Projects Releases Packages 1 Wiki Activity Actions

praying again

Browse source
This commit is contained in:
Andrea Toska 2025-04-27 11:39:33 +02:00
commit ca4354ba0a
No known key found for this signature in database
GPG key ID: 5B3C83807CCBE9A2
1 changed files with 39 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

82
src/nnid/people.rs
View file

@ -1,5 +1,6 @@
use std::env; use std::env;
use std::io::Cursor; use std::fs;
use std::path::Path;
use chrono::{NaiveDate, NaiveDateTime}; use chrono::{NaiveDate, NaiveDateTime};
use gxhash::{gxhash32, gxhash64}; use gxhash::{gxhash32, gxhash64};
use minio::s3::builders::{ObjectContent}; use minio::s3::builders::{ObjectContent};
@ -424,39 +425,45 @@ pub async fn change_mii(
Ok(()) Ok(())
} }
fn build_object_content(data: Vec<u8>) -> ObjectContent {
ObjectContent::SegmentedBytes(SegmentedBytes::new_from_bytes(data))
}
pub async fn generate_mii_images(client: Arc<Client>, bucket: &str, pid: i32, mii_data: &str) { pub async fn generate_mii_images(client: Arc<Client>, bucket: &str, pid: i32, mii_data: &str) {
let user_mii_key = format!("mii/{}", pid); let user_mii_key = format!("mii/{}", pid);
println!("Starting Mii image generation for PID {}", pid); println!("Starting Mii image generation for PID {}", pid);
if let Some(png_data) = get_image_png(mii_data).await { async fn save_upload_delete(
println!("Fetched PNG for PID {}, uploading...", pid); client: &Client,
let content = ObjectContent::from(Cursor::new(png_data.clone())); bucket: &str,
match client.put_object_content( key: &str,
bucket, data: &[u8],
&format!("{}/normal_face.png", user_mii_key), ) -> Result<(), Box<dyn std::error::Error>> {
content let temp_path = format!("/tmp/{}", key.replace("/", "_"));
).send().await { fs::write(&temp_path, data)?;
Ok(_) => println!("Uploaded normal_face.png for PID {}", pid),
Err(e) => println!("Failed to upload normal_face.png for PID {}: {:?}", pid, e), let content = ObjectContent::from(Path::new(&temp_path));
} client.put_object_content(bucket, key, content).send().await?;
fs::remove_file(&temp_path)?;
Ok(())
} }
// Upload normal face images
if let Some(png_data) = get_image_png(mii_data).await {
println!("Fetched PNG for PID {}, uploading...", pid);
if let Err(e) = save_upload_delete(&client, bucket, &format!("{}/normal_face.png", user_mii_key), &png_data).await {
println!("Failed to upload normal_face.png for PID {}: {:?}", pid, e);
} else {
println!("Uploaded normal_face.png for PID {}", pid);
}
} else {
println!("Failed to fetch PNG for PID {}", pid);
}
if let Some(tga_data) = get_image_tga(mii_data).await { if let Some(tga_data) = get_image_tga(mii_data).await {
println!("Fetched TGA for PID {}, uploading...", pid); println!("Fetched TGA for PID {}, uploading...", pid);
let object_content = ObjectContent::from(Cursor::new(tga_data.clone())); if let Err(e) = save_upload_delete(&client, bucket, &format!("{}/standard.tga", user_mii_key), &tga_data).await {
match client.put_object_content( println!("Failed to upload standard.tga for PID {}: {:?}", pid, e);
bucket, } else {
&format!("{}/standard.tga", user_mii_key), println!("Uploaded standard.tga for PID {}", pid);
object_content
).send().await {
Ok(_) => println!("Uploaded standard.tga for PID {}", pid),
Err(e) => println!("Failed to upload standard.tga for PID {}: {:?}", pid, e),
} }
} else { } else {
println!("Failed to fetch TGA for PID {}", pid); println!("Failed to fetch TGA for PID {}", pid);
@ -480,14 +487,10 @@ pub async fn generate_mii_images(client: Arc<Client>, bucket: &str, pid: i32, mi
if let Ok(resp) = reqwest::get(&url).await { if let Ok(resp) = reqwest::get(&url).await {
if let Ok(bytes) = resp.bytes().await { if let Ok(bytes) = resp.bytes().await {
println!("Fetched expression '{}', uploading...", expression); println!("Fetched expression '{}', uploading...", expression);
let object_content = ObjectContent::Bytes(bytes.to_vec()); if let Err(e) = save_upload_delete(&client, bucket, &format!("{}/{}.png", user_mii_key, expression), &bytes).await {
match client.put_object_content( println!("Failed to upload {}.png for PID {}: {:?}", expression, pid, e);
bucket, } else {
&format!("{}/{}.png", user_mii_key, expression), println!("Uploaded {}.png for PID {}", expression, pid);
object_content
).send().await {
Ok(_) => println!("Uploaded {}.png for PID {}", expression, pid),
Err(e) => println!("Failed to upload {}.png for PID {}: {:?}", expression, pid, e),
} }
} else { } else {
println!("Failed to read bytes for expression '{}' for PID {}", expression, pid); println!("Failed to read bytes for expression '{}' for PID {}", expression, pid);
@ -506,14 +509,10 @@ pub async fn generate_mii_images(client: Arc<Client>, bucket: &str, pid: i32, mi
if let Ok(resp) = reqwest::get(&body_url).await { if let Ok(resp) = reqwest::get(&body_url).await {
if let Ok(bytes) = resp.bytes().await { if let Ok(bytes) = resp.bytes().await {
println!("Fetched body image, uploading..."); println!("Fetched body image, uploading...");
let object_content = ObjectContent::Bytes(bytes.to_vec()); if let Err(e) = save_upload_delete(&client, bucket, &format!("{}/body.png", user_mii_key), &bytes).await {
match client.put_object_content( println!("Failed to upload body.png for PID {}: {:?}", pid, e);
bucket, } else {
&format!("{}/body.png", user_mii_key), println!("Uploaded body.png for PID {}", pid);
object_content
).send().await {
Ok(_) => println!("Uploaded body.png for PID {}", pid),
Err(e) => println!("Failed to upload body.png for PID {}: {:?}", pid, e),
} }
} else { } else {
println!("Failed to read body image bytes for PID {}", pid); println!("Failed to read body image bytes for PID {}", pid);
@ -527,3 +526,4 @@ pub async fn generate_mii_images(client: Arc<Client>, bucket: &str, pid: i32, mi