2FAuth/app/TwoFAccount.php

137 lines
2.4 KiB
PHP
Raw Normal View History

2019-05-20 07:37:41 +02:00
<?php
namespace App;
2020-01-24 12:56:38 +01:00
use OTPHP\HOTP;
use OTPHP\Factory;
2019-05-20 07:37:41 +02:00
use Illuminate\Database\Eloquent\Model;
2020-01-08 17:03:41 +01:00
use Illuminate\Support\Facades\Storage;
use Illuminate\Contracts\Encryption\DecryptException;
2019-05-20 07:37:41 +02:00
class TwoFAccount extends Model
2019-05-20 07:37:41 +02:00
{
2020-01-10 13:43:36 +01:00
/**
* model's array form.
*
* @var array
*/
protected $fillable = ['service', 'account', 'uri', 'icon'];
/**
* The table associated with the model.
*
* @var string
*/
protected $table = 'twofaccounts';
2020-01-08 17:03:41 +01:00
2020-01-24 12:56:38 +01:00
/**
* The accessors to append to the model's array form.
*
* @var array
*/
protected $appends = ['type', 'counter'];
2020-01-08 17:03:41 +01:00
/**
* Null empty icon resource has gone
*
* @param string $value
* @return string
*/
public function getIconAttribute($value)
{
2020-01-10 13:43:36 +01:00
if (\App::environment('testing') == false) {
if( !Storage::exists('public/icons/' . pathinfo($value)['basename']) ) {
2020-01-08 17:03:41 +01:00
2020-01-10 13:43:36 +01:00
return '';
}
2020-01-08 17:03:41 +01:00
}
return $value;
}
/**
* Prevent setting a missing icon
*
* @param string $value
* @return string
*/
public function setIconAttribute($value)
{
if( !Storage::exists('public/icons/' . pathinfo($value)['basename']) ) {
$this->attributes['icon'] = '';
}
else {
$this->attributes['icon'] = $value;
}
}
2020-01-24 12:56:38 +01:00
/**
* Get the account type.
*
* @return string
*/
public function getTypeAttribute()
{
return substr( $this->uri, 0, 15 ) === "otpauth://totp/" ? 'totp' : 'hotp';
}
/**
* Get the account counter in case of HOTP.
*
* @return integer
*/
public function getCounterAttribute()
{
if( $this->type === 'hotp' ) {
$otp = Factory::loadFromProvisioningUri($this->uri);
return $otp->getCounter();
}
return null;
}
/**
* Set the user's first name.
*
* @param string $value
* @return void
*/
// public function setUriAttribute($value)
// {
// $this->attributes['uri'] = encrypt($value);
// }
/**
* Get the user's first name.
*
* @param string $value
* @return string
*/
// public function getUriAttribute($value)
// {
// try {
// return decrypt($value);
// } catch (DecryptException $e) {
// return null;
// }
// }
2019-05-20 07:37:41 +02:00
}