mirror of
https://github.com/BookStackApp/BookStack.git
synced 2024-11-28 20:00:47 +08:00
eab0ca9648
Closes #316
134 lines
3.3 KiB
PHP
134 lines
3.3 KiB
PHP
<?php namespace BookStack\Auth\Access;
|
|
|
|
use BookStack\Auth\User;
|
|
use BookStack\Exceptions\UserTokenExpiredException;
|
|
use BookStack\Exceptions\UserTokenNotFoundException;
|
|
use Carbon\Carbon;
|
|
use Illuminate\Database\Connection as Database;
|
|
use stdClass;
|
|
|
|
class UserTokenService
|
|
{
|
|
|
|
/**
|
|
* Name of table where user tokens are stored.
|
|
* @var string
|
|
*/
|
|
protected $tokenTable = 'user_tokens';
|
|
|
|
/**
|
|
* Token expiry time in hours.
|
|
* @var int
|
|
*/
|
|
protected $expiryTime = 24;
|
|
|
|
protected $db;
|
|
|
|
/**
|
|
* UserTokenService constructor.
|
|
* @param Database $db
|
|
*/
|
|
public function __construct(Database $db)
|
|
{
|
|
$this->db = $db;
|
|
}
|
|
|
|
/**
|
|
* Delete all email confirmations that belong to a user.
|
|
* @param User $user
|
|
* @return mixed
|
|
*/
|
|
public function deleteByUser(User $user)
|
|
{
|
|
return $this->db->table($this->tokenTable)
|
|
->where('user_id', '=', $user->id)
|
|
->delete();
|
|
}
|
|
|
|
/**
|
|
* Get the user id from a token, while check the token exists and has not expired.
|
|
* @param string $token
|
|
* @return int
|
|
* @throws UserTokenNotFoundException
|
|
* @throws UserTokenExpiredException
|
|
*/
|
|
public function checkTokenAndGetUserId(string $token) : int
|
|
{
|
|
$entry = $this->getEntryByToken($token);
|
|
|
|
if (is_null($entry)) {
|
|
throw new UserTokenNotFoundException('Token "' . $token . '" not found');
|
|
}
|
|
|
|
if ($this->entryExpired($entry)) {
|
|
throw new UserTokenExpiredException("Token of id {$entry->id} has expired.", $entry->user_id);
|
|
}
|
|
|
|
return $entry->user_id;
|
|
}
|
|
|
|
/**
|
|
* Creates a unique token within the email confirmation database.
|
|
* @return string
|
|
*/
|
|
protected function generateToken() : string
|
|
{
|
|
$token = str_random(24);
|
|
while ($this->tokenExists($token)) {
|
|
$token = str_random(25);
|
|
}
|
|
return $token;
|
|
}
|
|
|
|
/**
|
|
* Generate and store a token for the given user.
|
|
* @param User $user
|
|
* @return string
|
|
*/
|
|
protected function createTokenForUser(User $user) : string
|
|
{
|
|
$token = $this->generateToken();
|
|
$this->db->table($this->tokenTable)->insert([
|
|
'user_id' => $user->id,
|
|
'token' => $token,
|
|
'created_at' => Carbon::now(),
|
|
'updated_at' => Carbon::now()
|
|
]);
|
|
return $token;
|
|
}
|
|
|
|
/**
|
|
* Check if the given token exists.
|
|
* @param string $token
|
|
* @return bool
|
|
*/
|
|
protected function tokenExists(string $token) : bool
|
|
{
|
|
return $this->db->table($this->tokenTable)
|
|
->where('token', '=', $token)->exists();
|
|
}
|
|
|
|
/**
|
|
* Get a token entry for the given token.
|
|
* @param string $token
|
|
* @return object|null
|
|
*/
|
|
protected function getEntryByToken(string $token)
|
|
{
|
|
return $this->db->table($this->tokenTable)
|
|
->where('token', '=', $token)
|
|
->first();
|
|
}
|
|
|
|
/**
|
|
* Check if the given token entry has expired.
|
|
* @param stdClass $tokenEntry
|
|
* @return bool
|
|
*/
|
|
protected function entryExpired(stdClass $tokenEntry) : bool
|
|
{
|
|
return Carbon::now()->subHours($this->expiryTime)
|
|
->gt(new Carbon($tokenEntry->created_at));
|
|
}
|
|
|
|
} |