repository = $repository; $this->backupManager = $backupManager; } /** * Returns the required presigned urls to upload a backup to S3 cloud storage. * * @return \Illuminate\Http\JsonResponse * * @throws \Exception * @throws \Throwable * @throws \Illuminate\Database\Eloquent\ModelNotFoundException */ public function __invoke(Request $request, string $backup) { // Get the size query parameter. $size = (int) $request->query('size'); if (empty($size)) { throw new BadRequestHttpException('A non-empty "size" query parameter must be provided.'); } /** @var \Pterodactyl\Models\Backup $backup */ $backup = Backup::query()->where('uuid', $backup)->firstOrFail(); // Prevent backups that have already been completed from trying to // be uploaded again. if (!is_null($backup->completed_at)) { throw new ConflictHttpException('This backup is already in a completed state.'); } // Ensure we are using the S3 adapter. $adapter = $this->backupManager->adapter(); if (!$adapter instanceof AwsS3Adapter) { throw new BadRequestHttpException('The configured backup adapter is not an S3 compatible adapter.'); } // The path where backup will be uploaded to $path = sprintf('%s/%s.tar.gz', $backup->server->uuid, $backup->uuid); // Get the S3 client $client = $adapter->getClient(); $expires = CarbonImmutable::now()->addMinutes(config('backups.presigned_url_lifespan', 60)); // Params for generating the presigned urls $params = [ 'Bucket' => $adapter->getBucket(), 'Key' => $path, 'ContentType' => 'application/x-gzip', ]; // Execute the CreateMultipartUpload request $result = $client->execute($client->getCommand('CreateMultipartUpload', $params)); // Get the UploadId from the CreateMultipartUpload request, this is needed to create // the other presigned urls. $params['UploadId'] = $result->get('UploadId'); // Retrieve configured part size $maxPartSize = $this->getConfiguredMaxPartSize(); // Create as many UploadPart presigned urls as needed $parts = []; for ($i = 0; $i < ($size / $maxPartSize); ++$i) { $parts[] = $client->createPresignedRequest( $client->getCommand('UploadPart', array_merge($params, ['PartNumber' => $i + 1])), $expires )->getUri()->__toString(); } // Set the upload_id on the backup in the database. $backup->update(['upload_id' => $params['UploadId']]); return new JsonResponse([ 'parts' => $parts, 'part_size' => $maxPartSize, ]); } /** * Get the configured maximum size of a single part in the multipart uplaod. * * The function tries to retrieve a configured value from the configuration. * If no value is specified, a fallback value will be used. * * Note if the received config cannot be converted to int (0), is zero or is negative, * the fallback value will be used too. * * The fallback value is {@see BackupRemoteUploadController::DEFAULT_MAX_PART_SIZE}. * * @return int */ private function getConfiguredMaxPartSize() { $maxPartSize = (int) config('backups.max_part_size', self::DEFAULT_MAX_PART_SIZE); if ($maxPartSize <= 0) { $maxPartSize = self::DEFAULT_MAX_PART_SIZE; } return $maxPartSize; } }