bot-mastodon-image-uploader/src/main.rs

use async_std;
use log;
use mastodon_async::entities::visibility::Visibility;
use mastodon_async::helpers::{cli, toml as masto_toml};
use mastodon_async::prelude::*;
use reqwest;
use serde::Deserialize;
use std::collections::HashSet;
use std::io::{Cursor, Write};
use std::process::exit;
use std::time::Duration;
use toml;

const CONFIG_FILENAME: &str = "config.toml";

type DynResult<T> = Result<T, Box<dyn std::error::Error>>;

#[derive(Deserialize)]
struct Config {
  bot: Bot,
  files: Files,
  errors: Errors,
}

#[derive(Deserialize)]
struct Bot {
  name: String,
  instance: String,
  bio: String,
}

#[derive(Deserialize)]
struct Files {
  posted: String,
  urls: String,
  tempfile: String,
}

#[derive(Deserialize)]
struct Errors {
  out_of_images: String,
  retry: u8,
}

#[tokio::main] // requires `features = ["mt"]
async fn main() -> DynResult<()> {
  stderrlog::new()
    .module(module_path!())
    .quiet(false)
    .verbosity(4) // Debug
    .timestamp(stderrlog::Timestamp::Second)
    .init()?;
  let config: Config = match parse_config(CONFIG_FILENAME) {
    Ok(config) => config,
    Err(err) => {
      log::error!("Config file parsing unsuccesful: {}", err);
      exit(1);
    }
  };

  let mastodon = if let Ok(data) = masto_toml::from_file("mastodon-data.toml") {
    Mastodon::from(data)
  } else {
    match register(&config).await {
      Ok(account) => account,
      Err(err) => {
        log::error!("Api registation unsuccesful: {}", err);
        exit(1);
      }
    }
  };

  match get_next_url(&config) {
    Ok(url) => match url {
      Some(url) => {
        let mut retry: u8 = 0;
        while let Err(err) = post_image(&mastodon, &url, &config).await {
          log::warn!("Cannot post image, retry: {}, {}", retry, err);
          async_std::task::sleep(Duration::new(1, 0)).await;
          retry += 1;
          if retry >= config.errors.retry {
            log::error!("Max ammount of retries reached on post_image");
            log::info!("Reverting file changes");
            while let Err(err) = pop_last_line_of_file(&config.files.posted) {
              log::warn!("Failed to revert, retrying: {}", err);
              async_std::task::sleep(Duration::new(1, 0)).await;
            }
            exit(1);
          }
        }
        let mut retry: u8 = 0;
        while let Err(err) = update_bio(&mastodon, &config).await {
          log::warn!("Cannot update bio, retry: {}, {}", retry, err);
          async_std::task::sleep(Duration::new(1, 0)).await;
          retry += 1;
          if retry >= config.errors.retry {
            log::error!("Max ammount of retries reached on update bio");
            exit(1);
          }
        }
      }
      None => {
        let mut retry: u8 = 0;
        while let Err(err) = post(&mastodon, &config.errors.out_of_images).await {
          log::warn!("Cannot post, retry: {}, {}", retry, err);
          async_std::task::sleep(Duration::new(1, 0)).await;
          retry += 1;
          if retry >= config.errors.retry {
            log::error!("Max ammount of retries reached on post");
            exit(1);
          }
        }
      }
    },
    Err(err) => {
      log::error!("Cannot get next image: {}", err);
      match post(&mastodon, &err.to_string()).await {
        Ok(_) => {}
        Err(err) => {
          log::error!("Cannot post error message: {}", err);
          exit(1);
        }
      };
    }
  }
  Ok(())
}

/// Parses the given filename to a config struct
fn parse_config(filename: &str) -> DynResult<Config> {
  let toml_file = std::fs::read_to_string(filename)?; //.expect("No config file, consider getting the original one and modifing it");
  Ok(toml::from_str(&toml_file)?) //("Malformed config file, check the original one for reference")
}

