84 lines
2.6 KiB
PHP
84 lines
2.6 KiB
PHP
<?php
|
|
|
|
namespace Pterodactyl\Repositories\SecurityKeys;
|
|
|
|
use Ramsey\Uuid\Uuid;
|
|
use Pterodactyl\Models\User;
|
|
use Illuminate\Container\Container;
|
|
use Webauthn\PublicKeyCredentialSource;
|
|
use Webauthn\PublicKeyCredentialUserEntity;
|
|
use Pterodactyl\Models\SecurityKey;
|
|
use Webauthn\PublicKeyCredentialSourceRepository as PublicKeyRepositoryInterface;
|
|
|
|
class PublicKeyCredentialSourceRepository implements PublicKeyRepositoryInterface
|
|
{
|
|
protected User $user;
|
|
|
|
public function __construct(User $user)
|
|
{
|
|
$this->user = $user;
|
|
}
|
|
|
|
/**
|
|
* Find a single hardware security token for a user by uzing the credential ID.
|
|
*/
|
|
public function findOneByCredentialId(string $id): ?PublicKeyCredentialSource
|
|
{
|
|
/** @var \Pterodactyl\Models\SecurityKey $key */
|
|
$key = $this->user->securityKeys()
|
|
->where('public_key_id', $id)
|
|
->first();
|
|
|
|
return $key ? $key->public_key_credential_source : null;
|
|
}
|
|
|
|
/**
|
|
* Find all of the hardware tokens that exist for the user using the given
|
|
* entity handle.
|
|
*/
|
|
public function findAllForUserEntity(PublicKeyCredentialUserEntity $entity): array
|
|
{
|
|
$results = $this->user->securityKeys()
|
|
->where('user_handle', $entity->getId())
|
|
->get();
|
|
|
|
return $results->map(function (SecurityKey $key) {
|
|
return $key->public_key_credential_source;
|
|
})->values()->toArray();
|
|
}
|
|
|
|
/**
|
|
* Save a credential to the database and link it with the user.
|
|
*
|
|
* @throws \Throwable
|
|
*/
|
|
public function saveCredentialSource(PublicKeyCredentialSource $source): void
|
|
{
|
|
$key = $this->user->securityKeys()->make();
|
|
|
|
$key->forceFill([
|
|
'uuid' => Uuid::uuid4(),
|
|
'user_id' => $this->user->id,
|
|
'public_key_id' => $source->getPublicKeyCredentialId(),
|
|
'public_key' => $source->getCredentialPublicKey(),
|
|
'aaguid' => $source->getAaguid(),
|
|
'type' => $source->getType(),
|
|
'transports' => $source->getTransports(),
|
|
'attestation_type' => $source->getAttestationType(),
|
|
'trust_path' => $source->getTrustPath(),
|
|
'user_handle' => $source->getUserHandle(),
|
|
'counter' => $source->getCounter(),
|
|
'other_ui' => $source->getOtherUI(),
|
|
]);
|
|
|
|
$key->saveOrFail();
|
|
}
|
|
|
|
/**
|
|
* Returns a new instance of the repository with the provided user attached.
|
|
*/
|
|
public static function factory(User $user): self
|
|
{
|
|
return Container::getInstance()->make(static::class, ['user' => $user]);
|
|
}
|
|
}
|