guagua_php/app/model/User.php

<?php

namespace app\model;

use think\facade\Session;
use think\Model;
use think\model\concern\SoftDelete;

class User extends Model
{
    use SoftDelete;
    protected $deleteTime = 'delete_time';
    protected $defaultSoftDelete = 0;

    /**
     * 注册用户
     * @param $data
     * @return bool
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public function register($data)
    {

        $user = $this->where('phone', $data['phone'])->find();
        if ($user) return false;

        // 生成盐值
        $salt = $this->generateSalt();
        // 密码加盐值后哈希存储
        $password = $this->generateHashedPassword($data['password'], $salt);

        $this->save([
            // 随机头像
            'avatar'    => rand_avatar(),
            'password'  => $password,
            'salt'      => $salt,
            'phone'     => $data['phone'],
            'aid'       => $data['aid'],
            'register_time' => date("Y-m-d H:i:s",time())
        ]);

        return true;
    }

    /**
     * 用户登陆
     * @param $data
     * @return array
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public function login($data)
    {
        // 根据用户名查询用户信息
        $user = $this->where('phone', $data['phone'])->field('id,avatar,phone,password,salt,status')->find();

        if ($user) {
            if ($user['status'] != 1) return ['status' => false, 'msg' => '用户已被停用'];
            // 使用相同的盐值对输入密码进行哈希验证
            $hashedPassword = $this->generateHashedPassword($data['password'], $user->salt);

            if ($user->password === $hashedPassword) {

                # 缓存用户信息
                $login_user_data = $user->toArray();
                unset($login_user_data['password'],$login_user_data['salt']);
                Session::set('login_user_data',$login_user_data);
                // 登陆成功
                return ['status' => true, 'msg' => '登陆成功', 'data' => $login_user_data];
            } else {

                return ['status' => false, 'msg' => '密码错误'];
            }
        }

        return ['status' => false, 'msg' => '手机号未注册'];
    }

    /**
     * 找回密码
     * @param $data
     * @return array
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public function retrieve($data)
    {
        // 根据用户名查询用户信息
        $user = [];
        $errorMsg = '';
        if (isset($data['phone'])) {
            $errorMsg = '手机号';
            $user = $this->where('phone', $data['phone'])->find();
        } elseif ($data['user_id']) {
            $errorMsg = '用户';
            $user = $this->find($data['user_id']);
        }

        if ($user) {

            // 生成盐值
            $salt = $this->generateSalt();
            $password = $this->generateHashedPassword($data['password'], $salt);
            // 密码加盐值后哈希存储
            $user->password = $password;
            $user->salt = $salt;
            $user->update_time = date("Y-m-d H:i:s",time());
            $user->save();

            return ['status' => true, 'msg' => '密码重制成功'];
        }

        return ['status' => false, 'msg' => $errorMsg.'未注册'];
    }

    /**
     * 修改密码
     * @param $data
     * @return array
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public function modifyPassword($data,$user_id)
    {
        $user = ['id' => $user_id];

        if ($user) {

            $userModel = $this->find($user['id']);
            $password = $this->generateHashedPassword($data['password'], $userModel->salt);

            if ($userModel->password == $password) {
                return ['status' => false, 'msg' => '新密码与原密码一致'];
            }

            // 生成盐值
            $salt = $this->generateSalt();
            $new_password = $this->generateHashedPassword($data['password'], $salt);

            // 密码加盐值后哈希存储
            $userModel->password = $new_password;
            $userModel->salt = $salt;
            $userModel->update_time = date("Y-m-d H:i:s",time());

            $this->save();

            return ['status' => true, 'msg' => '修改成功'];
        }

        return ['status' => false, 'msg' => '登陆状态有误'];
    }

    /**
     * 验证邀请码是否有效
     * @param $invite_code
     * @return bool
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public function verifyInviteCode($invite_code)
    {
        $AgentUser = new AgentUser();
        $codeRes = $AgentUser->where('invite_code',$invite_code)->find();

        if (empty($codeRes)) {
            return 0;
        }

        return $codeRes->id;
    }

    /**
     * 扣减余额（消费）
     * @param $user_id
     * @param $balance
     * @return float|mixed
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public static function decrBalance($user_id,$balance)
    {
        $userModel = new User();
        $user = $userModel->find($user_id);
        $user->balance = round($user->balance - $balance,2);
        $user->save();
        return $user->balance;
    }

    /**
     * 增加余额（充值、上分）
     * @param $user_id
     * @param $balance
     * @return float|mixed
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public static function incrBalance($user_id,$balance)
    {
        $userModel = new User();
        $user = $userModel->find($user_id);
        $user->balance = round($user->balance + $balance,2);
        $user->save();
        return $user->balance;
    }

    /**
     * 扣除可提余额 （提现、下分）
     * @param $user_id
     * @param $balance
     * @return float|mixed
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public static function decrWithdrawalBalance($user_id,$balance)
    {
        $userModel = new User();
        $user = $userModel->find($user_id);
        $user->withdrawal_balance = round($user->withdrawal_balance - $balance,2);
        $user->save();
        return $user->withdrawal_balance;
    }

    /**
     * 增加可提余额（中奖）
     * @param $user_id
     * @param $balance
     * @return float|mixed
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public static function incrWithdrawalBalance($user_id,$balance)
    {
        $userModel = new User();
        $user = $userModel->find($user_id);
        $user->withdrawal_balance = round($user->withdrawal_balance + $balance,2);
        $user->save();
        return $user->withdrawal_balance;
    }

    /**
     * 检查用户余额是否足够
     * @param $user_id
     * @param $balance
     * @return array|int[]
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public static function inspectUserBalance($user_id,$balance)
    {
        $userModel = new User();
        $user = $userModel->find($user_id);
        if ($user->withdrawal_balance < $balance) {
            return ['status' => 0, 'msg' => '用户可提余额不足'];
        }
        return ['status' => 1];
    }

    /**
     * 生成盐值
     * @return string
     */
    private function generateSalt()
    {
        return generate_random_str(6);
    }

    /**
     * 密码加盐值后哈希存储
     * @param $password
     * @param $salt
     * @return string
     */
    private function generateHashedPassword($password, $salt)
    {
        return md5(md5($password) . md5($salt));
    }
}