feat(cookies): Rework delete into a method and implement fromDBtoken

2 files changed, 11 insertions, 3 deletions

diff --git a/controllers/session.php b/controllers/session.php
index 179afe6..e02cc79 100644
--- a/controllers/session.php
+++ b/controllers/session.php
@@ -25,7 +25,8 @@ function on_post() {
 function on_delete() {
     global $TOKEN;
     try {
-        Database\Cookie::delete($TOKEN);
+        $token = Database\Cookie::fromDBtoken($TOKEN);
+        $token->delete();
     }
     catch(Exception $e) {}
 }
diff --git a/models/cookies.php b/models/cookies.php
index ddc7d1c..b38c5c0 100644
--- a/models/cookies.php
+++ b/models/cookies.php
@@ -25,10 +25,17 @@ class Cookie extends Table {
         return User::fromDBuid($UID);
     }
 
-    static function delete(string $token) {
+    static function fromDBtoken(string $token) : Cookie {
+        return Table::_fromDB(
+            "SELECT * FROM Cookies WHERE Token = \"$token\"",
+            'Database\Cookie'
+        );
+    }
+
+    function delete() {
         Table::_delete(
             'Cookies',
-            "Token = \"$token\""
+            "Token = \"$this->Token\""
         );
     }
 }