1 files changed, 50 insertions, 32 deletions

@@ -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.getUser().getID());
+    } else if (n.getType() == twitter::notification::type::unfollow)
+    {
+      streamed_friends.erase(n.getUser().getID());
+    } 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;
+            }
           }
         }
       }
@@ -66,7 +80,7 @@ int main(int argc, char** argv)
   client.startUserStream();
   for (;;)
   {
-    std::this_thread::sleep_for(std::chrono::hours(4));
+    std::this_thread::sleep_for(std::chrono::minutes(1));
     
     std::set<twitter::user_id> friends;
     std::set<twitter::user_id> followers;
@@ -76,22 +90,24 @@ int main(int argc, char** argv)
       resp = client.getFollowers(followers);
       if (resp == twitter::response::ok)
       {
-        auto frit = std::begin(friends);
-        auto foit = std::begin(followers);
+        std::list<twitter::user_id> old_friends, new_followers;
+        std::set_difference(std::begin(friends), std::end(friends), std::begin(followers), std::end(followers), std::back_inserter(old_friends));
+        std::set_difference(std::begin(followers), std::end(followers), std::begin(friends), std::end(friends), std::back_inserter(new_followers));
+        for (auto f : old_friends)
+        {
+          resp = client.unfollow(f);
+          if (resp != twitter::response::ok)
+          {
+            std::cout << "Twitter error while unfollowing" << std::endl;
+          }
+        }
         
-        while (frit != std::end(friends))
+        for (auto f : new_followers)
         {
-          auto match = std::mismatch(frit, std::end(friends), foit);
-          if (match.first != std::end(friends))
+          resp = client.follow(f);
+          if (resp != twitter::response::ok)
           {
-            resp = client.unfollow(*match.first);
-            if (resp != twitter::response::ok)
-            {
-              std::cout << "Twitter error while unfollowing" << std::endl;
-            }
-            
-            frit = match.first;
-            frit++;
+            std::cout << "Twitter error while following" << std::endl;
           }
         }
       } else {
@@ -100,6 +116,8 @@ int main(int argc, char** argv)
     } else {
       std::cout << "Twitter error while getting friends: " << resp << std::endl;
     }
+    
+    std::this_thread::sleep_for(std::chrono::hours(4));
   }
   
   client.stopUserStream();