refactor: make everything a little more modular

src/mastodon.rs

@@ -0,0 +1,106 @@
+// auto imports
+use crate::config::MastodonConfig;
+use crate::twitter::decode_urls;
+
+// std
+use std::{
+    borrow::Cow,
+    error::Error,
+    collections::HashMap,
+    io::stdin,
+};
+
+// htmlescape
+use htmlescape::decode_html;
+
+// egg-mode
+use egg_mode::{
+    tweet::Tweet,
+    entities::MentionEntity,
+};
+
+// elefren
+use elefren::{
+    prelude::*,
+    apps::App,
+    scopes::Scopes,
+};
+
+
+/// Decodes the Twitter mention to something that will make sense once Twitter has joined the
+/// Fediverse
+fn twitter_mentions(ums: &Vec<MentionEntity>) -> HashMap<String, String> {
+    let mut decoded_mentions = HashMap::new();
+
+    for um in ums {
+        decoded_mentions.insert(format!("@{}", um.screen_name), format!("@{}@twitter.com", um.screen_name));
+    }
+
+    decoded_mentions
+}
+
+/// Gets Mastodon Data
+pub fn get_mastodon_token(masto: &MastodonConfig) -> Mastodon {
+    let data = Data {
+        base: Cow::from(String::from(&masto.base)),
+        client_id: Cow::from(String::from(&masto.client_id)),
+        client_secret: Cow::from(String::from(&masto.client_secret)),
+        redirect: Cow::from(String::from(&masto.redirect)),
+        token: Cow::from(String::from(&masto.token)),
+    };
+
+    Mastodon::from(data)
+}
+
+/// Builds toot text from tweet
+pub fn build_basic_status(tweet: &Tweet) -> Result<String, Box<dyn Error>> {
+    let mut toot = String::from(&tweet.text);
+
+    let decoded_urls = decode_urls(&tweet.entities.urls);
+
+    for decoded_url in decoded_urls {
+        toot = toot.replace(&decoded_url.0, &decoded_url.1);
+    }
+
+    let decoded_mentions = twitter_mentions(&tweet.entities.user_mentions);
+
+    for decoded_mention in decoded_mentions {
+        toot = toot.replace(&decoded_mention.0, &decoded_mention.1);
+    }
+
+    if let Ok(t) = decode_html(&toot) {
+        toot = t;
+    }
+
+    Ok(toot)
+}
+
+/// Generic register function
+/// As this function is supposed to be run only once, it will panic for every error it encounters
+/// Most of this function is a direct copy/paste of the official `elefren` crate
+pub fn register(host: &str) {
+    let mut builder = App::builder();
+    builder.client_name(Cow::from(String::from(env!("CARGO_PKG_NAME"))))
+        .redirect_uris(Cow::from(String::from("urn:ietf:wg:oauth:2.0:oob")))
+        .scopes(Scopes::write_all())
+        .website(Cow::from(String::from("https://framagit.org/veretcle/scootaloo")));
+
+    let app = builder.build().expect("Cannot build the app");
+
+    let registration = Registration::new(host).register(app).expect("Cannot build registration object");
+    let url = registration.authorize_url().expect("Cannot generate registration URI!");
+
+    println!("Click this link to authorize on Mastodon: {}", url);
+    println!("Paste the returned authorization code: ");
+
+    let mut input = String::new();
+    stdin().read_line(&mut input).expect("Unable to read back registration code!");
+
+    let code = input.trim();
+    let mastodon = registration.complete(code).expect("Unable to create access token!");
+
+    let toml = toml::to_string(&*mastodon).unwrap();
+
+    println!("Please insert the following block at the end of your configuration file:\n[mastodon]\n{}", toml);
+}
+