Merge pull request #5 from basakest/updatable-adapter

feat: support Casbin UpdatableAdapter interface

Author: leeqvip

src/Adapter.php

@@ -5,6 +5,7 @@
 use Casbin\Exceptions\CasbinException;
 use Casbin\Persist\Adapter as AdapterContract;
 use Casbin\Persist\BatchAdapter as BatchAdapterContract;
+use Casbin\Persist\UpdatableAdapter as UpdatableAdapterContract;
 use Casbin\Persist\AdapterHelper;
 use Cake\ORM\TableRegistry;
 use CasbinAdapter\Cake\Model\Table\CasbinRuleTable;
@@ -14,7 +15,7 @@
  *
  * @author [email protected]
  */
-class Adapter implements AdapterContract, BatchAdapterContract
+class Adapter implements AdapterContract, BatchAdapterContract, UpdatableAdapterContract
 {
     use AdapterHelper;
 
@@ -134,6 +135,30 @@ public function removePolicies(string $sec, string $ptype, array $rules): void
         });
     }
 
+    /**
+     * Updates a policy rule from storage.
+     * This is part of the Auto-Save feature.
+     *
+     * @param string $sec
+     * @param string $ptype
+     * @param string[] $oldRule
+     * @param string[] $newPolicy
+     */
+    public function updatePolicy(string $sec, string $ptype, array $oldRule, array $newPolicy): void
+    {
+        $entity = $this->table->find()->where(['ptype' => $ptype]);
+        foreach ($oldRule as $k => $v) {
+            $entity->where(['v' . $k => $v]);
+        }
+        $first = $entity->first();
+
+        foreach ($newPolicy as $k => $v) {
+            $key = 'v' . $k;
+            $first->$key = $v;
+        }
+        $this->table->save($first);
+    }
+
     protected function getTable()
     {
         return  TableRegistry::getTableLocator()->get('CasbinRule', [

tests/DatabaseAdapterTest.php

@@ -133,4 +133,32 @@ public function testRemovePolicies()
             ['bob', 'data2', 'write']
         ], $e->getPolicy());
     }
+
+    public function testUpdatePolicy()
+    {
+        $e = $this->getEnforcer();
+        $this->assertEquals([
+            ['alice', 'data1', 'read'],
+            ['bob', 'data2', 'write'],
+            ['data2_admin', 'data2', 'read'],
+            ['data2_admin', 'data2', 'write'],
+        ], $e->getPolicy());
+
+        $e->updatePolicy(
+            ['alice', 'data1', 'read'],
+            ['alice', 'data1', 'write']
+        );
+
+        $e->updatePolicy(
+            ['bob', 'data2', 'write'],
+            ['bob', 'data2', 'read']
+        );
+
+        $this->assertEquals([
+            ['alice', 'data1', 'write'],
+            ['bob', 'data2', 'read'],
+            ['data2_admin', 'data2', 'read'],
+            ['data2_admin', 'data2', 'write'],
+        ], $e->getPolicy());
+    }
 }