Merge pull request #2 from iximeow/mute-tweaks

Plumb through support to mute retweets

1 files changed, 143 insertions, 1 deletions

diff --git a/src/tw/mod.rs b/src/tw/mod.rs
index a8db62f..f9fda69 100644
--- a/src/tw/mod.rs
+++ b/src/tw/mod.rs
@@ -99,6 +99,7 @@ enum TweetMuteType {
 
 #[derive(Serialize, Deserialize)]
 enum UserMuteType {
+    Retweets,
     Mentions,
     Everything
 }
@@ -230,6 +231,114 @@ mod tests {
     }
 
     #[test]
+    fn test_mute_behavior() {
+        let today = Utc::now();
+        let rt_muted_user = "1";
+        let everything_muted_user = "2";
+        let you = "3";
+        let rando = "4";
+
+        let you_tweet = Tweet {
+            id: "yours".to_owned(),
+            author_id: you.to_owned(),
+            text: "you said something".to_owned(),
+            created_at: "todayish".to_owned(),
+            recieved_at: today,
+            urls: HashMap::new(),
+            quoted_tweet_id: None,
+            rt_tweet: None,
+            reply_to_tweet: None,
+            internal_id: 0
+        };
+
+        let rt_muted_user_reply = Tweet {
+            id: "rt_muted_user_reply".to_owned(),
+            author_id: rt_muted_user.to_owned(),
+            text: "i said something in reply".to_owned(),
+            created_at: "todayish".to_owned(),
+            recieved_at: today,
+            urls: HashMap::new(),
+            quoted_tweet_id: None,
+            rt_tweet: None,
+            reply_to_tweet: Some("yours".to_owned()),
+            internal_id: 0
+        };
+
+        let everything_muted_user_reply = Tweet {
+            id: "everything_muted_user_reply".to_owned(),
+            author_id: everything_muted_user.to_owned(),
+            text: "i also said something in reply".to_owned(),
+            created_at: "todayish".to_owned(),
+            recieved_at: today,
+            urls: HashMap::new(),
+            quoted_tweet_id: None,
+            rt_tweet: None,
+            reply_to_tweet: Some("yours".to_owned()),
+            internal_id: 0
+        };
+
+        let rando_reply = Tweet {
+            id: "rando reply".to_owned(),
+            author_id: rando.to_owned(),
+            text: "some random reply".to_owned(),
+            created_at: "todayish".to_owned(),
+            recieved_at: today,
+            urls: HashMap::new(),
+            quoted_tweet_id: None,
+            rt_tweet: None,
+            reply_to_tweet: Some("yours".to_owned()),
+            internal_id: 0
+        };
+
+        let rt_of_rando_reply = Tweet {
+            id: "rt_of_rando_reply".to_owned(),
+            author_id: rt_muted_user.to_owned(),
+            text: "rando reply text".to_owned(),
+            created_at: "todayish".to_owned(),
+            recieved_at: today,
+            urls: HashMap::new(),
+            quoted_tweet_id: None,
+            rt_tweet: Some(rando_reply.id.to_owned()),
+            reply_to_tweet: Some("yours".to_owned()),
+            internal_id: 0
+        };
+
+        let muted_rt_of_you = Tweet {
+            id: "rt_of_yours".to_owned(),
+            author_id: rt_muted_user.to_owned(),
+            text: "you said something".to_owned(),
+            created_at: "todayish".to_owned(),
+            recieved_at: today,
+            urls: HashMap::new(),
+            quoted_tweet_id: None,
+            rt_tweet: Some("yours".to_owned()),
+            reply_to_tweet: None,
+            internal_id: 0
+        };
+
+        let tweets = vec![
+            &you_tweet, &rt_muted_user_reply, &everything_muted_user_reply, &rando_reply,
+            &rt_of_rando_reply, &muted_rt_of_you,
+        ];
+
+        let mut tweeter = TwitterCache::new();
+
+        for tweet in tweets {
+            tweeter.number_and_insert_tweet(tweet.to_owned());
+        }
+
+        tweeter.mute_user(rt_muted_user.to_owned(), UserMuteType::Retweets);
+        tweeter.mute_user(everything_muted_user.to_owned(), UserMuteType::Everything);
+
+        assert_eq!(tweeter.tweet_muted(&you_tweet), false);
+        assert_eq!(tweeter.tweet_muted(&rt_muted_user_reply), true);
+        assert_eq!(tweeter.tweet_muted(&everything_muted_user_reply), true);
+        assert_eq!(tweeter.tweet_muted(&rando_reply), false);
+        assert_eq!(tweeter.tweet_muted(&rt_of_rando_reply), true);
+        assert_eq!(tweeter.tweet_muted(&muted_rt_of_you), true);
+    }
+
+    #[test]
     fn test_display_id() {
         /*
          * ... I think this test only works if Local is -0800 or further from UTC.
@@ -594,6 +703,22 @@ impl TwitterCache {
         }
     }
 
+    fn mute_tweet(&mut self, twid: String, mute_setting: TweetMuteType) {
+        self.mutes.tweets.insert(twid, mute_setting);
+    }
+
+    fn mute_user(&mut self, userid: String, mute_setting: UserMuteType) {
+        self.mutes.users.insert(userid, mute_setting);
+    }
+
+    fn unmute_tweet(&mut self, twid: String) {
+        self.mutes.tweets.remove(&twid);
+    }
+
+    fn unmute_user(&mut self, userid: String) {
+        self.mutes.users.remove(&userid);
+    }
+
     fn tweet_muted(&self, tw: &Tweet) -> bool {
         match self.mutes.tweets.get(&tw.id) {
             Some(Conversation) => {
@@ -604,10 +729,27 @@ impl TwitterCache {
         };
         match self.mutes.users.get(&tw.author_id) {
             Some(Mentions) => { return true; },
-            Some(Everything) => { return true; }
+            Some(Everything) => { return true; },
+            Some(Retweets) => {
+                // if the author is muted for retweets, check if this is a retweet
+                if tw.rt_tweet.is_some() {
+                    return true;
+                }
+            }
             None => {}
         };
 
+        if let Some(ref rt_id) = tw.rt_tweet {
+            if let Some(rt) = self.retrieve_tweet(&TweetId::Twitter(rt_id.to_owned())) {
+                match self.mutes.users.get(&rt.author_id) {
+                    Some(Everything) => { return true; },
+                    Some(Retweet) => { /* Retweets don't show up as retweets of retweets, ever */ },
+                    Some(Mentions) => { /* what does this entail? would the rewteet have to mention you? */ },
+                    None => {}
+                }
+            }
+        }
+
         // Check reply_id for mutes independently because the mute
         // may be on a tweet we don't have cached
         if let Some(ref reply_id) = tw.reply_to_tweet {