Enable previews even when tweet has a url

dracarys18 · Apr 12, 2022 · 0864655 · 0864655
1 parent 1ca581b
commit 0864655
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/stream/stream_impl.rs b/src/stream/stream_impl.rs
@@ -1,6 +1,6 @@
 use crate::config::Config;
 use crate::error::TweetResult;
-use crate::utils::{get_tweet_url, tweet_has_media};
+use crate::utils::{get_tweet_url, tweet_has_media, tweet_has_url};
 use egg_mode::stream::{filter, StreamMessage};
 use futures::TryStreamExt;
 use teloxide::prelude2::*;
@@ -27,7 +27,7 @@ pub async fn stream_tweets(config: &Config, to_follow: &[u64]) -> TweetResult<()
                         && (to_follow.contains(&_user_id))
                         && tweet.in_reply_to_user_id.unwrap().eq(&_user_id))
                 {
-                    let preview = !tweet_has_media(&tweet).await;
+                    let preview = !tweet_has_media(&tweet).await || tweet_has_url(&tweet).await;
                     let contains_keyword = text
                         .to_lowercase()
                         .split_whitespace()

diff --git a/src/utils.rs b/src/utils.rs
@@ -19,7 +19,9 @@ pub async fn get_tweet_url(tweet: &Tweet) -> Option<String> {
         screen_name, tweet_id
     ))
 }
-
+pub async fn tweet_has_url(tweet: &Tweet) -> bool {
+    !tweet.entities.urls.is_empty()
+}
 pub async fn get_url_entity(tweet: &Tweet) -> Option<String> {
     let urls = tweet.entities.urls.clone();
     if urls.is_empty() {