feat: support Casbin UpdatableAdapter interface (#25)

Author: basakest

src/adapter/DatabaseAdapter.php

@@ -6,13 +6,15 @@
 use Casbin\Model\Model;
 use Casbin\Persist\Adapter;
 use Casbin\Persist\AdapterHelper;
+use Casbin\Persist\UpdatableAdapter;
+use think\facade\Db;
 
 /**
  * DatabaseAdapter.
  *
  * @author [email protected]
  */
-class DatabaseAdapter implements Adapter
+class DatabaseAdapter implements Adapter, UpdatableAdapter
 {
     use AdapterHelper;
 
@@ -151,4 +153,47 @@ public function removeFilteredPolicy(string $sec, string $ptype, int $fieldIndex
             }
         }
     }
+
+    /**
+     * 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 = $this->model->where('ptype', $ptype);
+        foreach ($oldRule as $key => $value) {
+            $instance->where('v' . strval($key), $value);
+        }
+        $instance = $instance->find();
+
+        foreach ($newPolicy as $key => $value) {
+            $column = 'v' . strval($key);
+            $instance->$column = $value;
+        }
+
+        $instance->save();
+    }
+
+    /**
+     * UpdatePolicies updates some policy rules to storage, like db, redis.
+     *
+     * @param string $sec
+     * @param string $ptype
+     * @param string[][] $oldRules
+     * @param string[][] $newRules
+     * @return void
+     */
+    public function updatePolicies(string $sec, string $ptype, array $oldRules, array $newRules): void
+    {
+        Db::transaction(function () use ($sec, $ptype, $oldRules, $newRules) {
+            foreach ($oldRules as $i => $oldRule) {
+                $this->updatePolicy($sec, $ptype, $oldRule, $newRules[$i]);
+            }
+        });
+    }
 }

tests/DatabaseAdapterTest.php

@@ -76,4 +76,63 @@ public function testRemoveFilteredPolicy()
             $this->assertFalse(Enforcer::enforce('alice', 'data2', 'write'));
         });
     }
-}
+
+    public function testUpdatePolicy()
+    {
+        $this->testing(function () {
+            $this->assertEquals([
+                ['alice', 'data1', 'read'],
+                ['bob', 'data2', 'write'],
+                ['data2_admin', 'data2', 'read'],
+                ['data2_admin', 'data2', 'write'],
+            ], Enforcer::getPolicy());
+    
+            Enforcer::updatePolicy(
+                ['alice', 'data1', 'read'],
+                ['alice', 'data1', 'write']
+            );
+    
+            Enforcer::updatePolicy(
+                ['bob', 'data2', 'write'],
+                ['bob', 'data2', 'read']
+            );
+    
+            $this->assertEquals([
+                ['alice', 'data1', 'write'],
+                ['bob', 'data2', 'read'],
+                ['data2_admin', 'data2', 'read'],
+                ['data2_admin', 'data2', 'write'],
+            ], Enforcer::getPolicy());
+        });
+    }
+
+    public function testUpdatePolicies()
+    {
+        $this->testing(function () {
+            $this->assertEquals([
+                ['alice', 'data1', 'read'],
+                ['bob', 'data2', 'write'],
+                ['data2_admin', 'data2', 'read'],
+                ['data2_admin', 'data2', 'write'],
+            ], Enforcer::getPolicy());
+    
+            $oldPolicies = [
+                ['alice', 'data1', 'read'],
+                ['bob', 'data2', 'write']
+            ];
+            $newPolicies = [
+                ['alice', 'data1', 'write'],
+                ['bob', 'data2', 'read']
+            ];
+    
+            Enforcer::updatePolicies($oldPolicies, $newPolicies);
+    
+            $this->assertEquals([
+                ['alice', 'data1', 'write'],
+                ['bob', 'data2', 'read'],
+                ['data2_admin', 'data2', 'read'],
+                ['data2_admin', 'data2', 'write'],
+            ], Enforcer::getPolicy());
+        });
+    }
+}