112 lines
2.6 KiB
PHP
112 lines
2.6 KiB
PHP
<?php
|
|
namespace Da\User\Model;
|
|
|
|
use Da\User\Helper\SecurityHelper;
|
|
use Da\User\Query\TokenQuery;
|
|
use Da\User\Traits\ContainerTrait;
|
|
use Da\User\Traits\ModuleTrait;
|
|
use yii\db\ActiveRecord;
|
|
use yii\helpers\Url;
|
|
use RuntimeException;
|
|
use Yii;
|
|
|
|
/**
|
|
* Token Active Record model.
|
|
*
|
|
* @property integer $user_id
|
|
* @property string $code
|
|
* @property integer $type
|
|
* @property string $url
|
|
* @property bool $isExpired
|
|
* @property integer $created_at
|
|
*
|
|
* @property User $user
|
|
*/
|
|
class Token extends ActiveRecord
|
|
{
|
|
use ModuleTrait;
|
|
use ContainerTrait;
|
|
|
|
const TYPE_CONFIRMATION = 0;
|
|
const TYPE_RECOVERY = 1;
|
|
const TYPE_CONFIRM_NEW_EMAIL = 2;
|
|
const TYPE_CONFIRM_OLD_EMAIL = 3;
|
|
|
|
protected $routes = [
|
|
self::TYPE_CONFIRMATION => '/user/registration/confirm',
|
|
self::TYPE_RECOVERY => '/usr/recovery/reset',
|
|
self::TYPE_CONFIRM_NEW_EMAIL => '/user/settings/confirm',
|
|
self::TYPE_CONFIRM_OLD_EMAIL => '/usr/settings/confirm'
|
|
];
|
|
|
|
/**
|
|
* @inheritdoc
|
|
*/
|
|
public function beforeSave($insert)
|
|
{
|
|
if ($insert) {
|
|
$this->setAttribute('code', $this->make(SecurityHelper::class)->generateRandomString());
|
|
static::deleteAll(['user_id' => $this->user_id, 'type' => $this->type]);
|
|
$this->setAttribute('created_at', time());
|
|
}
|
|
|
|
return parent::beforeSave($insert);
|
|
}
|
|
|
|
/**
|
|
* @inheritdoc
|
|
*/
|
|
public static function tableName()
|
|
{
|
|
return '{{%token}}';
|
|
}
|
|
|
|
/**
|
|
* @inheritdoc
|
|
*/
|
|
public static function primaryKey()
|
|
{
|
|
return ['user_id', 'code', 'type'];
|
|
}
|
|
|
|
/**
|
|
* @return \yii\db\ActiveQuery
|
|
*/
|
|
public function getUser()
|
|
{
|
|
return $this->hasOne($this->getClassMap()->get('User'), ['id' => 'user_id']);
|
|
}
|
|
|
|
/**
|
|
* @return string
|
|
*/
|
|
public function getUrl()
|
|
{
|
|
return Url::to([$this->routes[$this->type], 'id' => $this->user_id, 'code' => $this->code], true);
|
|
}
|
|
|
|
/**
|
|
* @return bool Whether token has expired.
|
|
*/
|
|
public function getIsExpired()
|
|
{
|
|
if ($this->type == static::TYPE_RECOVERY) {
|
|
$expirationTime = $this->getModule()->tokenRecoveryLifespan;
|
|
} elseif ($this->type >= static::TYPE_CONFIRMATION && $this->type <= static::TYPE_CONFIRM_OLD_EMAIL) {
|
|
$expirationTime = $this->getModule()->tokenConfirmationLifespan;
|
|
} else {
|
|
throw new RuntimeException();
|
|
}
|
|
|
|
return ($this->created_at + $expirationTime) < time();
|
|
}
|
|
|
|
/**
|
|
* @return TokenQuery
|
|
*/
|
|
public static function find()
|
|
{
|
|
return new TokenQuery(static::class);
|
|
}
|
|
}
|