<?php namespace Pterodactyl\Repositories\Wings; use Webmozart\Assert\Assert; use Pterodactyl\Models\Server; use Psr\Http\Message\ResponseInterface; use GuzzleHttp\Exception\TransferException; use Pterodactyl\Exceptions\Http\Server\FileSizeTooLargeException; use Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException; class DaemonFileRepository extends DaemonRepository { /** * Return the contents of a given file. * * @param string $path * @param int|null $notLargerThan the maximum content length in bytes * @return string * * @throws \GuzzleHttp\Exception\TransferException * @throws \Pterodactyl\Exceptions\Http\Server\FileSizeTooLargeException * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException */ public function getContent(string $path, int $notLargerThan = null): string { Assert::isInstanceOf($this->server, Server::class); try { $response = $this->getHttpClient()->get( sprintf('/api/servers/%s/files/contents', $this->server->uuid), [ 'query' => ['file' => $path], ] ); } catch (TransferException $exception) { throw new DaemonConnectionException($exception); } $length = (int) $response->getHeader('Content-Length')[0] ?? 0; if ($notLargerThan && $length > $notLargerThan) { throw new FileSizeTooLargeException; } return $response->getBody()->__toString(); } /** * Save new contents to a given file. This works for both creating and updating * a file. * * @param string $path * @param string $content * @return \Psr\Http\Message\ResponseInterface * * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException */ public function putContent(string $path, string $content): ResponseInterface { Assert::isInstanceOf($this->server, Server::class); try { return $this->getHttpClient()->post( sprintf('/api/servers/%s/files/write', $this->server->uuid), [ 'query' => ['file' => $path], 'body' => $content, ] ); } catch (TransferException $exception) { throw new DaemonConnectionException($exception); } } /** * Return a directory listing for a given path. * * @param string $path * @return array * * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException */ public function getDirectory(string $path): array { Assert::isInstanceOf($this->server, Server::class); try { $response = $this->getHttpClient()->get( sprintf('/api/servers/%s/files/list-directory', $this->server->uuid), [ 'query' => ['directory' => $path], ] ); } catch (TransferException $exception) { throw new DaemonConnectionException($exception); } return json_decode($response->getBody(), true); } /** * Creates a new directory for the server in the given $path. * * @param string $name * @param string $path * @return \Psr\Http\Message\ResponseInterface * * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException */ public function createDirectory(string $name, string $path): ResponseInterface { Assert::isInstanceOf($this->server, Server::class); try { return $this->getHttpClient()->post( sprintf('/api/servers/%s/files/create-directory', $this->server->uuid), [ 'json' => [ 'name' => $name, 'path' => $path, ], ] ); } catch (TransferException $exception) { throw new DaemonConnectionException($exception); } } /** * Renames or moves a file on the remote machine. * * @param string|null $root * @param array $files * @return \Psr\Http\Message\ResponseInterface * * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException */ public function renameFiles(?string $root, array $files): ResponseInterface { Assert::isInstanceOf($this->server, Server::class); try { return $this->getHttpClient()->put( sprintf('/api/servers/%s/files/rename', $this->server->uuid), [ 'json' => [ 'root' => $root ?? '/', 'files' => $files, ], ] ); } catch (TransferException $exception) { throw new DaemonConnectionException($exception); } } /** * Copy a given file and give it a unique name. * * @param string $location * @return \Psr\Http\Message\ResponseInterface * * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException */ public function copyFile(string $location): ResponseInterface { Assert::isInstanceOf($this->server, Server::class); try { return $this->getHttpClient()->post( sprintf('/api/servers/%s/files/copy', $this->server->uuid), [ 'json' => [ 'location' => $location, ], ] ); } catch (TransferException $exception) { throw new DaemonConnectionException($exception); } } /** * Delete a file or folder for the server. * * @param string|null $root * @param array $files * @return \Psr\Http\Message\ResponseInterface * * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException */ public function deleteFiles(?string $root, array $files): ResponseInterface { Assert::isInstanceOf($this->server, Server::class); try { return $this->getHttpClient()->post( sprintf('/api/servers/%s/files/delete', $this->server->uuid), [ 'json' => [ 'root' => $root ?? '/', 'files' => $files, ], ] ); } catch (TransferException $exception) { throw new DaemonConnectionException($exception); } } /** * Compress the given files or folders in the given root. * * @param string|null $root * @param array $files * @return array * * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException */ public function compressFiles(?string $root, array $files): array { Assert::isInstanceOf($this->server, Server::class); try { $response = $this->getHttpClient()->post( sprintf('/api/servers/%s/files/compress', $this->server->uuid), [ 'json' => [ 'root' => $root ?? '/', 'files' => $files, ], // Wait for up to 15 minutes for the archive to be completed when calling this endpoint // since it will likely take quite awhile for large directories. 'timeout' => 60 * 15, ] ); } catch (TransferException $exception) { throw new DaemonConnectionException($exception); } return json_decode($response->getBody(), true); } /** * Decompresses a given archive file. * * @param string|null $root * @param string $file * @return \Psr\Http\Message\ResponseInterface * * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException */ public function decompressFile(?string $root, string $file): ResponseInterface { Assert::isInstanceOf($this->server, Server::class); try { return $this->getHttpClient()->post( sprintf('/api/servers/%s/files/decompress', $this->server->uuid), [ 'json' => [ 'root' => $root ?? '/', 'file' => $file, ], ] ); } catch (TransferException $exception) { throw new DaemonConnectionException($exception); } } }