dingding/application/api/controller/Base.php

<?php

// +----------------------------------------------------------------------
// | ThinkAdmin
// +----------------------------------------------------------------------
// | 版权所有 2014~2019 广州楚才信息科技有限公司 [ http://www.cuci.cc ]
// +----------------------------------------------------------------------
// | 官方网站: http://demo.thinkadmin.top
// +----------------------------------------------------------------------
// | 开源协议 ( https://mit-license.org )
// +----------------------------------------------------------------------
// | gitee 代码仓库：https://gitee.com/zoujingli/ThinkAdmin
// | github 代码仓库：https://github.com/zoujingli/ThinkAdmin
// +----------------------------------------------------------------------

namespace app\api\controller;

use app\common\constant\CommonConstant;
use app\common\model\User;
use Firebase\JWT\JWT;
use think\Controller;
use think\Db;
use think\Exception;
use think\exception\HttpResponseException;
use think\Loader;
use think\Response;
use think\Validate;
use function AlibabaCloud\Client\value;

/**
 * 会员管理基类
 * Class Member
 * @package app\store\controller\api
 */
class Base extends Controller
{
    protected $user_id;
    protected $user;
    protected $page; // 页数
    protected $page_num;// 每页多少
    protected $off_set;
    protected $is_commit = true; // 事务是否提交
    protected $ret_msg = ''; // 返回提示信息
    protected $is_test = false;
    protected $need_login = [];

    public function initialize()
    {
        $modulename = $this->request->module();
        $controllername = Loader::parseName($this->request->controller());
        $actionname = strtolower($this->request->action());
        $this->checking($modulename, $controllername, $actionname);

        $this->page = input('page', 1);
        $this->page_num = input('page_num', 20);
        $this->off_set = $this->page * $this->page_num - $this->page_num;
        $this->is_test = input('test', 0);// 测试用的
        $path = explode('/', $this->request->path());
        if (input('user_id')) $this->user_id = input('user_id');
        if (input('user')) $this->user_id = input('user');
        if (!empty($this->need_login) && in_array(end($path), $this->need_login)) $this->checkLogin();
    }

    //校验jwt权限API
    protected function checkLogin()
    {
        $authorization = app()->request->header('Authorization');
        if (empty($authorization) || $authorization == null) {
            if ($this->is_test == 1) {
                $this->user_id = input('user_id') ? input('user_id') : 22;
                if (input('user')) $this->user_id = input('user');
                return true;
            }
            $this->error('Token不存在,拒绝访问--', '', 0, -1);
        }
        $key = md5(config('app.jwt'));
        try {
            $check_authorization = JWT::decode($authorization, $key, array('HS256'));
            if ($check_authorization['code'] != 200) $this->exception($check_authorization['msg']);
            $authInfo = json_decode(json_encode($check_authorization['data']), true);
            if (!empty($authInfo['uid'])) {
                $member = User::find($authInfo['uid']);
                if (empty($member) || $member['is_deleted']) $this->error('会员不存在', '', 0);
                //if(!empty($authInfo['facility_code']) && !in_array($authInfo['facility_code'],[$member['facility_1'],$member['facility_2'],$member['facility_3'],$member['facility_4'],$member['facility_5']]))    $this->error('该设备已被禁用','',0,-1);
                if ($member['status']) {
                    $this->user_id = $authInfo['uid'];
                    $this->user = $member;
                    return $this->user_id;
                } else {
                    $this->error('该会员已被禁用', '', 0, -1);
                }
            } else {
                $this->error('Token验证不通过,用户不存在', '', 0, -1);
            }
        } catch (\Exception $e) {
            $this->error($e->getMessage(), '', 0, -1);
        }
    }


    protected function setUid()
    {
        $authorization = app()->request->header('Authorization');
        $key = md5(config('app.jwt'));
        if (empty($authorization) || $authorization == null) return false;
        try {
            $check_authorization = JWT::decode($authorization, $key, array('HS256'));
            if ($check_authorization['code'] != 200) $this->exception($check_authorization['msg']);
            $authInfo = json_decode(json_encode($check_authorization['data']), true);
            if (!empty($authInfo['uid'])) {
                $member = Db::name('store_member')->field('status')->where('id', $authInfo['uid'])->find();
                if ($member['status']) {
                    $this->user_id = $authInfo['uid'];
                    $this->user_id;
                    return true;
                }
            }
        } catch (\Exception $e) {
            return false;
            //$this->error($e->getMessage(),'',0,-1);
        }
    }

    // 获取用户信息
    function userInfo($field = '')
    {
        if (!$this->user_id) return [];
        return $field ? Db::name('store_member')->field($field)->find($this->user_id) : Db::name('store_member')->find($this->user_id);
    }

    // 验证短信验证码
    protected function checkPhoneCode($phone, $code)
    {
        //return true;
        $sel_time = date('Y-m-d H:i:s', time() - 600);
        $store_member_sms = Db::name('store_member_sms')
            ->field('id,code')->where('phone', $phone)
            ->where('create_at', '> time', $sel_time)
            ->where('used', 0)->order('id desc')->find();
        return !empty($store_member_sms) && $store_member_sms['code'] == $code ? $store_member_sms['id'] : 0;
    }

    // 更新验证码状态
    protected function updatePhoneCode($code_id)
    {
        Db::name('store_member_sms')->where('id', $code_id)->update(['used' => 1]);
    }

