Modernized bot structure

This improves readability, and also will display a nicer error message
for things such as the verbly datafile being invalid.

1 files changed, 25 insertions, 27 deletions

@@ -1,41 +1,40 @@
-#include <verbly.h>
-#include <twitter.h>
+#include "fefisms.h"
 #include <yaml-cpp/yaml.h>
 #include <iostream>
 #include <chrono>
-#include <random>
 #include <thread>
 
-int main(int argc, char** argv)
+fefisms::fefisms(
+  std::string configFile,
+  std::mt19937& rng) :
+    rng_(rng)
 {
-  std::random_device random_device;
-  std::mt19937 random_engine{random_device()};
-  
-  if (argc != 2)
-  {
-    std::cout << "usage: fefisms [configfile]" << std::endl;
-    return -1;
-  }
+  // Load the config file.
+  YAML::Node config = YAML::LoadFile(configFile);
 
-  std::string configfile(argv[1]);
-  YAML::Node config = YAML::LoadFile(configfile);
+  // Set up the verbly database.
+  database_ = std::unique_ptr<verbly::database>(
+    new verbly::database(config["verbly_datafile"].as<std::string>()));
 
+  // Set up the Twitter client.
   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);
-
-  verbly::database database(config["verbly_datafile"].as<std::string>());
+  client_ = std::unique_ptr<twitter::client>(new twitter::client(auth));
+}
 
+void fefisms::run() const
+{
   verbly::filter formFilter =
     (verbly::form::complexity == 1)
     && (verbly::form::proper == false);
-  
+
+  // Blacklist ethnic slurs
   verbly::filter cleanFilter =
-    !(verbly::word::usageDomains %= (verbly::notion::wnid == 106718862)); // Blacklist ethnic slurs
+    !(verbly::word::usageDomains %= (verbly::notion::wnid == 106718862));
 
   for (;;)
   {
@@ -45,18 +44,18 @@ int main(int argc, char** argv)
 
     verbly::inflection nounInfl = verbly::inflection::base;
     verbly::inflection hypoInfl = verbly::inflection::base;
-    
-    if (std::bernoulli_distribution(1.0/2.0)(random_engine))
+
+    if (std::bernoulli_distribution(1.0/2.0)(rng_))
     {
       hypoInfl = verbly::inflection::plural;
     }
 
-    if (std::bernoulli_distribution(1.0/2.0)(random_engine))
+    if (std::bernoulli_distribution(1.0/2.0)(rng_))
     {
       nounInfl = verbly::inflection::plural;
     }
 
-    verbly::word noun = database.words(
+    verbly::word noun = database_->words(
       (verbly::notion::partOfSpeech == verbly::part_of_speech::noun)
       && (verbly::word::forms(nounInfl) %= formFilter)
       && cleanFilter
@@ -64,13 +63,13 @@ int main(int argc, char** argv)
         (verbly::word::forms(hypoInfl) %= formFilter)
         && cleanFilter)).first();
 
-    verbly::word hyponym = database.words(
+    verbly::word hyponym = database_->words(
       (verbly::notion::partOfSpeech == verbly::part_of_speech::noun)
       && (verbly::notion::hypernyms %= noun)
       && cleanFilter
       && (verbly::word::forms(hypoInfl) %= formFilter)).first();
 
-    if (std::bernoulli_distribution(1.0/2.0)(random_engine))
+    if (std::bernoulli_distribution(1.0/2.0)(rng_))
     {
       utterance << verbly::token(noun, nounInfl);
       utterance << verbly::token(hyponym, hypoInfl);
@@ -86,7 +85,7 @@ int main(int argc, char** argv)
     {
       std::cout << "Tweeting..." << std::endl;
 
-      client.updateStatus(action);
+      client_->updateStatus(action);
 
       std::cout << "Success!" << std::endl;
       std::cout << "Waiting..." << std::endl;
@@ -100,4 +99,3 @@ int main(int argc, char** argv)
     std::cout << std::endl;
   }
 }
-