Merge pull request #16 from basakest/UpdatableAdapter

feat: support Casbin UpdatableAdapter interface

Author: leeqvip

src/Adapters/DatabaseAdapter.php

@@ -15,8 +15,9 @@
 use Casbin\Persist\FilteredAdapter;
 use Casbin\Persist\Adapters\Filter;
 use Casbin\Exceptions\InvalidFilterTypeException;
+use Casbin\Persist\UpdatableAdapter;
 
-class DatabaseAdapter implements Adapter, BatchAdapter, FilteredAdapter
+class DatabaseAdapter implements Adapter, BatchAdapter, FilteredAdapter, UpdatableAdapter
 {
     use AdapterHelper;
 
@@ -261,4 +262,32 @@ public function setFiltered(bool $filtered): void
     {
         $this->filtered = $filtered;
     }
+
+    /**
+     * 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
+    {
+        $instance = RulesModel::create();
+        $where = [];
+        $update = [];
+        $where['ptype'] = $ptype;
+
+        foreach ($oldRule as $key => $value) {
+            $where['v' . $key] = $value;
+        }
+
+        $instance = $instance->where($where)->get();
+        foreach ($newPolicy as $key => $value) {
+            $update['v' . $key] = $value;
+        }
+
+        $instance->update($update);
+    }
 }

tests/DatabaseAdapterTest.php

@@ -189,4 +189,33 @@ public function testLoadFilteredPolicy()
             ['alice', 'data1', 'read'],
         ], $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());
+    }
 }