Got it bot-ready!

Using C++ randomization now, along with an update to rawr-ebooks that allows it to do the same. Cleaned up the code here and there. Added in the twitter client and the config file reader. We can use librawr's histogram class now, so we don't need our own. Also why did we not name this bot "aspartame". It's a Sugar replacer. Hahaha. I'm Pink Diamond.

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));
   }
 }