1 files changed, 45 insertions, 39 deletions

@@ -1,11 +1,12 @@
 #include <yaml-cpp/yaml.h>
+#include <verbly.h>
+#include <twitter.h>
+#include <fstream>
 #include <iostream>
-#include <cstdlib>
-#include <ctime>
 #include <sstream>
-#include <twitcurl.h>
-#include <unistd.h>
-#include <verbly.h>
+#include <chrono>
+#include <random>
+#include <thread>
 
 std::string capitalize(std::string input)
 {
@@ -33,28 +34,27 @@ std::string capitalize(std::string input)
 
 int main(int argc, char** argv)
 {
-  srand(time(NULL));
-  
-  YAML::Node config = YAML::LoadFile("config.yml");
+  std::random_device random_device;
+  std::mt19937 random_engine{random_device()};
   
   // Forms
   std::vector<std::string> forms;
-  std::ifstream formfile("forms.txt");
-  if (formfile.is_open())
   {
-    while (!formfile.eof())
+    std::ifstream formfile("forms.txt");
+    if (formfile.is_open())
     {
-      std::string l;
-      getline(formfile, l);
-      if (l.back() == '\r')
+      while (!formfile.eof())
       {
-        l.pop_back();
-      }
+        std::string l;
+        getline(formfile, l);
+        if (l.back() == '\r')
+        {
+          l.pop_back();
+        }
       
-      forms.push_back(l);
+        forms.push_back(l);
+      }
     }
-    
-    formfile.close();
   }
   
   if (forms.size() == 0)
@@ -67,31 +67,36 @@ int main(int argc, char** argv)
   // verbly
   verbly::data database("data.sqlite3");
   
-  // Twitter  
-  twitCurl twitter;
-  twitter.getOAuth().setConsumerKey(config["consumer_key"].as<std::string>());
-  twitter.getOAuth().setConsumerSecret(config["consumer_secret"].as<std::string>());
-  twitter.getOAuth().setOAuthTokenKey(config["access_key"].as<std::string>());
-  twitter.getOAuth().setOAuthTokenSecret(config["access_secret"].as<std::string>());
+  // Twitter
+  YAML::Node config = YAML::LoadFile("config.yml");
+  
+  twitter::auth auth;
+  auth.setConsumerKey(config["consumer_key"].as<std::string>());
+  auth.setConsumerSecret(config["consumer_secret"].as<std::string>());
+  auth.setAccessKey(config["access_key"].as<std::string>());
+  auth.setAccessSecret(config["access_secret"].as<std::string>());
+  
+  twitter::client client(auth);
   
   for (;;)
   {
-    std::cout << "Generating tweet" << std::endl;
+    std::cout << "Generating tweet..." << std::endl;
     
-    std::string form = forms[rand() % forms.size()];
+    int form_i = std::uniform_int_distribution<int>(0, forms.size()-1)(random_engine);
+    std::string form = forms[form_i];
     
     // Adjectives
     int i;
     while ((i = form.find("{adj}")) != std::string::npos)
     {
-      verbly::adjective adj = database.adjectives().random(true).limit(1).run().front();
+      verbly::adjective adj = database.adjectives().random().limit(1).run().front();
       form.replace(i, 5, capitalize(adj.base_form()));
     }
     
     // Nouns
     while ((i = form.find("{noun}")) != std::string::npos)
     {
-      verbly::noun n = database.nouns().is_not_proper(true).random(true).limit(1).run().front();
+      verbly::noun n = database.nouns().is_not_proper().random().limit(1).run().front();
       form.replace(i, 6, capitalize(n.singular_form()));
     }
     
@@ -99,18 +104,19 @@ int main(int argc, char** argv)
     {
       continue;
     }
-        
-    std::string replyMsg;
-    if (twitter.statusUpdate(form))
+    
+    try
     {
-      twitter.getLastWebResponse(replyMsg);
-      std::cout << "Twitter message: " << replyMsg << std::endl;
-    } else {
-      twitter.getLastCurlError(replyMsg);
-      std::cout << "Curl error: " << replyMsg << std::endl;
+      client.updateStatus(form);
+      
+      std::cout << "Tweeted!" << std::endl;
+    } catch (const twitter::twitter_error& e)
+    {
+      std::cout << "Twitter error: " << e.what() << std::endl;
     }
     
-    std::cout << "Waiting" << std::endl;
-    sleep(60 * 60 * 3);
+    std::cout << "Waiting..." << std::endl;
+    
+    std::this_thread::sleep_for(std::chrono::hours(1));
   }
 }