From 442dcfc5435c005eef11f22f92a1c290f4af22e8 Mon Sep 17 00:00:00 2001
From: nili <nili@baicizhan.com>
Date: Mon, 1 Jul 2024 16:14:51 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8A=A0=E4=B8=AA=E7=BB=91=E5=AE=9A=E9=82=80?=
 =?UTF-8?q?=E8=AF=B7=E7=A0=81=E7=9A=84=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../group/game/service/citrus/UserService.java     | 14 ++++++++++++++
 .../group/game/web/rest/citrus/UserController.java |  7 +++++++
 2 files changed, 21 insertions(+)

diff --git a/game-service/src/main/java/awesome/group/game/service/citrus/UserService.java b/game-service/src/main/java/awesome/group/game/service/citrus/UserService.java
index d5a5db8..a1f0384 100644
--- a/game-service/src/main/java/awesome/group/game/service/citrus/UserService.java
+++ b/game-service/src/main/java/awesome/group/game/service/citrus/UserService.java
@@ -186,6 +186,20 @@ public class UserService {
         return new TeamBo(members);
     }
 
+    public void bindInviteCode(Integer uid, String inviteCode) {
+        Assert.isTrue(StringUtils.hasText(inviteCode), "邀请码不能为空");
+        MatrixUser u = userMapper.selectById(uid);
+        Assert.isTrue(u.getUpUid() == null, "不能重复绑定");
+        MatrixUser up = userMapper.selectByInviteCode(inviteCode);
+        Assert.isTrue(up != null, "非法邀请码");
+        Assert.isTrue(up.getAppId().equals(u.getAppId()), "非法邀请码");
+        MatrixUser user = new MatrixUser();
+        user.setId(uid);
+        user.setUpUid(up.getId());
+        user.setAdminId(up.getAdminId());
+        userMapper.updateById(user);
+    }
+
     public void bindAliPay(Integer uid, String aliPayAccount, String name) {
         Assert.isTrue(StringUtils.hasText(aliPayAccount), "支付宝账号不能为空");
         Assert.isTrue(StringUtils.hasText(name), "支付宝账号姓名不能为空");
diff --git a/game-web/src/main/java/awesome/group/game/web/rest/citrus/UserController.java b/game-web/src/main/java/awesome/group/game/web/rest/citrus/UserController.java
index 69cd264..9a4f1e7 100644
--- a/game-web/src/main/java/awesome/group/game/web/rest/citrus/UserController.java
+++ b/game-web/src/main/java/awesome/group/game/web/rest/citrus/UserController.java
@@ -43,6 +43,13 @@ public class UserController {
         return new R<>(null);
     }
 
+    @PostMapping("/bindInviteCode")
+    @RestApi
+    public R<Void> bindInviteCode(@RequestParam String inviteCode) {
+        userService.bindInviteCode(RequestContext.getCitrusUid(), inviteCode);
+        return new R<>(null);
+    }
+
     @GetMapping("/cashRecords")
     @RestApi
     public R<List<CashRecord>> getCashRecord() {