139 lines
3.9 KiB
Rust
139 lines
3.9 KiB
Rust
use std::fs;
|
|
use serde::Deserialize;
|
|
use serde_json::{json, Value};
|
|
use std::error::Error;
|
|
use std::fs::File;
|
|
use std::io::BufReader;
|
|
use clap::{Parser, ValueEnum};
|
|
use std::path::PathBuf;
|
|
|
|
mod pixelfed;
|
|
pub mod image_description;
|
|
|
|
#[derive(Parser)]
|
|
#[command(name = "Pixelfed Image Bulk Uploader")]
|
|
#[command(version = "1.0")]
|
|
#[command(about = "Bulk uploads images to Pixelfed with image descriptions", long_about = None)]
|
|
#[command(version, about, long_about = None)]
|
|
struct Cli {
|
|
/// Image description mode
|
|
#[arg(short, long, default_value = "file")]
|
|
mode: Mode,
|
|
/// The title of the posting
|
|
#[arg(short, long)]
|
|
title: String,
|
|
/// The path to the file to read
|
|
#[arg(short, long)]
|
|
image_path: String,
|
|
/// Sets a custom config file
|
|
#[arg(short, long, value_name = "FILE")]
|
|
config: Option<String>,
|
|
}
|
|
|
|
|
|
#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]
|
|
enum Mode {
|
|
/// Use ChatGTP
|
|
ChatGPT,
|
|
/// Taking from a file
|
|
File,
|
|
/// Local LLM
|
|
Local,
|
|
}
|
|
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
struct Config {
|
|
pixelfed_url: String,
|
|
pixelfed_access_token: String,
|
|
pixelfed_visibility: String, // Should be "unlisted"
|
|
pixelfed_default_text: String,
|
|
pixelfed_batch_size: usize,
|
|
openai_api_key: String,
|
|
openai_api_url: String,
|
|
openai_model: String,
|
|
ollama_api_key: String,
|
|
ollama_api_url: String,
|
|
ollama_model: String,
|
|
caption_extension: String,
|
|
|
|
}
|
|
|
|
fn load_config(config_file: String) -> Result<Config, Box<dyn Error>> {
|
|
//let config_str = fs::read_to_string("config.json")?;
|
|
// Open the file in read-only mode with buffer.
|
|
|
|
let file = File::open(PathBuf::from(config_file))?;
|
|
let reader = BufReader::new(file);
|
|
|
|
// Read the JSON contents of the file as an instance of `User`.
|
|
|
|
|
|
let config: Config = serde_json::from_reader(reader)?;
|
|
Ok(config)
|
|
}
|
|
|
|
|
|
// get all the JPEG files from the give directory
|
|
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 main() -> Result<(), Box<dyn Error>> {
|
|
let args = Cli::parse();
|
|
|
|
//if args.len() < 2 || args.len() > 3 {
|
|
// eprintln!("Usage: {} <directory> [--title <title>]", args[0]);
|
|
// eprintln!("Usage: {} <directory> -ready [--title <title>]", args[0]);
|
|
// std::process::exit(1);
|
|
//}
|
|
|
|
let title = args.title;
|
|
let mut my_config: String;
|
|
|
|
|
|
|
|
match args.config {
|
|
Some(configstring) => { my_config = configstring},
|
|
None => {my_config = "config.json".to_string()},
|
|
}
|
|
println!("effictive config file: {}", my_config);
|
|
|
|
|
|
let config = load_config(my_config).unwrap();
|
|
println!("Config OK? true");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// get list of all the images in the gives path
|
|
let images = get_jpeg_files(&args.image_path);
|
|
println!("Images empty? {}", images.is_empty().to_string());
|
|
|
|
// knowing now the total number of images, calculate the number of batches
|
|
let total_batches = (images.len() + config.pixelfed_batch_size - 1) / config.pixelfed_batch_size;
|
|
println!("Found a total of {} images to upload. Will take {} batches", images.len(), total_batches);
|
|
|
|
// now iterate over all images in batches of batch_size
|
|
for (i, chunk) in images.chunks(config.pixelfed_batch_size).enumerate() {
|
|
println!("{}", i.clone());
|
|
pixelfed::bulk_upload_images(&config, chunk, i + 1, total_batches, &title, &args.mode);
|
|
}
|
|
println!("All images uploaded successfully.");
|
|
Ok(())
|
|
}
|