184 lines
		
	
	
		
			5.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			184 lines
		
	
	
		
			5.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
 | 
						|
namespace Pterodactyl\Repositories\Wings;
 | 
						|
 | 
						|
use Webmozart\Assert\Assert;
 | 
						|
use Pterodactyl\Models\User;
 | 
						|
use Pterodactyl\Models\Server;
 | 
						|
use GuzzleHttp\Exception\TransferException;
 | 
						|
use Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException;
 | 
						|
 | 
						|
class DaemonServerRepository extends DaemonRepository
 | 
						|
{
 | 
						|
    /**
 | 
						|
     * Returns details about a server from the Daemon instance.
 | 
						|
     *
 | 
						|
     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
 | 
						|
     */
 | 
						|
    public function getDetails(): array
 | 
						|
    {
 | 
						|
        Assert::isInstanceOf($this->server, Server::class);
 | 
						|
 | 
						|
        try {
 | 
						|
            $response = $this->getHttpClient()->get(
 | 
						|
                sprintf('/api/servers/%s', $this->server->uuid)
 | 
						|
            );
 | 
						|
        } catch (TransferException $exception) {
 | 
						|
            throw new DaemonConnectionException($exception, false);
 | 
						|
        }
 | 
						|
 | 
						|
        return json_decode($response->getBody()->__toString(), true);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Creates a new server on the Wings daemon.
 | 
						|
     *
 | 
						|
     * @param array $data
 | 
						|
     *
 | 
						|
     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
 | 
						|
     */
 | 
						|
    public function create(array $data): void
 | 
						|
    {
 | 
						|
        Assert::isInstanceOf($this->server, Server::class);
 | 
						|
 | 
						|
        try {
 | 
						|
            $this->getHttpClient()->post(
 | 
						|
                '/api/servers', [
 | 
						|
                    'json' => $data,
 | 
						|
                ]
 | 
						|
            );
 | 
						|
        } catch (TransferException $exception) {
 | 
						|
            throw new DaemonConnectionException($exception);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Updates details about a server on the Daemon.
 | 
						|
     *
 | 
						|
     * @param array $data
 | 
						|
     *
 | 
						|
     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
 | 
						|
     */
 | 
						|
    public function update(array $data): void
 | 
						|
    {
 | 
						|
        Assert::isInstanceOf($this->server, Server::class);
 | 
						|
 | 
						|
        try {
 | 
						|
            $this->getHttpClient()->patch('/api/servers/' . $this->server->uuid, ['json' => $data]);
 | 
						|
        } catch (TransferException $exception) {
 | 
						|
            throw new DaemonConnectionException($exception);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Delete a server from the daemon, forcibly if passed.
 | 
						|
     *
 | 
						|
     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
 | 
						|
     */
 | 
						|
    public function delete(): void
 | 
						|
    {
 | 
						|
        Assert::isInstanceOf($this->server, Server::class);
 | 
						|
 | 
						|
        try {
 | 
						|
            $this->getHttpClient()->delete('/api/servers/' . $this->server->uuid);
 | 
						|
        } catch (TransferException $exception) {
 | 
						|
            throw new DaemonConnectionException($exception);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Reinstall a server on the daemon.
 | 
						|
     *
 | 
						|
     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
 | 
						|
     */
 | 
						|
    public function reinstall(): void
 | 
						|
    {
 | 
						|
        Assert::isInstanceOf($this->server, Server::class);
 | 
						|
 | 
						|
        try {
 | 
						|
            $this->getHttpClient()->post(sprintf(
 | 
						|
                '/api/servers/%s/reinstall', $this->server->uuid
 | 
						|
            ));
 | 
						|
        } catch (TransferException $exception) {
 | 
						|
            throw new DaemonConnectionException($exception);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * By default this function will suspend a server instance on the daemon. However, passing
 | 
						|
     * "true" as the first argument will unsuspend the server.
 | 
						|
     *
 | 
						|
     * @param bool $unsuspend
 | 
						|
     *
 | 
						|
     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
 | 
						|
     */
 | 
						|
    public function suspend(bool $unsuspend = false): void
 | 
						|
    {
 | 
						|
        Assert::isInstanceOf($this->server, Server::class);
 | 
						|
 | 
						|
        try {
 | 
						|
            $this->getHttpClient()->patch(
 | 
						|
                '/api/servers/' . $this->server->uuid,
 | 
						|
                ['json' => ['suspended' => ! $unsuspend]]
 | 
						|
            );
 | 
						|
        } catch (TransferException $exception) {
 | 
						|
            throw new DaemonConnectionException($exception);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Requests the daemon to create a full archive of the server. Once the daemon is finished
 | 
						|
     * they will send a POST request to "/api/remote/servers/{uuid}/archive" with a boolean.
 | 
						|
     *
 | 
						|
     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
 | 
						|
     */
 | 
						|
    public function requestArchive(): void
 | 
						|
    {
 | 
						|
        Assert::isInstanceOf($this->server, Server::class);
 | 
						|
 | 
						|
        try {
 | 
						|
            $this->getHttpClient()->post(sprintf(
 | 
						|
                '/api/servers/%s/archive', $this->server->uuid
 | 
						|
            ));
 | 
						|
        } catch (TransferException $exception) {
 | 
						|
            throw new DaemonConnectionException($exception);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Revokes a single user's JTI by using their ID. This is simply a helper function to
 | 
						|
     * make it easier to revoke tokens on the fly. This ensures that the JTI key is formatted
 | 
						|
     * correctly and avoids any costly mistakes in the codebase.
 | 
						|
     *
 | 
						|
     * @param int $id
 | 
						|
     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
 | 
						|
     */
 | 
						|
    public function revokeUserJTI(int $id): void
 | 
						|
    {
 | 
						|
        Assert::isInstanceOf($this->server, Server::class);
 | 
						|
 | 
						|
        $this->revokeJTIs([ md5($id . $this->server->uuid) ]);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Revokes an array of JWT JTI's by marking any token generated before the current time on
 | 
						|
     * the Wings instance as being invalid.
 | 
						|
     *
 | 
						|
     * @param array $jtis
 | 
						|
     * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
 | 
						|
     */
 | 
						|
    protected function revokeJTIs(array $jtis): void
 | 
						|
    {
 | 
						|
        Assert::isInstanceOf($this->server, Server::class);
 | 
						|
 | 
						|
        try {
 | 
						|
            $this->getHttpClient()
 | 
						|
                ->post(sprintf('/api/servers/%s/ws/deny', $this->server->uuid), [
 | 
						|
                    'json' => ['jtis' => $jtis],
 | 
						|
                ]);
 | 
						|
        } catch (TransferException $exception) {
 | 
						|
            throw new DaemonConnectionException($exception);
 | 
						|
        }
 | 
						|
    }
 | 
						|
}
 |