Qiniu_Pro/app/services/wechat/WechatUserServices.php

<?php
// +----------------------------------------------------------------------
// | CRMEB [ CRMEB赋能开发者，助力企业发展 ]
// +----------------------------------------------------------------------
// | Copyright (c) 2016~2020 https://www.crmeb.com All rights reserved.
// +----------------------------------------------------------------------
// | Licensed CRMEB并不是自由软件，未经许可不能去掉CRMEB相关版权
// +----------------------------------------------------------------------
// | Author: CRMEB Team <admin@crmeb.com>
// +----------------------------------------------------------------------
declare (strict_types=1);

namespace app\services\wechat;

use qiniu\basic\BaseServices;
use app\model\wechat\WechatUser;
use qiniu\exceptions\AdminException;
use qiniu\exceptions\AuthException;
use qiniu\services\wechat\OfficialAccount;
use think\db\exception\DataNotFoundException;
use think\db\exception\DbException;
use think\db\exception\ModelNotFoundException;
use think\exception\ValidateException;

/**
 *
 * Class WechatUserServices
 * @package app\services\wechat
 * @mixin WechatUser
 */
class WechatUserServices extends BaseServices
{

    /**
     * WechatUserServices constructor.
     * @param WechatUser $model
     */
    public function __construct(WechatUser $model)
    {
        $this->model = $model;
    }


    public function getColumnUser($user_ids, $column, $key, string $user_type = 'wechat')
    {
        return $this->getColumn([['uid', 'IN', $user_ids], ['user_type', '=', $user_type]], $column, $key);
    }

    /**
     * 获取单个微信用户
     * @param array $where
     * @param string $field
     * @return array
     * @throws DataNotFoundException
     * @throws DbException
     * @throws ModelNotFoundException
     */
    public function getWechatUserInfo(array $where, $field = '*')
    {
        return $this->getOne($where, $field);
    }

    /**
     * 用uid获得 微信openid
     * @param int $uid
     * @param string $userType
     * @return mixed
     * @throws DataNotFoundException
     * @throws DbException
     * @throws ModelNotFoundException
     */
    public function uidToOpenid(int $uid, string $userType = 'wechat')
    {
        return $this->getOne(['uid' => $uid, 'user_type' => $userType])['openid'] ?? null;
    }


    /**
     *  用openid获得uid
     * @param $openid
     * @param string $openidType
     * @return mixed
     * @throws DataNotFoundException
     * @throws DbException
     * @throws ModelNotFoundException
     */
    public function openidToUid($openid, string $openidType = 'openid')
    {
        $uid = $this->getOne([[$openidType, '=', $openid], ['user_type', '<>', 'h5']])['uid'] ?? 0;
        if (!$uid)
            throw new AdminException('对应的uid不存在');
        return $uid;
    }


    /**
     * 获取用户信息
     * @param $openid
     * @return array|mixed
     * @throws DataNotFoundException
     * @throws DbException
     * @throws ModelNotFoundException
     */
    public function getAuthUserInfo($openid)
    {
        $user = [];
        if ($openid) {
            $uid = $this->model->getOneValue(['unionid|openid' => $openid], 'uid');
            /** @var UserServices $userServices */
            $userServices = app()->make(UserServices::class);
            $user = $userServices->getUserInfo($uid);
            if (isset($user['status']) && !$user['status'])
                throw new AuthException('您已被禁止登录，请联系管理员', 410020);
        }
        return $user;
    }

    /**
     * 更新微信用户信息
     * @param $event
     * @return bool
     */
    public function wechatUpdata($data)
    {
        [$uid, $userData] = $data;
        /** @var UserServices $userServices */
        $userServices = app()->make(UserServices::class);
        if (!$userInfo = $userServices->getUserInfo($uid)) {
            return false;
        }
        /** @var LoginServices $loginService */
        $loginService = app()->make(LoginServices::class);
        $loginService->updateUserWechatLogin($uid, $userData, $userInfo);
        //更新用户信息
        /** @var WechatUserServices $wechatUser */
        $wechatUser = app()->make(WechatUserServices::class);

        $wechatUserInfo = [];
        if (isset($userData['nickname']) && $userData['nickname']) $wechatUserInfo['nickname'] = filter_emoji($userData['nickname'] ?? '');//姓名
        if (isset($userData['headimgurl']) && $userData['headimgurl']) $wechatUserInfo['headimgurl'] = $userData['headimgurl'] ?? '';//头像
        if (isset($userData['sex']) && $userData['sex']) $wechatUserInfo['sex'] = $userData['gender'] ?? '';//性别
        if (isset($userData['language']) && $userData['language']) $wechatUserInfo['language'] = $userData['language'] ?? '';//语言
        if (isset($userData['city']) && $userData['city']) $wechatUserInfo['city'] = $userData['city'] ?? '';//城市
        if (isset($userData['province']) && $userData['province']) $wechatUserInfo['province'] = $userData['province'] ?? '';//省份
        if (isset($userData['country']) && $userData['country']) $wechatUserInfo['country'] = $userData['country'] ?? '';//国家
        if (!empty($wechatUserInfo['nickname']) || !empty($wechatUserInfo['headimgurl'])) {
            $wechatUserInfo['is_complete'] = 1;
        } else {
            $wechatUserInfo['is_complete'] = 0;
        }
        if ($wechatUserInfo) {
            if (isset($userData['openid']) && $userData['openid'] && false === $wechatUser->update(['uid' => $userInfo['uid'], 'openid' => $userData['openid']], $wechatUserInfo)) {
                throw new ValidateException('更新失败');
            }
        }
        return true;
    }

