Created bot

9 files changed, 411 insertions, 0 deletions

@@ -0,0 +1,6 @@
+res
+.DS_Store
+CMakeCache.txt
+CMakeFiles
+cmake_install.cmake
+Makefile
@@ -0,0 +1,3 @@
+[submodule "vendor/rawr-ebooks"]
+        path = vendor/rawr-ebooks
+        url = git@github.com:hatkirby/rawr-ebooks
@@ -0,0 +1,22 @@
+cmake_minimum_required (VERSION 3.1)
+project (fanmail)
+
+#add_subdirectory(vendor/libtwittercpp)
+add_subdirectory(vendor/rawr-ebooks EXCLUDE_FROM_ALL)
+
+find_package(PkgConfig)
+pkg_check_modules(yaml-cpp yaml-cpp REQUIRED)
+pkg_check_modules(GraphicsMagick GraphicsMagick++ REQUIRED)
+
+include_directories(
+  vendor/rawr-ebooks/vendor/libtwittercpp/src
+  vendor/rawr-ebooks
+  ${yaml-cpp_INCLUDE_DIRS}
+  ${GraphicsMagick_INCLUDE_DIRS})
+
+link_directories(${GraphicsMagick_LIBRARY_DIRS})
+
+add_executable(fanmail fanmail.cpp designer.cpp main.cpp)
+set_property(TARGET fanmail PROPERTY CXX_STANDARD 14)
+set_property(TARGET fanmail PROPERTY CXX_STANDARD_REQUIRED ON)
+target_link_libraries(fanmail ${yaml-cpp_LIBRARIES} ${GraphicsMagick_LIBRARIES} twitter++ rawr)
\ No newline at end of file
@@ -0,0 +1,181 @@
+#include "designer.h"
+#include <dirent.h>
+#include <sstream>
+#include <iostream>
+
+designer::designer(
+  std::string imagesDir) :
+    colDist_ {0, 1, 100, 30, 5, 1, 1}
+{
+  DIR* imgdir;
+  struct dirent* ent;
+  if ((imgdir = opendir(imagesDir.c_str())) == nullptr)
+  {
+    throw std::invalid_argument("Couldn't find images");
+  }
+
+  int maxStripLen = 0;
+  std::map<int, int> stripLength;
+
+  while ((ent = readdir(imgdir)) != nullptr)
+  {
+    std::string dname(ent->d_name);
+    std::stringstream nameParse;
+    int stripNum;
+    int panelNum;
+    char sep;
+    std::string ext;
+
+    nameParse << dname;
+
+    if (nameParse >> stripNum &&
+        nameParse >> sep &&
+        nameParse >> panelNum &&
+        nameParse >> ext &&
+        ext == ".png")
+    {
+      if (!stripLength.count(stripNum) || stripLength[stripNum] < panelNum)
+      {
+        stripLength[stripNum] = panelNum;
+      }
+
+      if (panelNum > maxStripLen)
+      {
+        maxStripLen = panelNum;
+      }
+
+      comics_[stripNum][panelNum] = imagesDir + "/" + dname;
+    }
+  }
+
+  closedir(imgdir);
+
+  std::vector<int> lenHist(maxStripLen, 0);
+  for (const auto& mapping : stripLength)
+  {
+    lenHist[mapping.second]++;
+  }
+
+  lenDist_ =
+    std::discrete_distribution<int>(
+      std::begin(lenHist),
+      std::end(lenHist));
+}
+
+Magick::Image designer::generate(std::mt19937& rng) const
+{
+  int numPanels = lenDist_(rng);
+  std::vector<Magick::Image> panels;
+
+  int numCols = std::min(colDist_(rng), numPanels);
+
+  int curCol = 0;
+  int numRows = 0;
+  int curRowWidth = 0;
+  int maxRowWidth = 0;
+
+  std::map<int, int> rowHeight;
+  int curRowHeight = 0;
+
+  for (int i = 0; i < numPanels; i++)
+  {
+    if (curCol == 0)
+    {
+      numRows++;
+    }
+
+    std::uniform_int_distribution<int> stripPick(1, comics_.size());
+    int stripNum = stripPick(rng);
+    const auto& strip = comics_.at(stripNum);
+
+    size_t low = i / static_cast<double>(numPanels) * strip.size();
+    size_t high = (i+1) / static_cast<double>(numPanels) * strip.size();
+    std::uniform_int_distribution<int> panelPick(
+      low,
+      std::min(high, strip.size() - 1));
+
+    int panelNum = panelPick(rng) + 1;
+    const std::string& panel = strip.at(panelNum);
+
+    std::cout << panel << std::endl;
+
+    Magick::Image curfile(panel);
+    curfile.borderColor("black");
+    curfile.backgroundColor("black");
+
+    if (std::bernoulli_distribution(1.0 / 50.0)(rng))
+    {
+      std::normal_distribution<double> rotDist(0.0, 20.0);
+      double rotAmt = rotDist(rng);
+
+      curfile.rotate(rotAmt);
+    }
+
+    if (std::bernoulli_distribution(1.0 / 2.0)(rng))
+    {
+      std::geometric_distribution<unsigned int> borderDist(1.0 / 8.0);
+      curfile.border(
+        Magick::Geometry{
+          borderDist(rng),
+          borderDist(rng)});
+    }
+
+    curRowWidth += curfile.columns();
+
+    if (curfile.rows() > curRowHeight)
+    {
+      curRowHeight = curfile.rows();
+    }
+
+    panels.emplace_back(std::move(curfile));
+
+    curCol++;
+
+    if (curCol >= numCols || (i == numPanels - 1))
+    {
+      if (curRowWidth > maxRowWidth)
+      {
+        maxRowWidth = curRowWidth;
+      }
+
+      rowHeight[numRows - 1] = curRowHeight;
+
+      curCol = 0;
+      curRowWidth = 0;
+      curRowHeight = 0;
+    }
+  }
+
+  int fileHeight = 0;
+  for (const auto& mapping : rowHeight)
+  {
+    fileHeight += mapping.second;
+  }
+
+  Magick::Image result{
+    Magick::Geometry(maxRowWidth, fileHeight),
+    "black"};
+
+  int curx = 0;
+  int cury = 0;
+  int thisCol = 0;
+  int thisRow = 0;
+  for (const Magick::Image& panel : panels)
+  {
+    if (thisCol == numCols)
+    {
+      thisCol = 0;
+      cury += rowHeight.at(thisRow);
+      thisRow++;
+      curx = 0;
+    }
+
+    result.composite(panel, curx, cury);
+
+    curx += panel.columns();
+
+    thisCol++;
+  }
+
+  return result;
+}
@@ -0,0 +1,22 @@
+#ifndef DESIGNER_H_863F77D3
+#define DESIGNER_H_863F77D3
+
+#include <random>
+#include <string>
+#include <Magick++.h>
+
+class designer {
+public:
+
+  designer(std::string imagesPath);
+
+  Magick::Image generate(std::mt19937& rng) const;
+
+private:
+
+  mutable std::discrete_distribution<int> colDist_;
+  mutable std::discrete_distribution<int> lenDist_;
+  std::map<int, std::map<int, std::string>> comics_;
+};
+
+#endif /* end of include guard: DESIGNER_H_863F77D3 */
@@ -0,0 +1,106 @@
+#include "fanmail.h"
+#include <yaml-cpp/yaml.h>
+#include <thread>
+#include <chrono>
+#include <fstream>
+#include <iostream>
+
+fanmail::fanmail(
+  std::string configFile,
+  std::mt19937& rng) :
+    rng_(rng)
+{
+  // Load the config file.
+  YAML::Node config = YAML::LoadFile(configFile);
+
+  // Set up the Twitter client.
+  auth_ =
+    std::unique_ptr<twitter::auth>(
+      new twitter::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>()));
+
+  client_ =
+    std::unique_ptr<twitter::client>(
+      new twitter::client(*auth_));
+
+  // Set up the layout designer.
+  layout_ = std::unique_ptr<designer>(
+    new designer(config["images"].as<std::string>()));
+
+  // Set up librawr.
+  std::ifstream infile(config["corpus"].as<std::string>());
+  std::string corpus;
+  std::string line;
+  while (getline(infile, line))
+  {
+    if (line.back() == '\r')
+    {
+      line.pop_back();
+    }
+
+    corpus += line + "\n";
+  }
+
+  kgramstats_.addCorpus(corpus);
+  kgramstats_.compile(3);
+}
+
+void fanmail::run() const
+{
+  for (;;)
+  {
+    std::cout << "Generating tweet..." << std::endl;
+
+    try
+    {
+      // Design the comic strip.
+      Magick::Image image = layout_->generate(rng_);
+
+      // Write the tweet.
+      std::string doc = kgramstats_.randomSentence(15);
+      doc.resize(40);
+
+      std::cout << doc << std::endl;
+
+      // Send the tweet.
+      std::cout << "Sending tweet..." << std::endl;
+
+      sendTweet(std::move(image), std::move(doc));
+
+      std::cout << "Tweeted!" << std::endl;
+
+      // Wait.
+      std::binomial_distribution<int> waitDist(35, 0.5);
+      int waitTime = waitDist(rng_) + 1;
+
+      std::cout << "Waiting for " << waitTime << " hours..." << std::endl;
+
+      std::this_thread::sleep_for(std::chrono::hours(waitTime));
+    } catch (const Magick::ErrorImage& ex)
+    {
+      std::cout << "Image error: " << ex.what() << std::endl;
+    } catch (const twitter::twitter_error& ex)
+    {
+      std::cout << "Twitter error: " << ex.what() << std::endl;
+
+      std::this_thread::sleep_for(std::chrono::hours(1));
+    }
+
+    std::cout << std::endl;
+  }
+}
+
+void fanmail::sendTweet(Magick::Image image, std::string doc) const
+{
+  Magick::Blob outputimg;
+  image.magick("png");
+  image.write(&outputimg);
+
+  long media_id = client_->uploadMedia("image/png",
+    static_cast<const char*>(outputimg.data()), outputimg.length());
+
+  client_->updateStatus(std::move(doc), {media_id});
+}
@@ -0,0 +1,33 @@
+#ifndef FANMAIL_H_11D8D668
+#define FANMAIL_H_11D8D668
+
+#include <random>
+#include <string>
+#include <memory>
+#include <Magick++.h>
+#include <twitter.h>
+#include <rawr.h>
+#include "designer.h"
+
+class fanmail {
+public:
+
+  fanmail(
+    std::string configFile,
+    std::mt19937& rng);
+
+  void run() const;
+
+private:
+
+  void sendTweet(Magick::Image image, std::string doc) const;
+
+  std::mt19937& rng_;
+  std::unique_ptr<twitter::auth> auth_;
+  std::unique_ptr<twitter::client> client_;
+  std::unique_ptr<designer> layout_;
+  rawr kgramstats_;
+
+};
+
+#endif /* end of include guard: SAP_H_11D8D668 */
@@ -0,0 +1,38 @@
+#include "fanmail.h"
+#include <iostream>
+
+int main(int argc, char** argv)
+{
+  Magick::InitializeMagick(nullptr);
+
+  std::random_device randomDevice;
+  std::mt19937 rng(randomDevice());
+
+  // We also have to seed the C-style RNG because librawr uses it.
+  srand(time(NULL));
+  rand(); rand(); rand(); rand();
+
+  if (argc != 2)
+  {
+    std::cout << "usage: fanmail [configfile]" << std::endl;
+    return -1;
+  }
+
+  std::string configfile(argv[1]);
+
+  try
+  {
+    fanmail 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;
+  }
+}
+Subproject 26d75f744913a8856e46f5fccbfda8f8336924a