Setup structure for storing middleware and added test todos

1 files changed, 12 insertions, 0 deletions

diff --git a/src/RouteDefinition.php b/src/RouteDefinition.php
index 860c918..b015ba8 100644
--- a/src/RouteDefinition.php
+++ b/src/RouteDefinition.php
@@ -4,6 +4,8 @@ declare(strict_types=1);
 
 namespace Lightscale\Router;
 
+use Lightscale\Router\Contracts\Middleware;
+
 class RouteDefinition
 {
     public function __construct(
@@ -31,6 +33,16 @@ class RouteDefinition
         return $this;
     }
 
+    /** @param Middleware|Middleware[] $value */
+    public function middleware(Middleware|array $value): static
+    {
+        $value = is_array($value) ? $value : [$value];
+
+        $this->route->addMiddleware($value);
+
+        return $this;
+    }
+
     public function getRoute(): Route
     {
         return $this->route;