From a77e37bb3114adccd6752ecdcf3ca3c00747b1c2 Mon Sep 17 00:00:00 2001
From: "iVampireSP.com" <im@ivampiresp.com>
Date: Fri, 28 Oct 2022 16:52:29 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E8=BF=9B=20=E8=AE=A1=E8=B4=B9?=
 =?UTF-8?q?=E8=A7=84=E5=88=99=EF=BC=8C=E6=94=B9=E4=B8=BA=E6=A8=A1=E5=9D=97?=
 =?UTF-8?q?=E9=9C=80=E8=A6=81=E5=88=9B=E5=BB=BA=E8=AE=A1=E8=B4=B9=E9=A1=B9?=
 =?UTF-8?q?=E7=9B=AE=E6=97=B6=E6=89=8D=E8=A6=81=E6=B1=82=E4=BD=99=E9=A2=9D?=
 =?UTF-8?q?=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/Http/Controllers/ForumController.php            |  4 ++--
 app/Http/Controllers/Remote/Host/HostController.php | 13 +++++++++++--
 app/Http/Controllers/Remote/ModuleController.php    | 12 ++++++------
 app/Http/Controllers/User/HostController.php        |  6 +++---
 4 files changed, 22 insertions(+), 13 deletions(-)

diff --git a/app/Http/Controllers/ForumController.php b/app/Http/Controllers/ForumController.php
index 8766d77..8c4e74a 100644
--- a/app/Http/Controllers/ForumController.php
+++ b/app/Http/Controllers/ForumController.php
@@ -3,8 +3,8 @@
 namespace App\Http\Controllers;
 
 use Closure;
-use Exception;
-use Illuminate\Http\Request;
+// use Exception;
+// use Illuminate\Http\Request;
 use Illuminate\Support\Facades\Http;
 use Illuminate\Support\Facades\Cache;
 
diff --git a/app/Http/Controllers/Remote/Host/HostController.php b/app/Http/Controllers/Remote/Host/HostController.php
index 54fafab..ae5e0bd 100644
--- a/app/Http/Controllers/Remote/Host/HostController.php
+++ b/app/Http/Controllers/Remote/Host/HostController.php
@@ -3,10 +3,12 @@
 namespace App\Http\Controllers\Remote\Host;
 
 use App\Models\Host;
+use App\Models\User;
 use Illuminate\Support\Str;
 use Illuminate\Http\Request;
+// use Illuminate\Support\Facades\Log;
 use App\Http\Controllers\Controller;
-use Illuminate\Support\Facades\Log;
+use App\Exceptions\User\BalanceNotEnoughException;
 
 class HostController extends Controller
 {
@@ -36,6 +38,13 @@ public function store(Request $request)
             'user_id' => 'required|integer|exists:users,id',
         ]);
 
+        //
+        $user = User::findOrFail($request->user_id);
+
+        if ($user->balance <= 1) {
+            throw new BalanceNotEnoughException("余额不足，无法开设新的计费项目。");
+        }
+
         // 如果没有 name，则随机
         $name = $request->input('name', Str::random(10));
 
@@ -43,7 +52,7 @@ public function store(Request $request)
             'name' => $name,
             'status' => $request->status,
             'price' => $request->price,
-            'user_id' => $request->user_id,
+            'user_id' => $user->id,
             'module_id' => auth('remote')->id()
         ];
 
diff --git a/app/Http/Controllers/Remote/ModuleController.php b/app/Http/Controllers/Remote/ModuleController.php
index d35ac5e..265159f 100644
--- a/app/Http/Controllers/Remote/ModuleController.php
+++ b/app/Http/Controllers/Remote/ModuleController.php
@@ -53,13 +53,13 @@ public function call(Request $request, Module $module)
         $method = Str::lower($request->method());
 
         // 如果 method 为 post, 检查用户余额
-        if ($method == 'post') {
-            $user = auth('api')->user();
+        // if ($method == 'post') {
+        //     $user = auth('api')->user();
 
-            if ($user->balance < 1) {
-                return $this->error('账户余额不足，请保证账户余额至少有 1 元。');
-            }
-        }
+        //     if ($user->balance < 1) {
+        //         return $this->error('账户余额不足，请保证账户余额至少有 1 元。');
+        //     }
+        // }
 
 
         $response = $module->remoteRequest($method, $path, $request->all());
diff --git a/app/Http/Controllers/User/HostController.php b/app/Http/Controllers/User/HostController.php
index 3b53343..c5274fe 100644
--- a/app/Http/Controllers/User/HostController.php
+++ b/app/Http/Controllers/User/HostController.php
@@ -23,9 +23,9 @@ public function update(Request $request, Host $host)
         $user = $request->user();
         if ($host->user_id == $user->id) {
 
-            if ($user->balance < 1) {
-                return $this->error('余额不足');
-            }
+            // if ($user->balance < 1) {
+            //     return $this->error('余额不足');
+            // }
 
             $host->update([
                 'status' => 'running'