Merge branch 'develop' into feature/bulk-reinstall-command

app/Repositories/Daemon/FileRepository.php

@@ -3,6 +3,7 @@
 namespace Pterodactyl\Repositories\Daemon;
 
 use stdClass;
+use RuntimeException;
 use Psr\Http\Message\ResponseInterface;
 use Pterodactyl\Contracts\Repository\Daemon\FileRepositoryInterface;
 
@@ -84,33 +85,20 @@ class FileRepository extends BaseRepository implements FileRepositoryInterface
     {
         $response = $this->getHttpClient()->request('GET', sprintf('server/directory/%s', rawurlencode($path)));
 
-        $contents = json_decode($response->getBody());
-        $files = $folders = [];
+        return json_decode($response->getBody());
+    }
 
-        foreach ($contents as $value) {
-            if ($value->directory) {
-                array_push($folders, [
-                    'entry' => $value->name,
-                    'directory' => trim($path, '/'),
-                    'size' => null,
-                    'date' => strtotime($value->modified),
-                    'mime' => $value->mime,
-                ]);
-            } elseif ($value->file) {
-                array_push($files, [
-                    'entry' => $value->name,
-                    'directory' => trim($path, '/'),
-                    'extension' => str_replace('\\', '/', pathinfo($value->name, PATHINFO_EXTENSION)),
-                    'size' => human_readable($value->size),
-                    'date' => strtotime($value->modified),
-                    'mime' => $value->mime,
-                ]);
-            }
-        }
-
-        return [
-            'files' => $files,
-            'folders' => $folders,
-        ];
+    /**
+     * Creates a new directory for the server in the given $path.
+     *
+     * @param string $name
+     * @param string $path
+     * @return \Psr\Http\Message\ResponseInterface
+     *
+     * @throws \RuntimeException
+     */
+    public function createDirectory(string $name, string $path): ResponseInterface
+    {
+        throw new RuntimeException('Not implemented.');
     }
 }

app/Repositories/Eloquent/DatabaseRepository.php

@@ -76,7 +76,7 @@ class DatabaseRepository extends EloquentRepository implements DatabaseRepositor
      */
     public function getDatabasesForServer(int $server): Collection
     {
-        return $this->getBuilder()->where('server_id', $server)->get($this->getColumns());
+        return $this->getBuilder()->with('host')->where('server_id', $server)->get($this->getColumns());
     }
 
     /**

app/Repositories/Eloquent/NodeRepository.php

@@ -6,10 +6,8 @@ use Generator;
 use Pterodactyl\Models\Node;
 use Illuminate\Support\Collection;
 use Pterodactyl\Repositories\Concerns\Searchable;
-use Illuminate\Database\Eloquent\ModelNotFoundException;
 use Illuminate\Contracts\Pagination\LengthAwarePaginator;
 use Pterodactyl\Contracts\Repository\NodeRepositoryInterface;
-use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
 
 class NodeRepository extends EloquentRepository implements NodeRepositoryInterface
 {
@@ -140,25 +138,6 @@ class NodeRepository extends EloquentRepository implements NodeRepositoryInterfa
         return $node;
     }
 
-    /**
-     * Return a node with all of the servers attached to that node.
-     *
-     * @param int $id
-     * @return \Pterodactyl\Models\Node
-     *
-     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
-     */
-    public function getNodeServers(int $id): Node
-    {
-        try {
-            return $this->getBuilder()->with([
-                'servers.user', 'servers.nest', 'servers.egg',
-            ])->findOrFail($id, $this->getColumns());
-        } catch (ModelNotFoundException $exception) {
-            throw new RecordNotFoundException;
-        }
-    }
-
     /**
      * Return a collection of nodes for all locations to use in server creation UI.
      *

app/Repositories/Eloquent/ServerRepository.php

@@ -2,6 +2,7 @@
 
 namespace Pterodactyl\Repositories\Eloquent;
 
+use Pterodactyl\Models\Node;
 use Pterodactyl\Models\User;
 use Webmozart\Assert\Assert;
 use Pterodactyl\Models\Server;
@@ -358,4 +359,20 @@ class ServerRepository extends EloquentRepository implements ServerRepositoryInt
     {
         return $this->getBuilder()->where('suspended', true)->count();
     }
+
+    /**
+     * Returns all of the servers that exist for a given node in a paginated response.
+     *
+     * @param int $node
+     * @param int $limit
+     *
+     * @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
+     */
+    public function loadAllServersForNode(int $node, int $limit): LengthAwarePaginator
+    {
+        return $this->getBuilder()
+            ->with(['user', 'nest', 'egg'])
+            ->where('node_id', '=', $node)
+            ->paginate($limit);
+    }
 }

app/Repositories/Wings/BaseWingsRepository.php

