From 93a33ffba33194e0f5bcc3c10a7b5779ca68b142 Mon Sep 17 00:00:00 2001
From: ctexthuang <ctexthuang@qq.com>
Date: Tue, 1 Apr 2025 10:02:10 +0800
Subject: [PATCH] feat: invite num

---
 app/Controller/Api/SystemController.php     |  3 +--
 app/Controller/Api/UserController.php       |  3 +--
 app/Service/Api/{ => User}/IndexService.php |  3 ++-
 app/Service/Api/User/MyPageService.php      | 24 ++++++++++++++++++---
 4 files changed, 25 insertions(+), 8 deletions(-)
 rename app/Service/Api/{ => User}/IndexService.php (96%)

diff --git a/app/Controller/Api/SystemController.php b/app/Controller/Api/SystemController.php
index c21864a..c38767f 100644
--- a/app/Controller/Api/SystemController.php
+++ b/app/Controller/Api/SystemController.php
@@ -6,15 +6,14 @@ namespace App\Controller\Api;
 
 use App\Controller\AbstractController;
 use App\Middleware\Api\JwtAuthMiddleware;
-use App\Service\Api\IndexService;
 use App\Service\Api\System\AliStsService;
 use App\Service\Api\System\CityListService;
 use App\Service\Api\System\MiniWxConfigService;
 use App\Service\Api\System\SiteListService;
 use App\Service\Api\System\SystemConfigService;
+use App\Service\Api\User\IndexService;
 use Hyperf\HttpServer\Annotation\Controller;
 use Hyperf\HttpServer\Annotation\Middleware;
-use Hyperf\HttpServer\Annotation\Middlewares;
 use Hyperf\HttpServer\Annotation\RequestMapping;
 use Hyperf\Validation\Annotation\Scene;
 use Psr\Container\ContainerExceptionInterface;
diff --git a/app/Controller/Api/UserController.php b/app/Controller/Api/UserController.php
index 9daae7c..4ee8034 100644
--- a/app/Controller/Api/UserController.php
+++ b/app/Controller/Api/UserController.php
@@ -9,6 +9,7 @@ use App\Middleware\Api\JwtAuthMiddleware;
 use App\Request\Api\UserRequest;
 use App\Service\Api\Login\LogOutService;
 use App\Service\Api\User\BindPhoneByWxService;
+use App\Service\Api\User\InviteInfoService;
 use App\Service\Api\User\InviteListService;
 use App\Service\Api\User\MyPageService;
 use App\Service\Api\User\SiteService;
@@ -148,6 +149,4 @@ class UserController extends AbstractController
     {
         return (new UpdateProfileService)->handle();
     }
-
-
 }
diff --git a/app/Service/Api/IndexService.php b/app/Service/Api/User/IndexService.php
similarity index 96%
rename from app/Service/Api/IndexService.php
rename to app/Service/Api/User/IndexService.php
index 2f98ebe..214b5b6 100644
--- a/app/Service/Api/IndexService.php
+++ b/app/Service/Api/User/IndexService.php
@@ -8,12 +8,13 @@
 
 declare(strict_types=1);
 
-namespace App\Service\Api;
+namespace App\Service\Api\User;
 
 use App\Constants\Common\BannerCode;
 use App\Constants\Common\CouponCode;
 use App\Model\Banner;
 use App\Model\UserCoupon;
+use App\Service\Api\BaseService;
 use App\Service\ServiceTrait\Api\GetUserInfoTrait;
 use App\Service\ServiceTrait\Common\OssTrait;
 use Hyperf\Di\Annotation\Inject;
diff --git a/app/Service/Api/User/MyPageService.php b/app/Service/Api/User/MyPageService.php
index 1e53b8b..0926b0a 100644
--- a/app/Service/Api/User/MyPageService.php
+++ b/app/Service/Api/User/MyPageService.php
@@ -14,6 +14,7 @@ use App\Constants\Common\CouponCode;
 use App\Extend\StringUtil;
 use App\Model\AdminUser;
 use App\Model\UserCoupon;
+use App\Model\UserInvite;
 use App\Service\Api\BaseService;
 use App\Service\ServiceTrait\Api\GetUserInfoTrait;
 use App\Service\ServiceTrait\Common\OssTrait;
@@ -44,13 +45,19 @@ class MyPageService extends BaseService
             'mobile' => StringUtil::maskMiddleStrByNum($userInfo->mobile,4),
             'point' => 0,
             'coupon_num' => $this->getCouponNum(),
-            'invite_num' => 0,
+            'invite_num' => $this->getInviteNum(),
             'role_id' => (!empty($adminInfo) && $adminInfo->role_id > 0) ? $adminInfo->role_id : 0,
         ];
 
         return $this->return->success('success', $res);
     }
 
+    /**
+     * @var UserCoupon
+     */
+    #[Inject]
+    private UserCoupon $userCouponModel;
+
     /**
      * @return int
      */
@@ -64,8 +71,19 @@ class MyPageService extends BaseService
     }
 
     /**
-     * @var UserCoupon
+     * @var UserInvite
      */
     #[Inject]
-    private UserCoupon $userCouponModel;
+    private UserInvite $userInviteModel;
+
+    /**
+     * @return int
+     */
+    private function getInviteNum(): int
+    {
+        return $this->userInviteModel
+            ->where('invitee_user_id',$this->userId)
+            ->count()?? 0;
+    }
+
 }
\ No newline at end of file