commit d526daa246e99d924c414d69e40390db4627b373
author 22301102 <22301102@bjtu.edu.cn> Sat Jun 07 10:53:27 2025 +0800
committer 22301102 <22301102@bjtu.edu.cn> Sat Jun 07 18:19:43 2025 +0800
tree 54d82f9aecd55a73471db64b21c322a6432371f4
parent 949f940ca8087fba837c46fc588676a54d2e1a3d
parent c8584eed27ec70586191bd5a282add3f52bc7019

添加邀请码注册机制

Change-Id: I1a104074a36564ffb332e928e131af182eb30240

src/main/java/com/pt/controller/UserController.java

diff --git a/src/main/java/com/pt/controller/UserController.java b/src/main/java/com/pt/controller/UserController.java
index 7603f9d..577b2a3 100644
--- a/src/main/java/com/pt/controller/UserController.java
+++ b/src/main/java/com/pt/controller/UserController.java
@@ -2,6 +2,7 @@
 
 import com.pt.constant.Constants;
 import com.pt.entity.User;
+import com.pt.service.InvitedCodeService;
 import com.pt.utils.JWTUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.ResponseEntity;
@@ -18,6 +19,7 @@
 
     @Autowired
     private UserService userService;
+    private InvitedCodeService invitedCodeService;
 
     @PostMapping("/register")
     public ResponseEntity<?> registerUser(@RequestBody Map<String, String> request) {
@@ -35,6 +37,17 @@
         } else {
             User newUser = new User();
 
+            if(request.get("invitedCode") != null) {
+                String invitedCode = request.get("invitedCode");
+                if (!invitedCodeService.isCodeValid(invitedCode)) {
+                    return ResponseEntity.badRequest().body("Invalid invited code");
+                }
+                else {
+                    newUser.setLevel(Constants.UserLevel.FRESH.getValue());
+                    newUser.setPoints(300);
+                }
+            }
+
             String uid = String.valueOf(System.currentTimeMillis());
             newUser.setUid(uid);
             newUser.setUsername(username);