@@ -0,0 +1,33 @@
+<?php
+
+namespace Pterodactyl\Repositories\Wings;
+
+use GuzzleHttp\Client;
+use Pterodactyl\Repositories\Daemon\BaseRepository;
+use Pterodactyl\Contracts\Repository\Daemon\BaseRepositoryInterface;
+
+abstract class BaseWingsRepository extends BaseRepository implements BaseRepositoryInterface
+{
+    /**
+     * Return an instance of the Guzzle HTTP Client to be used for requests.
+     *
+     * @param array $headers
+     * @return \GuzzleHttp\Client
+     */
+    public function getHttpClient(array $headers = []): Client
+    {
+        // We're just going to extend the parent client here since that logic is already quite
+        // sound and does everything we need it to aside from provide the correct base URL
+        // and authentication headers.
+        $client = parent::getHttpClient($headers);
+
+        return new Client(array_merge($client->getConfig(), [
+            'base_uri' => $this->getNode()->getConnectionAddress(),
+            'headers' => [
+                'Authorization' => 'Bearer ' . ($this->getToken() ?? $this->getNode()->daemonSecret),
+                'Accept' => 'application/json',
+                'Content-Type' => 'application/json',
+            ],
+        ]));
+    }
+}

app/Repositories/Wings/FileRepository.php

@@ -0,0 +1,173 @@
+<?php
+
+namespace Pterodactyl\Repositories\Wings;
+
+use stdClass;
+use Exception;
+use Psr\Http\Message\ResponseInterface;
+use Pterodactyl\Exceptions\Http\Server\FileSizeTooLargeException;
+use Pterodactyl\Contracts\Repository\Daemon\FileRepositoryInterface;
+
+class FileRepository extends BaseWingsRepository implements FileRepositoryInterface
+{
+    /**
+     * Return stat information for a given file.
+     *
+     * @param string $path
+     * @return \stdClass
+     *
+     * @throws \Exception
+     * @throws \GuzzleHttp\Exception\TransferException
+     */
+    public function getFileStat(string $path): stdClass
+    {
+        throw new Exception('Function not implemented.');
+    }
+
+    /**
+     * 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
+     */
+    public function getContent(string $path, int $notLargerThan = null): string
+    {
+        $response = $this->getHttpClient()->get(
+            sprintf('/api/servers/%s/files/contents', $this->getServer()->uuid),
+            [
+                'query' => ['file' => $path],
+            ]
+        );
+
+        $length = (int) $response->getHeader('Content-Length')[0] ?? 0;
+
+        if ($notLargerThan && $length > $notLargerThan) {
+            throw new FileSizeTooLargeException(
+                trans('server.files.exceptions.max_size')
+            );
+        }
+
+        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 \GuzzleHttp\Exception\TransferException
+     */
+    public function putContent(string $path, string $content): ResponseInterface
+    {
+        return $this->getHttpClient()->post(
+            sprintf('/api/servers/%s/files/write', $this->getServer()->uuid),
+            [
+                'query' => ['file' => $path],
+                'body' => $content,
+            ]
+        );
+    }
+
+    /**
+     * Return a directory listing for a given path.
+     *
+     * @param string $path
+     * @return array
+     *
+     * @throws \GuzzleHttp\Exception\TransferException
+     */
+    public function getDirectory(string $path): array
+    {
+        $response = $this->getHttpClient()->get(
+            sprintf('/api/servers/%s/files/list-directory', $this->getServer()->uuid),
+            [
+                'query' => ['directory' => $path],
+            ]
+        );
+
+        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
+     */
+    public function createDirectory(string $name, string $path): ResponseInterface
+    {
+        return $this->getHttpClient()->post(
+            sprintf('/api/servers/%s/files/create-directory', $this->getServer()->uuid),
+            [
+                'json' => [
+                    'name' => $name,
+                    'path' => $path,
+                ],
+            ]
+        );
+    }
+
+    /**
+     * Renames or moves a file on the remote machine.
+     *
+     * @param string $from
+     * @param string $to
+     * @return \Psr\Http\Message\ResponseInterface
+     */
+    public function renameFile(string $from, string $to): ResponseInterface
+    {
+        return $this->getHttpClient()->put(
+            sprintf('/api/servers/%s/files/rename', $this->getServer()->uuid),
+            [
+                'json' => [
+                    'rename_from' => $from,
+                    'rename_to' => $to,
+                ],
+            ]
+        );
+    }
+
+    /**
+     * Copy a given file and give it a unique name.
+     *
+     * @param string $location
+     * @return \Psr\Http\Message\ResponseInterface
+     */
+    public function copyFile(string $location): ResponseInterface
+    {
+        return $this->getHttpClient()->post(
+            sprintf('/api/servers/%s/files/copy', $this->getServer()->uuid),
+            [
+                'json' => [
+                    'location' => $location,
+                ],
+            ]
+        );
+    }
+
+    /**
+     * Delete a file or folder for the server.
+     *
+     * @param string $location
+     * @return \Psr\Http\Message\ResponseInterface
+     */
+    public function deleteFile(string $location): ResponseInterface
+    {
+        return $this->getHttpClient()->post(
+            sprintf('/api/servers/%s/files/delete', $this->getServer()->uuid),
+            [
+                'json' => [
+                    'location' => $location,
+                ],
+            ]
+        );
+    }
+}