ProxyPanel/app/Http/Controllers/Admin/UserController.php

<?php

namespace App\Http\Controllers\Admin;

use App\Components\Helpers;
use App\Components\IP;
use App\Http\Controllers\Controller;
use App\Http\Requests\Admin\UserStoreRequest;
use App\Http\Requests\Admin\UserUpdateRequest;
use App\Models\Level;
use App\Models\Node;
use App\Models\Order;
use App\Models\User;
use App\Models\UserGroup;
use App\Models\UserHourlyDataFlow;
use Auth;
use Exception;
use Illuminate\Http\JsonResponse;
use Illuminate\Http\Request;
use Log;
use Response;
use Session;
use Spatie\Permission\Models\Role;
use Str;

class UserController extends Controller
{
    // 用户列表
    public function index(Request $request)
    {
        $id = $request->input('id');
        $email = $request->input('email');
        $wechat = $request->input('wechat');
        $qq = $request->input('qq');
        $port = $request->input('port');
        $group = $request->input('group');
        $level = $request->input('level');
        $status = $request->input('status');
        $enable = $request->input('enable');
        $online = $request->input('online');
        $flowAbnormal = $request->input('flowAbnormal');
        $expireWarning = $request->input('expireWarning');
        $largeTraffic = $request->input('largeTraffic');

        $query = User::with('subscribe');
        if (isset($id)) {
            $query->whereId($id);
        }

        if (isset($email)) {
            $query->where('email', 'like', '%'.$email.'%');
        }

        if (isset($wechat)) {
            $query->where('wechat', 'like', '%'.$wechat.'%');
        }

        if (isset($qq)) {
            $query->where('qq', 'like', '%'.$qq.'%');
        }

        if (isset($port)) {
            $query->wherePort($port);
        }

        if (isset($status)) {
            $query->whereStatus($status);
        }

        if (isset($enable)) {
            $query->whereEnable($enable);
        }

        if (isset($group)) {
            $query->whereGroupId($group);
        }

        if (isset($level)) {
            $query->whereLevel($level);
        }

        // 流量超过100G的
        if ($largeTraffic) {
            $query->whereIn('status', [0, 1])->whereRaw('(u + d)/transfer_enable >= 0.9');
        }

        // 临近过期提醒
        if ($expireWarning) {
            $query->whereBetween('expired_at', [date('Y-m-d'), date('Y-m-d', strtotime('+'.sysConfig('expire_days').' days'))]);
        }

        // 当前在线
        if ($online) {
            $query->where('t', '>=', strtotime('-10 minutes'));
        }

        // 不活跃用户
        if ($request->has('unActive')) {
            $query->whereBetween('t', [1, strtotime('-'.sysConfig('expire_days').' days')])->whereEnable(1);
        }

        // 不活跃用户
        if ($request->has('paying')) {
            $payingUser = Order::whereStatus(2)->where('goods_id', '<>', 0)->whereIsExpire(0)->where('amount', '>', 0)->pluck('user_id')->unique();
            $query->whereIn('id', $payingUser);
        }

        // 1小时内流量异常用户
        if ($flowAbnormal) {
            $query->whereIn('id', (new UserHourlyDataFlow)->trafficAbnormal());
        }

        return view('admin.user.index', [
            'userList' => $query->orderByDesc('id')->paginate(15)->appends($request->except('page')),
            'userGroups' => UserGroup::all()->pluck('name', 'id')->toArray(),
            'levels' => Level::all()->pluck('name', 'level')->toArray(),
        ]);
    }

    // 添加账号页面
    public function create()
    {
        if (Auth::getUser()->hasRole('Super Admin')) {
            $roles = Role::all()->pluck('description', 'name');
        } elseif (Auth::getUser()->hasPermissionTo('give roles')) {
            $roles = Auth::getUser()->roles();
        }

        return view('admin.user.info', [
            'levels' => Level::orderBy('level')->get(),
            'userGroups' => UserGroup::orderBy('id')->get(),
            'roles' => $roles ?? [],
        ]);
    }

