From 0b72962ce56f7b537548c9bd2576c5d0a09fae55 Mon Sep 17 00:00:00 2001
From: overflowerror <mail@overflowerror.com>
Date: Sat, 27 Jul 2024 18:05:18 +0200
Subject: [PATCH] feat: Add new mobs to database

---
 .gitignore            |  3 ++-
 bin/cron.php          | 10 ++++++++--
 credentials.templ.php |  7 +++++++
 lib/database.php      |  7 +++++++
 lib/updateData.php    | 17 +++++++++++++++++
 5 files changed, 41 insertions(+), 3 deletions(-)
 create mode 100644 credentials.templ.php
 create mode 100644 lib/database.php

diff --git a/.gitignore b/.gitignore
index 723ef36..4cb256c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
-.idea
\ No newline at end of file
+.idea
+credentials.php
\ No newline at end of file
diff --git a/bin/cron.php b/bin/cron.php
index 8a3ea40..e05d5f2 100644
--- a/bin/cron.php
+++ b/bin/cron.php
@@ -36,8 +36,14 @@ $mobs = array_reduce($mobs, function ($mobs, $mob) {
 }, []);
 
 echo "Downloading images...\n";
-foreach ($mobs as $mob) {
+foreach ($mobs as &$mob) {
+    echo "   ... " . $mob["name"] . "\n";
     $filename = downloadImage($mob["image"], $mob["name"]);
     $mob["filename"] = $filename;
-    var_dump($mob);
+}
+
+echo "Adding to database...\n";
+foreach ($mobs as &$mob) {
+    echo "   ... " . $mob["name"] . "\n";
+    addOrUpdateMob($mob["name"], $mob["filename"]);
 }
\ No newline at end of file
diff --git a/credentials.templ.php b/credentials.templ.php
new file mode 100644
index 0000000..2347f3f
--- /dev/null
+++ b/credentials.templ.php
@@ -0,0 +1,7 @@
+<?php
+
+const POSTGRES_HOST = "%DBHOST%";
+const POSTGRES_PORT = 5432;
+const POSTGRES_DBNAME = "%DBNAME%";
+const POSTGRES_USER = "%DBUSER%";
+const POSTGRES_PASSWORD = "%DBPASSWORD%";
\ No newline at end of file
diff --git a/lib/database.php b/lib/database.php
new file mode 100644
index 0000000..74cf1fe
--- /dev/null
+++ b/lib/database.php
@@ -0,0 +1,7 @@
+<?php
+
+require_once __DIR__ . "/../credentials.php";
+
+$dsn = "pgsql:host=" . POSTGRES_HOST . ";dbname=" . POSTGRES_DBNAME . ";port=" . POSTGRES_PORT;
+
+$pdo = new PDO($dsn, POSTGRES_USER, POSTGRES_PASSWORD);
diff --git a/lib/updateData.php b/lib/updateData.php
index 55b47f7..d24c52c 100644
--- a/lib/updateData.php
+++ b/lib/updateData.php
@@ -1,6 +1,7 @@
 <?php
 
 require_once __DIR__ . "/request.php";
+require_once __DIR__ . "/database.php";
 
 function handleWikiRequest(string $args, bool $rvslots = true): array {
     $url = "https://minecraft.wiki/api.php?action=query&format=json&" . $args;
@@ -92,4 +93,20 @@ function downloadImage(string $url, string $mobname): string {
     fclose($file);
 
     return $name;
+}
+
+function addOrUpdateMob(string $name, string $filename) {
+    global $pdo;
+
+    $query = $pdo->prepare("SELECT name from mobs where name = ?");
+    $query->execute([$name]) or die("unable to check if mob exists");
+    if ($query->rowCount() == 0) {
+        $query = $pdo->prepare("INSERT INTO mobs (name, image) VALUES (?, ?)");
+        $query->execute([$name, $filename]) or die("unable to add new mob");
+        echo "      added\n";
+    } else {
+        $query = $pdo->prepare("UPDATE mobs SET image = ? WHERE name = ?");
+        $query->execute([$filename, $name]) or die("unable to update mob");
+        echo "      updated\n";
+    }
 }
\ No newline at end of file