    /**
     * 微信授权成功后
     * @param array $data
     * @return mixed
     * @throws DataNotFoundException
     * @throws DbException
     * @throws ModelNotFoundException
     */
    public function wechatOauthAfter(array $data)
    {
        [$openid, $wechatInfo, $spread_uid, $login_type, $userType] = $data;
        /** @var UserServices $userServices */
        $userServices = app()->make(UserServices::class);
        if ($spread_uid && !$userServices->userExist((int)$spread_uid)) {
            $spread_uid = 0;
        }
        //删除多余字段
        unset($wechatInfo['subscribe_scene'], $wechatInfo['qr_scene'], $wechatInfo['qr_scene_str']);
        if ($login_type) {
            $wechatInfo['login_type'] = $login_type;
        }
        if (!isset($wechatInfo['nickname']) || !$wechatInfo['nickname']) {
            if (isset($wechatInfo['phone']) && $wechatInfo['phone']) {
                $wechatInfo['nickname'] = substr_replace($wechatInfo['phone'], '****', 3, 4);
            } else {
                mt_srand();
                $wechatInfo['nickname'] = 'wx' . rand(100000, 999999);
            }
        } else {
            $wechatInfo['is_complete'] = 1;
            $wechatInfo['nickname'] = filter_emoji($wechatInfo['nickname']);
        }
        $userInfo = [];
        if (isset($wechatInfo['phone']) && $wechatInfo['phone']) {
            $userInfo = $userServices->getOne(['phone' => $wechatInfo['phone']]);
        }
        if (!$userInfo) {
            if (isset($wechatInfo['unionid']) && $wechatInfo['unionid']) {
                $uid = $this->getOneValue(['unionid' => $wechatInfo['unionid']], 'uid');
                if ($uid) {
                    $userInfo = $userServices->getOne(['uid' => $uid]);
                } else {
                    $userInfo = $this->getAuthUserInfo($openid);
                }
            } else {
                $userInfo = $this->getAuthUserInfo($openid);
            }
        }
        $uid = (int)($userInfo['uid'] ?? 0);
        $wechatInfo['user_type'] = $userType;
        return $this->transaction(function () use ($openid, $uid, $userInfo, $wechatInfo, $userServices, $spread_uid, $userType) {
            $wechatInfo['uid'] = $uid;
            if ($userInfo) {
                if (isset($userInfo['status']) && !$userInfo['status'])
                    throw new ValidateException('您已被禁止登录，请联系管理员');
                //更新用户表
                /** @var LoginServices $loginService */
                $loginService = app()->make(LoginServices::class);
                $loginService->updateUserWechatLogin($uid, $wechatInfo, $userInfo);
                if ($spread_uid > 0) {
                    /** @var UserSpreadServices $spreadService */
                    $spreadService = app()->make(UserSpreadServices::class);
                    $spreadService->setSpread($userInfo['uid'], $spread_uid);
                }
            } else {
                //新增用户表
                $userInfo = $userServices->setUserInfo($wechatInfo, (int)$spread_uid, $userType);
                if (!$userInfo) {
                    throw new AuthException('生成用户失败!');
                }
            }
            $uid = $userInfo['uid'];
            $wechatInfo['uid'] = $userInfo->uid;
            $wechatInfo['add_time'] = $userInfo->add_time;

            //判断该类性用户在wechatUser中是否存在
            $wechatUser = [];
            if (!empty($wechatInfo['unionid'])) {
                $wechatUser = $this->getOne(['unionid' => $wechatInfo['unionid']]);
            }
            if (!$wechatUser) {
                $wechatUser = $this->getOne(['openid' => $openid]);
            }
            if ($wechatUser) {
                //更换微信登录情况
                if (!$this->update($wechatUser['id'], $wechatInfo, 'id')) {
                    throw new ValidateException('更新数据失败');
                }
            } else {
                if (!$this->save($wechatInfo)) {
                    throw new AuthException('生成微信用户失败!');
                }
            }
            return $userInfo;
        });
    }

    /**
     * 更新用户信息
     * @param $openid
     * @param string $phone
     * @return bool
     */
    public function updateWecahtUser($openid)
    {
        try {
            $userInfo = OfficialAccount::getUserInfo($openid);
        } catch (\Throwable $e) {
            $userInfo = [];
        }
        if (isset($userInfo['nickname']) && $userInfo['nickname']) {
            $userInfo['nickname'] = filter_emoji($userInfo['nickname']);
        } else {
            if (isset($userInfo['nickname'])) unset($userInfo['nickname']);
        }
        if (isset($userInfo['tagid_list'])) {
            $userInfo['tagid_list'] = implode(',', $userInfo['tagid_list']);
        }
        if ($userInfo && !$this->update($openid, $userInfo, 'openid'))
            throw new AdminException('更新失败');
        return true;
    }


