Merge pull request #4 from basakest/updatable-adapter

feat: support Casbin UpdatableAdapter interface

Author: leeqvip

src/Adapter.php

@@ -4,6 +4,7 @@
 
 use Casbin\Persist\Adapter as AdapterContract;
 use Casbin\Persist\BatchAdapter as BatchAdapterContract;
+use Casbin\Persist\UpdatableAdapter as UpdatableAdapterContract;
 use Casbin\Persist\AdapterHelper;
 use Casbin\Model\Model;
 use Medoo\Medoo;
@@ -13,7 +14,7 @@
  *
  * @author [email protected]
  */
-class Adapter implements AdapterContract, BatchAdapterContract
+class Adapter implements AdapterContract, BatchAdapterContract, UpdatableAdapterContract
 {
     use AdapterHelper;
 
@@ -199,7 +200,7 @@ public function removePolicy(string $sec, string $ptype, array $rule): void
      * @param string[][] $rules
      */
     public function removePolicies(string $sec, string $ptype, array $rules): void
-    { 
+    {
         $this->database->action(function () use ($sec, $ptype, $rules) {
             foreach ($rules as $rule) {
                 $this->removePolicy($sec, $ptype, $rule);
@@ -233,6 +234,31 @@ public function removeFilteredPolicy(string $sec, string $ptype, int $fieldIndex
         $this->database->delete($this->casbinRuleTableName, ['AND' => $where]);
     }
 
+    /**
+     * 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
+    {
+        $where = ['ptype' => $ptype,];
+        
+        foreach ($oldRule as $k => $v) {
+            $where['v' . strval($k)] = $v;
+        }
+        
+        $columns = [];
+        foreach ($newPolicy as $k => $v) {
+            $columns['v' . strval($k)][$where['v' . strval($k)]] = $v;
+        }
+
+        $this->database->replace($this->casbinRuleTableName, $columns, $where);
+    }
+
     /**
      * Gets database.
      *

tests/AdapterTest.php

@@ -170,6 +170,34 @@ public function testRemoveFilteredPolicy()
         $this->assertFalse($e->enforce('alice', 'data2', 'write'));
     }
 
+    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());
+    }
+
     protected function env($key, $default = null)
     {
         $value = getenv($key);