middleware trait and added to router

1 files changed, 24 insertions, 0 deletions

diff --git a/src/Concerns/HasMiddleware.php b/src/Concerns/HasMiddleware.php
new file mode 100644
index 0000000..e0c6b1a
--- /dev/null
+++ b/src/Concerns/HasMiddleware.php
@@ -0,0 +1,24 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Lightscale\Router\Concerns;
+
+use Lightscale\Router\Contracts\Middleware;
+
+trait HasMiddleware
+{
+    /** @var Middleware[] */
+    private array $middleware = [];
+
+    /** @param Middleware[]|Middleware $middleware */
+    public function addMiddleware(array|Middleware $middleware): static
+    {
+        $middleware = is_array($middleware) ? $middleware : [$middleware];
+        $this->middleware = [
+            ...$this->middleware,
+            ...$middleware,
+        ];
+        return $this;
+    }
+}