Modernized bot

4 files changed, 95 insertions, 42 deletions

@@ -13,7 +13,7 @@ include_directories(
   ${yaml-cpp_INCLUDE_DIRS}
   vendor/hkutil)
 
-add_executable(cadence cadence.cpp)
+add_executable(cadence cadence.cpp main.cpp)
 set_property(TARGET cadence PROPERTY CXX_STANDARD 11)
 set_property(TARGET cadence PROPERTY CXX_STANDARD_REQUIRED ON)
 target_link_libraries(cadence ${sqlite3_LIBRARIES} twitter++ ${yaml-cpp_LIBRARIES})
@@ -1,45 +1,26 @@
+#include "cadence.h"
 #include <stdexcept>
-#include <vector>
-#include <string>
 #include <yaml-cpp/yaml.h>
-#include <sqlite3.h>
 #include <iostream>
 #include <fstream>
-#include <twitter.h>
 #include <chrono>
 #include <thread>
-#include <random>
 #include <list>
 #include <hkutil/string.h>
-#include <hkutil/database.h>
 
-int main(int argc, char** argv)
+cadence::cadence(
+  std::string configFile,
+  std::mt19937& rng) :
+    rng_(rng)
 {
-  if (argc != 2)
-  {
-    std::cout << "usage: cadence [configfile]" << std::endl;
-    return -1;
-  }
-
-  std::string configfile(argv[1]);
-  YAML::Node config = YAML::LoadFile(configfile);
-
-  // 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);
+  // Load the config file.
+  YAML::Node config = YAML::LoadFile(configFile);
 
   // Read in the forms file.
-  std::map<std::string, std::vector<std::string>> groups;
   std::ifstream datafile(config["forms"].as<std::string>());
   if (!datafile.is_open())
   {
-    std::cout << "Could not find forms file." << std::endl;
-    return 1;
+    throw std::invalid_argument("Could not find forms file");
   }
 
   bool newgroup = true;
@@ -61,30 +42,40 @@ int main(int argc, char** argv)
       {
         newgroup = true;
       } else {
-        groups[curgroup].push_back(line);
+        groups_[curgroup].push_back(line);
       }
     }
   }
 
-  // Initialize the random number generator.
-  std::random_device random_device;
-  std::mt19937 random_engine{random_device()};
-
   // Connect to the AcousticBrainz data file.
-  hatkirby::database abdb(
-    config["acoustic_datafile"].as<std::string>(),
-    hatkirby::dbmode::read);
+  database_ =
+    std::unique_ptr<hatkirby::database>(
+      new hatkirby::database(
+        config["acoustic_datafile"].as<std::string>(),
+        hatkirby::dbmode::read));
 
+  // 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>());
+
+  client_ = std::unique_ptr<twitter::client>(new twitter::client(auth));
+}
+
+void cadence::run() const
+{
   for (;;)
   {
     std::cout << "Generating tweet..." << std::endl;
 
     hatkirby::row songRow =
-      abdb.queryFirst(
+      database_->queryFirst(
         "SELECT song_id, title, artist FROM songs ORDER BY RANDOM() LIMIT 1");
 
     hatkirby::row moodRow =
-      abdb.queryFirst(
+      database_->queryFirst(
         "SELECT mood FROM moods WHERE song_id = ? ORDER BY RANDOM() LIMIT 1",
         { mpark::get<int>(songRow[0]) });
 
@@ -118,11 +109,11 @@ int main(int argc, char** argv)
       } else if (canontkn == "SONG")
       {
         result = "\"" + songTitle + "\"";
-      } else if (groups.count(canontkn)) {
-        auto& group = groups.at(canontkn);
+      } else if (groups_.count(canontkn)) {
+        auto& group = groups_.at(canontkn);
         std::uniform_int_distribution<int> dist(0, group.size() - 1);
 
-        result = group[dist(random_engine)];
+        result = group[dist(rng_)];
       } else {
         throw std::logic_error("No such form as " + canontkn);
       }
@@ -175,7 +166,7 @@ int main(int argc, char** argv)
     {
       try
       {
-        client.updateStatus(action);
+        client_->updateStatus(action);
 
         std::cout << action << std::endl;
       } catch (const twitter::twitter_error& e)
@@ -0,0 +1,30 @@
+#ifndef CADENCE_H_D4AE737D
+#define CADENCE_H_D4AE737D
+
+#include <random>
+#include <string>
+#include <map>
+#include <vector>
+#include <hkutil/database.h>
+#include <twitter.h>
+
+class cadence {
+public:
+
+  cadence(
+    std::string configFile,
+    std::mt19937& rng);
+
+  void run() const;
+
+private:
+
+  std::mt19937& rng_;
+  std::map<std::string, std::vector<std::string>> groups_;
+  std::unique_ptr<hatkirby::database> database_;
+  std::unique_ptr<twitter::client> client_;
+
+};
+
+
+#endif /* end of include guard: CADENCE_H_D4AE737D */
@@ -0,0 +1,32 @@
+#include "cadence.h"
+#include <iostream>
+
+int main(int argc, char** argv)
+{
+  std::random_device randomDevice;
+  std::mt19937 rng(randomDevice());
+
+  if (argc != 2)
+  {
+    std::cout << "usage: cadence [configfile]" << std::endl;
+    return -1;
+  }
+
+  std::string configfile(argv[1]);
+
+  try
+  {
+    cadence bot(configfile, rng);
+
+    try
+    {
+      bot.run();
+    } catch (const std::exception& ex)
+    {
+      std::cout << "Error running bot: " << ex.what() << std::endl;
+    }
+  } catch (const std::exception& ex)
+  {
+    std::cout << "Error initializing bot: " << ex.what() << std::endl;
+  }
+}