3 files changed, 166 insertions, 15 deletions

@@ -1,6 +1,8 @@
 #include "database.h"
 #include <sqlite3.h>
 #include <stdexcept>
+#include <sstream>
+#include <iomanip>
 
 database::database(std::string path)
 {
@@ -120,3 +122,55 @@ std::string database::getRandomImageForGame(int gameId) const
 
   return result;
 }
+
+did database::getRandomDidForAchievement(int achievementId) const
+{
+  std::string queryString = "SELECT profile_id, DATETIME(achieved_at) FROM dids WHERE achievement_id = ? ORDER BY RANDOM() LIMIT 1";
+
+  sqlite3_stmt* ppstmt;
+  if (sqlite3_prepare_v2(
+    ppdb_,
+    queryString.c_str(),
+    queryString.length(),
+    &ppstmt,
+    NULL) != SQLITE_OK)
+  {
+    std::string errorMsg = sqlite3_errmsg(ppdb_);
+    sqlite3_finalize(ppstmt);
+
+    throw std::logic_error(errorMsg);
+  }
+
+  if (sqlite3_bind_int(ppstmt, 1, achievementId) != SQLITE_OK)
+  {
+    std::string errorMsg = sqlite3_errmsg(ppdb_);
+    sqlite3_finalize(ppstmt);
+
+    throw std::logic_error(errorMsg);
+  }
+
+  if (sqlite3_step(ppstmt) != SQLITE_ROW)
+  {
+    std::string errorMsg = sqlite3_errmsg(ppdb_);
+    sqlite3_finalize(ppstmt);
+
+    throw std::logic_error(errorMsg);
+  }
+
+  did result;
+  result.profileId = sqlite3_column_int(ppstmt, 0);
+
+  std::tm achievedAt = {};
+
+  std::stringstream dateParser;
+  dateParser << reinterpret_cast<const char*>(sqlite3_column_text(ppstmt, 1));
+  dateParser >> std::get_time(&achievedAt, "%Y-%m-%d %H:%M:%S");
+
+  std::stringstream dateFormatter;
+  dateFormatter << std::put_time(&achievedAt, "%m/%d/%Y");
+  result.date = dateFormatter.str();
+
+  sqlite3_finalize(ppstmt);
+
+  return result;
+}
@@ -12,6 +12,11 @@ struct achievement {
   std::string color;
 };
 
+struct did {
+  int profileId;
+  std::string date;
+};
+
 class database {
 public:
 
@@ -43,6 +48,8 @@ public:
 
   std::string getRandomImageForGame(int gameId) const;
 
+  did getRandomDidForAchievement(int achievementId) const;
+
 private:
 
   database() = default;
@@ -6,8 +6,34 @@
 #include <chrono>
 #include <thread>
 #include <Magick++.h>
+#include <list>
+#include <string>
+#include <sstream>
 #include "database.h"
 
+template <class Container>
+Container split(std::string input, std::string delimiter)
+{
+  Container result;
+
+  while (!input.empty())
+  {
+    int divider = input.find(delimiter);
+    if (divider == std::string::npos)
+    {
+      result.push_back(input);
+
+      input = "";
+    } else {
+      result.push_back(input.substr(0, divider));
+
+      input = input.substr(divider+delimiter.length());
+    }
+  }
+
+  return result;
+}
+
 int main(int argc, char** argv)
 {
   if (argc != 2)
@@ -43,15 +69,88 @@ int main(int argc, char** argv)
     std::string imagePath = config["images"].as<std::string>()
       + "/" + imageName;
 
-    Magick::Image overlay;
-    overlay.read("res/overlay.png");
-
     Magick::Image moonColor;
     moonColor.read("res/" + ach.color + ".png");
 
     try
     {
-      // Start with the game image
+      // Start with the Odyssey text overlay
+      Magick::Image overlay;
+      overlay.read("res/overlay.png");
+
+      // Add the moon image
+      overlay.composite(moonColor, 672, 85, Magick::OverCompositeOp);
+
+      // Add the name of the achievement
+      overlay.fontPointsize(54);
+      overlay.fillColor("white");
+      overlay.font("@" + config["title_font"].as<std::string>());
+
+      std::list<std::string> words = split<std::list<std::string>>(
+        ach.title,
+        " ");
+      std::ostringstream wrappingStream;
+      std::string curline;
+      int lines = 1;
+
+      Magick::TypeMetric metric;
+      while (!words.empty())
+      {
+        std::string temp = curline;
+
+        if (!curline.empty())
+        {
+          temp += " ";
+        }
+
+        temp += words.front();
+
+        overlay.fontTypeMetrics(temp, &metric);
+
+        if (metric.textWidth() > 1200)
+        {
+          wrappingStream << std::endl;
+          curline = words.front();
+
+          lines++;
+        } else {
+          if (!curline.empty())
+          {
+            wrappingStream << " ";
+          }
+
+          curline = temp;
+        }
+
+        wrappingStream << words.front();
+        words.pop_front();
+      }
+
+      std::string wrapped = wrappingStream.str();
+
+      overlay.annotate(
+        wrapped,
+        Magick::Geometry(1600, 228, 0, 710),
+        Magick::GravityType::NorthGravity);
+
+      // Add the achievement date
+      did theDid = db.getRandomDidForAchievement(ach.achievementId);
+
+      overlay.fontTypeMetrics(wrapped, &metric);
+
+      overlay.fontPointsize(20);
+      overlay.font("@" + config["date_font"].as<std::string>());
+      overlay.annotate(
+        theDid.date,
+        Magick::Geometry(1600, 228, 0, 710 + metric.textHeight() * lines - 22),
+        Magick::GravityType::NorthGravity);
+
+      // Make a shadow copy
+      Magick::Image shadow(overlay);
+      shadow.negate();
+      shadow.blur(0, 12);
+
+      // Read the game image
       Magick::Image image;
       image.read(imagePath);
 
@@ -60,18 +159,9 @@ int main(int argc, char** argv)
       image.scale("80x45");
       image.scale("1600x900");
 
-      // Add the text and moon image from Odyssey
+      // Add the generated overlay to it
+      image.composite(shadow, 0, 0, Magick::OverCompositeOp);
       image.composite(overlay, 0, 0, Magick::OverCompositeOp);
-      image.composite(moonColor, 672, 85, Magick::OverCompositeOp);
-
-      // Add the name of the achievement
-      image.fontPointsize(36);
-      image.fillColor("white");
-      image.font("@" + config["font"].as<std::string>());
-      image.annotate(
-        ach.title,
-        Magick::Geometry(0, 0, 0, 672),
-        Magick::GravityType::NorthGravity);
 
       // Output for debug
       image.magick("png");