From 4e9d97268fc2e662378654e46562fe93be4348dc Mon Sep 17 00:00:00 2001
From: James Collins <james.collins@outlook.com.au>
Date: Sat, 6 May 2023 22:38:32 +1000
Subject: [PATCH] remove any query items

---
 public/shortlink.php | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/public/shortlink.php b/public/shortlink.php
index 8212219..9fdf11fd 100644
--- a/public/shortlink.php
+++ b/public/shortlink.php
@@ -7,7 +7,7 @@ $dotenv->load();
 
 $dbHost = $_ENV['DB_HOST'];
 $dbPort = $_ENV['DB_PORT'];
-$dbName = $_ENV['DB_NAME'];
+$dbName = $_ENV['DB_DATABASE'];
 $dbUser = $_ENV['DB_USERNAME'];
 $dbPass = $_ENV['DB_PASSWORD'];
 
@@ -22,6 +22,9 @@ if ($conn->connect_error) {
 // get code from URL
 $code = $_SERVER['REQUEST_URI'];
 $code = trim($code, '/');
+if (($pos = strpos($code, '?')) !== false) {
+    $code = substr($code, 0, $pos);
+}
 
 // lookup code in database
 $sql = "SELECT url, used FROM shortlinks WHERE code = '$code'";