From 96fa4ea92242c39cac53af4839c06b468bd9a0a3 Mon Sep 17 00:00:00 2001
From: mkm <727897186@qq.com>
Date: Sat, 13 Jan 2024 14:03:59 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/common/repositories/user/UserRepository.php |  7 ++++++-
 app/controller/api/Common.php                   | 10 ++++++++++
 route/api.php                                   |  1 +
 3 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/app/common/repositories/user/UserRepository.php b/app/common/repositories/user/UserRepository.php
index 06a66528..fabe3280 100644
--- a/app/common/repositories/user/UserRepository.php
+++ b/app/common/repositories/user/UserRepository.php
@@ -661,14 +661,19 @@ class UserRepository extends BaseRepository
     public function registr(string $phone, ?string $pwd, $user_type = 'h5')
     {
         $pwd = $pwd ? $this->encodePassword($pwd) : $this->encodePassword($this->dao->defaultPwd());
+        $ip=app('request')->ip();
+        $code=Cache::get('promote_'.$ip);
         $data = [
             'account' => $phone,
             'pwd' => $pwd,
             'nickname' => substr($phone, 0, 3) . '****' . substr($phone, 7, 4),
             'avatar' => '',
             'phone' => $phone,
-            'last_ip' => app('request')->ip()
+            'last_ip' => $ip
         ];
+        if($code){
+            $data['promotion_code']=$code;
+        }
         return $this->create($user_type, $data);
     }
 
diff --git a/app/controller/api/Common.php b/app/controller/api/Common.php
index fccbf30d..84cac40e 100644
--- a/app/controller/api/Common.php
+++ b/app/controller/api/Common.php
@@ -624,4 +624,14 @@ class Common extends BaseController
         }
         return app('json')->success($list);
     }
+
+    /**
+     * 推广写入
+     */
+    public function promote_writing()
+    {
+        $code = $this->request->param('code');
+        Cache::set('promote_' . app('request')->ip(), $code, 86400);
+        return app('json')->success();
+    }
 }
diff --git a/route/api.php b/route/api.php
index 354bac62..76a6cd65 100644
--- a/route/api.php
+++ b/route/api.php
@@ -21,6 +21,7 @@ use think\facade\Route;
 
 Route::group('api/', function () {
     Route::any('test', 'api.Auth/test');
+    Route::any('promote_writing', 'api.Common/promote_writing');
     Route::any('applet', 'api.Common/applet');
     Route::get('label_lst', 'api.Common/label_lst');
     Route::any('system_group_value', 'api.Common/system_group_value');