✨ 重置权限

src/app/cache.ts

@@ -35,4 +35,8 @@ export default class Cache {
   public del(key: string): void {
     this.map.delete(key);
   }
+
+  public list(): string[] {
+    return Array.from(this.map.keys());
+  }
 }

src/app/service/permission/controller.ts

@@ -53,4 +53,11 @@ export default class PermissionController {
       permission,
     });
   }
+
+  // 重置权限
+  resetPermission(scriptId: number) {
+    return this.msg.syncSend("resetPermission", {
+      scriptId,
+    });
+  }
 }

src/pages/components/ScriptSetting/Permission.tsx

@@ -17,15 +17,13 @@ import {
 } from "@arco-design/web-react";
 import Table, { ColumnProps } from "@arco-design/web-react/es/Table";
 import { IconDelete } from "@arco-design/web-react/icon";
-import ScriptController from "@App/app/service/script/controller";
 
 const PermissionManager: React.FC<{
   script: Script;
 }> = ({ script }) => {
   const permissionCtrl = IoC.instance(
     PermissionController
   ) as PermissionController;
-  const scriptCtrl = IoC.instance(ScriptController) as ScriptController;
   const [permission, setPermission] = useState<Permission[]>([]);
   const [permissionVisible, setPermissionVisible] = useState<boolean>(false);
   const [permissionValue, setPermissionValue] = useState<Permission>();
@@ -176,7 +174,9 @@ const PermissionManager: React.FC<{
           <Popconfirm
             title={t("confirm_reset")}
             onOk={() => {
-              scriptCtrl.resetExclude(script.id, undefined).then(() => {});
+              permissionCtrl.resetPermission(script.id).then(() => {
+                setPermission([]);
+              });
             }}
           >
             <Button type="primary" size="small" status="warning">

src/runtime/background/permission_verify.ts

@@ -204,6 +204,25 @@ export default class PermissionVerify {
         return Promise.resolve(true);
       }
     );
+    // 监听重置权限
+    message.setHandler(
+      "resetPermission",
+      async (_action, data: { scriptId: number }) => {
+        // 先删除缓存
+        Cache.getInstance()
+          .list()
+          .forEach((key) => {
+            if (key.startsWith(`permission:${data.scriptId.toString()}:`)) {
+              Cache.getInstance().del(key);
+            }
+          });
+        // 从数据库中查询是否有此权限
+        await this.permissionDAO.delete({
+          scriptId: data.scriptId,
+        });
+        return Promise.resolve(true);
+      }
+    );
     this.dealConfirmQueue();
   }