    // 添加账号
    public function store(UserStoreRequest $request): JsonResponse
    {
        try {
            $data = $request->except('_token', 'uuid', 'roles');
            $data['password'] = $data['password'] ?? Str::random();
            $data['port'] = $data['port'] ?? Helpers::getPort();
            $data['passwd'] = $data['passwd'] ?? Str::random();
            $data['vmess_id'] = $request->input('uuid') ?? Str::uuid();
            $data['expired_at'] = $data['expired_at'] ?? date('Y-m-d', strtotime('+365 days'));
            $data['remark'] = str_replace(['atob', 'eval'], '', $data['remark']);
            $data['reg_ip'] = IP::getClientIp();
            $data['reset_time'] = $data['reset_time'] > date('Y-m-d') ? $data['reset_time'] : null;
            $user = User::create($data);

            $roles = $request->input('roles') ?? [];
            if ($roles && (Auth::getUser()->hasPermissionTo('give roles') || (in_array('Super Admin', $roles, true) && Auth::getUser()->hasRole('Super Admin'))
                    || Auth::getUser()->hasRole('Super Admin'))) {
                $user->assignRole($roles);
            }

            if ($user) {
                // 写入用户流量变动记录
                Helpers::addUserTrafficModifyLog($user->id, null, 0, $data['transfer_enable'], '后台手动添加用户');

                return Response::json(['status' => 'success', 'message' => '添加成功']);
            }
        } catch (Exception $e) {
            Log::error('添加用户错误：'.$e->getMessage());

            return Response::json(['status' => 'fail', 'message' => $e->getMessage()]);
        }

        return Response::json(['status' => 'fail', 'message' => '添加失败']);
    }

    // 编辑账号页面
    public function edit(User $user)
    {
        if (Auth::getUser()->hasRole('Super Admin')) {
            $roles = Role::all()->pluck('description', 'name');
        } elseif (Auth::getUser()->hasPermissionTo('give roles')) {
            $roles = Auth::getUser()->roles();
        }

        return view('admin.user.info', [
            'user' => $user->load('inviter:id,email'),
            'levels' => Level::orderBy('level')->get(),
            'userGroups' => UserGroup::orderBy('id')->get(),
            'roles' => $roles ?? [],
        ]);
    }

    // 编辑账号
    public function update(UserUpdateRequest $request, User $user)
    {
        try {
            $data = $request->except('_token', 'password', 'uuid', 'password', 'roles');
            $data['passwd'] = $request->input('passwd') ?? Str::random();
            $data['vmess_id'] = $request->input('uuid') ?? Str::uuid();
            $data['transfer_enable'] *= GB;
            $data['enable'] = $data['status'] < 0 ? 0 : $data['enable'];
            $data['expired_at'] = $data['expired_at'] ?? date('Y-m-d', strtotime('+365 days'));
            $data['remark'] = str_replace(['atob', 'eval'], '', $data['remark']);

            // 只有超级管理员才能赋予超级管理员
            $roles = $request->input('roles') ?? [];

            if ($roles && (Auth::getUser()->hasPermissionTo('give roles') || (in_array('Super Admin', $roles, true) && Auth::getUser()->hasRole('Super Admin')) ||
                    Auth::getUser()->hasRole('Super Admin'))) {
                $user->syncRoles($roles);
            }

            // Input checking for dummy
            if ($data['enable'] === '1') {
                if ($data['status'] === '-1' || $data['transfer_enable'] === 0 || $data['expired_at'] < date('Y-m-d')) {
                    $data['enable'] = 0;
                }
            }

            // 非演示环境才可以修改管理员密码
            $password = $request->input('password');
            if (! empty($password) && ! (env('APP_DEMO') && $user->id === 1)) {
                $data['password'] = $password;
            }

            // 写入用户流量变动记录
            if ($user->transfer_enable !== $data['transfer_enable']) {
                Helpers::addUserTrafficModifyLog($user->id, null, $user->transfer_enable, $data['transfer_enable'], '后台手动编辑用户');
            }

            if ($user->update($data)) {
                return Response::json(['status' => 'success', 'message' => '编辑成功']);
            }
        } catch (Exception $e) {
            Log::error('编辑用户信息异常：'.$e->getMessage());

            return Response::json(['status' => 'fail', 'message' => '编辑用户信息错误：'.$e->getMessage()]);
        }

        return Response::json(['status' => 'fail', 'message' => '编辑失败']);
    }

