From b44b174f656aac1fe03e7f96851e564c3982f67d Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期五, 28 三月 2025 16:52:25 +0800
Subject: [PATCH] Merge branch 'master' of https://gitee.com/boymi/YiFeiNN

---
 medicalWaste-admin/src/main/java/com/sinata/web/controller/applet/AppLoginController.java |   88 +++++++++++++++++++++++++++++++++++--------
 1 files changed, 71 insertions(+), 17 deletions(-)

diff --git a/medicalWaste-admin/src/main/java/com/sinata/web/controller/applet/AppLoginController.java b/medicalWaste-admin/src/main/java/com/sinata/web/controller/applet/AppLoginController.java
index 138ad68..8c82857 100644
--- a/medicalWaste-admin/src/main/java/com/sinata/web/controller/applet/AppLoginController.java
+++ b/medicalWaste-admin/src/main/java/com/sinata/web/controller/applet/AppLoginController.java
@@ -2,6 +2,7 @@
 
 import com.alibaba.fastjson2.JSON;
 import com.alibaba.fastjson2.JSONObject;
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.sinata.common.constant.Constants;
 import com.sinata.common.core.domain.AjaxResult;
@@ -20,6 +21,7 @@
 import com.sinata.system.domain.SysDepartment;
 import com.sinata.system.domain.SysUserRole;
 import com.sinata.system.domain.dto.UserInfoDTO;
+import com.sinata.system.enums.DepartmentEnum;
 import com.sinata.system.mapper.SysUserRoleMapper;
 import com.sinata.system.service.ISysMenuService;
 import com.sinata.system.service.ISysRoleService;
@@ -40,6 +42,7 @@
 import javax.annotation.Resource;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.Random;
 import java.util.Set;
 
@@ -123,15 +126,34 @@
         String openid = map.get("openid").toString();
         String sessionKey = map.get("sessionKey").toString();
 
-
         String decrypt = WXCore.decrypt(EncryptedData_phone, sessionKey, Iv_phone);
         if (StringUtils.isEmpty(decrypt)) {
             return R.fail("获取手机信息失败");
         }
         JSONObject phone = JSON.parseObject(decrypt);
         String purePhoneNumber = phone.getString("purePhoneNumber");
-        return R.ok(purePhoneNumber);
+        
+        // 查询手机号对应的用户
+        SysUser user = userService.lambdaQuery()
+                .eq(SysUser::getUserName, purePhoneNumber)
+                .one();
+        if(user == null) {
+            return R.fail("该手机号未注册");
+        }
 
+        // 创建LoginUser对象
+        LoginUser loginUser = new LoginUser();
+        loginUser.setUserId(user.getUserId());
+        loginUser.setUser(user);
+
+        // 生成token
+        String token = tokenService.createToken(loginUser);
+        
+        JSONObject result = new JSONObject();
+        result.put("phone", purePhoneNumber);
+        result.put(Constants.TOKEN, token);
+        
+        return R.ok(result);
     }
     /**
      * 登录方法
@@ -139,32 +161,54 @@
      * @param loginBody 登录信息
      * @return 结果
      */
+//     @PostMapping("/login")
+//     @ApiOperation("验证码登录")
+//     public AjaxResult login(@RequestBody LoginBody loginBody)
+//     {
+//         AjaxResult ajax = AjaxResult.success();
+//         // 生成令牌
+//         LoginUser loginUser = loginService.login1(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(),
+//                 loginBody.getUuid());
+//         ajax.put(Constants.TOKEN, tokenService.createToken(loginUser));
+// //        List<SysRole> roles = loginUser.getUser().getRoles();
+// //        if (CollectionUtils.isEmpty(roles)) {
+// //            return AjaxResult.error("请关联角色!");
+// //        }
+
+//         List<SysMenu> menus = roleService.roleInfoFromUserId(loginUser.getUserId());
+//         String string = redisCache.getCacheObject(loginBody.getUsername()).toString();
+//         if (StringUtils.isBlank(string)) {
+//             return AjaxResult.error("验证码已失效");
+//         }
+//         if (!loginBody.getCode().equals(string)){
+//             return AjaxResult.error("验证码错误,请重试");
+//         }
+// //        ajax.put("menus", menus);
+//         ajax.put("userInfo", loginUser);
+//         return ajax;
+//     }
+
     @PostMapping("/login")
-    @ApiOperation("验证码登录")
+    @ApiOperation("管理后台登录")
     public AjaxResult login(@RequestBody LoginBody loginBody)
     {
         AjaxResult ajax = AjaxResult.success();
         // 生成令牌
-        LoginUser loginUser = loginService.login1(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(),
+        LoginUser loginUser = loginService.login(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(),
                 loginBody.getUuid());
         ajax.put(Constants.TOKEN, tokenService.createToken(loginUser));
-//        List<SysRole> roles = loginUser.getUser().getRoles();
-//        if (CollectionUtils.isEmpty(roles)) {
-//            return AjaxResult.error("请关联角色!");
-//        }
-
+        List<SysRole> roles = loginUser.getUser().getRoles();
+        if (CollectionUtils.isEmpty(roles)) {
+            return AjaxResult.error("请关联角色!");
+        }
         List<SysMenu> menus = roleService.roleInfoFromUserId(loginUser.getUserId());
-        String string = redisCache.getCacheObject(loginBody.getUsername()).toString();
-        if (StringUtils.isBlank(string)) {
-            return AjaxResult.error("验证码已失效");
-        }
-        if (!loginBody.getCode().equals(string)){
-            return AjaxResult.error("验证码错误,请重试");
-        }
-//        ajax.put("menus", menus);
+        Set<Long> permissions = roleService.getMenuIdSets(loginUser.getUserId());
+        ajax.put("menus", menus);
         ajax.put("userInfo", loginUser);
+        ajax.put("permissions", permissions);
         return ajax;
     }
+
 
     @PostMapping("/get/department")
     @ApiOperation("登录后获取部门")
@@ -188,6 +232,16 @@
         userInfoDTO.setNickName(user.getNickName());
         userInfoDTO.setDepartment(byId2);
         userInfoDTO.setRoles(roles);
+        if (byId2.getOrgType().equals(DepartmentEnum.REGULATORY_UNIT.getCode())) {
+            SysDepartment department = departmentService.getById(byId2.getParentId());
+            if (Objects.nonNull(department)) {
+                userInfoDTO.setCurrentDepartmentId(department.getId());
+                userInfoDTO.setCurrentDepartmentName(department.getDepartmentName());
+            }
+        } else {
+            userInfoDTO.setCurrentDepartmentId(byId2.getId());
+            userInfoDTO.setCurrentDepartmentName(byId2.getDepartmentName());
+        }
         SysUserRole sysUserRole = sysUserRoleMapper.selectOne(Wrappers.<SysUserRole>lambdaQuery().eq(SysUserRole::getUserId, user.getUserId()));
         if (sysUserRole != null) {
             SysRole byId = roleService.getById(sysUserRole.getRoleId());

--
Gitblit v1.7.1