insta-import-pixelfed/src/pixelfed.rs

use reqwest::{self};
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 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::blocking::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.clone(),
    };
    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)?;
                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)?;
                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)?;
                println!("Description generated by OLLAMA: {}", description.clone());
                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 = reqwest::blocking::multipart::Form::new().text("description", description.clone())
            .file("file", image_path)?;
        // upload the form to Pixelfed
        let res = client.post(&url)
            .bearer_auth(&pxl_config.pixelfed_access_token)
            .multipart(form)
            .send();

        let res_json: serde_json::Value = res.unwrap().json()?;
        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", pxl_config.pixelfed_url);
        let post_text = format_post_text(&pxl_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": pxl_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());
        let res = client.post(&post_url)
            .bearer_auth(&pxl_config.pixelfed_access_token)
            .json(&body)
            .send();
    }
    Ok(())
}