83 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			83 lines
		
	
	
		
			2.5 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.
 | |
|      */
 | |
|     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.
 | |
|      *
 | |
|      * @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),
 | |
|         ]);
 | |
|     }
 | |
| }
 | 