fn pop_last_line_of_file(filename: &str) -> DynResult<()> {
  let binding = std::fs::read_to_string(filename)?;
  let mut posted: Vec<_> = binding.lines().collect();
  posted.pop();
  std::fs::write(filename, posted.concat())?;
  log::info!("Success reverting changes");
  Ok(())
}

fn get_next_url(config: &Config) -> DynResult<Option<String>> {
  let binding = std::fs::read_to_string(&config.files.posted)?; //.expect("Posted file not found");
  let posted = binding.lines().collect::<HashSet<&str>>();
  let binding = std::fs::read_to_string(&config.files.urls)?; //.expect("Urls file not found");
  let urls = binding.lines().collect::<HashSet<&str>>();
  let urls = urls.difference(&posted).collect::<Vec<_>>();
  if urls.is_empty() {
    Ok(None)
  } else {
    let mut file = std::fs::OpenOptions::new()
      .write(true)
      .append(true) // This is needed to append to file
      .open(&config.files.posted)?; //.expect("Cannot open posted file"); // Maybe we should retry just in case
    write!(file, "{}\n", urls[0])?; //.expect("Cannot write to posted file"); // maybe we should retry tbh
    Ok(Some(urls[0].to_string().clone()))
  }
}

async fn post_image(account: &Mastodon, url: &String, config: &Config) -> DynResult<()> {
  fetch_url(url, &config.files.tempfile).await?; //.expect("Error fetching url");
  let attachment = account
    .media(&config.files.tempfile, Some(url.to_string()))
    .await?; //.expect("Attachment upload error");
  let attachment = account
    .wait_for_processing(attachment, Default::default())
    .await?; //.expect("Attachment processing error");
  let status = StatusBuilder::new()
    .media_ids(&[attachment.id])
    .visibility(Visibility::Unlisted)
    .sensitive(true)
    .build()?; //.expect("Could not build status"); // we should retry
  account.new_status(status).await?; //.expect("Error generating status"); // we should retry or delete last url in posted
  log::info!("Image status posted: {}", url);
  Ok(())
}

async fn update_bio(account: &Mastodon, config: &Config) -> DynResult<()> {
  let binding = std::fs::read_to_string(&config.files.posted)?; //.expect("Posted file not found");
  let posted = binding.lines().collect::<HashSet<&str>>();
  let binding = std::fs::read_to_string(&config.files.urls)?; //.expect("Url file not found");
  let urls = binding.lines().collect::<HashSet<&str>>();

  let remaining = urls.difference(&posted).count();
  let client = reqwest::Client::new();

  let response = client
    .patch(&format!(
      "{}/api/v1/accounts/update_credentials",
      config.bot.instance
    ))
    .bearer_auth(&account.data.token)
    .body(format!(
      "note={}\n\n{} new images remaining",
      config.bot.bio, remaining
    ))
    .send()
    .await?;
  log::info!("Bio updated, response status: {}", response.status());
  Ok(())
}

async fn post(account: &Mastodon, msg: &str) -> DynResult<()> {
  let status = StatusBuilder::new()
    .visibility(Visibility::Direct)
    .status(msg)
    .build()?; //.expect("Error building error status");
  account.new_status(status).await?; //.expect("Error posting error status");
  log::info!("Text status posted: {}", msg);
  Ok(())
}

async fn fetch_url(url: &String, file_name: &String) -> DynResult<()> {
  let response = reqwest::get(url);
  let mut file = std::fs::File::create(file_name)?;
  let mut content = Cursor::new(response.await?.bytes().await?);
  std::io::copy(&mut content, &mut file)?;
  Ok(())
}

async fn register(config: &Config) -> DynResult<Mastodon> {
  let registration = Registration::new(&config.bot.instance)
    .client_name(&config.bot.name)
    .scopes(Scopes::write_all())
    .build()
    .await?;
  let mastodon = cli::authenticate(registration).await?;

  // Save app data for using on the next run.
  masto_toml::to_file(&mastodon.data, "mastodon-data.toml")?;

  Ok(mastodon)
}