From 5c891551df4b46076081ffd02e264e7ff6e4d2f6 Mon Sep 17 00:00:00 2001
From: William <william@willy.club>
Date: Mon, 28 Feb 2022 04:14:53 +0100
Subject: [PATCH] Commit

---
 app/core/ErrorHandler.php               |  3 ++-
 app/core/User.php                       | 28 ++++++++++++++++---------
 app/view/pages/teamtable/edit/index.php |  3 +--
 3 files changed, 21 insertions(+), 13 deletions(-)

diff --git a/app/core/ErrorHandler.php b/app/core/ErrorHandler.php
index 86f9618..ed1fe11 100644
--- a/app/core/ErrorHandler.php
+++ b/app/core/ErrorHandler.php
@@ -1,6 +1,7 @@
 <?php
 
-// Janky error handler page thingy that looks cool
+// Bit of a janky way to display a custom page on error or exception.
+// But looks pretty cool and professional!
 class ErrorHandler 
 {
     public array $errors; // Holds error messages (even though for now we only display a maximum of 1)
diff --git a/app/core/User.php b/app/core/User.php
index 89cf8b5..9d7e801 100644
--- a/app/core/User.php
+++ b/app/core/User.php
@@ -1,8 +1,8 @@
 <?php
 
-// TODO: ...
 class User 
 {
+    private const    SESSION_KEY = 'UserClass';
     private Session  $session;
     private Database $database;
 
@@ -19,7 +19,7 @@ class User
         $this->session  = $session;
         $this->database = $database;
 
-        $user = $this->session->get('user');
+        $user = $this->session->get(self::SESSION_KEY);
 
         // Check if user session has been set
         if (!$user)
@@ -38,17 +38,25 @@ class User
         }
 
         // All is good, we should be logged in now! (hopefully)
-        $this->loggedIn = TRUE;
-        $this->username = $user['username'];
-        $this->password = $user['password'];
+        $this->loggedIn   = TRUE;
+        $this->username   = $user['username'];
+        $this->password   = $user['password'];
+        $this->powerLevel = $this->getPowerLevel();
+    }
 
-        // Set powerLevel
+    // Get current user power level
+    private function getPowerLevel(): int
+    {
+        if (!$this->loggedIn)
+        {
+            throw new Exception("Can't get power level without being logged in!");
+        }
         $sth = $this->database->conn->prepare(
-            'SELECT * FROM brukertabell WHERE Navn = ? AND Passord = ?'
+            'SELECT Nivå FROM brukertabell WHERE Navn = ? AND Passord = ?'
         );
         $sth->execute([$this->username, $this->password]);
         $row = $sth->fetch(PDO::FETCH_ASSOC);
-        $this->powerLevel = $row['Nivå'];
+        return $row['Nivå'];
     }
 
     // Set session if user and password match
@@ -56,7 +64,7 @@ class User
     {
         if ($this->authenticate($username, $password))
         {
-            $this->session->set('user', [
+            $this->session->set(self::SESSION_KEY, [
                 'loggedIn' => TRUE,
                 'username' => $username,
                 'password' => $password
@@ -82,6 +90,6 @@ class User
 
     public function logout(): void
     {
-        $this->session->remove('user');
+        $this->session->remove(self::SESSION_KEY);
     }
 }
\ No newline at end of file
diff --git a/app/view/pages/teamtable/edit/index.php b/app/view/pages/teamtable/edit/index.php
index 45ceee1..9a0f789 100644
--- a/app/view/pages/teamtable/edit/index.php
+++ b/app/view/pages/teamtable/edit/index.php
@@ -36,5 +36,4 @@
         echo '</tr>';
     }
     ?>
-</table>
-<span class="float-right">[&nbsp;<a class="success" href="add.php">Opprett lag</a>&nbsp;]</span>
\ No newline at end of file
+</table>
\ No newline at end of file