    // 删除用户
    public function destroy(User $user)
    {
        if ($user->id === 1) {
            return Response::json(['status' => 'fail', 'message' => '系统管理员不可删除']);
        }

        try {
            if ($user->delete()) {
                return Response::json(['status' => 'success', 'message' => '删除成功']);
            }
        } catch (Exception $e) {
            Log::error('删除用户信息异常：'.$e->getMessage());

            return Response::json(['status' => 'fail', 'message' => '删除失败'.$e->getMessage()]);
        }

        return Response::json(['status' => 'fail', 'message' => '删除失败']);
    }

    // 批量生成账号
    public function batchAddUsers()
    {
        $preset = ['transfer_enable' => 1024 * GB, 'expired_at' => date('Y-m-d', strtotime('+365 days'))];

        try {
            for ($i = 0; $i < (int) request('amount', 1); $i++) {
                $user = Helpers::addUser(Str::random(8).'@auto.generate', Str::random(), 1024 * GB, 365);
                // 写入用户流量变动记录
                Helpers::addUserTrafficModifyLog($user->id, null, 0, 1024 * GB, '后台批量生成用户');
            }

            return Response::json(['status' => 'success', 'message' => '批量生成账号成功']);
        } catch (Exception $e) {
            return Response::json(['status' => 'fail', 'message' => '批量生成账号失败：'.$e->getMessage()]);
        }
    }

    // 转换成某个用户的身份
    public function switchToUser(User $user): JsonResponse
    {
        // 存储当前管理员ID，并将当前登录信息改成要切换的用户的身份信息
        Session::put('admin', Auth::id());
        Session::put('user', $user->id);

        return Response::json(['status' => 'success', 'message' => '身份切换成功']);
    }

    // 重置用户流量
    public function resetTraffic(User $user): JsonResponse
    {
        try {
            $user->update(['u' => 0, 'd' => 0]);
        } catch (Exception $e) {
            Log::error('流量重置失败：'.$e->getMessage());

            return Response::json(['status' => 'fail', 'message' => '流量重置失败']);
        }

        return Response::json(['status' => 'success', 'message' => '流量重置成功']);
    }

    // 操作用户余额
    public function handleUserCredit(Request $request, User $user): JsonResponse
    {
        $amount = (int) $request->input('amount');

        if (empty($amount)) {
            return Response::json(['status' => 'fail', 'message' => '充值异常']);
        }

        // 加减余额
        if ($user->updateCredit($amount)) {
            Helpers::addUserCreditLog($user->id, null, $user->credit, $user->credit + $amount, $amount, '后台手动充值');  // 写入余额变动日志

            return Response::json(['status' => 'success', 'message' => '充值成功']);
        }

        return Response::json(['status' => 'fail', 'message' => '充值失败']);
    }

    // 导出配置信息
    public function export(User $user)
    {
        return view('admin.user.export', [
            'user' => $user,
            'nodeList' => Node::whereStatus(1)->orderByDesc('sort')->orderBy('id')->paginate(15)->appends(\request('page')),
        ]);
    }

    public function exportProxyConfig(Request $request, User $user): JsonResponse
    {
        $node = Node::find($request->input('id'));
        if ($node->type === 1) {
            if ($node->compatible) {
                $proxyType = 'SS';
            } else {
                $proxyType = 'SSR';
            }
        } else {
            $proxyType = 'V2Ray';
        }

        $data = $this->getUserNodeInfo($user->id, $node->id, $request->input('type') !== 'text' ? 0 : 1);

        return Response::json(['status' => 'success', 'data' => $data, 'title' => $proxyType]);
    }
}