Added utility function to generate tweet reply prefill

6 files changed, 45 insertions, 0 deletions

@@ -420,6 +420,22 @@ namespace twitter {
     _user_stream.stop();
   }
   
+  std::string client::generateReplyPrefill(tweet _tweet) const
+  {
+    std::ostringstream output;
+    output << "@" << _tweet.getAuthor().getScreenName() << " ";
+    
+    for (auto mention : _tweet.getMentions())
+    {
+      if ((mention.first != _tweet.getAuthor().getID()) && (mention.first != _current_user.getID()))
+      {
+        output << "@" << mention.second << " ";
+      }
+    }
+    
+    return output.str();
+  }
+  
   bool client::performGet(std::string url, long& response_code, std::string& result)
   {
     std::ostringstream output;
@@ -84,6 +84,8 @@ namespace twitter {
       void startUserStream();
       void stopUserStream();
       
+      std::string generateReplyPrefill(tweet t) const;
+      
     private:
       friend class stream;
       
@@ -18,6 +18,18 @@ namespace twitter {
     _text = _data.at("text");
     _author = user(_data.at("user").dump());
     _retweeted = _data.at("retweeted");
+    
+    if (_data.find("entities") != _data.end())
+    {
+      auto _entities = _data.at("entities");
+      if (_entities.find("user_mentions") != _entities.end())
+      {
+        for (auto _mention : _entities.at("user_mentions"))
+        {
+          _mentions.push_back(std::make_pair(_mention.at("id"), _mention.at("screen_name").get<std::string>()));
+        }
+      }
+    }
   }
   
   tweet_id tweet::getID() const
@@ -48,6 +60,11 @@ namespace twitter {
     return _retweeted;
   }
   
+  std::vector<std::pair<user_id, std::string>> tweet::getMentions() const
+  {
+    return _mentions;
+  }
+  
   tweet::operator bool() const
   {
     return _valid;
@@ -3,6 +3,8 @@
 
 #include <string>
 #include "user.h"
+#include <vector>
+#include <utility>
 
 namespace twitter {
   
@@ -17,6 +19,7 @@ namespace twitter {
       std::string getText() const;
       const user& getAuthor() const;
       bool isRetweet() const;
+      std::vector<std::pair<user_id, std::string>> getMentions() const;
       
       operator bool() const;
       
@@ -26,6 +29,7 @@ namespace twitter {
       std::string _text;
       user _author;
       bool _retweeted;
+      std::vector<std::pair<user_id, std::string>> _mentions;
   };
   
 };
@@ -42,5 +42,10 @@ namespace twitter {
   {
     return _id == other._id;
   }
+  
+  bool user::operator!=(const user& other) const
+  {
+    return _id != other._id;
+  }
 
 };
@@ -18,6 +18,7 @@ namespace twitter {
       
       operator bool() const;
       bool operator==(const user& other) const;
+      bool operator!=(const user& other) const;
       
     private:
       bool _valid = false;