    /**
     * 操作成功返回的数据
     * @param string $msg 提示信息
     * @param mixed $data 要返回的数据
     * @param int $code 错误码，默认为1
     * @param string $type 输出类型
     * @param array $header 发送的 Header 信息
     */
    protected function success($msg = 'ok', $data = null, $is_login = 1, $code = 1, $type = null, array $header = [])
    {
        $this->results($msg, $data, $is_login, $code, $type, $header);
    }

    /**
     * 操作失败返回的数据
     * @param string $msg 提示信息
     * @param mixed $data 要返回的数据
     * @param int $code 错误码，默认为0
     * @param string $type 输出类型
     * @param array $header 发送的 Header 信息
     */
    protected function error($msg = '', $data = null, $is_login = 1, $code = 0, $type = null, array $header = [])
    {
        if (empty($this->user_id)) {
            $is_login = 0;
        }
        $this->results($msg, $data, $is_login, $code, $type, $header);
    }

    /**
     * 返回封装后的 API 数据到客户端
     * @access protected
     * @param mixed $msg 提示信息
     * @param mixed $data 要返回的数据
     * @param int $code 错误码，默认为0
     * @param string $type 输出类型，支持json/xml/jsonp
     * @param array $header 发送的 Header 信息
     * @return void
     * @throws HttpResponseException
     */
    protected function results($msg, $data = null, $is_login, $code = 0, $type = null, array $header = [])
    {
        $result = [
            'code' => $code,
            'is_login' => $is_login,
            'msg' => $msg,
            'time' => \think\facade\Request::instance()->server('REQUEST_TIME'),
            'data' => $data,
        ];
        // 如果未设置类型则自动判断
        $type = $type ? $type : 'json';

        if (isset($header['statuscode'])) {
            $code = $header['statuscode'];
            unset($header['statuscode']);
        } else {
            //未设置状态码,根据code值判断
            $code = $code >= 1000 || $code < 200 ? 200 : $code;
        }
        $response = Response::create($result, $type, $code)->header($header);
        throw new HttpResponseException($response);
    }


    //token加密
    public function createJwt($uid, $facility_code = '')
    {
        $key = md5(config('app.jwt')); //jwt的签发密钥，验证token的时候需要用到
        $time = time(); //签发时间
        $expire = $time + config('app.jwt_time'); //过期时间
        $token = array(
            "uid" => $uid,
            "iss" => "https://zain.com",//签发组织
            "aud" => "https://zain.com", //签发作者
            "iat" => $time,
            "nbf" => $time,
            "exp" => $expire,
            "facility_code" => $facility_code,
        );
        $jwt = JWT::encode($token, $key);
        return $jwt;
    }

    protected function exception($msg)
    {
        throw new Exception($msg);
    }

    // 事务返回
    protected function transReturn($data = [])
    {
        $this->is_commit ? $this->success($this->ret_msg, $data) : $this->error($this->ret_msg);
    }

    protected function checking($modulename, $controllername, $action_name)
    {
        $params = $this->request->post();
        if (in_array($action_name, ['get_data', 'create', 'edit'])) {
            $get_module_list = CommonConstant::get_module_list();
            $rule = [
                'module|模块类型' => 'require|in:' . implode(',', array_keys($get_module_list)),
            ];
            $message = [
                'module.in' => '请选择正确的模块类型',
            ];
            $validate = new Validate($rule, $message);
            if (!$validate->check($params)) {
                $this->error($validate->getError());
            }
            if ($action_name == 'create') {
                $id = $this->request->post('id');
                $module = $this->request->post('module');
                $validates = CommonConstant::get_module_validate_list();
                $validate = $validates[$module];
                $validate = new $validate;
                if (!$validate->check($params, [], $id > 0 ? 'update' : 'create')) {
                    $this->error($validate->getError());
                }
            }
            if ($action_name == 'edit') {
                $module = $this->request->post('module');
                $validates = CommonConstant::get_module_validate_list();
                $validate = $validates[$module];
                $validate = new $validate;
                if (!$validate->check($params, [], 'edit')) {
                    $this->error($validate->getError());
                }
            }
        }
        if ($controllername == 'approveinfo') {
            if (in_array($action_name, ['get_detail', 'get_info', 'urging', 'cancel'])) {
                $rule = [
                    'id|申请参数' => 'require|gt:0',
                ];
                $message = [];
                $validate = new Validate($rule, $message);
                if (!$validate->check($params)) {
                    $this->error($validate->getError());
                }
            }
        }
        if ($controllername == 'approve') {
            if (in_array($action_name, ['get_detail', 'get_info'])) {
                $rule = [
                    'approve_id|审批参数' => 'require|gt:0',
                ];
                $message = [];
                $validate = new Validate($rule, $message);
                if (!$validate->check($params)) {
                    $this->error($validate->getError());
                }
            }
            if ($action_name == 'audit') {
                $rule = [
                    'approve_id|审批参数' => 'require|gt:0',
                    'status|审批状态' => 'require|in:' . CommonConstant::STATUS_3 . ',' . CommonConstant::STATUS_4,
                ];
                $message = [
                    'status.in' => '请选择正确的审批状态',
                ];
                $validate = new Validate($rule, $message);
                if (!$validate->check($params)) {
                    $this->error($validate->getError());
                }
            }
        }
    }

}