88 lines
2.8 KiB
PHP
88 lines
2.8 KiB
PHP
<?php
|
|
|
|
namespace Pterodactyl\Http\Controllers\Admin;
|
|
|
|
use Illuminate\Http\Request;
|
|
use Pterodactyl\Models\Node;
|
|
use Pterodactyl\Models\ApiKey;
|
|
use Illuminate\Http\JsonResponse;
|
|
use Pterodactyl\Http\Controllers\Controller;
|
|
use Illuminate\Contracts\Encryption\Encrypter;
|
|
use Pterodactyl\Services\Api\KeyCreationService;
|
|
use Pterodactyl\Repositories\Eloquent\ApiKeyRepository;
|
|
|
|
class NodeAutoDeployController extends Controller
|
|
{
|
|
/**
|
|
* @var \Pterodactyl\Services\Api\KeyCreationService
|
|
*/
|
|
private $keyCreationService;
|
|
|
|
/**
|
|
* @var \Pterodactyl\Repositories\Eloquent\ApiKeyRepository
|
|
*/
|
|
private $repository;
|
|
|
|
/**
|
|
* @var \Illuminate\Contracts\Encryption\Encrypter
|
|
*/
|
|
private $encrypter;
|
|
|
|
/**
|
|
* NodeAutoDeployController constructor.
|
|
*
|
|
* @param \Pterodactyl\Repositories\Eloquent\ApiKeyRepository $repository
|
|
* @param \Illuminate\Contracts\Encryption\Encrypter $encrypter
|
|
* @param \Pterodactyl\Services\Api\KeyCreationService $keyCreationService
|
|
*/
|
|
public function __construct(
|
|
ApiKeyRepository $repository,
|
|
Encrypter $encrypter,
|
|
KeyCreationService $keyCreationService
|
|
) {
|
|
$this->keyCreationService = $keyCreationService;
|
|
$this->repository = $repository;
|
|
$this->encrypter = $encrypter;
|
|
}
|
|
|
|
/**
|
|
* Generates a new API key for the logged in user with only permission to read
|
|
* nodes, and returns that as the deployment key for a node.
|
|
*
|
|
* @param \Illuminate\Http\Request $request
|
|
* @param \Pterodactyl\Models\Node $node
|
|
* @return \Illuminate\Http\JsonResponse
|
|
*
|
|
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
|
|
*/
|
|
public function __invoke(Request $request, Node $node)
|
|
{
|
|
/** @var \Pterodactyl\Models\ApiKey|null $key */
|
|
$key = $this->repository->getApplicationKeys($request->user())
|
|
->filter(function (ApiKey $key) {
|
|
foreach ($key->getAttributes() as $permission => $value) {
|
|
if ($permission === 'r_nodes' && $value === 1) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
})
|
|
->first();
|
|
|
|
// We couldn't find a key that exists for this user with only permission for
|
|
// reading nodes. Go ahead and create it now.
|
|
if (! $key) {
|
|
$key = $this->keyCreationService->setKeyType(ApiKey::TYPE_APPLICATION)->handle([
|
|
'user_id' => $request->user()->id,
|
|
'memo' => 'Automatically generated node deployment key.',
|
|
'allowed_ips' => [],
|
|
], ['r_nodes' => 1]);
|
|
}
|
|
|
|
return JsonResponse::create([
|
|
'node' => $node->id,
|
|
'token' => $key->identifier . $this->encrypter->decrypt($key->token),
|
|
]);
|
|
}
|
|
}
|