From c9e9737a7584ebf84e12ecc956a337c3f8f0d67b Mon Sep 17 00:00:00 2001
From: “mitao” <“2763622819@qq.com”>
Date: 星期一, 02 二月 2026 16:33:09 +0800
Subject: [PATCH] fix:小程序码生成增加环境判断
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCompanyController.java | 32 ++++++++++++++++++++++++++------
1 files changed, 26 insertions(+), 6 deletions(-)
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCompanyController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCompanyController.java
index 3a2aa2d..dbe0eee 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCompanyController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCompanyController.java
@@ -82,7 +82,8 @@
private final TAppUserService appUserService ;
private final TBigGiftPackageOrderService bigGiftPackageOrderService;
-
+ @Value("${spring.profiles.active}")
+ private String env;
public TCompanyController(TCompanyService companyService, ISysUserService userService, RedisTemplate redisTemplate, TAppUserService appUserService, TBigGiftPackageOrderService bigGiftPackageOrderService) {
this.companyService = companyService;
this.userService = userService;
@@ -118,6 +119,10 @@
public R<String> add(@RequestBody TCompany company) {
company.setStatus(1);
company.setCreateUserId(getUserId());
+ if(company.getAccount()!=null && "admin".equals(company.getAccount())){
+ return R.fail("账号不能为admin");
+ }
+
SysUser user = new SysUser();
user.setRoleType(2);
user.setDeptId(124l);
@@ -126,7 +131,8 @@
user.setPassword(company.getPassword());
user.setPhonenumber(company.getChargePersonPhone());
user.setNickName(company.getChargePerson());
- user.setRoleId(1l);
+ user.setCode(company.getCode());
+ user.setRoleId(2l);
if (!userService.checkUserNameUnique(user)) {
return R.fail("新增用户'" + user.getUserName() + "'失败,登录账号已存在");
} else if (StringUtils.isNotEmpty(user.getPhonenumber()) && !userService.checkPhoneUnique(user)) {
@@ -152,6 +158,9 @@
@PostMapping(value = "/edit")
@Log(title = "修改公司-修改", businessType = BusinessType.UPDATE)
public R<String> edit(@RequestBody TCompany company) {
+ if(company.getAccount()!=null && "admin".equals(company.getAccount())){
+ return R.fail("账号不能为admin");
+ }
SysUser user = userService.getUserByCompanyId(company.getId());
if (!userService.checkUserNameUnique(user)) {
return R.fail("修改用户'" + user.getUserName() + "'失败,登录账号已存在");
@@ -159,8 +168,9 @@
return R.fail("修改用户'" + user.getUserName() + "'失败,手机号码已存在");
}
user.setUserName(company.getAccount());
- user.setPassword(SecurityUtils.encryptPassword(user.getPassword()));
+ user.setPassword(SecurityUtils.encryptPassword(company.getPassword()));
company.setUpdateUserId(getUserId());
+ user.setRoleId(2L);
int count = userService.updateUser(user);
if (count > 0) {
boolean update = companyService.updateById(company);
@@ -210,7 +220,7 @@
param.put("page", "pages/home/home");
param.put("width", 430); //二维码尺寸
param.put("scene", "code=" + sysUser.getCode()+"&type=1"); //二维码尺寸
- param.put("env_version", "trial"); //二维码尺寸
+ param.put("env_version", env.equals("prod") ? "release" : "trial"); //二维码尺寸
param.put("check_path", false); //二维码尺寸
HttpRequest post = HttpUtil.createPost(url);
post.body(JSON.toJSONString(param));
@@ -254,6 +264,9 @@
@PostMapping("/system/user/add")
public R<String> add(@RequestBody SysUser user)
{
+ if(user.getUserName()!=null && "admin".equals(user.getUserName())){
+ return R.fail("账号不能为admin");
+ }
user.setUserName(user.getUserName());
if (!userService.checkUserNameUnique(user))
{
@@ -267,6 +280,8 @@
{
return R.fail("新增用户'" + user.getUserName() + "'失败,推广号已存在");
}
+
+
user.setCreateBy(getUsername());
user.setPassword(SecurityUtils.encryptPassword(user.getPassword()));
@@ -281,7 +296,8 @@
param.put("page", "pages/home/home");
param.put("width", 430); //二维码尺寸
param.put("scene", "code=" + user.getCode()+"&type=1"); //二维码尺寸
- param.put("env_version", "trial"); //二维码尺寸
+ param.put("env_version", env.equals("prod") ? "release" : "trial"); //二维码尺寸
+ param.put("check_path", false); //二维码尺寸
HttpRequest post = HttpUtil.createPost(url);
post.body(JSON.toJSONString(param));
@@ -299,6 +315,9 @@
@PostMapping("/system/user/edit")
public R edit(@RequestBody SysUser user)
{
+ if(user.getUserName()!=null && "admin".equals(user.getUserName())){
+ return R.fail("账号不能为admin");
+ }
user.setUserName(user.getPhonenumber());
// userService.checkUserAllowed(user);
// userService.checkUserDataScope(user.getUserId());
@@ -314,6 +333,7 @@
if(StringUtils.isNotEmpty(user.getPassword())){
user.setPassword(SecurityUtils.encryptPassword(user.getPassword()));
}
+ user.setRoleId(2L);
// 移除之前的二维码
redisTemplate.delete("ADMIN_WX_IMG" + user.getUserId());
return R.ok(userService.updateUser(user));
@@ -408,7 +428,7 @@
sysUser.setCreateBy(getUsername());
sysUser.setRoleType(3);
- sysUser.setRoleId(1L);
+ sysUser.setRoleId(2L);
// sysUsers.add(sysUser);
userService.insertUser(sysUser);
i++;
--
Gitblit v1.7.1