initial working version

This commit is contained in:
Falko Zurell 2025-01-17 15:25:44 +01:00
commit c0d428297e
3 changed files with 102 additions and 0 deletions

1
.gitignore vendored Normal file
View file

@ -0,0 +1 @@
/target

10
Cargo.toml Normal file
View file

@ -0,0 +1,10 @@
[package]
name = "pixelfed_batch_uploader"
version = "0.1.0"
edition = "2021"
[dependencies]
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
reqwest = { version = "0.11", features = ["blocking", "json", "multipart"] }
tokio = { version = "1", features = ["full"] }

91
src/main.rs Normal file
View file

@ -0,0 +1,91 @@
use std::fs;
use std::path::Path;
use serde::{Deserialize, Serialize};
use reqwest::blocking::Client;
use std::error::Error;
#[derive(Debug, Deserialize)]
struct Config {
pixelfed_url: String,
access_token: String,
visibility: String, // Should be "unlisted"
default_text: String,
}
fn load_config() -> Result<Config, Box<dyn Error>> {
let config_str = fs::read_to_string("config.json")?;
let config: Config = serde_json::from_str(&config_str)?;
Ok(config)
}
fn get_jpeg_files(directory: &str) -> Vec<String> {
let mut images = Vec::new();
if let Ok(entries) = fs::read_dir(directory) {
for entry in entries.flatten() {
let path = entry.path();
if let Some(ext) = path.extension() {
if ext.eq_ignore_ascii_case("jpg") || ext.eq_ignore_ascii_case("jpeg") {
images.push(path.to_string_lossy().to_string());
}
}
}
}
images
}
fn upload_images_batch(client: &Client, config: &Config, images: &[String], batch_num: usize, total_batches: usize) -> Result<(), Box<dyn Error>> {
let url = format!("{}/api/v1/media", config.pixelfed_url);
let mut media_ids = Vec::new();
for image_path in images {
let form = reqwest::blocking::multipart::Form::new()
.file("file", image_path)?;
let res = client.post(&url)
.bearer_auth(&config.access_token)
.multipart(form)
.send()?;
let json: serde_json::Value = res.json()?;
if let Some(id) = json["id"].as_str() {
media_ids.push(id.to_string());
}
}
if !media_ids.is_empty() {
let post_url = format!("{}/api/v1/statuses", config.pixelfed_url);
let post_text = format!("{} (Batch {} out of {})", config.default_text, batch_num, total_batches);
let body = serde_json::json!({
"status": post_text,
"media_ids": media_ids,
"visibility": config.visibility,
});
client.post(&post_url)
.bearer_auth(&config.access_token)
.json(&body)
.send()?;
}
Ok(())
}
fn main() -> Result<(), Box<dyn Error>> {
let args: Vec<String> = std::env::args().collect();
if args.len() < 2 {
eprintln!("Usage: {} <directory>", args[0]);
std::process::exit(1);
}
let config = load_config()?;
let images = get_jpeg_files(&args[1]);
let client = Client::new();
let total_batches = (images.len() + 19) / 20;
for (i, chunk) in images.chunks(20).enumerate() {
upload_images_batch(&client, &config, chunk, i + 1, total_batches)?;
}
println!("All images uploaded successfully.");
Ok(())
}