diff --git a/src/lib.rs b/src/lib.rs index cd4f0b4..412f379 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -23,7 +23,7 @@ use elefren::{prelude::*, status_builder::StatusBuilder, Language}; use futures::StreamExt; use log::info; use rusqlite::Connection; -use std::sync::Arc; +use std::{collections::HashMap, sync::Arc}; use tokio::{spawn, sync::Mutex}; const DEFAULT_RATE_LIMIT: usize = 4; @@ -42,6 +42,22 @@ pub async fn run(config: Config) { }), )); + let scootaloo_mentions: HashMap = config + .mastodon + .values() + .filter(|s| s.mastodon_screen_name.is_some()) + .map(|s| { + ( + format!("@{}", s.twitter_screen_name), + format!( + "{}@{}", + s.mastodon_screen_name.as_ref().unwrap(), + s.base.split('/').last().unwrap() + ), + ) + }) + .collect(); + let mut stream = futures::stream::iter(config.mastodon.into_values()) .map(|mastodon_config| { // calculate Twitter page size @@ -51,6 +67,7 @@ pub async fn run(config: Config) { // create temporary value for each task let scootaloo_cache_path = config.scootaloo.cache_path.clone(); + let scootaloo_mentions = scootaloo_mentions.clone(); let token = get_oauth2_token(&config.twitter); let task_conn = conn.clone(); @@ -91,7 +108,7 @@ pub async fn run(config: Config) { drop(lconn); // build basic status by just yielding text and dereferencing contained urls - let mut status_text = build_basic_status(tweet); + let mut status_text = build_basic_status(tweet, &scootaloo_mentions); // building associative media list let (media_url, status_medias) = diff --git a/src/mastodon.rs b/src/mastodon.rs index 5434f40..7bd7d0a 100644 --- a/src/mastodon.rs +++ b/src/mastodon.rs @@ -48,14 +48,18 @@ pub fn get_mastodon_token(masto: &MastodonConfig) -> Mastodon { } /// Builds toot text from tweet -pub fn build_basic_status(tweet: &Tweet) -> String { +pub fn build_basic_status(tweet: &Tweet, mentions: &HashMap) -> String { let mut toot = tweet.text.to_owned(); for decoded_url in decode_urls(&tweet.entities.urls) { toot = toot.replace(&decoded_url.0, &decoded_url.1); } - for decoded_mention in twitter_mentions(&tweet.entities.user_mentions) { + for decoded_mention in twitter_mentions(&tweet.entities.user_mentions) + .into_iter() + .chain(mentions.to_owned()) + .collect::>() + { toot = toot.replace(&decoded_mention.0, &decoded_mention.1); } @@ -192,7 +196,13 @@ mod tests { range: (80, 95), name: "Nintendo France".to_string(), screen_name: "NintendoFrance".to_string(), - } + }, + MentionEntity { + id: 999999999, + range: (80, 95), + name: "Willy Wonka".to_string(), + screen_name: "WillyWonka".to_string(), + }, ], media: None, }, @@ -213,7 +223,7 @@ mod tests { retweeted: None, retweeted_status: None, source: None, - text: "Mother 1 & 2 sur le NES/SNES online !\nDispo maintenant. cc @NintendoFrance https://t.co/zXw0FfX2Nt".to_string(), + text: "Mother 1 & 2 sur le NES/SNES online !\nDispo maintenant. cc @NintendoFrance @WillyWonka https://t.co/zXw0FfX2Nt".to_string(), truncated: false, user: None, withheld_copyright: false, @@ -221,8 +231,13 @@ mod tests { withheld_scope: None, }; - let t_out = build_basic_status(&t); + let s: HashMap = HashMap::from([( + "@WillyWonka".to_string(), + "@WillyWonka@chocolatefactory.org".to_string(), + )]); - assert_eq!(&t_out, "Mother 1 & 2 sur le NES/SNES online !\nDispo maintenant. cc @NintendoFrance@twitter.com https://www.youtube.com/watch?v=w5TrSaoYmZ8"); + let t_out = build_basic_status(&t, &s); + + assert_eq!(&t_out, "Mother 1 & 2 sur le NES/SNES online !\nDispo maintenant. cc @NintendoFrance@twitter.com @WillyWonka@chocolatefactory.org https://www.youtube.com/watch?v=w5TrSaoYmZ8"); } }