From 7f90d586f1eeca0354b0e4500645e72b5b9f1021 Mon Sep 17 00:00:00 2001
From: Kelly Rauchenberger <fefferburbia@gmail.com>
Date: Fri, 20 May 2016 16:27:58 -0400
Subject: Maintain a list of friends so that we can filter notifications

---
 snitch.cpp | 50 ++++++++++++++++++++++++++++++++------------------
 1 file changed, 32 insertions(+), 18 deletions(-)

diff --git a/snitch.cpp b/snitch.cpp
index 3fcb3d0..cd5153b 100644
--- a/snitch.cpp
+++ b/snitch.cpp
@@ -23,30 +23,44 @@ int main(int argc, char** argv)
   img_file.close();
   
   twitter::client client(auth);
+  std::set<twitter::user_id> streamed_friends;
   client.setUserStreamNotifyCallback([&] (twitter::notification n) {
-    if (n.getType() == twitter::notification::type::tweet)
+    if (n.getType() == twitter::notification::type::friends)
     {
-      std::string orig = n.getTweet().getText();
-      std::string canonical;
-      std::transform(std::begin(orig), std::end(orig), std::back_inserter(canonical), [] (char ch) {
-        return std::tolower(ch);
-      });
-      
-      if (canonical.find("calling the cops") != std::string::npos)
+      streamed_friends = n.getFriends();
+    } else if (n.getType() == twitter::notification::type::follow)
+    {
+      streamed_friends.insert(n.getUserID());
+    } else if (n.getType() == twitter::notification::type::unfollow)
+    {
+      streamed_friends.erase(n.getUserID());
+    } else if (n.getType() == twitter::notification::type::tweet)
+    {
+      // Only monitor people you are following
+      if (streamed_friends.count(n.getTweet().getAuthor().getID()) == 1)
       {
-        std::cout << "Calling the cops on @" << n.getTweet().getAuthor().getScreenName() << std::endl;
-        
-        long media_id;
-        twitter::response resp = client.uploadMedia("image/jpeg", (const char*) img_buf, img_len, media_id);
-        if (resp != twitter::response::ok)
+        std::string orig = n.getTweet().getText();
+        std::string canonical;
+        std::transform(std::begin(orig), std::end(orig), std::back_inserter(canonical), [] (char ch) {
+          return std::tolower(ch);
+        });
+      
+        if (canonical.find("calling the cops") != std::string::npos)
         {
-          std::cout << "Twitter error while uploading image: " << resp << std::endl;
-        } else {
-          twitter::tweet tw;
-          resp = client.updateStatus("@" + n.getTweet().getAuthor().getScreenName(), tw, n.getTweet(), {media_id});
+          std::cout << "Calling the cops on @" << n.getTweet().getAuthor().getScreenName() << std::endl;
+        
+          long media_id;
+          twitter::response resp = client.uploadMedia("image/jpeg", (const char*) img_buf, img_len, media_id);
           if (resp != twitter::response::ok)
           {
-            std::cout << "Twitter error while tweeting: " << resp << std::endl;
+            std::cout << "Twitter error while uploading image: " << resp << std::endl;
+          } else {
+            twitter::tweet tw;
+            resp = client.updateStatus("@" + n.getTweet().getAuthor().getScreenName(), tw, n.getTweet(), {media_id});
+            if (resp != twitter::response::ok)
+            {
+              std::cout << "Twitter error while tweeting: " << resp << std::endl;
+            }
           }
         }
       }
-- 
cgit 1.4.1