From 901b2ab3e0e7c592602848cfa256f34c5b0c7bb8 Mon Sep 17 00:00:00 2001
From: Junjie <540245094@qq.com>
Date: 星期四, 11 一月 2024 09:57:01 +0800
Subject: [PATCH] #

---
 zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java |   62 ++++++++++++++++++++++++++++---
 1 files changed, 56 insertions(+), 6 deletions(-)

diff --git a/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java b/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java
index df4a984..80c67e9 100644
--- a/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java
+++ b/zy-asrs-common/src/main/java/com/zy/asrs/common/web/AuthController.java
@@ -44,6 +44,10 @@
     private ResourceService resourceService;
     @Autowired
     private RoleResourceService roleResourceService;
+    @Autowired
+    private HostService hostService;
+    @Autowired
+    private ConfigService configService;
 
     @RequestMapping("/login.action")
     @ManagerAuth(value = ManagerAuth.Auth.NONE, memo = "鐧诲綍")
@@ -79,11 +83,47 @@
         userLogin.setToken(token);
         userLogin.setCreateTime(new Date());
         userLogin.setSystem(system);
+        if (user.getRoleId() == 2) {
+            userLogin.setHostId(hostService.getTop1().getId());
+        }
         userLoginService.save(userLogin);
         Map<String, Object> res = new HashMap<>();
         res.put("username", user.getUsername());
         res.put("token", token);
         return R.ok(res);
+    }
+
+    @RequestMapping("/show/host.action")
+    @ManagerAuth
+    public R showHosts() {
+        Long hostId = getHostId();
+        String hostName = null;
+        if (hostId != null) {
+            Host host = hostService.getById(hostId);
+            if (host != null) {
+                hostName = host.getName();
+            }
+        }
+        return R.ok().add(Cools
+                .add("root", getUser().getRoleId() == 2)
+                .add("host", getHostId() == null)
+                .add("hostName", hostName)
+        );
+    }
+
+    @RequestMapping(value = "/root/change/host/auth")
+    @ManagerAuth
+    public R rootChangeHost(@RequestParam Long hostId) {
+        UserLogin userLogin = userLoginService.getOne(new LambdaQueryWrapper<UserLogin>().eq(UserLogin::getUserId, getUserId()).eq(UserLogin::getSystem, String.valueOf(LoginSystemType.WMS)));
+        if (userLogin != null) {
+            userLogin.setHostId(hostId);
+            if (!userLoginService.updateById(userLogin)) {
+                throw new CoolException("淇敼鍟嗘埛澶辫触");
+            }
+            return R.ok();
+        } else {
+            return R.error();
+        }
     }
 
     @RequestMapping("/code/switch.action")
@@ -130,10 +170,6 @@
             oneLevel = resourceService.list(new LambdaQueryWrapper<Resource>().eq(Resource::getLevel, 1).orderByAsc(Resource::getSort));
             resourceWrapper = new LambdaQueryWrapper<Resource>().eq(Resource::getLevel, 2).eq(Resource::getStatus, 1).orderByAsc(Resource::getSort);
         } else {
-            // 婵�娲荤爜楠岃瘉
-            if (!SystemProperties.SYSTEM_ACTIVATION) {
-                return R.ok();
-            }
             oneLevel = resourceService.list(new LambdaQueryWrapper<Resource>().eq(Resource::getLevel, 1).eq(Resource::getStatus, 1).orderByAsc(Resource::getSort));
             // 鑾峰彇褰撳墠鐢ㄦ埛鐨勬墍鏈変簩绾ц彍鍗�
             user = userService.getById(getUserId());
@@ -158,11 +194,11 @@
                     // 鏄惁鎷ユ湁鏌ョ湅鏉冮檺
                     if (getUserId() != 9527) {
                         Resource view = resourceService.getOne(new LambdaQueryWrapper<Resource>().eq(Resource::getResourceId, resource.getId()).like(Resource::getCode, "#view"));
-                        if (!Cools.isEmpty(view)){
+                        if (!Cools.isEmpty(view)) {
                             RoleResource param = new RoleResource();
                             param.setResourceId(view.getId());
                             param.setRoleId(user.getRoleId());
-                            if (null == roleResourceService.getOne(new LambdaQueryWrapper<>(param))){
+                            if (null == roleResourceService.getOne(new LambdaQueryWrapper<>(param))) {
                                 continue;
                             }
                         }
@@ -324,6 +360,20 @@
         }
     }
 
+    @RequestMapping("/loginInformation")
+    public R getLoginPageInformation() {
+        Config loginLogo = configService.selectByCode("loginLogo");
+        Config loginCopyrightText = configService.selectByCode("loginCopyrightText");
+        Config loginCopyrightImg1 = configService.selectByCode("loginCopyrightImg1");
+
+        HashMap<String, Object> map = new HashMap<>();
+        map.put("loginLogo", loginLogo.getValue());
+        map.put("loginCopyrightText", loginCopyrightText.getValue());
+        map.put("loginCopyrightImg1", loginCopyrightImg1.getStatus() == 1 ? loginCopyrightImg1.getValue() : "");
+
+        return R.ok().add(map);
+    }
+
     @PostMapping(value = "/system/secret/auth")
     @ManagerAuth
     public R systemSecret(@RequestParam(value = "secret") String secret) {

--
Gitblit v1.9.1