feat: register middleware

leafsphp · Dec 14, 2024 · 8fda649 · 8fda649
1 parent 23ff45b
commit 8fda649
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/src/Auth.php b/src/Auth.php
@@ -608,6 +608,38 @@ public function middleware(string $middleware, callable $callback)
             });
         }
 
+        if ($middleware === 'is') {
+            return app()->registerMiddleware('is', function ($role) use ($callback) {
+                if ($this->user()?->isNot($role)) {
+                    $callback();
+                }
+            });
+        }
+
+        if ($middleware === 'isNot') {
+            return app()->registerMiddleware('isNot', function ($role) use ($callback) {
+                if ($this->user()?->is($role)) {
+                    $callback();
+                }
+            });
+        }
+
+        if ($middleware === 'can') {
+            return app()->registerMiddleware('can', function ($role) use ($callback) {
+                if ($this->user()?->can($role)) {
+                    $callback();
+                }
+            });
+        }
+
+        if ($middleware === 'cannot') {
+            return app()->registerMiddleware('cannot', function ($role) use ($callback) {
+                if ($this->user()?->cannot($role)) {
+                    $callback();
+                }
+            });
+        }
+
         app()->registerMiddleware($middleware, $callback);
     }