thinkphp6-template/app/BaseController.php

<?php
declare(strict_types=1);

namespace app;

use think\App;
use think\exception\ValidateException;
use think\Validate;
use think\facade\Db;
use think\cache\driver\Redis;


/**
 * 控制器基础类
 */
abstract class BaseController
{
    /**
     * Request实例
     * @var \think\Request
     */
    protected $request;

    /**
     * 应用实例
     * @var \think\App
     */
    protected $app;

    /**
     * 是否批量验证
     * @var bool
     */
    protected $batchValidate = false;

    /**
     * 控制器中间件
     * @var array
     */
    protected $middleware = [];

    /**
     * 无需登录鉴权的方法名
     * @var array
     */
    protected $noNeedLogin = [];
    /**
     * redis对象
     * @var array
     */
    protected $redis;

    private $token;
    private $userData = null;


    /**
     * 构造方法
     * @access public
     * @param  App  $app  应用对象
     */
    public function __construct(App $app)
    {

        $this->app = $app;
        $this->request = $this->app->request;
        $this->redis = new Redis();
        // 控制器初始化
        $this->initialize();
    }

    // 初始化
    protected function initialize()
    {
        //noNeedLogin全部转换小写
        foreach ($this->noNeedLogin as &$action) {
            $action = strtolower($action);
        }
        //验证需要鉴权的方法名
        $callMethodName = strtolower($this->request->action());
        if (!in_array($callMethodName, $this->noNeedLogin)) {
            $userToken = Db::table('tb_user_token')
                ->where(['token' => $this->request->header('token')])
                ->where('expired_at', '>', time())
                ->find();
            if (!$userToken) {
                $this->fail(401, '请先登录后再操作');
                header('Location: /login');
            }
            $this->token = $this->request->header('token');
        }
        session_start();
    }


    protected function getUser()
    {
        if ($this->userData == null) {
            $this->userData = Db::table('tb_user_token')
                ->where(['token' => $this->token])
                ->find();
        }
        return (object) $this->userData;
    }

    /**
     * 验证数据
     * @access protected
     * @param  array        $data     数据
     * @param  string|array $validate 验证器名或者验证规则数组
     * @param  array        $message  提示信息
     * @param  bool         $batch    是否批量验证
     * @return array|string|true
     * @throws ValidateException
     */
    protected function validate(array $data, $validate, array $message = [], bool $batch = false)
    {
        if (is_array($validate)) {
            $v = new Validate();
            $v->rule($validate);
        } else {
            if (strpos($validate, '.')) {
                // 支持场景
                [$validate, $scene] = explode('.', $validate);
            }
            $class = false !== strpos($validate, '\\') ? $validate : $this->app->parseClass('validate', $validate);
            $v = new $class();
            if (!empty($scene)) {
                $v->scene($scene);
            }
        }

        $v->message($message);

        // 是否批量验证
        if ($batch || $this->batchValidate) {
            $v->batch(true);
        }

        return $v->failException(true)->check($data);
    }

    protected function success($msg, $data)
    {
        header('Content-Type:application/json; charset=utf-8');
        echo json_encode([
            'code' => 200,
            'msg' => $msg,
            'data' => $data,
            'time' => time()
        ]);
        exit();
    }

    protected function fail($code, $msg)
    {
        header('Content-Type:application/json; charset=utf-8');
        echo json_encode([
            'code' => $code,
            'msg' => $msg,
            'time' => time()
        ]);
        exit();
    }

    //发送TCP消息
    protected function sendTcpMessage()
    {

    }

    //金币增加
    protected function addGold($value)
    {
        $row = Db::table('tb_user')
            ->where(['id' => $this->getUser()->user_id])
            ->find();
        Db::table('tb_user')
            ->where(['id' => $this->getUser()->user_id])
            ->update([
                'gold' => $row['gold'] + $value,
            ]);
    }

    //增加经验
    protected function addExp($value)
    {
        $row = Db::table('tb_user')
            ->where(['id' => $this->getUser()->user_id])
            ->find();
        Db::table('tb_user')
            ->where(['id' => $this->getUser()->user_id])
            ->update([
                'exp' => $row['exp'] + $value,
            ]);
    }

    //增加积分
    protected function addScore($value)
    {
        $row = Db::table('tb_user')
            ->where(['id' => $this->getUser()->user_id])
            ->find();
        Db::table('tb_user')
            ->where(['id' => $this->getUser()->user_id])
            ->update([
                'score' => $row['score'] + $value,
            ]);
    }

    // 解析获取头像Url
    public function getUserAvatar($avatar)
    {
        if (strpos($avatar, 'system://') === 0) {
            $data = explode('//', $avatar);
            $avatarId = isset($data[1]) ? intval($data[1]) : 0;
            if ($avatarId <= 0) {
                return $avatar;
            }
            $avatarRow = Db::table('tb_system_avatar')
                ->where(['id' => $avatarId])
                ->find();
            return $avatarRow ? $avatarRow['image_url'] : $avatar;
        }
        return $avatar;
    }

}