From 501880160ac69f90143bc38add541731a9242ab0 Mon Sep 17 00:00:00 2001
From: Kelly Rauchenberger <fefferburbia@gmail.com>
Date: Wed, 25 May 2016 21:34:57 -0400
Subject: Added option to receive all replies in user stream

---
 src/client.cpp | 40 +++++++++++++++++++++++++++++++---------
 src/client.h   |  5 ++++-
 2 files changed, 35 insertions(+), 10 deletions(-)

diff --git a/src/client.cpp b/src/client.cpp
index c6bcd65..b7d1624 100644
--- a/src/client.cpp
+++ b/src/client.cpp
@@ -410,6 +410,11 @@ namespace twitter {
     _user_stream.setNotifyCallback(callback);
   }
   
+  void client::setUserStreamReceiveAllReplies(bool _arg)
+  {
+    _user_stream.setReceiveAllReplies(_arg);
+  }
+  
   void client::startUserStream()
   {
     _user_stream.start();
@@ -614,8 +619,22 @@ namespace twitter {
   
   void client::stream::setNotifyCallback(notify_callback _n)
   {
-    std::lock_guard<std::mutex> _notify_lock(_notify_mutex);
-    _notify = _n;
+    std::lock_guard<std::mutex> _running_lock(_running_mutex);
+    
+    if (!_thread.joinable())
+    {
+      _notify = _n;
+    }
+  }
+  
+  void client::stream::setReceiveAllReplies(bool _arg)
+  {
+    std::lock_guard<std::mutex> _running_lock(_running_mutex);
+    
+    if (!_thread.joinable())
+    {
+      _receive_all_replies = _arg;
+    }
   }
     
   void client::stream::start()
@@ -643,8 +662,15 @@ namespace twitter {
   void client::stream::run()
   {
     curl::curl_easy conn;
-    std::string url = "https://userstream.twitter.com/1.1/user.json";
+    std::ostringstream urlstr;
+    urlstr << "https://userstream.twitter.com/1.1/user.json";
     
+    if (_receive_all_replies)
+    {
+      urlstr << "?replies=all";
+    }
+    
+    std::string url = urlstr.str();
     curl::curl_header headers;
     std::string oauth_header = _client._oauth_client->getFormattedHttpHeader(OAuth::Http::Get, url, "");
     if (!oauth_header.empty())
@@ -768,13 +794,9 @@ namespace twitter {
             _backoff_amount = std::chrono::milliseconds(0);
           }
           
+          if (_notify)
           {
-            std::lock_guard<std::mutex> _notify_lock(_notify_mutex);
-        
-            if (_notify)
-            {
-              _notify(n);
-            }
+            _notify(n);
           }
                     
           _buffer = "";
diff --git a/src/client.h b/src/client.h
index 901c7b5..c1c6344 100644
--- a/src/client.h
+++ b/src/client.h
@@ -31,6 +31,7 @@ namespace twitter {
           stream(client& _client);
           
           void setNotifyCallback(notify_callback _n);
+          void setReceiveAllReplies(bool _arg);
           
           bool isRunning() const;
           void start();
@@ -54,13 +55,13 @@ namespace twitter {
           bool _stop = false;
           std::thread _thread;
           std::mutex _running_mutex;
-          std::mutex _notify_mutex;
           std::mutex _stall_mutex;
           std::string _buffer;
           time_t _last_write;
           bool _established = false;
           backoff _backoff_type = backoff::none;
           std::chrono::milliseconds _backoff_amount;
+          bool _receive_all_replies = false;
       };
       
       client(const auth& _auth);
@@ -80,7 +81,9 @@ namespace twitter {
       
       const user& getUser() const;
       
+      // NOTE: stream setting function calls will fail silently when stream is running
       void setUserStreamNotifyCallback(stream::notify_callback callback);
+      void setUserStreamReceiveAllReplies(bool _arg);
       void startUserStream();
       void stopUserStream();
       
-- 
cgit 1.4.1