1 files changed, 47 insertions, 19 deletions

@@ -2,19 +2,17 @@
 #include "identifier.h"
 #include "histogram.h"
 #include <rawr.h>
-#include <cstdlib>
-#include <ctime>
 #include <map>
 #include <string>
 #include <iostream>
 #include <sstream>
-
-
+#include <random>
+#include <twitter.h>
+#include <yaml-cpp/yaml.h>
 
 using speakerstore = identifier<std::string>;
 using speaker_id = speakerstore::key_type;
 
-
 struct speaker_data {
 
   std::string name;
@@ -24,21 +22,39 @@ struct speaker_data {
 
 };
 
+int main(int argc, char** argv)
+{
+  if (argc != 2)
+  {
+    std::cout << "usage: garnet [configfile]" << std::endl;
+    return -1;
+  }
+
+  std::random_device randomDevice;
+  std::mt19937 rng(randomDevice());
 
+  std::string configfile(argv[1]);
+  YAML::Node config = YAML::LoadFile(configfile);
 
+  twitter::auth auth(
+    config["consumer_key"].as<std::string>(),
+    config["consumer_secret"].as<std::string>(),
+    config["access_key"].as<std::string>(),
+    config["access_secret"].as<std::string>());
 
-int main(int, char**)
-{
-  srand(time(NULL));
-  rand(); rand(); rand(); rand();
+  twitter::client client(auth);
 
   speakerstore speakers;
   std::map<speaker_id, speaker_data> speakerData;
   histogram<speaker_id> allSpeakers;
 
+  using csv =
+    io::CSVReader<
+      2,
+      io::trim_chars<' ', '\t'>,
+      io::double_quote_escape<',', '"'>>;
 
-
-  io::CSVReader<2,io::trim_chars<' ', '\t'>,io::double_quote_escape<',', '"'>> in("../dialogue.csv");
+  csv in(config["transcript"].as<std::string>());
   std::string speaker;
   std::string line;
 
@@ -85,11 +101,10 @@ int main(int, char**)
   {
     std::set<speaker_id> pastSpeakers;
 
-
-    speaker_id curSpeaker = allSpeakers.next();
+    speaker_id curSpeaker = allSpeakers.next(rng);
 
     std::ostringstream theEnd;
-    int maxLines = rand() % 4 + 3;
+    int maxLines = std::uniform_int_distribution<int>(3, 6)(rng);
 
     for (int i = 0; i < maxLines; i++)
     {
@@ -102,7 +117,8 @@ int main(int, char**)
         theEnd << curSd.name << ": ";
       }
 
-      std::string curLine = curSd.chain.randomSentence(rand() % 30 + 1);
+      int maxL = std::uniform_int_distribution<int>(1, 30)(rng);
+      std::string curLine = curSd.chain.randomSentence(maxL, rng);
 
       if (curSd.name == "" &&
           curLine[0] != '[' &&
@@ -116,13 +132,16 @@ int main(int, char**)
 
       theEnd << std::endl;
 
-      speaker_id repeatSpeaker = *std::next(std::begin(pastSpeakers), rand() % pastSpeakers.size());
+      int psi =
+        std::uniform_int_distribution<int>(0, pastSpeakers.size()-1)(rng);
+
+      speaker_id repeatSpeaker = *std::next(std::begin(pastSpeakers), psi);
       if (repeatSpeaker != curSpeaker &&
-          rand() % 3 == 0)
+          std::bernoulli_distribution(1.0 / 3.0)(rng))
       {
         curSpeaker = repeatSpeaker;
       } else {
-        curSpeaker = curSd.nextSpeaker.next();
+        curSpeaker = curSd.nextSpeaker.next(rng);
       }
     }
 
@@ -131,9 +150,18 @@ int main(int, char**)
     output = output.substr(0, output.find_last_of('\n'));
     std::cout << output;
 
+    try
+    {
+      client.updateStatus(output);
+    } catch (const twitter::twitter_error& error)
+    {
+      std::cout << "Twitter error while tweeting: " << error.what()
+        << std::endl;
+    }
+
     std::cout << std::endl;
     std::cout << std::endl;
 
-    getc(stdin);
+    std::this_thread::sleep_for(std::chrono::hours(4));
   }
 }