From 55ff555f46867705cb26cef97885543d00f17acc Mon Sep 17 00:00:00 2001
From: flashwave <me@flash.moe>
Date: Tue, 31 Jan 2023 00:24:21 +0000
Subject: [PATCH] Added clicks counter to redirects.

---
 database/2023_01_31_001812_add_clicks_counter.php | 12 ++++++++++++
 src/RedirectorRoutes.php                          |  8 ++++++--
 2 files changed, 18 insertions(+), 2 deletions(-)
 create mode 100644 database/2023_01_31_001812_add_clicks_counter.php

diff --git a/database/2023_01_31_001812_add_clicks_counter.php b/database/2023_01_31_001812_add_clicks_counter.php
new file mode 100644
index 0000000..d66902a
--- /dev/null
+++ b/database/2023_01_31_001812_add_clicks_counter.php
@@ -0,0 +1,12 @@
+<?php
+use Index\Data\IDbConnection;
+use Index\Data\Migration\IDbMigration;
+
+final class AddClicksCounter_20230131_001812 implements IDbMigration {
+    public function migrate(IDbConnection $conn): void {
+        $conn->execute('
+            ALTER TABLE awk_redirects
+                ADD COLUMN redir_clicks BIGINT UNSIGNED NOT NULL DEFAULT 0 AFTER redir_url;
+        ');
+    }
+}
diff --git a/src/RedirectorRoutes.php b/src/RedirectorRoutes.php
index 1baf860..ded8926 100644
--- a/src/RedirectorRoutes.php
+++ b/src/RedirectorRoutes.php
@@ -47,7 +47,7 @@ final class RedirectorRoutes {
     }
 
     public function redirectDatabase($response, $request, $id) {
-        $getInfo = $this->dbConn->prepare('SELECT redir_url FROM awk_redirects WHERE redir_id = ? OR redir_vanity = ?');
+        $getInfo = $this->dbConn->prepare('SELECT redir_id, redir_url FROM awk_redirects WHERE redir_id = ? OR redir_vanity = ?');
         $getInfo->addParameter(1, $id, DbType::INTEGER);
         $getInfo->addParameter(2, $id, DbType::STRING);
         $getInfo->execute();
@@ -56,7 +56,11 @@ final class RedirectorRoutes {
         if(!$info->next())
             return 404;
 
-        $this->redirect($response, $request, $info->getString(0));
+        $incClicks = $this->dbConn->prepare('UPDATE awk_redirects SET redir_clicks = redir_clicks + 1 WHERE redir_id = ?');
+        $incClicks->addParameter(1, $info->getInteger(0), DbType::INTEGER);
+        $incClicks->execute();
+
+        $this->redirect($response, $request, $info->getString(1));
     }
 
     private function redirectSimple($response, $request, string $format, string $argument) {