catchAdmin/extend/catcher/CatchAuth.php
2020-09-09 10:40:44 +08:00

311 lines
6.4 KiB
PHP

<?php
namespace catcher;
use catchAdmin\permissions\model\Users;
use catcher\exceptions\FailedException;
use catcher\exceptions\LoginFailedException;
use thans\jwt\facade\JWTAuth;
use think\facade\Session;
use think\helper\Str;
class CatchAuth
{
protected $auth;
protected $guard;
// 默认获取
protected $username = 'email';
// 校验字段
protected $password = 'password';
// 保存用户信息
protected $user = [];
public function __construct()
{
$this->auth = config('catch.auth');
$this->guard = $this->auth['default']['guard'];
}
/**
* set guard
*
* @time 2020年01月07日
* @param $guard
* @return $this
*/
public function guard($guard)
{
$this->guard = $guard;
return $this;
}
/**
*
* @time 2020年01月07日
* @param $condition
* @return mixed
*/
public function attempt($condition)
{
try {
$user = $this->authenticate($condition);
if (!$user) {
throw new LoginFailedException();
}
if ($user->status == Users::DISABLE) {
throw new LoginFailedException('该用户已被禁用|' . $user->username, Code::USER_FORBIDDEN);
}
if (!password_verify($condition['password'], $user->password)) {
throw new LoginFailedException('登录失败|' . $user->username);
}
$token = $this->{$this->getDriver()}($user);
$this->afterLoginSuccess($user);
// 登录事件
$this->loginEvent($user->username);
return $token;
} catch (\Exception $exception) {
$message = $exception->getMessage();
if (strpos($message, '|') !== false) {
$username = explode('|', $message)[1];
} else {
$username = $condition['email'];
}
$this->loginEvent($username, false);
throw new LoginFailedException('登录失败', $exception->getCode());
}
}
/**
* 用户登录成功后
*
* @time 2020年09月09日
* @param $user
* @return void
*/
protected function afterLoginSuccess($user)
{
$user->last_login_ip = request()->ip();
$user->last_login_time = time();
$user->save();
}
/**
* 登录事件
*
* @time 2020年09月09日
* @param $name
* @param bool $success
* @return void
*/
protected function loginEvent($name, $success = true)
{
$params['login_name'] = $name;
$params['success'] = $success ? 1 : 2;
event('loginLog', $params);
}
/**
* user
*
* @time 2020年09月09日
* @return mixed
*/
public function user()
{
$user = $this->user[$this->guard] ?? null;
if (!$user) {
switch ($this->getDriver()) {
case 'jwt':
$model = app($this->getProvider()['model']);
$user = $model->where($model->getPk(), JWTAuth::auth()[$this->jwtKey()])->find();
break;
case 'session':
$user = Session::get($this->sessionUserKey(), null);
break;
default:
throw new FailedException('user not found');
}
$this->user[$this->guard] = $user;
return $user;
}
return $user;
}
/**
*
* @time 2020年01月07日
* @return mixed
*/
public function logout()
{
switch ($this->getDriver()) {
case 'jwt':
return true;
case 'session':
Session::delete($this->sessionUserKey());
return true;
default:
throw new FailedException('user not found');
}
}
/**
*
* @time 2020年01月07日
* @param $user
* @return string
*/
protected function jwt($user)
{
$token = JWTAuth::builder([$this->jwtKey() => $user->id]);
JWTAuth::setToken($token);
return $token;
}
/**
*
* @time 2020年01月07日
* @param $user
* @return void
*/
protected function session($user)
{
Session::set($this->sessionUserKey(), $user);
}
/**
*
* @time 2020年01月07日
* @return string
*/
protected function sessionUserKey()
{
return $this->guard . '_user';
}
/**
*
* @time 2020年01月07日
* @return string
*/
protected function jwtKey()
{
return $this->guard . '_id';
}
/**
*
* @time 2020年01月07日
* @return mixed
*/
protected function getDriver()
{
return $this->auth['guards'][$this->guard]['driver'];
}
/**
*
* @time 2020年01月07日
* @return mixed
*/
protected function getProvider()
{
if (!isset($this->auth['guards'][$this->guard])) {
throw new FailedException('Auth Guard Not Found');
}
return $this->auth['providers'][$this->auth['guards'][$this->guard]['provider']];
}
/**
*
* @time 2020年01月07日
* @param $condition
* @return mixed
*/
protected function authenticate($condition)
{
$provider = $this->getProvider();
return $this->{$provider['driver']}($condition);
}
/**
*
* @time 2020年01月07日
* @param $condition
* @return void
*/
protected function database($condition): void
{}
/**
*
* @time 2020年01月07日
* @param $condition
* @return mixed
*/
protected function orm($condition)
{
return app($this->getProvider()['model'])->where($this->filter($condition))->find();
}
/**
*
* @time 2020年01月07日
* @param $condition
* @return array
*/
protected function filter($condition): array
{
$where = [];
foreach ($condition as $field => $value) {
if ($field != $this->password) {
$where[$field] = $value;
}
}
return $where;
}
/**
*
* @time 2020年01月07日
* @param $field
* @return $this
*/
public function username($field): self
{
$this->username = $field;
return $this;
}
/**
*
* @time 2020年01月07日
* @param $field
* @return $this
*/
public function password($field): self
{
$this->password = $field;
return $this;
}
}