102 lines
4.5 KiB
Rust
102 lines
4.5 KiB
Rust
use reqwest::{self, multipart};
|
|
use std::error::Error;
|
|
|
|
struct PixelfedConfig {
|
|
pixelfed_url: String,
|
|
pixelfed_access_token: String,
|
|
pixelfed_visibility: String, // Should be "unlisted"
|
|
pixelfed_default_text: String,
|
|
pixelfed_batch_size: usize,
|
|
}
|
|
|
|
fn format_post_text(template: &str, batch_num: usize, total_batches: usize, title: &str) -> String {
|
|
template
|
|
.replace("@batch@", &format!("Batch {} out of {}", batch_num, total_batches))
|
|
.replace("@title@", title)
|
|
}
|
|
|
|
// upload a single image to pixelfed
|
|
pub async fn bulk_upload_images(config: &super::Config, images: &[String], batch_num: usize, total_batches: usize, title: &str, caption_mode: &super::Mode) -> Result<(), Box<dyn Error>> {
|
|
let mut media_ids = Vec::new();
|
|
let mut media_descriptions = Vec::new();
|
|
let client = reqwest::Client::new();
|
|
let pxl_config = PixelfedConfig {
|
|
pixelfed_url: config.pixelfed_url.clone(),
|
|
pixelfed_access_token: config.pixelfed_access_token.clone(),
|
|
pixelfed_visibility: config.pixelfed_visibility.clone(),
|
|
pixelfed_default_text: config.pixelfed_default_text.clone(),
|
|
pixelfed_batch_size: config.pixelfed_batch_size,
|
|
};
|
|
let url = format!("{}/api/v1/media", config.pixelfed_url.clone());
|
|
|
|
for image_path in images {
|
|
let description: String;
|
|
println!("Uploading image {}", image_path.to_string());
|
|
// get image description depending on the caption_mode
|
|
match caption_mode {
|
|
super::Mode::ChatGPT => {
|
|
let im_config = super::image_description::ChatGPTConfig {
|
|
openai_model: config.openai_model.clone(),
|
|
openai_api_key: config.openai_api_key.clone(),
|
|
openai_api_url: config.openai_api_url.clone(),
|
|
};
|
|
description = super::image_description::get_description_from_chatgpt(image_path.to_string(), im_config).await?;
|
|
media_descriptions.push(description.clone());
|
|
},
|
|
super::Mode::File => {
|
|
let im_config = super::image_description::FileConfig {
|
|
caption_extension: config.caption_extension.clone(),
|
|
};
|
|
println!("Fetching image description from File for {}", image_path.to_string());
|
|
description = super::image_description::get_description_from_file(image_path.to_string(), im_config).await?;
|
|
media_descriptions.push(description.clone());
|
|
},
|
|
super::Mode::Local => {
|
|
let im_config = super::image_description::OllamaConfig {
|
|
ollama_api_key: config.ollama_api_key.clone(),
|
|
ollama_api_url: config.ollama_api_url.clone(),
|
|
ollama_model: config.ollama_model.clone(),
|
|
};
|
|
println!("Fetching image description from OLLAMA for {}", image_path.to_string());
|
|
description = super::image_description::get_description_from_ollama(image_path.to_string(), im_config).await?;
|
|
media_descriptions.push(description.clone());
|
|
},
|
|
}
|
|
|
|
println!("Uploading image {}", image_path.to_string());
|
|
|
|
// construct the upload form for Pixelfed Upload of a single image including image description
|
|
let form = multipart::Form::new()
|
|
.file("file", image_path).await?;
|
|
// upload the form to Pixelfed
|
|
let res = client.post(&url)
|
|
.bearer_auth(&config.pixelfed_access_token)
|
|
.multipart(form)
|
|
.send()
|
|
.await?;
|
|
|
|
let res_json: serde_json::Value = res.json().await?;
|
|
let image_id = res_json["id"].as_str().unwrap().to_string();
|
|
media_ids.push(image_id);
|
|
}
|
|
|
|
if !media_ids.is_empty() {
|
|
let post_url = format!("{}/api/v1/statuses", config.pixelfed_url);
|
|
let post_text = format_post_text(&config.pixelfed_default_text, batch_num, total_batches, title);
|
|
let body = serde_json::json!({
|
|
"status": post_text,
|
|
"media_ids": media_ids,
|
|
"alt_texts": media_descriptions,
|
|
"visibility": config.pixelfed_visibility,
|
|
});
|
|
println!("Body: \n{}", body.to_string());
|
|
println!("MediaIDs: {}", media_ids.len());
|
|
println!("Alt_texts: {}", media_descriptions.len());
|
|
println!("Posting batch {} out of {} with media {}", batch_num, total_batches, media_ids.len());
|
|
client.post(&post_url)
|
|
.bearer_auth(&config.pixelfed_access_token)
|
|
.json(&body)
|
|
.send().await?;
|
|
}
|
|
Ok(())
|
|
}
|