Merge pull request #64 from sj-distributor/enhance-login-permission

Enhance login permission
sj-distributor · Nov 13, 2024 · 192aec9 · 192aec9
2 parents 7274772 + c5acc01
commit 192aec9
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 2 deletions.
diff --git a/web/src/dtos/mine.ts b/web/src/dtos/mine.ts
@@ -42,4 +42,5 @@ export enum FrontRolePermissionEnum {
   CanViewCameraAiFeedbackListPage = "CanViewCameraAiFeedbackListPage",
   CanExportExcelCameraAiFeedback = "CanExportExcelCameraAiFeedback",
   CanViewDetailCameraAiFeedback = "CanViewDetailCameraAiFeedback",
+  CanEnterCameraAi = "CanEnterCameraAi",
 }
diff --git a/web/src/pages/login/hook.ts b/web/src/pages/login/hook.ts
@@ -4,6 +4,8 @@ import { useState } from "react";
 import { IUserInfo } from "@/dtos";
 import { useAuth } from "@/hooks/use-auth";
 import { Login } from "@/services/home";
+import { GetMineRoleList } from "@/services/default";
+import { FrontRolePermissionEnum } from "@/dtos/mine";
 
 export const useAction = () => {
   const { signIn, message } = useAuth();
@@ -22,6 +24,16 @@ export const useAction = () => {
     }));
   };
 
+  const hanldeNoPermission = () => {
+    message.error("您没有访问权限");
+
+    localStorage.removeItem(
+      (window as any).appsettings?.tokenKey ?? "tokenKey"
+    );
+
+    localStorage.removeItem((window as any).appsettings?.userNameKey);
+  };
+
   const onLogin = () => {
     setLoginLoading(true);
     if (
@@ -31,14 +43,38 @@ export const useAction = () => {
       Login(userInfo)
         .then((res) => {
           if (res) {
-            message.success("登录成功");
             localStorage.setItem((window as any).appsettings?.tokenKey, res);
             localStorage.setItem(
               (window as any).appsettings?.userNameKey,
               userInfo.userName
             );
 
-            signIn(res, userInfo.userName);
+            GetMineRoleList()
+              .then((response) => {
+                if (
+                  response.rolePermissionData.some((item) =>
+                    item.permissions.some(
+                      (permission) =>
+                        permission.name ===
+                        FrontRolePermissionEnum.CanEnterCameraAi
+                    )
+                  )
+                ) {
+                  message.success("登录成功");
+
+                  signIn(
+                    localStorage.getItem(
+                      (window as any).appsettings?.tokenKey ?? "tokenKey"
+                    ) ?? "",
+                    userInfo.userName
+                  );
+                } else {
+                  hanldeNoPermission();
+                }
+              })
+              .catch(() => {
+                hanldeNoPermission();
+              });
           }
         })
         .catch(() => {