Execute middleware

Author: na2axl

src/WaterPipe/HTTP/Response/Response.php

@@ -33,6 +33,7 @@
 namespace ElementaryFramework\WaterPipe\HTTP\Response;
 
 use ElementaryFramework\WaterPipe\Exceptions\FileNotFoundException;
+use ElementaryFramework\WaterPipe\WaterPipe;
 use ElementaryFramework\WaterPipe\WaterPipeConfig;
 
 class Response
@@ -69,6 +70,9 @@ public function __construct()
      */
     public function send()
     {
+        // Execute middleware
+        WaterPipe::triggerBeforeSendEvent($this);
+
         // Set status code
         $code = $this->_status->getCode();
         $text = $this->_status->getDescription();

src/WaterPipe/Routing/Middleware/Middleware.php

@@ -33,25 +33,11 @@
 namespace ElementaryFramework\WaterPipe\Routing\Middleware;
 
 use ElementaryFramework\WaterPipe\HTTP\Request\Request;
+use ElementaryFramework\WaterPipe\HTTP\Response\Response;
 
 abstract class Middleware
 {
-    /**
-     * @var Request
-     */
-    protected $_request;
+    public abstract function beforeExecute(Request &$request);
 
-    /**
-     * RouteAction constructor.
-     *
-     * @throws \Exception
-     */
-    public function __construct()
-    {
-        $this->_request =& Request::getInstance();
-    }
-
-    public abstract function beforeExecute();
-
-    public abstract function afterExecute();
+    public abstract function beforeSend(Response &$response);
 }

src/WaterPipe/WaterPipe.php

@@ -42,6 +42,14 @@
 
 class WaterPipe
 {
+    /**
+     * Stores the only running instance
+     * of WaterPipe.
+     *
+     * @var WaterPipe
+     */
+    private static $_runningInstance = null;
+
     /**
      * Defines if the pipe is running or not.
      *
@@ -74,6 +82,40 @@ class WaterPipe
 
     private $_pipesRegistry;
 
+    /**
+     * @return WaterPipe
+     */
+    public static function getRunningInstance(): WaterPipe
+    {
+        return self::$_runningInstance;
+    }
+
+    /**
+     * Triggers all middlewares with before send event
+     * defined on the running pipe.
+     *
+     * @param Response $response The response on which trigger the event.
+     */
+    public static function triggerBeforeSendEvent(Response &$response)
+    {
+        foreach (self::$_runningInstance->_middlewareRegistry as $middleware) {
+            $middleware->beforeSend($response);
+        }
+    }
+
+    /**
+     * Triggers all middlewares with before execute event
+     * defined on the running pipe.
+     *
+     * @param Request $request The request to execute.
+     */
+    public static function triggerBeforeExecuteEvent(Request &$request)
+    {
+        foreach (self::$_runningInstance->_middlewareRegistry as $middleware) {
+            $middleware->beforeExecute($request);
+        }
+    }
+
     public function __construct()
     {
         $this->_isRunning = false;
@@ -154,6 +196,8 @@ public function run()
     {
         $this->_isRunning = true;
 
+        self::$_runningInstance = $this;
+
         Request::capture();
 
         $pipe = $this->_findSubPipe();
@@ -165,12 +209,19 @@ public function run()
         }
     }
 
+    /**
+     * @param string $baseUri
+     * @throws \Exception
+     */
     private function _runBase(string $baseUri)
     {
         $this->_baseUri = $baseUri;
         $this->run();
     }
 
+    /**
+     * @return array|null
+     */
     private function _findSubPipe()
     {
         foreach ($this->_pipesRegistry as $baseUri => $pipe) {
@@ -223,22 +274,28 @@ private function _executeRequest()
             throw new \Exception("404 Error");
         }
 
-        // NOTE: No code will normally not be executed after this block...
+        if (!is_callable($runner) && !is_array($runner) && !is_subclass_of($runner, RouteAction::class)) {
+            // TODO: Proper exception
+            throw new \Exception("Malformed route action");
+        }
+
+        // Execute middleware
+        self::triggerBeforeExecuteEvent(Request::getInstance());
+
+        // NOTE: No code will be executed after this block...
         if (is_callable($runner) || is_array($runner)) {
             call_user_func_array($runner, array(Request::getInstance(), new Response()));
         } elseif (is_subclass_of($runner, RouteAction::class)) {
             if (is_string($runner)) {
                 $runner = new $runner;
             }
             $runner->execute();
-        } else {
-            throw new \Exception("Malformed route action");
         }
     }
 
     /**
      * @param array $routes
-     * @return callable
+     * @return callable|RouteAction
      * @throws Exceptions\RequestUriBuilderException
      */
     private function _getActionForRoutes(array $routes)