From c51d5aeb4f788a3e988556287d437faa57ada186 Mon Sep 17 00:00:00 2001
From: "iVampireSP.com" <im@ivampiresp.com>
Date: Tue, 29 Nov 2022 18:43:31 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=20=E4=BD=99=E9=A2=9D?=
 =?UTF-8?q?=E6=89=A3=E9=99=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/Http/Controllers/Admin/UserController.php | 13 +++++++++----
 resources/views/admin/users/edit.blade.php    | 13 ++++++++-----
 2 files changed, 17 insertions(+), 9 deletions(-)

diff --git a/app/Http/Controllers/Admin/UserController.php b/app/Http/Controllers/Admin/UserController.php
index c24e3a8..91691b0 100644
--- a/app/Http/Controllers/Admin/UserController.php
+++ b/app/Http/Controllers/Admin/UserController.php
@@ -2,6 +2,7 @@
 
 namespace App\Http\Controllers\Admin;
 
+use App\Exceptions\ChargeException;
 use App\Http\Controllers\Controller;
 use App\Models\Balance;
 use App\Models\Host;
@@ -93,10 +94,6 @@ public function update(Request $request, User $user)
 
         $transaction = new Transaction();
 
-        if ($request->filled('balance')) {
-            $transaction->addAmount($user->id, 'console', $request->balance, '管理员汇入', true);
-        }
-
         if ($request->is_banned) {
             $user->banned_at = Carbon::now();
 
@@ -116,6 +113,14 @@ public function update(Request $request, User $user)
                 $user->hosts()->update(['status' => 'suspended', 'suspended_at' => now()]);
             } else if ($request->one_time_action == 'stop_all_hosts') {
                 $user->hosts()->update(['status' => 'stopped', 'suspended_at' => null]);
+            } else if ($request->one_time_action == 'add_balance') {
+                try {
+                    $transaction->addAmount($user->id, 'console', $request->balance ?? 0, '管理员添加。');
+                } catch (ChargeException $e) {
+                    return back()->with('error', $e->getMessage());
+                }
+            } else if ($request->one_time_action == 'reduce_balance') {
+                $transaction->reduceAmount($user->id, $request->balance ?? 0, '管理员扣除。');
             }
 
         }
diff --git a/resources/views/admin/users/edit.blade.php b/resources/views/admin/users/edit.blade.php
index 63c87f9..4b09a2c 100644
--- a/resources/views/admin/users/edit.blade.php
+++ b/resources/views/admin/users/edit.blade.php
@@ -130,11 +130,6 @@
         @csrf
         @method('PATCH')
 
-        <div class="form-group">
-            <label for="balance">充值余额(元)</label>
-            <input type="text" class="form-control" id="balance" name="balance" placeholder="充值金额">
-        </div>
-
         {{-- 封禁 --}}
         <div class="form-group">
             <label for="is_banned">封禁</label>
@@ -151,11 +146,19 @@
                    value="{{ $user->banned_reason }}">
         </div>
 
+
+        <div class="form-group">
+            <label for="balance">余额(元)</label>
+            <input type="text" class="form-control" id="balance" name="balance" placeholder="需要在下方选择一次性操作。">
+        </div>
+
         {{-- 一次性操作 --}}
         <div class="form-group">
             <label for="one_time_action">一次性操作</label>
             <select class="form-control" id="one_time_action" name="one_time_action">
                 <option value="">无</option>
+                <option value="add_balance">充值余额</option>
+                <option value="reduce_balance">扣除余额</option>
                 <option value="clear_all_keys">清除所有密钥</option>
                 <option value="suspend_all_hosts">暂停所有主机(3天后不恢复，将会自动删除)</option>
                 <option value="stop_all_hosts">停止所有主机(从暂停中恢复或者将其设置为 停止，需要用户手动启动)</option>