dineng/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 Firebase\JWT\JWT;
use think\Controller;
use think\Db;
use think\Exception;
use think\exception\HttpResponseException;
use think\Request;
use think\Response;
use function AlibabaCloud\Client\value;

/**
 * 会员管理基类
 * Class Member
 * @package app\store\controller\api
 */
class Base extends Controller
{
    protected $user_id;
    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(){
        $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(!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;
                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 = Db::name('store_member')->where('id',$authInfo['uid'])->find();
                if(empty($member) || $member['is_deleted'])  $this->exception('会员不存在');
                //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'];
                    return $this->user_id;
                }else{
                    $this->exception('该会员已被禁用');
                }
            } else {
                $this->exception('Token验证不通过,用户不存在');
            }
        } 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'];
                    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);
    }


}