Implement session class and display registration success message

Author: thecodeholic

controllers/SiteController.php

@@ -8,6 +8,7 @@
 namespace app\controllers;
 
 
+use app\core\Application;
 use app\core\Controller;
 use app\core\Request;
 use app\models\User;
@@ -39,6 +40,8 @@ public function register(Request $request)
         if ($request->getMethod() === 'post') {
             $registerModel->loadData($request->getBody());
             if ($registerModel->validate() && $registerModel->save()) {
+                Application::$app->session->setFlash('success', 'Thanks for registering');
+                Application::$app->response->redirect('/');
                 return 'Show success page';
             }
 

core/Application.php

@@ -21,6 +21,7 @@ class Application
     public Response $response;
     public Controller $controller;
     public Database $db;
+    public Session $session;
 
     public function __construct($rootDir, $config)
     {
@@ -30,6 +31,7 @@ public function __construct($rootDir, $config)
         $this->response = new Response();
         $this->router = new Router($this->request, $this->response);
         $this->db = new Database($config['db']);
+        $this->session = new Session();
     }
 
     public function run()

core/Session.php

@@ -0,0 +1,59 @@
+<?php
+/**
+ * User: TheCodeholic
+ * Date: 7/24/2020
+ * Time: 11:18 PM
+ */
+
+namespace app\core;
+
+
+/**
+ * Class Session
+ *
+ * @author  Zura Sekhniashvili <[email protected]>
+ * @package app\core
+ */
+class Session
+{
+    protected const FLASH_KEY = 'flash_messages';
+
+    public function __construct()
+    {
+        session_start();
+        $flashMessages = $_SESSION[self::FLASH_KEY] ?? [];
+        foreach ($flashMessages as $key => &$flashMessage) {
+            $flashMessage['remove'] = true;
+        }
+        $_SESSION[self::FLASH_KEY] = $flashMessages;
+    }
+
+    public function setFlash($key, $message)
+    {
+        $_SESSION[self::FLASH_KEY][$key] = [
+            'remove' => false,
+            'value' => $message
+        ];
+    }
+
+    public function getFlash($key)
+    {
+        return $_SESSION[self::FLASH_KEY][$key]['value'] ?? false;
+    }
+
+    public function __destruct()
+    {
+        $this->removeFlashMessages();
+    }
+
+    private function removeFlashMessages()
+    {
+        $flashMessages = $_SESSION[self::FLASH_KEY] ?? [];
+        foreach ($flashMessages as $key => $flashMessage) {
+            if ($flashMessage['remove']) {
+                unset($flashMessages[$key]);
+            }
+        }
+        $_SESSION[self::FLASH_KEY] = $flashMessages;
+    }
+}

views/layouts/main.php

@@ -43,6 +43,11 @@
 </nav>
 
 <div class="container">
+    <?php if (\app\core\Application::$app->session->getFlash('success')): ?>
+        <div class="alert alert-success">
+            <p><?php echo \app\core\Application::$app->session->getFlash('success') ?></p>
+        </div>
+    <?php endif; ?>
     {{content}}
 </div>