    /**
     * 添加新用户
     * @param $openid
     * @param int $spread_uid
     * @param array $append 追加字段
     * @return \app\services\user\User|\think\Model
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public function setNewUser($openid, $spread_uid = 0, array $append = [])
    {

        try {
            $wechatInfo = OfficialAccount::getUserInfo($openid);
        } catch (\Throwable $e) {
            $wechatInfo = [];
        }
        if (!isset($wechatInfo['openid']))
            throw new ValidateException('请关注公众号!');

        if (isset($wechatInfo['nickname']) && $wechatInfo['nickname']) {
            $wechatInfo['is_complete'] = 1;
        } else {
            //昵称不存在的信息不完整
            $wechatInfo['is_complete'] = 0;
            mt_srand();
            $wechatInfo['nickname'] = 'wx' . rand(100000, 999999);
        }
        if (isset($wechatInfo['tagid_list'])) {
            $wechatInfo['tagid_list'] = implode(',', $wechatInfo['tagid_list']);
        }

        $uid = 0;
        $userType = 'wechat';
        $userInfo = [];
        /** @var UserServices $userServices */
        $userServices = app()->make(UserServices::class);
        if (isset($append['phone']) && $append['phone']) {
            $userInfo = $userServices->getOne(['phone' => $append['phone']]);
            $wechatInfo['phone'] = $append['phone'];
        }
        if (!$userInfo) {
            if (isset($wechatInfo['unionid']) && $wechatInfo['unionid']) {
                $uid = $this->getOneValue(['unionid' => $wechatInfo['unionid']], 'uid');
                if ($uid) {
                    $userInfo = $userServices->getOne(['uid' => $uid]);
                }
            } else {
                $userInfo = $this->getAuthUserInfo($openid);
            }
        }
        if ($userInfo) {
            $uid = (int)$userInfo['uid'];
            if (isset($userInfo['status']) && !$userInfo['status'])
                throw new ValidateException('您已被禁止登录，请联系管理员');
        }
        $wechatInfo['user_type'] = $userType;
        if ($userInfo) {
            //更新用户表和wechat_user表
            //判断该类性用户在wechatUser中是否存在
            $wechatUser = $this->getOne(['uid' => $uid, 'user_type' => $userType]);
            if ($wechatUser) {
                $wechatUser = $this->getOne(['openid' => $openid]);
            }
            /** @var LoginServices $loginService */
            $loginService = app()->make(LoginServices::class);
            $this->transaction(function () use ($openid, $loginService, $wechatInfo, $userInfo, $uid, $userType, $spread_uid, $wechatUser) {
                $wechatInfo['spread_uid'] = $spread_uid;
                $wechatInfo['uid'] = $uid;
                $loginService->updateUserWechatLogin($wechatInfo, $userInfo);
                if ($wechatUser) {
                    //更换微信登录情况
                    if (!$this->update($wechatUser['id'], $wechatInfo, 'id')) {
                        throw new ValidateException('更新数据失败');
                    }
                } else {
                    if (!$this->save($wechatInfo)) {
                        throw new ValidateException('写入信息失败');
                    }
                }
            });
        } else {
            //user表没有用户,wechat_user表没有用户创建新用户
            //不存在则创建用户
            $userInfo = $this->transaction(function () use ($openid, $userServices, $wechatInfo, $spread_uid, $userType) {
                $userInfo = $userServices->setUserInfo($wechatInfo, (int)$spread_uid, $userType);
                if (!$userInfo) {
                    throw new AuthException('生成User用户失败!');
                }
                $wechatInfo['uid'] = $userInfo->uid;
                $wechatInfo['add_time'] = $userInfo->add_time;
                $wechatUser = $this->getOne(['openid' => $openid]);
                if ($wechatUser) {
                    if (!$this->update($wechatUser['id'], $wechatInfo, 'id')) {
                        throw new ValidateException('更新数据失败');
                    }
                } else {
                    if (!$this->save($wechatInfo)) {
                        throw new AuthException('生成微信用户失败!');
                    }
                }
                return $userInfo;
            });
        }
        return $userInfo;
    }


    /**
     * 用户存在就更新 不存在就添加
     * @param $openid
     * @param int $spread_uid
     * @param string $phone
     * @return \app\services\user\User|array|\think\Model|null
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public function saveUser($openid, $spread_uid = 0, $phone = '')
    {
        /** @var UserServices $userServices */
        $userServices = app()->make(UserServices::class);
        $wechatUser = $this->getOne(['openid' => $openid]);
        if ($wechatUser && $wechatUser['uid']) {
            $userInfo = $userServices->getUserInfo((int)$wechatUser['uid']);
            $this->updateWecahtUser($openid);
        } else {
            $userInfo = $this->setNewUser($openid, $spread_uid, ['phone' => $phone]);
        }
        return $userInfo;
    }
}