5 files changed, 61 insertions, 10 deletions

@@ -82,6 +82,47 @@ achievement database::getRandomAchievement() const
   return result;
 }
 
+bool database::doesGameHaveImages(int gameId) const
+{
+  std::string queryString = "SELECT COUNT(*) FROM images WHERE game_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, gameId) != 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);
+  }
+
+  int result = sqlite3_column_int(ppstmt, 0);
+
+  sqlite3_finalize(ppstmt);
+
+  return (result > 0);
+}
+
 std::string database::getRandomImageForGame(int gameId) const
 {
   std::string queryString = "SELECT filename FROM images WHERE game_id = ? ORDER BY RANDOM() LIMIT 1";
@@ -46,6 +46,8 @@ public:
 
   achievement getRandomAchievement() const;
 
+  bool doesGameHaveImages(int gameId) const;
+
   std::string getRandomImageForGame(int gameId) const;
 
   did getRandomDidForAchievement(int achievementId) const;
@@ -65,9 +65,6 @@ int main(int argc, char** argv)
     std::cout << "Generating tweet" << std::endl;
 
     achievement ach = db.getRandomAchievement();
-    std::string imageName = db.getRandomImageForGame(ach.gameId);
-    std::string imagePath = config["images"].as<std::string>()
-      + "/" + imageName;
 
     Magick::Image moonColor;
     moonColor.read("res/" + ach.color + ".png");
@@ -150,14 +147,25 @@ int main(int argc, char** argv)
       shadow.negate();
       shadow.blur(0, 12);
 
-      // Read the game image
+      // Read the game image, using a default if the game has no images
       Magick::Image image;
-      image.read(imagePath);
 
-      // Stretch and pixelate it
-      image.transform("1600x900!");
-      image.scale("80x45");
-      image.scale("1600x900");
+      if (db.doesGameHaveImages(ach.gameId))
+      {
+        std::string imageName = db.getRandomImageForGame(ach.gameId);
+        std::string imagePath = config["images"].as<std::string>()
+          + "/" + imageName;
+
+        image.read(imagePath);
+
+        // Stretch and pixelate it
+        image.transform("1600x900!");
+        image.scale("80x45");
+        image.scale("1600x900");
+      } else {
+        image.read("res/default.png");
+        image.transform("1600x900!");
+      }
 
       // Add the generated overlay to it
       image.composite(shadow, 0, 0, Magick::OverCompositeOp);
@@ -1 +1 @@
-The images in this directory are originally from the game Super Mario Odyssey, which is copyrighted by Nintendo.
\ No newline at end of file
+The images in this directory, apart from overlay.png and overlay.psd, are originally from the game Super Mario Odyssey, which is copyrighted by Nintendo.
\ No newline at end of file