Feature: Override custom IAE handler

geggleto · Jan 8, 2016 · 0f36da7 · 0f36da7
1 parent 74973db
commit 0f36da7
Showing 1 changed file with 26 additions and 10 deletions.
diff --git a/src/AclRepository.php b/src/AclRepository.php
@@ -11,6 +11,7 @@
 use Psr\Http\Message\ResponseInterface;
 use Psr\Http\Message\ServerRequestInterface;
 use Zend\Permissions\Acl\Acl;
+use Zend\Permissions\Acl\Exception\InvalidArgumentException;
 use Zend\Permissions\Acl\Role\GenericRole as Role;
 use Zend\Permissions\Acl\Resource\GenericResource as Resource;
 
@@ -26,6 +27,9 @@ class AclRepository
      */
     protected $role;
 
+
+    protected $handler;
+
     /**
      * AclRepository constructor.
      *
@@ -80,6 +84,23 @@ public function __construct(array $role, array $config = [])
                 }
             }
         }
+
+        $this->handler = function (ServerRequestInterface $requestInterface) {
+            $route = $requestInterface->getAttribute('route');
+            if (!empty($route)) {
+                foreach ($this->role as $role) {
+                    if ($this->isAllowed($role, $route->getPattern())) {
+                        return true;
+                    }
+                }
+            }
+            return false;
+        };
+
+    }
+
+    public function setCustomHandler(callable $handler) {
+        $this->handler = $handler;
     }
 
     /**
@@ -149,20 +170,15 @@ public function __invoke(ServerRequestInterface $requestInterface, ResponseInter
 
         $route = '/' . ltrim($requestInterface->getUri()->getPath(), '/');
 
-        foreach ($this->role as $role) {
-            if ($this->isAllowed($role, $route)) {
-                $allowed = true;
-            }
-        }
-
-        //This is likely Slim 3 specific code...
-        $route = $requestInterface->getAttribute('route');
-        if (!empty($route)) {
+        try {
             foreach ($this->role as $role) {
-                if ($this->isAllowed($role, $route->getPattern())) {
+                if ($this->isAllowed($role, $route)) {
                     $allowed = true;
                 }
             }
+        } catch (InvalidArgumentException $iae) {
+            $fn = $this->handler;
+            $allowed = $fn($requestInterface);
         }
 
         if ($allowed) {