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 +++++++++++++++++++++++++++++++---------
 1 file changed, 31 insertions(+), 9 deletions(-)

(limited to 'src/client.cpp')

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 = "";
-- 
cgit 1.4.1