From b59765f927fcdc21892dee19446012a9dadad130 Mon Sep 17 00:00:00 2001
From: PIGCLOUD <wangiegie@gmail.com>
Date: Tue, 6 Jan 2026 10:53:58 +0800
Subject: [PATCH] =?UTF-8?q?fix(ai-register-user):=20=E8=B7=B3=E8=BF=87?=
 =?UTF-8?q?=E5=A4=96=E9=83=A8=20AI=20=E6=B3=A8=E5=86=8C=E7=9F=AD=E4=BF=A1?=
 =?UTF-8?q?=E9=AA=8C=E8=AF=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 添加 SecurityConstants 导入
- 使用 WebUtils 获取 请求头
- 当 header 为 FROM_IN 时跳过 短信验证
- 保持内部注册验证逻辑
---
 .../service/impl/SysUserServiceImpl.java      | 21 ++++++++++++-------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/service/impl/SysUserServiceImpl.java b/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/service/impl/SysUserServiceImpl.java
index 95c26468a..4d9941a5e 100644
--- a/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/service/impl/SysUserServiceImpl.java
+++ b/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/service/impl/SysUserServiceImpl.java
@@ -41,9 +41,11 @@ import com.pig4cloud.pigx.admin.service.*;
 import com.pig4cloud.pigx.common.audit.annotation.Audit;
 import com.pig4cloud.pigx.common.core.constant.CacheConstants;
 import com.pig4cloud.pigx.common.core.constant.CommonConstants;
+import com.pig4cloud.pigx.common.core.constant.SecurityConstants;
 import com.pig4cloud.pigx.common.core.constant.enums.LoginTypeEnum;
 import com.pig4cloud.pigx.common.core.util.MsgUtils;
 import com.pig4cloud.pigx.common.core.util.R;
+import com.pig4cloud.pigx.common.core.util.WebUtils;
 import com.pig4cloud.pigx.common.data.cache.RedisUtils;
 import com.pig4cloud.pigx.common.data.datascope.DataScope;
 import com.pig4cloud.pigx.common.data.resolver.ParamResolver;
@@ -534,14 +536,17 @@ public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> impl
     @Override
     @Transactional(rollbackFor = Exception.class)
     public R<Boolean> registerUser(RegisterUserDTO userDto) {
-        // 校验短信验证码
-        if (StrUtil.isBlank(userDto.getCode())) {
-            return R.failed(MsgUtils.getMessage(UpmsErrorCodes.SYS_PARAM_ILLEGAL));
-        }
-        String codeObj = RedisUtils
-                .get(CacheConstants.DEFAULT_CODE_KEY + LoginTypeEnum.SMS.getType() + StringPool.AT + userDto.getPhone());
-        if (!StrUtil.equals(codeObj, userDto.getCode())) {
-            return R.failed(MsgUtils.getMessage(UpmsErrorCodes.SYS_APP_SMS_ERROR));
+        String feignHeader = WebUtils.getRequest().getHeader(SecurityConstants.FROM);
+        // 外部注册校验短信验证码,对 AI 的内部注册跳过验证码处理
+        if (!SecurityConstants.FROM_IN.equals(feignHeader)){
+            if (StrUtil.isBlank(userDto.getCode())) {
+                return R.failed(MsgUtils.getMessage(UpmsErrorCodes.SYS_PARAM_ILLEGAL));
+            }
+            String codeObj = RedisUtils
+                    .get(CacheConstants.DEFAULT_CODE_KEY + LoginTypeEnum.SMS.getType() + StringPool.AT + userDto.getPhone());
+            if (!StrUtil.equals(codeObj, userDto.getCode())) {
+                return R.failed(MsgUtils.getMessage(UpmsErrorCodes.SYS_APP_SMS_ERROR));
+            }
         }
 
         // 判断用户名是否存在