Merge branch 'develop' into feature/api-v1

# Conflicts:
#	app/Contracts/Repository/RepositoryInterface.php
#	app/Repositories/Eloquent/EloquentRepository.php
#	app/Services/Nodes/NodeUpdateService.php
#	tests/Unit/Services/Nodes/NodeUpdateServiceTest.php
This commit is contained in:
Dane Everitt 2018-01-10 20:55:22 -06:00
commit 800e2df6b2
No known key found for this signature in database
GPG key ID: EEA66103B3D71F53
187 changed files with 1878 additions and 3143 deletions

View file

@ -11,10 +11,18 @@ This project follows [Semantic Versioning](http://semver.org) guidelines.
* `[beta.3]` — Fixes error handling of the settings service provider when no migrations have been run. * `[beta.3]` — Fixes error handling of the settings service provider when no migrations have been run.
* `[beta.3]` — Fixes validation error when trying to use 'None' as the 'Copy Script From' option for an egg script. * `[beta.3]` — Fixes validation error when trying to use 'None' as the 'Copy Script From' option for an egg script.
* Fixes a design bug in the database that prevented the storage of negative numbers, thus preventing a server from being assigned unlimited swap. * Fixes a design bug in the database that prevented the storage of negative numbers, thus preventing a server from being assigned unlimited swap.
* Fixes a bug where the 'Assign New Allocations' box would only show IPs that were present in the current pagination block.
### Added ### Added
* Nest and Egg listings now show the associated ID in order to make API requests easier. * Nest and Egg listings now show the associated ID in order to make API requests easier.
### Changed
* Changed behavior of allocation IP Address/Ports box to automatically store the value entered if a user unfocuses the field without hitting space.
* Changed order in which allocations are displayed to prioritize those with servers attached (in ascending IP & port order) followed by ascending IP & port order where no server is attached.
### Removed
* OOM exceptions can no longer be disabled on servers due to a startling number of users that were using it to avoid allocating proper amounts of resources to servers.
## v0.7.0-beta.3 (Derelict Dermodactylus) ## v0.7.0-beta.3 (Derelict Dermodactylus)
### Fixed ### Fixed
* `[beta.2]` — Fixes a bug that would cause an endless exception message stream in the console when attemping to setup environment settings in certain instances. * `[beta.2]` — Fixes a bug that would cause an endless exception message stream in the console when attemping to setup environment settings in certain instances.

View file

@ -77,7 +77,7 @@ class RebuildServerCommand extends Command
$json = array_merge($this->configurationStructureService->handle($server), ['rebuild' => true]); $json = array_merge($this->configurationStructureService->handle($server), ['rebuild' => true]);
try { try {
$this->daemonRepository->setNode($server->node_id)->setAccessServer($server->uuid)->update($json); $this->daemonRepository->setServer($server)->update($json);
} catch (RequestException $exception) { } catch (RequestException $exception) {
$this->output->error(trans('command/messages.server.rebuild_failed', [ $this->output->error(trans('command/messages.server.rebuild_failed', [
'name' => $server->name, 'name' => $server->name,

View file

@ -61,7 +61,7 @@ class DeleteUserCommand extends Command
$search = $this->option('user') ?? $this->ask(trans('command/messages.user.search_users')); $search = $this->option('user') ?? $this->ask(trans('command/messages.user.search_users'));
Assert::notEmpty($search, 'Search term must be a non-null value, received %s.'); Assert::notEmpty($search, 'Search term must be a non-null value, received %s.');
$results = $this->repository->search($search)->all(); $results = $this->repository->setSearchTerm($search)->all();
if (count($results) < 1) { if (count($results) < 1) {
$this->error(trans('command/messages.user.no_users_found')); $this->error(trans('command/messages.user.no_users_found'));
if ($this->input->isInteractive()) { if ($this->input->isInteractive()) {

View file

@ -54,9 +54,9 @@ class DisableTwoFactorCommand extends Command
} }
$email = $this->option('email') ?? $this->ask(trans('command/messages.user.ask_email')); $email = $this->option('email') ?? $this->ask(trans('command/messages.user.ask_email'));
$user = $this->repository->withColumns(['id', 'email'])->findFirstWhere([['email', '=', $email]]); $user = $this->repository->setColumns(['id', 'email'])->findFirstWhere([['email', '=', $email]]);
$this->repository->withoutFresh()->update($user->id, [ $this->repository->withoutFreshModel()->update($user->id, [
'use_totp' => false, 'use_totp' => false,
'totp_secret' => null, 'totp_secret' => null,
]); ]);

View file

@ -1,14 +1,9 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Illuminate\Support\Collection;
interface AllocationRepositoryInterface extends RepositoryInterface interface AllocationRepositoryInterface extends RepositoryInterface
{ {
/** /**
@ -18,13 +13,21 @@ interface AllocationRepositoryInterface extends RepositoryInterface
* @param array $ids * @param array $ids
* @return int * @return int
*/ */
public function assignAllocationsToServer($server, array $ids); public function assignAllocationsToServer(int $server = null, array $ids): int;
/** /**
* Return all of the allocations for a specific node. * Return all of the allocations for a specific node.
* *
* @param int $node * @param int $node
* @return \Illuminate\Database\Eloquent\Collection * @return \Illuminate\Support\Collection
*/ */
public function getAllocationsForNode($node); public function getAllocationsForNode(int $node): Collection;
/**
* Return all of the unique IPs that exist for a given node.
*
* @param int $node
* @return \Illuminate\Support\Collection
*/
public function getUniqueAllocationIpsForNode(int $node): Collection;
} }

View file

@ -1,21 +1,38 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository\Attributes; namespace Pterodactyl\Contracts\Repository\Attributes;
interface SearchableInterface interface SearchableInterface
{ {
/** /**
* Filter results by search term. * Set the search term.
* *
* @param string $term * @param string|null $term
* @return $this * @return $this
* @deprecated
*/ */
public function search($term); public function search($term);
/**
* Set the search term to use when requesting all records from
* the model.
*
* @param string|null $term
* @return $this
*/
public function setSearchTerm(string $term = null);
/**
* Determine if a valid search term is set on this repository.
*
* @return bool
*/
public function hasSearchTerm(): bool;
/**
* Return the search term.
*
* @return string|null
*/
public function getSearchTerm();
} }

View file

@ -1,68 +1,65 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository\Daemon; namespace Pterodactyl\Contracts\Repository\Daemon;
use GuzzleHttp\Client;
use Pterodactyl\Models\Node;
use Pterodactyl\Models\Server;
interface BaseRepositoryInterface interface BaseRepositoryInterface
{ {
/** /**
* Set the node model to be used for this daemon connection. * Set the node model to be used for this daemon connection.
* *
* @param int $id * @param \Pterodactyl\Models\Node $node
* @return $this * @return $this
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function setNode($id); public function setNode(Node $node);
/** /**
* Return the node model being used. * Return the node model being used.
* *
* @return \Pterodactyl\Models\Node * @return \Pterodactyl\Models\Node|null
*/ */
public function getNode(); public function getNode();
/** /**
* Set the UUID for the server to be used in the X-Access-Server header for daemon requests. * Set the Server model to use when requesting information from the Daemon.
* *
* @param null|string $server * @param \Pterodactyl\Models\Server $server
* @return $this * @return $this
*/ */
public function setAccessServer($server = null); public function setServer(Server $server);
/** /**
* Return the UUID of the server being used in requests. * Return the Server model.
* *
* @return string * @return \Pterodactyl\Models\Server|null
*/ */
public function getAccessServer(); public function getServer();
/** /**
* Set the token to be used in the X-Access-Token header for requests to the daemon. * Set the token to be used in the X-Access-Token header for requests to the daemon.
* *
* @param null|string $token * @param string $token
* @return $this * @return $this
*/ */
public function setAccessToken($token = null); public function setToken(string $token);
/** /**
* Return the access token being used for requests. * Return the access token being used for requests.
* *
* @return string * @return string|null
*/ */
public function getAccessToken(); public function getToken();
/** /**
* Return an instance of the Guzzle HTTP Client to be used for requests. * Return an instance of the Guzzle HTTP Client to be used for requests.
* *
* @param array $headers * @param array $headers
* @return \GuzzleHttp\Client * @return \GuzzleHttp\Client
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getHttpClient(array $headers = []); public function getHttpClient(array $headers = []): Client;
} }

View file

@ -1,14 +1,9 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository\Daemon; namespace Pterodactyl\Contracts\Repository\Daemon;
use Psr\Http\Message\ResponseInterface;
interface CommandRepositoryInterface extends BaseRepositoryInterface interface CommandRepositoryInterface extends BaseRepositoryInterface
{ {
/** /**
@ -17,5 +12,5 @@ interface CommandRepositoryInterface extends BaseRepositoryInterface
* @param string $command * @param string $command
* @return \Psr\Http\Message\ResponseInterface * @return \Psr\Http\Message\ResponseInterface
*/ */
public function send($command); public function send(string $command): ResponseInterface;
} }

View file

@ -1,14 +1,9 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository\Daemon; namespace Pterodactyl\Contracts\Repository\Daemon;
use Psr\Http\Message\ResponseInterface;
interface ConfigurationRepositoryInterface extends BaseRepositoryInterface interface ConfigurationRepositoryInterface extends BaseRepositoryInterface
{ {
/** /**
@ -17,5 +12,5 @@ interface ConfigurationRepositoryInterface extends BaseRepositoryInterface
* @param array $overrides * @param array $overrides
* @return \Psr\Http\Message\ResponseInterface * @return \Psr\Http\Message\ResponseInterface
*/ */
public function update(array $overrides = []); public function update(array $overrides = []): ResponseInterface;
} }

View file

@ -1,35 +1,31 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository\Daemon; namespace Pterodactyl\Contracts\Repository\Daemon;
use stdClass;
use Psr\Http\Message\ResponseInterface;
interface FileRepositoryInterface extends BaseRepositoryInterface interface FileRepositoryInterface extends BaseRepositoryInterface
{ {
/** /**
* Return stat information for a given file. * Return stat information for a given file.
* *
* @param string $path * @param string $path
* @return object * @return \stdClass
* *
* @throws \GuzzleHttp\Exception\RequestException * @throws \GuzzleHttp\Exception\RequestException
*/ */
public function getFileStat($path); public function getFileStat(string $path): stdClass;
/** /**
* Return the contents of a given file if it can be edited in the Panel. * Return the contents of a given file if it can be edited in the Panel.
* *
* @param string $path * @param string $path
* @return object * @return \stdClass
* *
* @throws \GuzzleHttp\Exception\RequestException * @throws \GuzzleHttp\Exception\RequestException
*/ */
public function getContent($path); public function getContent(string $path): stdClass;
/** /**
* Save new contents to a given file. * Save new contents to a given file.
@ -40,7 +36,7 @@ interface FileRepositoryInterface extends BaseRepositoryInterface
* *
* @throws \GuzzleHttp\Exception\RequestException * @throws \GuzzleHttp\Exception\RequestException
*/ */
public function putContent($path, $content); public function putContent(string $path, string $content): ResponseInterface;
/** /**
* Return a directory listing for a given path. * Return a directory listing for a given path.
@ -50,5 +46,5 @@ interface FileRepositoryInterface extends BaseRepositoryInterface
* *
* @throws \GuzzleHttp\Exception\RequestException * @throws \GuzzleHttp\Exception\RequestException
*/ */
public function getDirectory($path); public function getDirectory(string $path): array;
} }

View file

@ -1,14 +1,9 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository\Daemon; namespace Pterodactyl\Contracts\Repository\Daemon;
use Psr\Http\Message\ResponseInterface;
interface PowerRepositoryInterface extends BaseRepositoryInterface interface PowerRepositoryInterface extends BaseRepositoryInterface
{ {
const SIGNAL_START = 'start'; const SIGNAL_START = 'start';
@ -24,5 +19,5 @@ interface PowerRepositoryInterface extends BaseRepositoryInterface
* *
* @throws \Pterodactyl\Exceptions\Repository\Daemon\InvalidPowerSignalException * @throws \Pterodactyl\Exceptions\Repository\Daemon\InvalidPowerSignalException
*/ */
public function sendSignal($signal); public function sendSignal(string $signal): ResponseInterface;
} }

View file

@ -1,11 +1,4 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository\Daemon; namespace Pterodactyl\Contracts\Repository\Daemon;
@ -30,7 +23,7 @@ interface ServerRepositoryInterface extends BaseRepositoryInterface
* @param array $data * @param array $data
* @return \Psr\Http\Message\ResponseInterface * @return \Psr\Http\Message\ResponseInterface
*/ */
public function update(array $data); public function update(array $data): ResponseInterface;
/** /**
* Mark a server to be reinstalled on the system. * Mark a server to be reinstalled on the system.
@ -38,42 +31,42 @@ interface ServerRepositoryInterface extends BaseRepositoryInterface
* @param array|null $data * @param array|null $data
* @return \Psr\Http\Message\ResponseInterface * @return \Psr\Http\Message\ResponseInterface
*/ */
public function reinstall($data = null); public function reinstall(array $data = null): ResponseInterface;
/** /**
* Mark a server as needing a container rebuild the next time the server is booted. * Mark a server as needing a container rebuild the next time the server is booted.
* *
* @return \Psr\Http\Message\ResponseInterface * @return \Psr\Http\Message\ResponseInterface
*/ */
public function rebuild(); public function rebuild(): ResponseInterface;
/** /**
* Suspend a server on the daemon. * Suspend a server on the daemon.
* *
* @return \Psr\Http\Message\ResponseInterface * @return \Psr\Http\Message\ResponseInterface
*/ */
public function suspend(); public function suspend(): ResponseInterface;
/** /**
* Un-suspend a server on the daemon. * Un-suspend a server on the daemon.
* *
* @return \Psr\Http\Message\ResponseInterface * @return \Psr\Http\Message\ResponseInterface
*/ */
public function unsuspend(); public function unsuspend(): ResponseInterface;
/** /**
* Delete a server on the daemon. * Delete a server on the daemon.
* *
* @return \Psr\Http\Message\ResponseInterface * @return \Psr\Http\Message\ResponseInterface
*/ */
public function delete(); public function delete(): ResponseInterface;
/** /**
* Return detials on a specific server. * Return detials on a specific server.
* *
* @return \Psr\Http\Message\ResponseInterface * @return \Psr\Http\Message\ResponseInterface
*/ */
public function details(); public function details(): ResponseInterface;
/** /**
* Revoke an access key on the daemon before the time is expired. * Revoke an access key on the daemon before the time is expired.
@ -83,5 +76,5 @@ interface ServerRepositoryInterface extends BaseRepositoryInterface
* *
* @throws \GuzzleHttp\Exception\RequestException * @throws \GuzzleHttp\Exception\RequestException
*/ */
public function revokeAccessKey($key); public function revokeAccessKey($key): ResponseInterface;
} }

View file

@ -1,26 +1,4 @@
<?php <?php
/*
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
@ -44,14 +22,6 @@ interface DaemonKeyRepositoryInterface extends RepositoryInterface
*/ */
public function loadServerAndUserRelations(DaemonKey $key, bool $refresh = false): DaemonKey; public function loadServerAndUserRelations(DaemonKey $key, bool $refresh = false): DaemonKey;
/**
* Gets the daemon keys associated with a specific server.
*
* @param int $server
* @return \Illuminate\Support\Collection
*/
public function getServerKeys($server);
/** /**
* Return a daemon key with the associated server relation attached. * Return a daemon key with the associated server relation attached.
* *
@ -60,7 +30,7 @@ interface DaemonKeyRepositoryInterface extends RepositoryInterface
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getKeyWithServer($key); public function getKeyWithServer(string $key): DaemonKey;
/** /**
* Get all of the keys for a specific user including the information needed * Get all of the keys for a specific user including the information needed

View file

@ -1,30 +1,28 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Illuminate\Support\Collection;
use Pterodactyl\Models\DatabaseHost;
interface DatabaseHostRepositoryInterface extends RepositoryInterface interface DatabaseHostRepositoryInterface extends RepositoryInterface
{ {
/** /**
* Return database hosts with a count of databases and the node information for which it is attached. * Return database hosts with a count of databases and the node
* information for which it is attached.
* *
* @return \Illuminate\Support\Collection * @return \Illuminate\Support\Collection
*/ */
public function getWithViewDetails(); public function getWithViewDetails(): Collection;
/** /**
* Return a database host with the databases and associated servers that are attached to said databases. * Return a database host with the databases and associated servers
* that are attached to said databases.
* *
* @param int $id * @param int $id
* @return mixed * @return \Pterodactyl\Models\DatabaseHost
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getWithServers($id); public function getWithServers(int $id): DatabaseHost;
} }

View file

@ -9,6 +9,7 @@
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Pterodactyl\Models\Database;
use Illuminate\Support\Collection; use Illuminate\Support\Collection;
interface DatabaseRepositoryInterface extends RepositoryInterface interface DatabaseRepositoryInterface extends RepositoryInterface
@ -43,12 +44,12 @@ interface DatabaseRepositoryInterface extends RepositoryInterface
* the provided details. * the provided details.
* *
* @param array $data * @param array $data
* @return mixed * @return \Pterodactyl\Models\Database
* *
* @throws \Pterodactyl\Exceptions\DisplayException
* @throws \Pterodactyl\Exceptions\Model\DataValidationException * @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\DuplicateDatabaseNameException
*/ */
public function createIfNotExists(array $data); public function createIfNotExists(array $data): Database;
/** /**
* Create a new database on a given connection. * Create a new database on a given connection.
@ -56,7 +57,7 @@ interface DatabaseRepositoryInterface extends RepositoryInterface
* @param string $database * @param string $database
* @return bool * @return bool
*/ */
public function createDatabase($database); public function createDatabase(string $database): bool;
/** /**
* Create a new database user on a given connection. * Create a new database user on a given connection.
@ -66,7 +67,7 @@ interface DatabaseRepositoryInterface extends RepositoryInterface
* @param string $password * @param string $password
* @return bool * @return bool
*/ */
public function createUser($username, $remote, $password); public function createUser(string $username, string $remote, string $password): bool;
/** /**
* Give a specific user access to a given database. * Give a specific user access to a given database.
@ -76,14 +77,14 @@ interface DatabaseRepositoryInterface extends RepositoryInterface
* @param string $remote * @param string $remote
* @return bool * @return bool
*/ */
public function assignUserToDatabase($database, $username, $remote); public function assignUserToDatabase(string $database, string $username, string $remote): bool;
/** /**
* Flush the privileges for a given connection. * Flush the privileges for a given connection.
* *
* @return mixed * @return bool
*/ */
public function flush(); public function flush(): bool;
/** /**
* Drop a given database on a specific connection. * Drop a given database on a specific connection.
@ -91,7 +92,7 @@ interface DatabaseRepositoryInterface extends RepositoryInterface
* @param string $database * @param string $database
* @return bool * @return bool
*/ */
public function dropDatabase($database); public function dropDatabase(string $database): bool;
/** /**
* Drop a given user on a specific connection. * Drop a given user on a specific connection.
@ -100,5 +101,5 @@ interface DatabaseRepositoryInterface extends RepositoryInterface
* @param string $remote * @param string $remote
* @return mixed * @return mixed
*/ */
public function dropUser($username, $remote); public function dropUser(string $username, string $remote): bool;
} }

View file

@ -1,14 +1,9 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Pterodactyl\Models\Location;
use Illuminate\Support\Collection;
use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface; use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface;
interface LocationRepositoryInterface extends RepositoryInterface, SearchableInterface interface LocationRepositoryInterface extends RepositoryInterface, SearchableInterface
@ -16,16 +11,16 @@ interface LocationRepositoryInterface extends RepositoryInterface, SearchableInt
/** /**
* Return locations with a count of nodes and servers attached to it. * Return locations with a count of nodes and servers attached to it.
* *
* @return mixed * @return \Illuminate\Support\Collection
*/ */
public function getAllWithDetails(); public function getAllWithDetails(): Collection;
/** /**
* Return all of the available locations with the nodes as a relationship. * Return all of the available locations with the nodes as a relationship.
* *
* @return \Illuminate\Support\Collection * @return \Illuminate\Support\Collection
*/ */
public function getAllWithNodes(); public function getAllWithNodes(): Collection;
/** /**
* Return all of the nodes and their respective count of servers for a location. * Return all of the nodes and their respective count of servers for a location.
@ -35,7 +30,7 @@ interface LocationRepositoryInterface extends RepositoryInterface, SearchableInt
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getWithNodes($id); public function getWithNodes(int $id): Location;
/** /**
* Return a location and the count of nodes in that location. * Return a location and the count of nodes in that location.
@ -45,5 +40,5 @@ interface LocationRepositoryInterface extends RepositoryInterface, SearchableInt
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getWithNodeCount($id); public function getWithNodeCount(int $id): Location;
} }

View file

@ -1,68 +1,65 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Pterodactyl\Models\Node;
use Illuminate\Support\Collection;
use Illuminate\Contracts\Pagination\LengthAwarePaginator;
use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface; use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface;
interface NodeRepositoryInterface extends RepositoryInterface, SearchableInterface interface NodeRepositoryInterface extends RepositoryInterface, SearchableInterface
{ {
const THRESHOLD_PERCENTAGE_LOW = 75;
const THRESHOLD_PERCENTAGE_MEDIUM = 90;
/** /**
* Return the usage stats for a single node. * Return the usage stats for a single node.
* *
* @param int $id * @param \Pterodactyl\Models\Node $node
* @return array * @return array
*/ */
public function getUsageStats($id); public function getUsageStats(Node $node): array;
/** /**
* Return all available nodes with a searchable interface. * Return all available nodes with a searchable interface.
* *
* @param int $count
* @return \Illuminate\Contracts\Pagination\LengthAwarePaginator * @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
*/ */
public function getNodeListingData($count = 25); public function getNodeListingData(): LengthAwarePaginator;
/** /**
* Return a single node with location and server information. * Return a single node with location and server information.
* *
* @param int $id * @param \Pterodactyl\Models\Node $node
* @return mixed * @param bool $refresh
* * @return \Pterodactyl\Models\Node
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getSingleNode($id); public function loadLocationAndServerCount(Node $node, bool $refresh = false): Node;
/** /**
* Return a node with all of the associated allocations and servers that are attached to said allocations. * Attach a paginated set of allocations to a node mode including
* any servers that are also attached to those allocations.
* *
* @param int $id * @param \Pterodactyl\Models\Node $node
* @return mixed * @param bool $refresh
* * @return \Pterodactyl\Models\Node
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getNodeAllocations($id); public function loadNodeAllocations(Node $node, bool $refresh = false): Node;
/** /**
* Return a node with all of the servers attached to that node. * Return a node with all of the servers attached to that node.
* *
* @param int $id * @param int $id
* @return mixed * @return \Pterodactyl\Models\Node
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getNodeServers($id); public function getNodeServers(int $id): Node;
/** /**
* Return a collection of nodes for all locations to use in server creation UI. * Return a collection of nodes for all locations to use in server creation UI.
* *
* @return mixed * @return \Illuminate\Support\Collection
*/ */
public function getNodesForServerCreation(); public function getNodesForServerCreation(): Collection;
} }

View file

@ -1,44 +1,26 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Pterodactyl\Models\Pack;
use Illuminate\Contracts\Pagination\LengthAwarePaginator;
use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface; use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface;
interface PackRepositoryInterface extends RepositoryInterface, SearchableInterface interface PackRepositoryInterface extends RepositoryInterface, SearchableInterface
{ {
/**
* Return a paginated listing of packs with their associated egg and server count.
*
* @param int $paginate
* @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
*/
public function paginateWithEggAndServerCount($paginate = 50);
/** /**
* Return a pack with the associated server models attached to it. * Return a pack with the associated server models attached to it.
* *
* @param int $id * @param \Pterodactyl\Models\Pack $pack
* @return \Illuminate\Database\Eloquent\Collection * @param bool $refresh
* * @return \Pterodactyl\Models\Pack
* @throws \Illuminate\Database\Eloquent\ModelNotFoundException
*/ */
public function getWithServers($id); public function loadServerData(Pack $pack, bool $refresh = false): Pack;
/** /**
* Return all of the file archives for a given pack. * Return a paginated listing of packs with their associated egg and server count.
* *
* @param int $id * @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
* @param bool $collection
* @return object|\Illuminate\Support\Collection
*
* @throws \Illuminate\Database\Eloquent\ModelNotFoundException
*/ */
public function getFileArchives($id, $collection = false); public function paginateWithEggAndServerCount(): LengthAwarePaginator;
} }

View file

@ -1,14 +1,9 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Illuminate\Support\Collection;
interface RepositoryInterface interface RepositoryInterface
{ {
/** /**
@ -42,17 +37,34 @@ interface RepositoryInterface
/** /**
* An array of columns to filter the response by. * An array of columns to filter the response by.
* *
* @param array $columns * @param array|string $columns
* @return $this * @return $this
*/ */
public function withColumns($columns = ['*']); public function setColumns($columns = ['*']);
/** /**
* Disable returning a fresh model when data is inserted or updated. * Stop repository update functions from returning a fresh
* model when changes are committed.
* *
* @return $this * @return $this
*/ */
public function withoutFresh(); public function withoutFreshModel();
/**
* Return a fresh model with a repository updates a model.
*
* @return $this
*/
public function withFreshModel();
/**
* Set wether or not the repository should return a fresh model
* when changes are committed.
*
* @param bool $fresh
* @return $this
*/
public function setFreshModel(bool $fresh = true);
/** /**
* Create a new model instance and persist it to the database. * Create a new model instance and persist it to the database.
@ -64,23 +76,7 @@ interface RepositoryInterface
* *
* @throws \Pterodactyl\Exceptions\Model\DataValidationException * @throws \Pterodactyl\Exceptions\Model\DataValidationException
*/ */
public function create(array $fields, $validate = true, $force = false); public function create(array $fields, bool $validate = true, bool $force = false);
/**
* Delete a given record from the database.
*
* @param int $id
* @return int
*/
public function delete($id);
/**
* Delete records matching the given attributes.
*
* @param array $attributes
* @return int
*/
public function deleteWhere(array $attributes);
/** /**
* Find a model that has the specific ID passed. * Find a model that has the specific ID passed.
@ -90,15 +86,15 @@ interface RepositoryInterface
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function find($id); public function find(int $id);
/** /**
* Find a model matching an array of where clauses. * Find a model matching an array of where clauses.
* *
* @param array $fields * @param array $fields
* @return mixed * @return \Illuminate\Support\Collection
*/ */
public function findWhere(array $fields); public function findWhere(array $fields): Collection;
/** /**
* Find and return the first matching instance for the given fields. * Find and return the first matching instance for the given fields.
@ -116,7 +112,23 @@ interface RepositoryInterface
* @param array $fields * @param array $fields
* @return int * @return int
*/ */
public function findCountWhere(array $fields); public function findCountWhere(array $fields): int;
/**
* Delete a given record from the database.
*
* @param int $id
* @return int
*/
public function delete(int $id): int;
/**
* Delete records matching the given attributes.
*
* @param array $attributes
* @return int
*/
public function deleteWhere(array $attributes): int;
/** /**
* Update a given ID with the passed array of fields. * Update a given ID with the passed array of fields.
@ -130,7 +142,7 @@ interface RepositoryInterface
* @throws \Pterodactyl\Exceptions\Model\DataValidationException * @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function update($id, array $fields, $validate = true, $force = false); public function update($id, array $fields, bool $validate = true, bool $force = false);
/** /**
* Perform a mass update where matching records are updated using whereIn. * Perform a mass update where matching records are updated using whereIn.
@ -141,7 +153,7 @@ interface RepositoryInterface
* @param array $fields * @param array $fields
* @return int * @return int
*/ */
public function updateWhereIn($column, array $values, array $fields); public function updateWhereIn(string $column, array $values, array $fields): int;
/** /**
* Update a record if it exists in the database, otherwise create it. * Update a record if it exists in the database, otherwise create it.
@ -154,24 +166,14 @@ interface RepositoryInterface
* *
* @throws \Pterodactyl\Exceptions\Model\DataValidationException * @throws \Pterodactyl\Exceptions\Model\DataValidationException
*/ */
public function updateOrCreate(array $where, array $fields, $validate = true, $force = false); public function updateOrCreate(array $where, array $fields, bool $validate = true, bool $force = false);
/** /**
* Update multiple records matching the passed clauses. * Return all records associated with the given model.
* *
* @param array $where * @return Collection
* @param array $fields
* @return mixed
*/ */
public function massUpdate(array $where, array $fields); public function all(): Collection;
/**
* Return all records from the model.
*
* @param null $paginate
* @return mixed
*/
public function all($paginate = null);
/** /**
* Insert a single or multiple records into the database at once skipping * Insert a single or multiple records into the database at once skipping
@ -180,7 +182,7 @@ interface RepositoryInterface
* @param array $data * @param array $data
* @return bool * @return bool
*/ */
public function insert(array $data); public function insert(array $data): bool;
/** /**
* Insert multiple records into the database and ignore duplicates. * Insert multiple records into the database and ignore duplicates.
@ -188,5 +190,5 @@ interface RepositoryInterface
* @param array $values * @param array $values
* @return bool * @return bool
*/ */
public function insertIgnore(array $values); public function insertIgnore(array $values): bool;
} }

View file

@ -1,11 +1,4 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
@ -22,6 +15,16 @@ interface ScheduleRepositoryInterface extends RepositoryInterface
*/ */
public function findServerSchedules(int $server): Collection; public function findServerSchedules(int $server): Collection;
/**
* Load the tasks relationship onto the Schedule module if they are not
* already present.
*
* @param \Pterodactyl\Models\Schedule $schedule
* @param bool $refresh
* @return \Pterodactyl\Models\Schedule
*/
public function loadTasks(Schedule $schedule, bool $refresh = false): Schedule;
/** /**
* Return a schedule model with all of the associated tasks as a relationship. * Return a schedule model with all of the associated tasks as a relationship.
* *
@ -38,5 +41,5 @@ interface ScheduleRepositoryInterface extends RepositoryInterface
* @param string $timestamp * @param string $timestamp
* @return \Illuminate\Support\Collection * @return \Illuminate\Support\Collection
*/ */
public function getSchedulesToProcess($timestamp); public function getSchedulesToProcess(string $timestamp): Collection;
} }

View file

@ -1,15 +1,11 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Pterodactyl\Models\User;
use Pterodactyl\Models\Server; use Pterodactyl\Models\Server;
use Illuminate\Support\Collection;
use Illuminate\Contracts\Pagination\LengthAwarePaginator;
use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface; use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface;
interface ServerRepositoryInterface extends RepositoryInterface, SearchableInterface interface ServerRepositoryInterface extends RepositoryInterface, SearchableInterface
@ -17,19 +13,10 @@ interface ServerRepositoryInterface extends RepositoryInterface, SearchableInter
/** /**
* Returns a listing of all servers that exist including relationships. * Returns a listing of all servers that exist including relationships.
* *
* @param int|null $paginate * @param int $paginate
* @return mixed * @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
*/ */
public function getAllServers($paginate); public function getAllServers(int $paginate): LengthAwarePaginator;
/**
* Return a collection of servers with their associated data for rebuild operations.
*
* @param int|null $server
* @param int|null $node
* @return \Illuminate\Database\Eloquent\Collection
*/
public function getDataForRebuild($server = null, $node = null);
/** /**
* Load the egg relations onto the server model. * Load the egg relations onto the server model.
@ -40,15 +27,35 @@ interface ServerRepositoryInterface extends RepositoryInterface, SearchableInter
*/ */
public function loadEggRelations(Server $server, bool $refresh = false): Server; public function loadEggRelations(Server $server, bool $refresh = false): Server;
/**
* Return a collection of servers with their associated data for rebuild operations.
*
* @param int|null $server
* @param int|null $node
* @return \Illuminate\Support\Collection
*/
public function getDataForRebuild(int $server = null, int $node = null): Collection;
/** /**
* Return a server model and all variables associated with the server. * Return a server model and all variables associated with the server.
* *
* @param int $id * @param int $id
* @return mixed * @return \Pterodactyl\Models\Server
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function findWithVariables($id); public function findWithVariables(int $id): Server;
/**
* Get the primary allocation for a given server. If a model is passed into
* the function, load the allocation relationship onto it. Otherwise, find and
* return the server from the database.
*
* @param \Pterodactyl\Models\Server $server
* @param bool $refresh
* @return \Pterodactyl\Models\Server
*/
public function getPrimaryAllocation(Server $server, bool $refresh = false): Server;
/** /**
* Return all of the server variables possible and default to the variable * Return all of the server variables possible and default to the variable
@ -60,20 +67,7 @@ interface ServerRepositoryInterface extends RepositoryInterface, SearchableInter
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getVariablesWithValues($id, $returnAsObject = false); public function getVariablesWithValues(int $id, bool $returnAsObject = false);
/**
* Get the primary allocation for a given server. If a model is passed into
* the function, load the allocation relationship onto it. Otherwise, find and
* return the server from the database.
*
* @param int|\Pterodactyl\Models\Server $server
* @param bool $refresh
* @return \Pterodactyl\Models\Server
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/
public function getPrimaryAllocation($server, bool $refresh = false): Server;
/** /**
* Return enough data to be used for the creation of a server via the daemon. * Return enough data to be used for the creation of a server via the daemon.
@ -85,14 +79,13 @@ interface ServerRepositoryInterface extends RepositoryInterface, SearchableInter
public function getDataForCreation(Server $server, bool $refresh = false): Server; public function getDataForCreation(Server $server, bool $refresh = false): Server;
/** /**
* Return a server as well as associated databases and their hosts. * Load associated databases onto the server model.
* *
* @param int $id * @param \Pterodactyl\Models\Server $server
* @return mixed * @param bool $refresh
* * @return \Pterodactyl\Models\Server
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getWithDatabases($id); public function loadDatabaseRelations(Server $server, bool $refresh = false): Server;
/** /**
* Get data for use when updating a server on the Daemon. Returns an array of * Get data for use when updating a server on the Daemon. Returns an array of
@ -105,24 +98,14 @@ interface ServerRepositoryInterface extends RepositoryInterface, SearchableInter
*/ */
public function getDaemonServiceData(Server $server, bool $refresh = false): array; public function getDaemonServiceData(Server $server, bool $refresh = false): array;
/**
* Return an array of server IDs that a given user can access based on owner and subuser permissions.
*
* @param int $user
* @return array
*/
public function getUserAccessServers($user);
/** /**
* Return a paginated list of servers that a user can access at a given level. * Return a paginated list of servers that a user can access at a given level.
* *
* @param int $user * @param \Pterodactyl\Models\User $user
* @param string $level * @param int $level
* @param bool $admin
* @param array $relations
* @return \Illuminate\Pagination\LengthAwarePaginator * @return \Illuminate\Pagination\LengthAwarePaginator
*/ */
public function filterUserAccessServers($user, $admin = false, $level = 'all', array $relations = []); public function filterUserAccessServers(User $user, int $level): LengthAwarePaginator;
/** /**
* Return a server by UUID. * Return a server by UUID.
@ -132,5 +115,5 @@ interface ServerRepositoryInterface extends RepositoryInterface, SearchableInter
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getByUuid($uuid); public function getByUuid(string $uuid): Server;
} }

View file

@ -1,11 +1,4 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;

View file

@ -1,16 +1,19 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Illuminate\Support\Collection;
interface SessionRepositoryInterface extends RepositoryInterface interface SessionRepositoryInterface extends RepositoryInterface
{ {
/**
* Return all of the active sessions for a user.
*
* @param int $user
* @return \Illuminate\Support\Collection
*/
public function getUserSessions(int $user): Collection;
/** /**
* Delete a session for a given user. * Delete a session for a given user.
* *
@ -18,13 +21,5 @@ interface SessionRepositoryInterface extends RepositoryInterface
* @param int $session * @param int $session
* @return null|int * @return null|int
*/ */
public function deleteUserSession($user, $session); public function deleteUserSession(int $user, int $session);
/**
* Return all of the active sessions for a user.
*
* @param int $user
* @return \Illuminate\Support\Collection
*/
public function getUserSessions($user);
} }

View file

@ -9,7 +9,9 @@ interface SettingsRepositoryInterface extends RepositoryInterface
* *
* @param string $key * @param string $key
* @param string $value * @param string $value
* @return mixed *
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function set(string $key, string $value); public function set(string $key, string $value);
@ -26,7 +28,6 @@ interface SettingsRepositoryInterface extends RepositoryInterface
* Remove a key from the database cache. * Remove a key from the database cache.
* *
* @param string $key * @param string $key
* @return mixed
*/ */
public function forget(string $key); public function forget(string $key);
} }

View file

@ -34,25 +34,4 @@ interface SubuserRepositoryInterface extends RepositoryInterface
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getWithPermissionsUsingUserAndServer(int $user, int $server): Subuser; public function getWithPermissionsUsingUserAndServer(int $user, int $server): Subuser;
/**
* Find a subuser and return with server and permissions relationships.
*
* @param int $id
* @return \Illuminate\Database\Eloquent\Collection
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/
public function getWithServerAndPermissions($id);
/**
* Return a subuser and their associated connection key for a server.
*
* @param int $user
* @param int $server
* @return \Pterodactyl\Models\Subuser
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/
public function getWithKey($user, $server);
} }

View file

@ -1,14 +1,9 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Pterodactyl\Models\Task;
interface TaskRepositoryInterface extends RepositoryInterface interface TaskRepositoryInterface extends RepositoryInterface
{ {
/** /**
@ -19,14 +14,14 @@ interface TaskRepositoryInterface extends RepositoryInterface
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getTaskWithServer($id); public function getTaskForJobProcess(int $id): Task;
/** /**
* Returns the next task in a schedule. * Returns the next task in a schedule.
* *
* @param int $schedule the ID of the schedule to select the next task from * @param int $schedule
* @param int $index the index of the current task * @param int $index
* @return null|\Pterodactyl\Models\Task * @return null|\Pterodactyl\Models\Task
*/ */
public function getNextTask($schedule, $index); public function getNextTask(int $schedule, int $index);
} }

View file

@ -1,14 +1,9 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Contracts\Repository; namespace Pterodactyl\Contracts\Repository;
use Illuminate\Support\Collection;
use Illuminate\Contracts\Pagination\LengthAwarePaginator;
use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface; use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface;
interface UserRepositoryInterface extends RepositoryInterface, SearchableInterface interface UserRepositoryInterface extends RepositoryInterface, SearchableInterface
@ -18,13 +13,13 @@ interface UserRepositoryInterface extends RepositoryInterface, SearchableInterfa
* *
* @return \Illuminate\Contracts\Pagination\LengthAwarePaginator * @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
*/ */
public function getAllUsersWithCounts(); public function getAllUsersWithCounts(): LengthAwarePaginator;
/** /**
* Return all matching models for a user in a format that can be used for dropdowns. * Return all matching models for a user in a format that can be used for dropdowns.
* *
* @param string $query * @param string $query
* @return \Illuminate\Database\Eloquent\Collection * @return \Illuminate\Support\Collection
*/ */
public function filterUsersByQuery($query); public function filterUsersByQuery(string $query): Collection;
} }

View file

@ -139,7 +139,6 @@ class DatabaseController extends Controller
* @param int $host * @param int $host
* @return \Illuminate\Http\RedirectResponse * @return \Illuminate\Http\RedirectResponse
* *
* @throws \Pterodactyl\Exceptions\DisplayException
* @throws \Pterodactyl\Exceptions\Model\DataValidationException * @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */

View file

@ -126,7 +126,7 @@ class NodesController extends Controller
public function index(Request $request) public function index(Request $request)
{ {
return view('admin.nodes.index', [ return view('admin.nodes.index', [
'nodes' => $this->repository->search($request->input('query'))->getNodeListingData(), 'nodes' => $this->repository->setSearchTerm($request->input('query'))->getNodeListingData(),
]); ]);
} }
@ -166,15 +166,15 @@ class NodesController extends Controller
/** /**
* Shows the index overview page for a specific node. * Shows the index overview page for a specific node.
* *
* @param int $node * @param \Pterodactyl\Models\Node $node
* @return \Illuminate\View\View * @return \Illuminate\View\View
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function viewIndex($node) public function viewIndex(Node $node)
{ {
return view('admin.nodes.view.index', [ return view('admin.nodes.view.index', [
'node' => $this->repository->getSingleNode($node), 'node' => $this->repository->loadLocationAndServerCount($node),
'stats' => $this->repository->getUsageStats($node), 'stats' => $this->repository->getUsageStats($node),
'version' => $this->versionService, 'version' => $this->versionService,
]); ]);
@ -208,17 +208,18 @@ class NodesController extends Controller
/** /**
* Shows the allocation page for a specific node. * Shows the allocation page for a specific node.
* *
* @param int $node * @param \Pterodactyl\Models\Node $node
* @return \Illuminate\View\View * @return \Illuminate\View\View
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function viewAllocation($node) public function viewAllocation(Node $node)
{ {
$node = $this->repository->getNodeAllocations($node); $this->repository->loadNodeAllocations($node);
Javascript::put(['node' => collect($node)->only(['id'])]); Javascript::put(['node' => collect($node)->only(['id'])]);
return view('admin.nodes.view.allocation', ['node' => $node]); return view('admin.nodes.view.allocation', [
'allocations' => $this->allocationRepository->setColumns(['ip'])->getUniqueAllocationIpsForNode($node->id),
'node' => $node,
]);
} }
/** /**

View file

@ -114,9 +114,7 @@ class PackController extends Controller
public function index(Request $request) public function index(Request $request)
{ {
return view('admin.packs.index', [ return view('admin.packs.index', [
'packs' => $this->repository->search($request->input('query'))->paginateWithEggAndServerCount( 'packs' => $this->repository->setSearchTerm($request->input('query'))->paginateWithEggAndServerCount(),
$this->config->get('pterodactyl.paginate.admin.packs')
),
]); ]);
} }
@ -177,14 +175,14 @@ class PackController extends Controller
/** /**
* Display pack view template to user. * Display pack view template to user.
* *
* @param int $pack * @param \Pterodactyl\Models\Pack $pack
* @return \Illuminate\View\View * @return \Illuminate\View\View
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function view($pack) public function view(Pack $pack)
{ {
return view('admin.packs.view', [ return view('admin.packs.view', [
'pack' => $this->repository->getWithServers($pack), 'pack' => $this->repository->loadServerData($pack),
'nests' => $this->serviceRepository->getWithEggs(), 'nests' => $this->serviceRepository->getWithEggs(),
]); ]);
} }

View file

@ -254,6 +254,7 @@ class ServersController extends Controller
* @throws \Pterodactyl\Exceptions\DisplayException * @throws \Pterodactyl\Exceptions\DisplayException
* @throws \Pterodactyl\Exceptions\Model\DataValidationException * @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
* @throws \Illuminate\Validation\ValidationException
*/ */
public function store(ServerFormRequest $request) public function store(ServerFormRequest $request)
{ {
@ -352,14 +353,12 @@ class ServersController extends Controller
/** /**
* Display the database management page for a specific server. * Display the database management page for a specific server.
* *
* @param int $server * @param \Pterodactyl\Models\Server $server
* @return \Illuminate\View\View * @return \Illuminate\View\View
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function viewDatabase($server) public function viewDatabase(Server $server)
{ {
$server = $this->repository->getWithDatabases($server); $this->repository->loadDatabaseRelations($server);
return view('admin.servers.view.database', [ return view('admin.servers.view.database', [
'hosts' => $this->databaseHostRepository->all(), 'hosts' => $this->databaseHostRepository->all(),
@ -459,12 +458,10 @@ class ServersController extends Controller
* *
* @param \Pterodactyl\Models\Server $server * @param \Pterodactyl\Models\Server $server
* @return \Illuminate\Http\RedirectResponse * @return \Illuminate\Http\RedirectResponse
*
* @throws \Pterodactyl\Exceptions\DisplayException
*/ */
public function rebuildContainer(Server $server) public function rebuildContainer(Server $server)
{ {
$this->containerRebuildService->rebuild($server); $this->containerRebuildService->handle($server);
$this->alert->success(trans('admin/server.alerts.rebuild_on_boot'))->flash(); $this->alert->success(trans('admin/server.alerts.rebuild_on_boot'))->flash();
return redirect()->route('admin.servers.view.manage', $server->id); return redirect()->route('admin.servers.view.manage', $server->id);

View file

@ -83,7 +83,7 @@ class UserController extends Controller
*/ */
public function index(Request $request) public function index(Request $request)
{ {
$users = $this->repository->search($request->input('query'))->getAllUsersWithCounts(); $users = $this->repository->setSearchTerm($request->input('query'))->getAllUsersWithCounts();
return view('admin.users.index', ['users' => $users]); return view('admin.users.index', ['users' => $users]);
} }

View file

@ -1,31 +1,9 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>
* Some Modifications (c) 2015 Dylan Seidt <dylan.seidt@gmail.com>.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
namespace Pterodactyl\Http\Controllers\Base; namespace Pterodactyl\Http\Controllers\Base;
use Illuminate\Http\Request; use Illuminate\Http\Request;
use Pterodactyl\Models\User;
use GuzzleHttp\Exception\RequestException; use GuzzleHttp\Exception\RequestException;
use Pterodactyl\Http\Controllers\Controller; use Pterodactyl\Http\Controllers\Controller;
use Symfony\Component\HttpKernel\Exception\HttpException; use Symfony\Component\HttpKernel\Exception\HttpException;
@ -75,8 +53,8 @@ class IndexController extends Controller
*/ */
public function getIndex(Request $request) public function getIndex(Request $request)
{ {
$servers = $this->repository->search($request->input('query'))->filterUserAccessServers( $servers = $this->repository->setSearchTerm($request->input('query'))->filterUserAccessServers(
$request->user()->id, $request->user()->root_admin, 'all', ['user'] $request->user(), User::FILTER_LEVEL_ALL
); );
return view('base.index', ['servers' => $servers]); return view('base.index', ['servers' => $servers]);
@ -102,10 +80,7 @@ class IndexController extends Controller
} }
try { try {
$response = $this->daemonRepository->setNode($server->node_id) $response = $this->daemonRepository->setServer($server)->setToken($token)->details();
->setAccessServer($server->uuid)
->setAccessToken($token)
->details();
} catch (RequestException $exception) { } catch (RequestException $exception) {
throw new HttpException(500, $exception->getMessage()); throw new HttpException(500, $exception->getMessage());
} }

View file

@ -96,7 +96,6 @@ class FileActionsController extends Controller
* @return \Illuminate\View\View * @return \Illuminate\View\View
* *
* @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function view(UpdateFileContentsFormRequest $request, string $uuid, string $file): View public function view(UpdateFileContentsFormRequest $request, string $uuid, string $file): View
{ {
@ -104,9 +103,7 @@ class FileActionsController extends Controller
$dirname = pathinfo($file, PATHINFO_DIRNAME); $dirname = pathinfo($file, PATHINFO_DIRNAME);
try { try {
$content = $this->repository->setNode($server->node_id)->setAccessServer($server->uuid) $content = $this->repository->setServer($server)->setToken($request->attributes->get('server_token'))->getContent($file);
->setAccessToken($request->attributes->get('server_token'))
->getContent($file);
} catch (RequestException $exception) { } catch (RequestException $exception) {
throw new DaemonConnectionException($exception); throw new DaemonConnectionException($exception);
} }

View file

@ -66,10 +66,7 @@ class RemoteRequestController extends Controller
} }
try { try {
$listing = $this->repository->setNode($server->node_id) $listing = $this->repository->setServer($server)->setToken($request->attributes->get('server_token'))->getDirectory($requestDirectory);
->setAccessServer($server->uuid)
->setAccessToken($request->attributes->get('server_token'))
->getDirectory($requestDirectory);
} catch (RequestException $exception) { } catch (RequestException $exception) {
throw new DaemonConnectionException($exception, true); throw new DaemonConnectionException($exception, true);
} }
@ -90,7 +87,6 @@ class RemoteRequestController extends Controller
* *
* @throws \Illuminate\Auth\Access\AuthorizationException * @throws \Illuminate\Auth\Access\AuthorizationException
* @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException * @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function store(Request $request): Response public function store(Request $request): Response
{ {
@ -98,9 +94,7 @@ class RemoteRequestController extends Controller
$this->authorize('save-files', $server); $this->authorize('save-files', $server);
try { try {
$this->repository->setNode($server->node_id) $this->repository->setServer($server)->setToken($request->attributes->get('server_token'))
->setAccessServer($server->uuid)
->setAccessToken($request->attributes->get('server_token'))
->putContent($request->input('file'), $request->input('contents') ?? ''); ->putContent($request->input('file'), $request->input('contents') ?? '');
return response('', 204); return response('', 204);

View file

@ -0,0 +1,70 @@
<?php
namespace Pterodactyl\Http\Controllers\Server\Tasks;
use Carbon\Carbon;
use Illuminate\Http\Request;
use Illuminate\Http\Response;
use Pterodactyl\Http\Controllers\Controller;
use Pterodactyl\Services\Schedules\ProcessScheduleService;
use Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface;
class ActionController extends Controller
{
private $processScheduleService;
/**
* @var \Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface
*/
private $repository;
public function __construct(ProcessScheduleService $processScheduleService, ScheduleRepositoryInterface $repository)
{
$this->processScheduleService = $processScheduleService;
$this->repository = $repository;
}
/**
* Toggle a task to be active or inactive for a given server.
*
* @param \Illuminate\Http\Request $request
* @return \Illuminate\Http\Response
*
* @throws \Illuminate\Auth\Access\AuthorizationException
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/
public function toggle(Request $request): Response
{
$server = $request->attributes->get('server');
$schedule = $request->attributes->get('schedule');
$this->authorize('toggle-schedule', $server);
$this->repository->update($schedule->id, [
'is_active' => ! $schedule->is_active,
]);
return response('', 204);
}
/**
* Trigger a schedule to run now.
*
* @param \Illuminate\Http\Request $request
* @return \Illuminate\Http\Response
*
* @throws \Illuminate\Auth\Access\AuthorizationException
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/
public function trigger(Request $request): Response
{
$server = $request->attributes->get('server');
$this->authorize('toggle-schedule', $server);
$this->processScheduleService->setRunTimeOverride(Carbon::now())->handle(
$request->attributes->get('schedule')
);
return response('', 204);
}
}

View file

@ -112,7 +112,7 @@ class TaskManagementController extends Controller
$server = $request->attributes->get('server'); $server = $request->attributes->get('server');
$schedule = $this->creationService->handle($server, $request->normalize(), $request->getTasks()); $schedule = $this->creationService->handle($server, $request->normalize(), $request->getTasks());
$this->alert->success(trans('server.schedules.task_created'))->flash(); $this->alert->success(trans('server.schedule.task_created'))->flash();
return redirect()->route('server.schedules.view', [ return redirect()->route('server.schedules.view', [
'server' => $server->uuidShort, 'server' => $server->uuidShort,

View file

@ -34,6 +34,7 @@ class DaemonAuthenticate
* Create a new filter instance. * Create a new filter instance.
* *
* @param \Pterodactyl\Contracts\Repository\NodeRepositoryInterface $repository * @param \Pterodactyl\Contracts\Repository\NodeRepositoryInterface $repository
* @deprecated
*/ */
public function __construct(NodeRepositoryInterface $repository) public function __construct(NodeRepositoryInterface $repository)
{ {
@ -47,6 +48,7 @@ class DaemonAuthenticate
* @param \Closure $next * @param \Closure $next
* @return mixed * @return mixed
* *
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
* @throws \Symfony\Component\HttpKernel\Exception\AccessDeniedHttpException * @throws \Symfony\Component\HttpKernel\Exception\AccessDeniedHttpException
*/ */
public function handle(Request $request, Closure $next) public function handle(Request $request, Closure $next)
@ -59,7 +61,7 @@ class DaemonAuthenticate
throw new AccessDeniedHttpException; throw new AccessDeniedHttpException;
} }
$node = $this->repository->findWhere(['daemonSecret' => $request->header('X-Access-Node')]); $node = $this->repository->findFirstWhere(['daemonSecret' => $request->header('X-Access-Node')]);
$request->attributes->set('node', $node); $request->attributes->set('node', $node);
return $next($request); return $next($request);

View file

@ -20,7 +20,7 @@ class AllocationFormRequest extends AdminFormRequest
{ {
return [ return [
'allocation_ip' => 'required|string', 'allocation_ip' => 'required|string',
'allocation_alias' => 'sometimes|string|max:255', 'allocation_alias' => 'sometimes|nullable|string|max:255',
'allocation_ports' => 'required|array', 'allocation_ports' => 'required|array',
]; ];
} }

View file

@ -21,7 +21,10 @@ class ServerFormRequest extends AdminFormRequest
*/ */
public function rules() public function rules()
{ {
return Server::getCreateRules(); $rules = Server::getCreateRules();
$rules['description'][] = 'nullable';
return $rules;
} }
/** /**

View file

@ -69,7 +69,6 @@ class UpdateFileContentsFormRequest extends ServerFormRequest
* @throws \Pterodactyl\Exceptions\DisplayException * @throws \Pterodactyl\Exceptions\DisplayException
* @throws \Pterodactyl\Exceptions\Http\Server\FileSizeTooLargeException * @throws \Pterodactyl\Exceptions\Http\Server\FileSizeTooLargeException
* @throws \Pterodactyl\Exceptions\Http\Server\FileTypeNotEditableException * @throws \Pterodactyl\Exceptions\Http\Server\FileTypeNotEditableException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
private function checkFileCanBeEdited($server, $token) private function checkFileCanBeEdited($server, $token)
{ {
@ -77,9 +76,7 @@ class UpdateFileContentsFormRequest extends ServerFormRequest
$repository = app()->make(FileRepositoryInterface::class); $repository = app()->make(FileRepositoryInterface::class);
try { try {
$stats = $repository->setNode($server->node_id)->setAccessServer($server->uuid) $stats = $repository->setServer($server)->setToken($token)->getFileStat($this->route()->parameter('file'));
->setAccessToken($token)
->getFileStat($this->route()->parameter('file'));
} catch (RequestException $exception) { } catch (RequestException $exception) {
switch ($exception->getCode()) { switch ($exception->getCode()) {
case 404: case 404:

View file

@ -1,18 +1,10 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Jobs\Schedule; namespace Pterodactyl\Jobs\Schedule;
use Exception; use Exception;
use Carbon\Carbon; use Carbon\Carbon;
use Pterodactyl\Jobs\Job; use Pterodactyl\Jobs\Job;
use Webmozart\Assert\Assert;
use InvalidArgumentException; use InvalidArgumentException;
use Illuminate\Queue\SerializesModels; use Illuminate\Queue\SerializesModels;
use Illuminate\Queue\InteractsWithQueue; use Illuminate\Queue\InteractsWithQueue;
@ -59,12 +51,9 @@ class RunTaskJob extends Job implements ShouldQueue
* @param int $task * @param int $task
* @param int $schedule * @param int $schedule
*/ */
public function __construct($task, $schedule) public function __construct(int $task, int $schedule)
{ {
Assert::integerish($task, 'First argument passed to constructor must be integer, received %s.'); $this->queue = config('pterodactyl.queues.standard');
Assert::integerish($schedule, 'Second argument passed to constructor must be integer, received %s.');
$this->queue = app()->make('config')->get('pterodactyl.queues.standard');
$this->task = $task; $this->task = $task;
$this->schedule = $schedule; $this->schedule = $schedule;
} }
@ -91,26 +80,32 @@ class RunTaskJob extends Job implements ShouldQueue
$this->powerRepository = $powerRepository; $this->powerRepository = $powerRepository;
$this->taskRepository = $taskRepository; $this->taskRepository = $taskRepository;
$task = $this->taskRepository->getTaskWithServer($this->task); $task = $this->taskRepository->getTaskForJobProcess($this->task);
$server = $task->getRelation('server'); $server = $task->getRelation('server');
$user = $server->getRelation('user'); $user = $server->getRelation('user');
// Do not process a task that is not set to active.
if (! $task->getRelation('schedule')->is_active) {
$this->markTaskNotQueued();
$this->markScheduleComplete();
return;
}
// Perform the provided task aganist the daemon. // Perform the provided task aganist the daemon.
switch ($task->action) { switch ($task->action) {
case 'power': case 'power':
$this->powerRepository->setNode($server->node_id) $this->powerRepository->setServer($server)
->setAccessServer($server->uuid) ->setToken($keyProviderService->handle($server, $user))
->setAccessToken($keyProviderService->handle($server, $user))
->sendSignal($task->payload); ->sendSignal($task->payload);
break; break;
case 'command': case 'command':
$this->commandRepository->setNode($server->node_id) $this->commandRepository->setServer($server)
->setAccessServer($server->uuid) ->setToken($keyProviderService->handle($server, $user))
->setAccessToken($keyProviderService->handle($server, $user))
->send($task->payload); ->send($task->payload);
break; break;
default: default:
throw new InvalidArgumentException('Cannot run a task that points to a non-existant action.'); throw new InvalidArgumentException('Cannot run a task that points to a non-existent action.');
} }
$this->markTaskNotQueued(); $this->markTaskNotQueued();
@ -161,7 +156,7 @@ class RunTaskJob extends Job implements ShouldQueue
private function markScheduleComplete() private function markScheduleComplete()
{ {
$repository = app()->make(ScheduleRepositoryInterface::class); $repository = app()->make(ScheduleRepositoryInterface::class);
$repository->withoutFresh()->update($this->schedule, [ $repository->withoutFreshModel()->update($this->schedule, [
'is_processing' => false, 'is_processing' => false,
'last_run_at' => Carbon::now()->toDateTimeString(), 'last_run_at' => Carbon::now()->toDateTimeString(),
]); ]);

View file

@ -27,12 +27,13 @@ namespace Pterodactyl\Models;
use Sofa\Eloquence\Eloquence; use Sofa\Eloquence\Eloquence;
use Sofa\Eloquence\Validable; use Sofa\Eloquence\Validable;
use Illuminate\Database\Eloquent\Model; use Illuminate\Database\Eloquent\Model;
use Znck\Eloquent\Traits\BelongsToThrough;
use Sofa\Eloquence\Contracts\CleansAttributes; use Sofa\Eloquence\Contracts\CleansAttributes;
use Sofa\Eloquence\Contracts\Validable as ValidableContract; use Sofa\Eloquence\Contracts\Validable as ValidableContract;
class DaemonKey extends Model implements CleansAttributes, ValidableContract class DaemonKey extends Model implements CleansAttributes, ValidableContract
{ {
use Eloquence, Validable; use BelongsToThrough, Eloquence, Validable;
/** /**
* @var string * @var string
@ -91,6 +92,17 @@ class DaemonKey extends Model implements CleansAttributes, ValidableContract
return $this->belongsTo(Server::class); return $this->belongsTo(Server::class);
} }
/**
* Return the node relation.
*
* @return \Znck\Eloquent\Relations\BelongsToThrough
* @throws \Exception
*/
public function node()
{
return $this->belongsToThrough(Node::class, Server::class);
}
/** /**
* Return the user relation. * Return the user relation.
* *

View file

@ -119,13 +119,13 @@ class Server extends Model implements CleansAttributes, ValidableContract
* @var array * @var array
*/ */
protected $searchableColumns = [ protected $searchableColumns = [
'name' => 10, 'name' => 50,
'uuidShort' => 9, 'uuidShort' => 10,
'uuid' => 8, 'uuid' => 10,
'pack.name' => 7, 'pack.name' => 5,
'user.email' => 6, 'user.email' => 20,
'user.username' => 6, 'user.username' => 20,
'node.name' => 2, 'node.name' => 10,
]; ];
/** /**

View file

@ -32,6 +32,11 @@ class User extends Model implements
const USER_LEVEL_USER = 0; const USER_LEVEL_USER = 0;
const USER_LEVEL_ADMIN = 1; const USER_LEVEL_ADMIN = 1;
const FILTER_LEVEL_ALL = 0;
const FILTER_LEVEL_OWNER = 1;
const FILTER_LEVEL_ADMIN = 2;
const FILTER_LEVEL_SUBUSER = 3;
/** /**
* Level of servers to display when using access() on a user. * Level of servers to display when using access() on a user.
* *

View file

@ -39,7 +39,7 @@ class APIKeyPolicy
})->values(); })->values();
}); });
return $permissions->search($permission, true) !== false; return $permissions->setSearchTerm($permission, true) !== false;
} }
/** /**

View file

@ -32,7 +32,7 @@ class ServerPolicy
})->values(); })->values();
}); });
return $permissions->search($permission, true) !== false; return $permissions->setSearchTerm($permission, true) !== false;
} }
/** /**

View file

@ -1,30 +1,36 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Concerns; namespace Pterodactyl\Repositories\Concerns;
trait Searchable trait Searchable
{ {
/** /**
* The term to search. * The search term to use when filtering results.
* *
* @var bool|string * @var null|string
*/ */
protected $searchTerm = false; protected $searchTerm;
/** /**
* Perform a search of the model using the given term. * Set the search term.
* *
* @param string $term * @param string|null $term
* @return $this * @return $this
* @deprecated
*/ */
public function search($term) public function search($term)
{
return $this->setSearchTerm($term);
}
/**
* Set the search term to use when requesting all records from
* the model.
*
* @param string|null $term
* @return $this
*/
public function setSearchTerm(string $term = null)
{ {
if (empty($term)) { if (empty($term)) {
return $this; return $this;
@ -35,4 +41,24 @@ trait Searchable
return $clone; return $clone;
} }
/**
* Determine if a valid search term is set on this repository.
*
* @return bool
*/
public function hasSearchTerm(): bool
{
return ! empty($this->searchTerm);
}
/**
* Return the search term.
*
* @return string|null
*/
public function getSearchTerm()
{
return $this->searchTerm;
}
} }

View file

@ -1,84 +1,71 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Daemon; namespace Pterodactyl\Repositories\Daemon;
use RuntimeException;
use GuzzleHttp\Client; use GuzzleHttp\Client;
use Webmozart\Assert\Assert; use Pterodactyl\Models\Node;
use Pterodactyl\Models\Server;
use Illuminate\Foundation\Application; use Illuminate\Foundation\Application;
use Pterodactyl\Contracts\Repository\NodeRepositoryInterface; use Pterodactyl\Contracts\Repository\NodeRepositoryInterface;
use Illuminate\Contracts\Config\Repository as ConfigRepository;
use Pterodactyl\Contracts\Repository\Daemon\BaseRepositoryInterface; use Pterodactyl\Contracts\Repository\Daemon\BaseRepositoryInterface;
class BaseRepository implements BaseRepositoryInterface abstract class BaseRepository implements BaseRepositoryInterface
{ {
/** /**
* @var \Illuminate\Foundation\Application * @var \Illuminate\Foundation\Application
*/ */
protected $app; private $app;
/** /**
* @var * @var \Pterodactyl\Models\Server
*/ */
protected $accessServer; private $server;
/** /**
* @var * @var string|null
*/ */
protected $accessToken; private $token;
/** /**
* @var * @var \Pterodactyl\Models\Node|null
*/ */
protected $node; private $node;
/**
* @var \Illuminate\Contracts\Config\Repository
*/
protected $config;
/** /**
* @var \Pterodactyl\Contracts\Repository\NodeRepositoryInterface * @var \Pterodactyl\Contracts\Repository\NodeRepositoryInterface
*/ */
protected $nodeRepository; private $nodeRepository;
/** /**
* BaseRepository constructor. * BaseRepository constructor.
* *
* @param \Illuminate\Foundation\Application $app * @param \Illuminate\Foundation\Application $app
* @param \Illuminate\Contracts\Config\Repository $config
* @param \Pterodactyl\Contracts\Repository\NodeRepositoryInterface $nodeRepository * @param \Pterodactyl\Contracts\Repository\NodeRepositoryInterface $nodeRepository
*/ */
public function __construct( public function __construct(Application $app, NodeRepositoryInterface $nodeRepository)
Application $app, {
ConfigRepository $config,
NodeRepositoryInterface $nodeRepository
) {
$this->app = $app; $this->app = $app;
$this->config = $config;
$this->nodeRepository = $nodeRepository; $this->nodeRepository = $nodeRepository;
} }
/** /**
* {@inheritdoc} * Set the node model to be used for this daemon connection.
*
* @param \Pterodactyl\Models\Node $node
* @return $this
*/ */
public function setNode($id) public function setNode(Node $node)
{ {
Assert::numeric($id, 'The first argument passed to setNode must be numeric, received %s.'); $this->node = $node;
$this->node = $this->nodeRepository->find($id);
return $this; return $this;
} }
/** /**
* {@inheritdoc} * Return the node model being used.
*
* @return \Pterodactyl\Models\Node|null
*/ */
public function getNode() public function getNode()
{ {
@ -86,64 +73,80 @@ class BaseRepository implements BaseRepositoryInterface
} }
/** /**
* {@inheritdoc} * Set the Server model to use when requesting information from the Daemon.
*
* @param \Pterodactyl\Models\Server $server
* @return $this
*/ */
public function setAccessServer($server = null) public function setServer(Server $server)
{ {
Assert::nullOrString($server, 'The first argument passed to setAccessServer must be null or a string, received %s.'); $this->server = $server;
$this->accessServer = $server;
return $this; return $this;
} }
/** /**
* {@inheritdoc} * Return the Server model.
*
* @return \Pterodactyl\Models\Server|null
*/ */
public function getAccessServer() public function getServer()
{ {
return $this->accessServer; return $this->server;
} }
/** /**
* {@inheritdoc} * Set the token to be used in the X-Access-Token header for requests to the daemon.
*
* @param string $token
* @return $this
*/ */
public function setAccessToken($token = null) public function setToken(string $token)
{ {
Assert::nullOrString($token, 'The first argument passed to setAccessToken must be null or a string, received %s.'); $this->token = $token;
$this->accessToken = $token;
return $this; return $this;
} }
/** /**
* {@inheritdoc} * Return the access token being used for requests.
*
* @return string|null
*/ */
public function getAccessToken() public function getToken()
{ {
return $this->accessToken; return $this->token;
} }
/** /**
* {@inheritdoc} * Return an instance of the Guzzle HTTP Client to be used for requests.
*
* @param array $headers
* @return \GuzzleHttp\Client
*/ */
public function getHttpClient(array $headers = []) public function getHttpClient(array $headers = []): Client
{ {
if (! is_null($this->accessServer)) { // If no node is set, load the relationship onto the Server model
$headers['X-Access-Server'] = $this->getAccessServer(); // and pass that to the setNode function.
if (! $this->getNode() instanceof Node) {
if (! $this->getServer() instanceof Server) {
throw new RuntimeException('An instance of ' . Node::class . ' or ' . Server::class . ' must be set on this repository in order to return a client.');
} }
if (! is_null($this->accessToken)) { $this->getServer()->loadMissing('node');
$headers['X-Access-Token'] = $this->getAccessToken(); $this->setNode($this->getServer()->getRelation('node'));
} elseif (! is_null($this->node)) {
$headers['X-Access-Token'] = $this->getNode()->daemonSecret;
} }
if ($this->getServer() instanceof Server) {
$headers['X-Access-Server'] = $this->getServer()->uuid;
}
$headers['X-Access-Token'] = $this->getToken() ?? $this->getNode()->daemonSecret;
return new Client([ return new Client([
'base_uri' => sprintf('%s://%s:%s/v1/', $this->getNode()->scheme, $this->getNode()->fqdn, $this->getNode()->daemonListen), 'base_uri' => sprintf('%s://%s:%s/v1/', $this->getNode()->scheme, $this->getNode()->fqdn, $this->getNode()->daemonListen),
'timeout' => $this->config->get('pterodactyl.guzzle.timeout'), 'timeout' => config('pterodactyl.guzzle.timeout'),
'connect_timeout' => $this->config->get('pterodactyl.guzzle.connect_timeout'), 'connect_timeout' => config('pterodactyl.guzzle.connect_timeout'),
'headers' => $headers, 'headers' => $headers,
]); ]);
} }

View file

@ -1,26 +1,21 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Daemon; namespace Pterodactyl\Repositories\Daemon;
use Webmozart\Assert\Assert; use Psr\Http\Message\ResponseInterface;
use Pterodactyl\Contracts\Repository\Daemon\CommandRepositoryInterface; use Pterodactyl\Contracts\Repository\Daemon\CommandRepositoryInterface;
class CommandRepository extends BaseRepository implements CommandRepositoryInterface class CommandRepository extends BaseRepository implements CommandRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Send a command to a server.
*
* @param string $command
* @return \Psr\Http\Message\ResponseInterface
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function send($command) public function send(string $command): ResponseInterface
{ {
Assert::stringNotEmpty($command, 'First argument passed to send must be a non-empty string, received %s.');
return $this->getHttpClient()->request('POST', 'server/command', [ return $this->getHttpClient()->request('POST', 'server/command', [
'json' => [ 'json' => [
'command' => $command, 'command' => $command,

View file

@ -1,22 +1,19 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Daemon; namespace Pterodactyl\Repositories\Daemon;
use Psr\Http\Message\ResponseInterface;
use Pterodactyl\Contracts\Repository\Daemon\ConfigurationRepositoryInterface; use Pterodactyl\Contracts\Repository\Daemon\ConfigurationRepositoryInterface;
class ConfigurationRepository extends BaseRepository implements ConfigurationRepositoryInterface class ConfigurationRepository extends BaseRepository implements ConfigurationRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Update the configuration details for the specified node using data from the database.
*
* @param array $overrides
* @return \Psr\Http\Message\ResponseInterface
*/ */
public function update(array $overrides = []) public function update(array $overrides = []): ResponseInterface
{ {
$node = $this->getNode(); $node = $this->getNode();
$structure = [ $structure = [

View file

@ -1,23 +1,23 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Daemon; namespace Pterodactyl\Repositories\Daemon;
use Webmozart\Assert\Assert; use stdClass;
use Psr\Http\Message\ResponseInterface;
use Pterodactyl\Contracts\Repository\Daemon\FileRepositoryInterface; use Pterodactyl\Contracts\Repository\Daemon\FileRepositoryInterface;
class FileRepository extends BaseRepository implements FileRepositoryInterface class FileRepository extends BaseRepository implements FileRepositoryInterface
{ {
public function getFileStat($path) /**
* Return stat information for a given file.
*
* @param string $path
* @return \stdClass
*
* @throws \GuzzleHttp\Exception\RequestException
*/
public function getFileStat(string $path): stdClass
{ {
Assert::stringNotEmpty($path, 'First argument passed to getStat must be a non-empty string, received %s.');
$file = pathinfo($path); $file = pathinfo($path);
$file['dirname'] = in_array($file['dirname'], ['.', './', '/']) ? null : trim($file['dirname'], '/') . '/'; $file['dirname'] = in_array($file['dirname'], ['.', './', '/']) ? null : trim($file['dirname'], '/') . '/';
@ -30,12 +30,15 @@ class FileRepository extends BaseRepository implements FileRepositoryInterface
} }
/** /**
* {@inheritdoc} * Return the contents of a given file if it can be edited in the Panel.
*
* @param string $path
* @return \stdClass
*
* @throws \GuzzleHttp\Exception\RequestException
*/ */
public function getContent($path) public function getContent(string $path): stdClass
{ {
Assert::stringNotEmpty($path, 'First argument passed to getContent must be a non-empty string, received %s.');
$file = pathinfo($path); $file = pathinfo($path);
$file['dirname'] = in_array($file['dirname'], ['.', './', '/']) ? null : trim($file['dirname'], '/') . '/'; $file['dirname'] = in_array($file['dirname'], ['.', './', '/']) ? null : trim($file['dirname'], '/') . '/';
@ -48,13 +51,16 @@ class FileRepository extends BaseRepository implements FileRepositoryInterface
} }
/** /**
* {@inheritdoc} * Save new contents to a given file.
*
* @param string $path
* @param string $content
* @return \Psr\Http\Message\ResponseInterface
*
* @throws \GuzzleHttp\Exception\RequestException
*/ */
public function putContent($path, $content) public function putContent(string $path, string $content): ResponseInterface
{ {
Assert::stringNotEmpty($path, 'First argument passed to putContent must be a non-empty string, received %s.');
Assert::string($content, 'Second argument passed to putContent must be a string, received %s.');
$file = pathinfo($path); $file = pathinfo($path);
$file['dirname'] = in_array($file['dirname'], ['.', './', '/']) ? null : trim($file['dirname'], '/') . '/'; $file['dirname'] = in_array($file['dirname'], ['.', './', '/']) ? null : trim($file['dirname'], '/') . '/';
@ -67,20 +73,19 @@ class FileRepository extends BaseRepository implements FileRepositoryInterface
} }
/** /**
* {@inheritdoc} * Return a directory listing for a given path.
*
* @param string $path
* @return array
*
* @throws \GuzzleHttp\Exception\RequestException
*/ */
public function getDirectory($path) public function getDirectory(string $path): array
{ {
Assert::string($path, 'First argument passed to getDirectory must be a string, received %s.'); $response = $this->getHttpClient()->request('GET', sprintf('server/directory/%s', rawurlencode($path)));
$response = $this->getHttpClient()->request('GET', sprintf(
'server/directory/%s',
rawurlencode($path)
));
$contents = json_decode($response->getBody()); $contents = json_decode($response->getBody());
$files = []; $files = $folders = [];
$folders = [];
foreach ($contents as $value) { foreach ($contents as $value) {
if ($value->directory) { if ($value->directory) {

View file

@ -1,27 +1,23 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Daemon; namespace Pterodactyl\Repositories\Daemon;
use Webmozart\Assert\Assert; use Psr\Http\Message\ResponseInterface;
use Pterodactyl\Contracts\Repository\Daemon\PowerRepositoryInterface; use Pterodactyl\Contracts\Repository\Daemon\PowerRepositoryInterface;
use Pterodactyl\Exceptions\Repository\Daemon\InvalidPowerSignalException; use Pterodactyl\Exceptions\Repository\Daemon\InvalidPowerSignalException;
class PowerRepository extends BaseRepository implements PowerRepositoryInterface class PowerRepository extends BaseRepository implements PowerRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Send a power signal to a server.
*
* @param string $signal
* @return \Psr\Http\Message\ResponseInterface
*
* @throws \Pterodactyl\Exceptions\Repository\Daemon\InvalidPowerSignalException
*/ */
public function sendSignal($signal) public function sendSignal(string $signal): ResponseInterface
{ {
Assert::stringNotEmpty($signal, 'The first argument passed to sendSignal must be a non-empty string, received %s.');
switch ($signal) { switch ($signal) {
case self::SIGNAL_START: case self::SIGNAL_START:
case self::SIGNAL_STOP: case self::SIGNAL_STOP:
@ -32,9 +28,8 @@ class PowerRepository extends BaseRepository implements PowerRepositoryInterface
'action' => $signal, 'action' => $signal,
], ],
]); ]);
break;
default: default:
throw new InvalidPowerSignalException('The signal ' . $signal . ' is not defined and could not be processed.'); throw new InvalidPowerSignalException('The signal "' . $signal . '" is not defined and could not be processed.');
} }
} }
} }

View file

@ -1,11 +1,4 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Daemon; namespace Pterodactyl\Repositories\Daemon;
@ -26,9 +19,8 @@ class ServerRepository extends BaseRepository implements ServerRepositoryInterfa
*/ */
public function create(array $structure, array $overrides = []): ResponseInterface public function create(array $structure, array $overrides = []): ResponseInterface
{ {
// Loop through overrides.
foreach ($overrides as $key => $value) { foreach ($overrides as $key => $value) {
array_set($structure, $key, $value); $structure[$key] = value($value);
} }
return $this->getHttpClient()->request('POST', 'servers', [ return $this->getHttpClient()->request('POST', 'servers', [
@ -37,9 +29,12 @@ class ServerRepository extends BaseRepository implements ServerRepositoryInterfa
} }
/** /**
* {@inheritdoc} * Update server details on the daemon.
*
* @param array $data
* @return \Psr\Http\Message\ResponseInterface
*/ */
public function update(array $data) public function update(array $data): ResponseInterface
{ {
return $this->getHttpClient()->request('PATCH', 'server', [ return $this->getHttpClient()->request('PATCH', 'server', [
'json' => $data, 'json' => $data,
@ -47,65 +42,77 @@ class ServerRepository extends BaseRepository implements ServerRepositoryInterfa
} }
/** /**
* {@inheritdoc} * Mark a server to be reinstalled on the system.
*
* @param array|null $data
* @return \Psr\Http\Message\ResponseInterface
*/ */
public function reinstall($data = null) public function reinstall(array $data = null): ResponseInterface
{ {
Assert::nullOrIsArray($data, 'First argument passed to reinstall must be null or an array, received %s.');
if (is_null($data)) {
return $this->getHttpClient()->request('POST', 'server/reinstall');
}
return $this->getHttpClient()->request('POST', 'server/reinstall', [ return $this->getHttpClient()->request('POST', 'server/reinstall', [
'json' => $data, 'json' => $data ?? [],
]); ]);
} }
/** /**
* {@inheritdoc} * Mark a server as needing a container rebuild the next time the server is booted.
*
* @return \Psr\Http\Message\ResponseInterface
*/ */
public function rebuild() public function rebuild(): ResponseInterface
{ {
return $this->getHttpClient()->request('POST', 'server/rebuild'); return $this->getHttpClient()->request('POST', 'server/rebuild');
} }
/** /**
* {@inheritdoc} * Suspend a server on the daemon.
*
* @return \Psr\Http\Message\ResponseInterface
*/ */
public function suspend() public function suspend(): ResponseInterface
{ {
return $this->getHttpClient()->request('POST', 'server/suspend'); return $this->getHttpClient()->request('POST', 'server/suspend');
} }
/** /**
* {@inheritdoc} * Un-suspend a server on the daemon.
*
* @return \Psr\Http\Message\ResponseInterface
*/ */
public function unsuspend() public function unsuspend(): ResponseInterface
{ {
return $this->getHttpClient()->request('POST', 'server/unsuspend'); return $this->getHttpClient()->request('POST', 'server/unsuspend');
} }
/** /**
* {@inheritdoc} * Delete a server on the daemon.
*
* @return \Psr\Http\Message\ResponseInterface
*/ */
public function delete() public function delete(): ResponseInterface
{ {
return $this->getHttpClient()->request('DELETE', 'servers'); return $this->getHttpClient()->request('DELETE', 'servers');
} }
/** /**
* {@inheritdoc} * Return detials on a specific server.
*
* @return \Psr\Http\Message\ResponseInterface
*/ */
public function details() public function details(): ResponseInterface
{ {
return $this->getHttpClient()->request('GET', 'server'); return $this->getHttpClient()->request('GET', 'server');
} }
/** /**
* {@inheritdoc} * Revoke an access key on the daemon before the time is expired.
*
* @param string|array $key
* @return \Psr\Http\Message\ResponseInterface
*
* @throws \GuzzleHttp\Exception\RequestException
*/ */
public function revokeAccessKey($key) public function revokeAccessKey($key): ResponseInterface
{ {
if (is_array($key)) { if (is_array($key)) {
return $this->getHttpClient()->request('POST', 'keys', [ return $this->getHttpClient()->request('POST', 'keys', [

View file

@ -1,21 +1,17 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Illuminate\Support\Collection;
use Pterodactyl\Models\Allocation; use Pterodactyl\Models\Allocation;
use Pterodactyl\Contracts\Repository\AllocationRepositoryInterface; use Pterodactyl\Contracts\Repository\AllocationRepositoryInterface;
class AllocationRepository extends EloquentRepository implements AllocationRepositoryInterface class AllocationRepository extends EloquentRepository implements AllocationRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -23,18 +19,39 @@ class AllocationRepository extends EloquentRepository implements AllocationRepos
} }
/** /**
* {@inheritdoc} * Set an array of allocation IDs to be assigned to a specific server.
*
* @param int|null $server
* @param array $ids
* @return int
*/ */
public function assignAllocationsToServer($server, array $ids) public function assignAllocationsToServer(int $server = null, array $ids): int
{ {
return $this->getBuilder()->whereIn('id', $ids)->update(['server_id' => $server]); return $this->getBuilder()->whereIn('id', $ids)->update(['server_id' => $server]);
} }
/** /**
* {@inheritdoc} * Return all of the allocations for a specific node.
*
* @param int $node
* @return \Illuminate\Support\Collection
*/ */
public function getAllocationsForNode($node) public function getAllocationsForNode(int $node): Collection
{ {
return $this->getBuilder()->where('node_id', $node)->get(); return $this->getBuilder()->where('node_id', $node)->get($this->getColumns());
}
/**
* Return all of the unique IPs that exist for a given node.
*
* @param int $node
* @return \Illuminate\Support\Collection
*/
public function getUniqueAllocationIpsForNode(int $node): Collection
{
return $this->getBuilder()->where('node_id', $node)
->groupBy('ip')
->orderByRaw('INET_ATON(ip) ASC')
->get($this->getColumns());
} }
} }

View file

@ -1,11 +1,4 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
@ -15,7 +8,9 @@ use Pterodactyl\Contracts\Repository\ApiKeyRepositoryInterface;
class ApiKeyRepository extends EloquentRepository implements ApiKeyRepositoryInterface class ApiKeyRepository extends EloquentRepository implements ApiKeyRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {

View file

@ -1,11 +1,4 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
@ -15,7 +8,9 @@ use Pterodactyl\Contracts\Repository\ApiPermissionRepositoryInterface;
class ApiPermissionRepository extends EloquentRepository implements ApiPermissionRepositoryInterface class ApiPermissionRepository extends EloquentRepository implements ApiPermissionRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {

View file

@ -1,26 +1,4 @@
<?php <?php
/*
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
@ -28,13 +6,16 @@ use Pterodactyl\Models\User;
use Webmozart\Assert\Assert; use Webmozart\Assert\Assert;
use Pterodactyl\Models\DaemonKey; use Pterodactyl\Models\DaemonKey;
use Illuminate\Support\Collection; use Illuminate\Support\Collection;
use Illuminate\Database\Eloquent\ModelNotFoundException;
use Pterodactyl\Exceptions\Repository\RecordNotFoundException; use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
use Pterodactyl\Contracts\Repository\DaemonKeyRepositoryInterface; use Pterodactyl\Contracts\Repository\DaemonKeyRepositoryInterface;
class DaemonKeyRepository extends EloquentRepository implements DaemonKeyRepositoryInterface class DaemonKeyRepository extends EloquentRepository implements DaemonKeyRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -62,28 +43,22 @@ class DaemonKeyRepository extends EloquentRepository implements DaemonKeyReposit
} }
/** /**
* {@inheritdoc} * Return a daemon key with the associated server relation attached.
*
* @param string $key
* @return \Pterodactyl\Models\DaemonKey
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getServerKeys($server) public function getKeyWithServer(string $key): DaemonKey
{ {
Assert::integerish($server, 'First argument passed to getServerKeys must be integer, received %s.'); Assert::notEmpty($key, 'Expected non-empty string as first argument passed to ' . __METHOD__);
return $this->getBuilder()->where('server_id', $server)->get($this->getColumns()); try {
} return $this->getBuilder()->with('server')->where('secret', '=', $key)->firstOrFail($this->getColumns());
} catch (ModelNotFoundException $exception) {
/**
* {@inheritdoc}
*/
public function getKeyWithServer($key)
{
Assert::stringNotEmpty($key, 'First argument passed to getServerByKey must be string, received %s.');
$instance = $this->getBuilder()->with('server')->where('secret', '=', $key)->first();
if (is_null($instance)) {
throw new RecordNotFoundException; throw new RecordNotFoundException;
} }
return $instance;
} }
/** /**
@ -95,7 +70,7 @@ class DaemonKeyRepository extends EloquentRepository implements DaemonKeyReposit
*/ */
public function getKeysForRevocation(User $user): Collection public function getKeysForRevocation(User $user): Collection
{ {
return $this->getBuilder()->with('server:id,uuid,node_id')->where('user_id', $user->id)->get($this->getColumns()); return $this->getBuilder()->with('node')->where('user_id', $user->id)->get($this->getColumns());
} }
/** /**

View file

@ -1,23 +1,19 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Webmozart\Assert\Assert; use Illuminate\Support\Collection;
use Pterodactyl\Models\DatabaseHost; use Pterodactyl\Models\DatabaseHost;
use Illuminate\Database\Eloquent\ModelNotFoundException;
use Pterodactyl\Exceptions\Repository\RecordNotFoundException; use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface; use Pterodactyl\Contracts\Repository\DatabaseHostRepositoryInterface;
class DatabaseHostRepository extends EloquentRepository implements DatabaseHostRepositoryInterface class DatabaseHostRepository extends EloquentRepository implements DatabaseHostRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -25,25 +21,31 @@ class DatabaseHostRepository extends EloquentRepository implements DatabaseHostR
} }
/** /**
* {@inheritdoc} * Return database hosts with a count of databases and the node
* information for which it is attached.
*
* @return \Illuminate\Support\Collection
*/ */
public function getWithViewDetails() public function getWithViewDetails(): Collection
{ {
return $this->getBuilder()->withCount('databases')->with('node')->get(); return $this->getBuilder()->withCount('databases')->with('node')->get();
} }
/** /**
* {@inheritdoc} * Return a database host with the databases and associated servers
* that are attached to said databases.
*
* @param int $id
* @return \Pterodactyl\Models\DatabaseHost
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getWithServers($id) public function getWithServers(int $id): DatabaseHost
{ {
Assert::numeric($id, 'First argument passed to getWithServers must be numeric, recieved %s.'); try {
return $this->getBuilder()->with('databases.server')->findOrFail($id, $this->getColumns());
$instance = $this->getBuilder()->with('databases.server')->find($id, $this->getColumns()); } catch (ModelNotFoundException $exception) {
if (! $instance) { throw new RecordNotFoundException;
throw new RecordNotFoundException(); }
}
return $instance;
} }
} }

View file

@ -1,11 +1,4 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
@ -34,17 +27,17 @@ class DatabaseRepository extends EloquentRepository implements DatabaseRepositor
* @param \Illuminate\Foundation\Application $application * @param \Illuminate\Foundation\Application $application
* @param \Illuminate\Database\DatabaseManager $database * @param \Illuminate\Database\DatabaseManager $database
*/ */
public function __construct( public function __construct(Application $application, DatabaseManager $database)
Application $application, {
DatabaseManager $database
) {
parent::__construct($application); parent::__construct($application);
$this->database = $database; $this->database = $database;
} }
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -86,18 +79,24 @@ class DatabaseRepository extends EloquentRepository implements DatabaseRepositor
} }
/** /**
* {@inheritdoc} * Create a new database if it does not already exist on the host with
* @return bool|\Illuminate\Database\Eloquent\Model * the provided details.
*
* @param array $data
* @return \Pterodactyl\Models\Database
*
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\DuplicateDatabaseNameException
*/ */
public function createIfNotExists(array $data) public function createIfNotExists(array $data): Database
{ {
$instance = $this->getBuilder()->where([ $count = $this->getBuilder()->where([
['server_id', '=', array_get($data, 'server_id')], ['server_id', '=', array_get($data, 'server_id')],
['database_host_id', '=', array_get($data, 'database_host_id')], ['database_host_id', '=', array_get($data, 'database_host_id')],
['database', '=', array_get($data, 'database')], ['database', '=', array_get($data, 'database')],
])->count(); ])->count();
if ($instance > 0) { if ($count > 0) {
throw new DuplicateDatabaseNameException('A database with those details already exists for the specified server.'); throw new DuplicateDatabaseNameException('A database with those details already exists for the specified server.');
} }
@ -105,27 +104,40 @@ class DatabaseRepository extends EloquentRepository implements DatabaseRepositor
} }
/** /**
* {@inheritdoc} * Create a new database on a given connection.
*
* @param string $database
* @return bool
*/ */
public function createDatabase($database) public function createDatabase(string $database): bool
{ {
return $this->runStatement(sprintf('CREATE DATABASE IF NOT EXISTS `%s`', $database)); return $this->run(sprintf('CREATE DATABASE IF NOT EXISTS `%s`', $database));
} }
/** /**
* {@inheritdoc} * Create a new database user on a given connection.
*
* @param string $username
* @param string $remote
* @param string $password
* @return bool
*/ */
public function createUser($username, $remote, $password) public function createUser(string $username, string $remote, string $password): bool
{ {
return $this->runStatement(sprintf('CREATE USER `%s`@`%s` IDENTIFIED BY \'%s\'', $username, $remote, $password)); return $this->run(sprintf('CREATE USER `%s`@`%s` IDENTIFIED BY \'%s\'', $username, $remote, $password));
} }
/** /**
* {@inheritdoc} * Give a specific user access to a given database.
*
* @param string $database
* @param string $username
* @param string $remote
* @return bool
*/ */
public function assignUserToDatabase($database, $username, $remote) public function assignUserToDatabase(string $database, string $username, string $remote): bool
{ {
return $this->runStatement(sprintf( return $this->run(sprintf(
'GRANT SELECT, INSERT, UPDATE, DELETE, CREATE, DROP, ALTER, INDEX, EXECUTE ON `%s`.* TO `%s`@`%s`', 'GRANT SELECT, INSERT, UPDATE, DELETE, CREATE, DROP, ALTER, INDEX, EXECUTE ON `%s`.* TO `%s`@`%s`',
$database, $database,
$username, $username,
@ -134,27 +146,36 @@ class DatabaseRepository extends EloquentRepository implements DatabaseRepositor
} }
/** /**
* {@inheritdoc} * Flush the privileges for a given connection.
*
* @return bool
*/ */
public function flush() public function flush(): bool
{ {
return $this->runStatement('FLUSH PRIVILEGES'); return $this->run('FLUSH PRIVILEGES');
} }
/** /**
* {@inheritdoc} * Drop a given database on a specific connection.
*
* @param string $database
* @return bool
*/ */
public function dropDatabase($database) public function dropDatabase(string $database): bool
{ {
return $this->runStatement(sprintf('DROP DATABASE IF EXISTS `%s`', $database)); return $this->run(sprintf('DROP DATABASE IF EXISTS `%s`', $database));
} }
/** /**
* {@inheritdoc} * Drop a given user on a specific connection.
*
* @param string $username
* @param string $remote
* @return mixed
*/ */
public function dropUser($username, $remote) public function dropUser(string $username, string $remote): bool
{ {
return $this->runStatement(sprintf('DROP USER IF EXISTS `%s`@`%s`', $username, $remote)); return $this->run(sprintf('DROP USER IF EXISTS `%s`@`%s`', $username, $remote));
} }
/** /**
@ -163,7 +184,7 @@ class DatabaseRepository extends EloquentRepository implements DatabaseRepositor
* @param string $statement * @param string $statement
* @return bool * @return bool
*/ */
protected function runStatement($statement) private function run(string $statement): bool
{ {
return $this->database->connection($this->getConnection())->statement($statement); return $this->database->connection($this->getConnection())->statement($statement);
} }

View file

@ -1,24 +1,20 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Pterodactyl\Models\Egg; use Pterodactyl\Models\Egg;
use Webmozart\Assert\Assert; use Webmozart\Assert\Assert;
use Illuminate\Database\Eloquent\Collection; use Illuminate\Database\Eloquent\Collection;
use Illuminate\Database\Eloquent\ModelNotFoundException;
use Pterodactyl\Contracts\Repository\EggRepositoryInterface; use Pterodactyl\Contracts\Repository\EggRepositoryInterface;
use Pterodactyl\Exceptions\Repository\RecordNotFoundException; use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
class EggRepository extends EloquentRepository implements EggRepositoryInterface class EggRepository extends EloquentRepository implements EggRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -35,13 +31,11 @@ class EggRepository extends EloquentRepository implements EggRepositoryInterface
*/ */
public function getWithVariables(int $id): Egg public function getWithVariables(int $id): Egg
{ {
/** @var \Pterodactyl\Models\Egg $instance */ try {
$instance = $this->getBuilder()->with('variables')->find($id, $this->getColumns()); return $this->getBuilder()->with('variables')->findOrFail($id, $this->getColumns());
if (! $instance) { } catch (ModelNotFoundException $exception) {
throw new RecordNotFoundException; throw new RecordNotFoundException;
} }
return $instance;
} }
/** /**
@ -67,13 +61,11 @@ class EggRepository extends EloquentRepository implements EggRepositoryInterface
{ {
Assert::true((is_digit($value) || is_string($value)), 'First argument passed to getWithCopyAttributes must be an integer or string, received %s.'); Assert::true((is_digit($value) || is_string($value)), 'First argument passed to getWithCopyAttributes must be an integer or string, received %s.');
/** @var \Pterodactyl\Models\Egg $instance */ try {
$instance = $this->getBuilder()->with('scriptFrom', 'configFrom')->where($column, '=', $value)->first($this->getColumns()); return $this->getBuilder()->with('scriptFrom', 'configFrom')->where($column, '=', $value)->firstOrFail($this->getColumns());
if (! $instance) { } catch (ModelNotFoundException $exception) {
throw new RecordNotFoundException; throw new RecordNotFoundException;
} }
return $instance;
} }
/** /**
@ -86,13 +78,11 @@ class EggRepository extends EloquentRepository implements EggRepositoryInterface
*/ */
public function getWithExportAttributes(int $id): Egg public function getWithExportAttributes(int $id): Egg
{ {
/** @var \Pterodactyl\Models\Egg $instance */ try {
$instance = $this->getBuilder()->with('scriptFrom', 'configFrom', 'variables')->find($id, $this->getColumns()); return $this->getBuilder()->with('scriptFrom', 'configFrom', 'variables')->findOrFail($id, $this->getColumns());
if (! $instance) { } catch (ModelNotFoundException $exception) {
throw new RecordNotFoundException; throw new RecordNotFoundException;
} }
return $instance;
} }
/** /**

View file

@ -1,11 +1,4 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
@ -16,7 +9,9 @@ use Pterodactyl\Contracts\Repository\EggVariableRepositoryInterface;
class EggVariableRepository extends EloquentRepository implements EggVariableRepositoryInterface class EggVariableRepository extends EloquentRepository implements EggVariableRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {

View file

@ -1,17 +1,12 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Webmozart\Assert\Assert; use Webmozart\Assert\Assert;
use Illuminate\Support\Collection;
use Pterodactyl\Repositories\Repository; use Pterodactyl\Repositories\Repository;
use Illuminate\Database\Query\Expression; use Illuminate\Database\Query\Expression;
use Illuminate\Database\Eloquent\ModelNotFoundException;
use Pterodactyl\Contracts\Repository\RepositoryInterface; use Pterodactyl\Contracts\Repository\RepositoryInterface;
use Pterodactyl\Exceptions\Model\DataValidationException; use Pterodactyl\Exceptions\Model\DataValidationException;
use Pterodactyl\Exceptions\Repository\RecordNotFoundException; use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
@ -20,7 +15,19 @@ use Pterodactyl\Contracts\Repository\Attributes\SearchableInterface;
abstract class EloquentRepository extends Repository implements RepositoryInterface abstract class EloquentRepository extends Repository implements RepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return an instance of the eloquent model bound to this
* repository instance.
*
* @return \Illuminate\Database\Eloquent\Model
*/
public function getModel()
{
return $this->model;
}
/**
* Return an instance of the builder to use for this repository.
*
* @return \Illuminate\Database\Eloquent\Builder * @return \Illuminate\Database\Eloquent\Builder
*/ */
public function getBuilder() public function getBuilder()
@ -29,22 +36,19 @@ abstract class EloquentRepository extends Repository implements RepositoryInterf
} }
/** /**
* {@inheritdoc} * Create a new record in the database and return the associated model.
*
* @param array $fields
* @param bool $validate
* @param bool $force * @param bool $force
* @return \Illuminate\Database\Eloquent\Model|bool * @return \Illuminate\Database\Eloquent\Model|bool
*
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
*/ */
public function create(array $fields, $validate = true, $force = false) public function create(array $fields, bool $validate = true, bool $force = false)
{ {
Assert::boolean($validate, 'Second argument passed to create must be boolean, recieved %s.');
Assert::boolean($force, 'Third argument passed to create must be boolean, received %s.');
$instance = $this->getBuilder()->newModelInstance(); $instance = $this->getBuilder()->newModelInstance();
($force) ? $instance->forceFill($fields) : $instance->fill($fields);
if ($force) {
$instance->forceFill($fields);
} else {
$instance->fill($fields);
}
if (! $validate) { if (! $validate) {
$saved = $instance->skipValidation()->save(); $saved = $instance->skipValidation()->save();
@ -58,99 +62,108 @@ abstract class EloquentRepository extends Repository implements RepositoryInterf
} }
/** /**
* {@inheritdoc} * Find a model that has the specific ID passed.
* @return \Illuminate\Database\Eloquent\Builder|\Illuminate\Database\Eloquent\Model *
* @param int $id
* @return \Illuminate\Database\Eloquent\Model
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function find($id) public function find(int $id)
{ {
Assert::numeric($id, 'First argument passed to find must be numeric, received %s.'); try {
return $this->getBuilder()->findOrFail($id, $this->getColumns());
$instance = $this->getBuilder()->find($id, $this->getColumns()); } catch (ModelNotFoundException $exception) {
throw new RecordNotFoundException;
if (! $instance) {
throw new RecordNotFoundException();
} }
return $instance;
} }
/** /**
* {@inheritdoc} * Find a model matching an array of where clauses.
* @return \Illuminate\Database\Eloquent\Collection *
* @param array $fields
* @return \Illuminate\Support\Collection
*/ */
public function findWhere(array $fields) public function findWhere(array $fields): Collection
{ {
return $this->getBuilder()->where($fields)->get($this->getColumns()); return $this->getBuilder()->where($fields)->get($this->getColumns());
} }
/** /**
* {@inheritdoc} * Find and return the first matching instance for the given fields.
*
* @param array $fields
* @return \Illuminate\Database\Eloquent\Model * @return \Illuminate\Database\Eloquent\Model
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function findFirstWhere(array $fields) public function findFirstWhere(array $fields)
{ {
$instance = $this->getBuilder()->where($fields)->first($this->getColumns()); try {
return $this->getBuilder()->where($fields)->firstOrFail($this->getColumns());
if (! $instance) { } catch (ModelNotFoundException $exception) {
throw new RecordNotFoundException; throw new RecordNotFoundException;
} }
return $instance;
} }
/** /**
* {@inheritdoc}. * Return a count of records matching the passed arguments.
*
* @param array $fields
* @return int
*/ */
public function findCountWhere(array $fields) public function findCountWhere(array $fields): int
{ {
return $this->getBuilder()->where($fields)->count($this->getColumns()); return $this->getBuilder()->where($fields)->count($this->getColumns());
} }
/** /**
* {@inheritdoc} * Delete a given record from the database.
*
* @param int $id
* @param bool $destroy
* @return int
*/ */
public function delete($id, $destroy = false) public function delete(int $id, bool $destroy = false): int
{ {
Assert::numeric($id, 'First argument passed to delete must be numeric, received %s.'); return $this->deleteWhere(['id' => $id], $destroy);
Assert::boolean($destroy, 'Second argument passed to delete must be boolean, received %s.');
$instance = $this->getBuilder()->where($this->getModel()->getKeyName(), $id);
return ($destroy) ? $instance->forceDelete() : $instance->delete();
} }
/** /**
* {@inheritdoc} * Delete records matching the given attributes.
*
* @param array $attributes
* @param bool $force
* @return int
*/ */
public function deleteWhere(array $attributes, $force = false) public function deleteWhere(array $attributes, bool $force = false): int
{ {
Assert::boolean($force, 'Second argument passed to deleteWhere must be boolean, received %s.');
$instance = $this->getBuilder()->where($attributes); $instance = $this->getBuilder()->where($attributes);
return ($force) ? $instance->forceDelete() : $instance->delete(); return ($force) ? $instance->forceDelete() : $instance->delete();
} }
/** /**
* {@inheritdoc} * Update a given ID with the passed array of fields.
*
* @param int $id
* @param array $fields
* @param bool $validate
* @param bool $force
* @return \Illuminate\Database\Eloquent\Model|bool
*
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function update($id, array $fields, $validate = true, $force = false) public function update($id, array $fields, bool $validate = true, bool $force = false)
{ {
Assert::numeric($id, 'First argument passed to update must be numeric, received %s.'); try {
Assert::boolean($validate, 'Third argument passed to update must be boolean, received %s.'); $instance = $this->getBuilder()->where('id', $id)->firstOrFail();
Assert::boolean($force, 'Fourth argument passed to update must be boolean, received %s.'); } catch (ModelNotFoundException $exception) {
throw new RecordNotFoundException;
$instance = $this->getBuilder()->where('id', $id)->first();
if (! $instance) {
throw new RecordNotFoundException();
} }
if ($force) { ($force) ? $instance->forceFill($fields) : $instance->fill($fields);
$instance->forceFill($fields);
} else {
$instance->fill($fields);
}
if (! $validate) { if (! $validate) {
$saved = $instance->skipValidation()->save(); $saved = $instance->skipValidation()->save();
@ -164,46 +177,71 @@ abstract class EloquentRepository extends Repository implements RepositoryInterf
} }
/** /**
* {@inheritdoc} * Perform a mass update where matching records are updated using whereIn.
* This does not perform any model data validation.
*
* @param string $column
* @param array $values
* @param array $fields
* @return int
*/ */
public function updateWhereIn($column, array $values, array $fields) public function updateWhereIn(string $column, array $values, array $fields): int
{ {
Assert::stringNotEmpty($column, 'First argument passed to updateWhereIn must be a non-empty string, received %s.'); Assert::notEmpty($column, 'First argument passed to updateWhereIn must be a non-empty string.');
return $this->getBuilder()->whereIn($column, $values)->update($fields); return $this->getBuilder()->whereIn($column, $values)->update($fields);
} }
/** /**
* {@inheritdoc} * Update a record if it exists in the database, otherwise create it.
*
* @param array $where
* @param array $fields
* @param bool $validate
* @param bool $force
* @return \Illuminate\Database\Eloquent\Model
*
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function massUpdate(array $where, array $fields) public function updateOrCreate(array $where, array $fields, bool $validate = true, bool $force = false)
{ {
// TODO: Implement massUpdate() method. foreach ($where as $item) {
Assert::true(is_scalar($item) || is_null($item), 'First argument passed to updateOrCreate should be an array of scalar or null values, received an array value of %s.');
}
try {
$instance = $this->setColumns('id')->findFirstWhere($where);
} catch (RecordNotFoundException $exception) {
return $this->create(array_merge($where, $fields), $validate, $force);
}
return $this->update($instance->id, $fields, $validate, $force);
} }
/** /**
* {@inheritdoc} * Return all records associated with the given model.
*
* @return \Illuminate\Support\Collection
*/ */
public function all($paginate = null) public function all(): Collection
{ {
Assert::nullOrIntegerish($paginate, 'First argument passed to all must be null or integer, received %s.');
$instance = $this->getBuilder(); $instance = $this->getBuilder();
if (is_subclass_of(get_called_class(), SearchableInterface::class)) { if (is_subclass_of(get_called_class(), SearchableInterface::class) && $this->hasSearchTerm()) {
$instance = $instance->search($this->searchTerm); $instance = $instance->search($this->getSearchTerm());
} }
if (is_null($paginate)) {
return $instance->get($this->getColumns()); return $instance->get($this->getColumns());
} }
return $instance->paginate($paginate, $this->getColumns());
}
/** /**
* {@inheritdoc} * Insert a single or multiple records into the database at once skipping
* validation and mass assignment checking.
*
* @param array $data
* @return bool
*/ */
public function insert(array $data) public function insert(array $data): bool
{ {
return $this->getBuilder()->insert($data); return $this->getBuilder()->insert($data);
} }
@ -214,20 +252,16 @@ abstract class EloquentRepository extends Repository implements RepositoryInterf
* @param array $values * @param array $values
* @return bool * @return bool
*/ */
public function insertIgnore(array $values) public function insertIgnore(array $values): bool
{ {
if (empty($values)) { if (empty($values)) {
return true; return true;
} }
if (! is_array(reset($values))) {
$values = [$values];
} else {
foreach ($values as $key => $value) { foreach ($values as $key => $value) {
ksort($value); ksort($value);
$values[$key] = $value; $values[$key] = $value;
} }
}
$bindings = array_values(array_filter(array_flatten($values, 1), function ($binding) { $bindings = array_values(array_filter(array_flatten($values, 1), function ($binding) {
return ! $binding instanceof Expression; return ! $binding instanceof Expression;
@ -245,26 +279,4 @@ abstract class EloquentRepository extends Repository implements RepositoryInterf
return $this->getBuilder()->getConnection()->statement($statement, $bindings); return $this->getBuilder()->getConnection()->statement($statement, $bindings);
} }
/**
* {@inheritdoc}
* @return bool|\Illuminate\Database\Eloquent\Model
*/
public function updateOrCreate(array $where, array $fields, $validate = true, $force = false)
{
Assert::boolean($validate, 'Third argument passed to updateOrCreate must be boolean, received %s.');
Assert::boolean($force, 'Fourth argument passed to updateOrCreate must be boolean, received %s.');
foreach ($where as $item) {
Assert::true(is_scalar($item) || is_null($item), 'First argument passed to updateOrCreate should be an array of scalar or null values, received an array value of %s.');
}
$instance = $this->withColumns('id')->findWhere($where)->first();
if (! $instance) {
return $this->create(array_merge($where, $fields), $validate, $force);
}
return $this->update($instance->id, $fields, $validate, $force);
}
} }

View file

@ -1,16 +1,11 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Pterodactyl\Models\Location; use Pterodactyl\Models\Location;
use Illuminate\Support\Collection;
use Pterodactyl\Repositories\Concerns\Searchable; use Pterodactyl\Repositories\Concerns\Searchable;
use Illuminate\Database\Eloquent\ModelNotFoundException;
use Pterodactyl\Exceptions\Repository\RecordNotFoundException; use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
use Pterodactyl\Contracts\Repository\LocationRepositoryInterface; use Pterodactyl\Contracts\Repository\LocationRepositoryInterface;
@ -19,7 +14,9 @@ class LocationRepository extends EloquentRepository implements LocationRepositor
use Searchable; use Searchable;
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -27,44 +24,56 @@ class LocationRepository extends EloquentRepository implements LocationRepositor
} }
/** /**
* {@inheritdoc} * Return locations with a count of nodes and servers attached to it.
*
* @return \Illuminate\Support\Collection
*/ */
public function getAllWithDetails() public function getAllWithDetails(): Collection
{ {
return $this->getBuilder()->withCount('nodes', 'servers')->get($this->getColumns()); return $this->getBuilder()->withCount('nodes', 'servers')->get($this->getColumns());
} }
/** /**
* {@inheritdoc} * Return all of the available locations with the nodes as a relationship.
*
* @return \Illuminate\Support\Collection
*/ */
public function getAllWithNodes() public function getAllWithNodes(): Collection
{ {
return $this->getBuilder()->with('nodes')->get($this->getColumns()); return $this->getBuilder()->with('nodes')->get($this->getColumns());
} }
/** /**
* {@inheritdoc} * Return all of the nodes and their respective count of servers for a location.
*
* @param int $id
* @return mixed
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getWithNodes($id) public function getWithNodes(int $id): Location
{ {
$instance = $this->getBuilder()->with('nodes.servers')->find($id, $this->getColumns()); try {
if (! $instance) { return $this->getBuilder()->with('nodes.servers')->findOrFail($id, $this->getColumns());
} catch (ModelNotFoundException $exception) {
throw new RecordNotFoundException; throw new RecordNotFoundException;
} }
return $instance;
} }
/** /**
* {@inheritdoc} * Return a location and the count of nodes in that location.
*
* @param int $id
* @return mixed
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getWithNodeCount($id) public function getWithNodeCount(int $id): Location
{ {
$instance = $this->getBuilder()->withCount('nodes')->find($id, $this->getColumns()); try {
if (! $instance) { return $this->getBuilder()->withCount('nodes')->findOrFail($id, $this->getColumns());
} catch (ModelNotFoundException $exception) {
throw new RecordNotFoundException; throw new RecordNotFoundException;
} }
return $instance;
} }
} }

View file

@ -16,7 +16,9 @@ use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
class NestRepository extends EloquentRepository implements NestRepositoryInterface class NestRepository extends EloquentRepository implements NestRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {

View file

@ -1,16 +1,12 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Pterodactyl\Models\Node; use Pterodactyl\Models\Node;
use Illuminate\Support\Collection;
use Pterodactyl\Repositories\Concerns\Searchable; use Pterodactyl\Repositories\Concerns\Searchable;
use Illuminate\Database\Eloquent\ModelNotFoundException;
use Illuminate\Contracts\Pagination\LengthAwarePaginator;
use Pterodactyl\Contracts\Repository\NodeRepositoryInterface; use Pterodactyl\Contracts\Repository\NodeRepositoryInterface;
use Pterodactyl\Exceptions\Repository\RecordNotFoundException; use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
@ -18,11 +14,10 @@ class NodeRepository extends EloquentRepository implements NodeRepositoryInterfa
{ {
use Searchable; use Searchable;
const THRESHOLD_PERCENTAGE_LOW = 75;
const THRESHOLD_PERCENTAGE_MEDIUM = 90;
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -30,23 +25,18 @@ class NodeRepository extends EloquentRepository implements NodeRepositoryInterfa
} }
/** /**
* {@inheritdoc} * Return the usage stats for a single node.
*
* @param \Pterodactyl\Models\Node $node
* @return array
*/ */
public function getUsageStats($id) public function getUsageStats(Node $node): array
{ {
$node = $this->getBuilder()->select([
'nodes.disk_overallocate',
'nodes.memory_overallocate',
'nodes.disk',
'nodes.memory',
])->where('id', $id)->first();
$stats = $this->getBuilder()->select( $stats = $this->getBuilder()->select(
$this->getBuilder()->raw('IFNULL(SUM(servers.memory), 0) as sum_memory, IFNULL(SUM(servers.disk), 0) as sum_disk') $this->getBuilder()->raw('IFNULL(SUM(servers.memory), 0) as sum_memory, IFNULL(SUM(servers.disk), 0) as sum_disk')
)->join('servers', 'servers.node_id', '=', 'nodes.id')->where('node_id', $id)->first(); )->join('servers', 'servers.node_id', '=', 'nodes.id')->where('node_id', $node->id)->first();
return collect(['disk' => $stats->sum_disk, 'memory' => $stats->sum_memory]) return collect(['disk' => $stats->sum_disk, 'memory' => $stats->sum_memory])->mapWithKeys(function ($value, $key) use ($node) {
->mapWithKeys(function ($value, $key) use ($node) {
$maxUsage = $node->{$key}; $maxUsage = $node->{$key};
if ($node->{$key . '_overallocate'} > 0) { if ($node->{$key . '_overallocate'} > 0) {
$maxUsage = $node->{$key} * (1 + ($node->{$key . '_overallocate'} / 100)); $maxUsage = $node->{$key} * (1 + ($node->{$key . '_overallocate'} / 100));
@ -62,81 +52,94 @@ class NodeRepository extends EloquentRepository implements NodeRepositoryInterfa
'css' => ($percent <= self::THRESHOLD_PERCENTAGE_LOW) ? 'green' : (($percent > self::THRESHOLD_PERCENTAGE_MEDIUM) ? 'red' : 'yellow'), 'css' => ($percent <= self::THRESHOLD_PERCENTAGE_LOW) ? 'green' : (($percent > self::THRESHOLD_PERCENTAGE_MEDIUM) ? 'red' : 'yellow'),
], ],
]; ];
}) })->toArray();
->toArray();
} }
/** /**
* {@inheritdoc} * Return all available nodes with a searchable interface.
*
* @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
*/ */
public function getNodeListingData($count = 25) public function getNodeListingData(): LengthAwarePaginator
{ {
$instance = $this->getBuilder()->with('location')->withCount('servers'); $instance = $this->getBuilder()->with('location')->withCount('servers');
if ($this->searchTerm) { if ($this->hasSearchTerm()) {
$instance->search($this->searchTerm); $instance->setSearchTerm($this->getSearchTerm());
} }
return $instance->paginate($count, $this->getColumns()); return $instance->paginate(25, $this->getColumns());
} }
/** /**
* {@inheritdoc} * Return a single node with location and server information.
*
* @param \Pterodactyl\Models\Node $node
* @param bool $refresh
* @return \Pterodactyl\Models\Node
*/ */
public function getSingleNode($id) public function loadLocationAndServerCount(Node $node, bool $refresh = false): Node
{ {
$instance = $this->getBuilder()->with('location')->withCount('servers')->find($id, $this->getColumns()); if (! $node->relationLoaded('location') || $refresh) {
$node->load('location');
if (! $instance) {
throw new RecordNotFoundException();
} }
return $instance; // This is quite ugly and can probably be improved down the road.
// And by probably, I mean it should.
if (is_null($node->servers_count) || $refresh) {
$node->load('servers');
$node->setRelation('servers_count', count($node->getRelation('servers')));
unset($node->servers);
}
return $node;
} }
/** /**
* {@inheritdoc} * Attach a paginated set of allocations to a node mode including
* any servers that are also attached to those allocations.
*
* @param \Pterodactyl\Models\Node $node
* @param bool $refresh
* @return \Pterodactyl\Models\Node
*/ */
public function getNodeAllocations($id) public function loadNodeAllocations(Node $node, bool $refresh = false): Node
{ {
$instance = $this->getBuilder()->find($id, $this->getColumns()); $node->setRelation('allocations',
$node->allocations()->orderByRaw('server_id IS NOT NULL DESC, server_id IS NULL')->orderByRaw('INET_ATON(ip) ASC')->orderBy('port', 'asc')->with('server:id,name')->paginate(50)
if (! $instance) {
throw new RecordNotFoundException();
}
$instance->setRelation(
'allocations',
$instance->allocations()->orderBy('ip', 'asc')->orderBy('port', 'asc')->with('server')->paginate(50)
); );
return $instance; return $node;
} }
/** /**
* {@inheritdoc} * 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($id) public function getNodeServers(int $id): Node
{ {
$instance = $this->getBuilder()->with('servers.user', 'servers.nest', 'servers.egg') try {
->find($id, $this->getColumns()); return $this->getBuilder()->with([
'servers.user', 'servers.nest', 'servers.egg',
if (! $instance) { ])->findOrFail($id, $this->getColumns());
throw new RecordNotFoundException(); } catch (ModelNotFoundException $exception) {
throw new RecordNotFoundException;
} }
return $instance;
} }
/** /**
* {@inheritdoc} * Return a collection of nodes for all locations to use in server creation UI.
*
* @return \Illuminate\Support\Collection
*/ */
public function getNodesForServerCreation() public function getNodesForServerCreation(): Collection
{ {
$instance = $this->getBuilder()->with('allocations')->get(); return $this->getBuilder()->with('allocations')->get()->map(function (Node $item) {
$filtered = $item->getRelation('allocations')->where('server_id', null)->map(function ($map) {
return $instance->map(function ($item) {
$filtered = $item->allocations->where('server_id', null)->map(function ($map) {
return collect($map)->only(['id', 'ip', 'port']); return collect($map)->only(['id', 'ip', 'port']);
}); });

View file

@ -1,27 +1,20 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Pterodactyl\Models\Pack; use Pterodactyl\Models\Pack;
use Webmozart\Assert\Assert;
use Pterodactyl\Repositories\Concerns\Searchable; use Pterodactyl\Repositories\Concerns\Searchable;
use Illuminate\Database\Eloquent\ModelNotFoundException; use Illuminate\Contracts\Pagination\LengthAwarePaginator;
use Pterodactyl\Contracts\Repository\PackRepositoryInterface; use Pterodactyl\Contracts\Repository\PackRepositoryInterface;
use Illuminate\Contracts\Filesystem\Factory as FilesystemFactory;
class PackRepository extends EloquentRepository implements PackRepositoryInterface class PackRepository extends EloquentRepository implements PackRepositoryInterface
{ {
use Searchable; use Searchable;
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -29,58 +22,32 @@ class PackRepository extends EloquentRepository implements PackRepositoryInterfa
} }
/** /**
* {@inheritdoc} * Return a pack with the associated server models attached to it.
*
* @param \Pterodactyl\Models\Pack $pack
* @param bool $refresh
* @return \Pterodactyl\Models\Pack
*/ */
public function getFileArchives($id, $collection = false) public function loadServerData(Pack $pack, bool $refresh = false): Pack
{ {
Assert::numeric($id, 'First argument passed to getFileArchives must be numeric, received %s.'); if ($refresh) {
Assert::boolean($collection, 'Second argument passed to getFileArchives must be boolean, received %s.'); $pack->load(['servers.node', 'servers.user']);
$pack = $this->getBuilder()->find($id, ['id', 'uuid']);
if (! $pack) {
throw new ModelNotFoundException;
} }
$storage = $this->app->make(FilesystemFactory::class); $pack->loadMissing(['servers.node', 'servers.user']);
$files = collect($storage->disk('default')->files('packs/' . $pack->uuid));
$files = $files->map(function ($file) { return $pack;
$path = storage_path('app/' . $file);
return (object) [
'name' => basename($file),
'hash' => sha1_file($path),
'size' => human_readable($path),
];
});
return ($collection) ? $files : (object) $files->all();
} }
/** /**
* {@inheritdoc} * Return a paginated listing of packs with their associated egg and server count.
*
* @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
*/ */
public function getWithServers($id) public function paginateWithEggAndServerCount(): LengthAwarePaginator
{ {
Assert::numeric($id, 'First argument passed to getWithServers must be numeric, received %s.');
$instance = $this->getBuilder()->with('servers.node', 'servers.user')->find($id, $this->getColumns());
if (! $instance) {
throw new ModelNotFoundException;
}
return $instance;
}
/**
* {@inheritdoc}
*/
public function paginateWithEggAndServerCount($paginate = 50)
{
Assert::integer($paginate, 'First argument passed to paginateWithOptionAndServerCount must be integer, received %s.');
return $this->getBuilder()->with('egg')->withCount('servers') return $this->getBuilder()->with('egg')->withCount('servers')
->search($this->searchTerm) ->search($this->getSearchTerm())
->paginate($paginate, $this->getColumns()); ->paginate(50, $this->getColumns());
} }
} }

View file

@ -1,11 +1,4 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
@ -15,7 +8,9 @@ use Pterodactyl\Contracts\Repository\PermissionRepositoryInterface;
class PermissionRepository extends EloquentRepository implements PermissionRepositoryInterface class PermissionRepository extends EloquentRepository implements PermissionRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {

View file

@ -1,23 +1,19 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Pterodactyl\Models\Schedule; use Pterodactyl\Models\Schedule;
use Illuminate\Support\Collection; use Illuminate\Support\Collection;
use Illuminate\Database\Eloquent\ModelNotFoundException;
use Pterodactyl\Exceptions\Repository\RecordNotFoundException; use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
use Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface; use Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface;
class ScheduleRepository extends EloquentRepository implements ScheduleRepositoryInterface class ScheduleRepository extends EloquentRepository implements ScheduleRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -35,6 +31,23 @@ class ScheduleRepository extends EloquentRepository implements ScheduleRepositor
return $this->getBuilder()->withCount('tasks')->where('server_id', '=', $server)->get($this->getColumns()); return $this->getBuilder()->withCount('tasks')->where('server_id', '=', $server)->get($this->getColumns());
} }
/**
* Load the tasks relationship onto the Schedule module if they are not
* already present.
*
* @param \Pterodactyl\Models\Schedule $schedule
* @param bool $refresh
* @return \Pterodactyl\Models\Schedule
*/
public function loadTasks(Schedule $schedule, bool $refresh = false): Schedule
{
if (! $schedule->relationLoaded('tasks') || $refresh) {
$schedule->load('tasks');
}
return $schedule;
}
/** /**
* Return a schedule model with all of the associated tasks as a relationship. * Return a schedule model with all of the associated tasks as a relationship.
* *
@ -45,19 +58,20 @@ class ScheduleRepository extends EloquentRepository implements ScheduleRepositor
*/ */
public function getScheduleWithTasks(int $schedule): Schedule public function getScheduleWithTasks(int $schedule): Schedule
{ {
/** @var \Pterodactyl\Models\Schedule $instance */ try {
$instance = $this->getBuilder()->with('tasks')->find($schedule, $this->getColumns()); return $this->getBuilder()->with('tasks')->findOrFail($schedule, $this->getColumns());
if (! $instance) { } catch (ModelNotFoundException $exception) {
throw new RecordNotFoundException; throw new RecordNotFoundException;
} }
return $instance;
} }
/** /**
* {@inheritdoc} * Return all of the schedules that should be processed.
*
* @param string $timestamp
* @return \Illuminate\Support\Collection
*/ */
public function getSchedulesToProcess($timestamp) public function getSchedulesToProcess(string $timestamp): Collection
{ {
return $this->getBuilder()->with('tasks') return $this->getBuilder()->with('tasks')
->where('is_active', true) ->where('is_active', true)

View file

@ -1,17 +1,14 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Pterodactyl\Models\User;
use Webmozart\Assert\Assert; use Webmozart\Assert\Assert;
use Pterodactyl\Models\Server; use Pterodactyl\Models\Server;
use Illuminate\Support\Collection;
use Pterodactyl\Repositories\Concerns\Searchable; use Pterodactyl\Repositories\Concerns\Searchable;
use Illuminate\Database\Eloquent\ModelNotFoundException;
use Illuminate\Contracts\Pagination\LengthAwarePaginator;
use Pterodactyl\Exceptions\Repository\RecordNotFoundException; use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
use Pterodactyl\Contracts\Repository\ServerRepositoryInterface; use Pterodactyl\Contracts\Repository\ServerRepositoryInterface;
@ -20,7 +17,9 @@ class ServerRepository extends EloquentRepository implements ServerRepositoryInt
use Searchable; use Searchable;
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -28,15 +27,16 @@ class ServerRepository extends EloquentRepository implements ServerRepositoryInt
} }
/** /**
* {@inheritdoc} * Returns a listing of all servers that exist including relationships.
*
* @param int $paginate
* @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
*/ */
public function getAllServers($paginate = 25) public function getAllServers(int $paginate): LengthAwarePaginator
{ {
Assert::nullOrIntegerish($paginate, 'First argument passed to getAllServers must be integer or null, received %s.'); $instance = $this->getBuilder()->with('node', 'user', 'allocation')->search($this->getSearchTerm());
$instance = $this->getBuilder()->with('node', 'user', 'allocation')->search($this->searchTerm); return $instance->paginate($paginate, $this->getColumns());
return is_null($paginate) ? $instance->get($this->getColumns()) : $instance->paginate($paginate, $this->getColumns());
} }
/** /**
@ -56,14 +56,15 @@ class ServerRepository extends EloquentRepository implements ServerRepositoryInt
} }
/** /**
* {@inheritdoc} * Return a collection of servers with their associated data for rebuild operations.
*
* @param int|null $server
* @param int|null $node
* @return \Illuminate\Support\Collection
*/ */
public function getDataForRebuild($server = null, $node = null) public function getDataForRebuild(int $server = null, int $node = null): Collection
{ {
Assert::nullOrIntegerish($server, 'First argument passed to getDataForRebuild must be null or integer, received %s.'); $instance = $this->getBuilder()->with(['allocation', 'allocations', 'pack', 'egg', 'node']);
Assert::nullOrIntegerish($node, 'Second argument passed to getDataForRebuild must be null or integer, received %s.');
$instance = $this->getBuilder()->with('allocation', 'allocations', 'pack', 'egg', 'node');
if (! is_null($server) && is_null($node)) { if (! is_null($server) && is_null($node)) {
$instance = $instance->where('id', '=', $server); $instance = $instance->where('id', '=', $server);
@ -75,22 +76,22 @@ class ServerRepository extends EloquentRepository implements ServerRepositoryInt
} }
/** /**
* {@inheritdoc} * Return a server model and all variables associated with the server.
* @return \Illuminate\Database\Eloquent\Model *
* @param int $id
* @return \Pterodactyl\Models\Server
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function findWithVariables($id) public function findWithVariables(int $id): Server
{ {
Assert::integerish($id, 'First argument passed to findWithVariables must be integer, received %s.'); try {
return $this->getBuilder()->with('egg.variables', 'variables')
$instance = $this->getBuilder()->with('egg.variables', 'variables')
->where($this->getModel()->getKeyName(), '=', $id) ->where($this->getModel()->getKeyName(), '=', $id)
->first($this->getColumns()); ->firstOrFail($this->getColumns());
} catch (ModelNotFoundException $exception) {
if (is_null($instance)) { throw new RecordNotFoundException;
throw new RecordNotFoundException();
} }
return $instance;
} }
/** /**
@ -98,51 +99,45 @@ class ServerRepository extends EloquentRepository implements ServerRepositoryInt
* the function, load the allocation relationship onto it. Otherwise, find and * the function, load the allocation relationship onto it. Otherwise, find and
* return the server from the database. * return the server from the database.
* *
* @param int|\Pterodactyl\Models\Server $server * @param \Pterodactyl\Models\Server $server
* @param bool $refresh * @param bool $refresh
* @return \Pterodactyl\Models\Server * @return \Pterodactyl\Models\Server
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getPrimaryAllocation($server, bool $refresh = false): Server public function getPrimaryAllocation(Server $server, bool $refresh = false): Server
{ {
$instance = $server; if (! $server->relationLoaded('allocation') || $refresh) {
if (! $instance instanceof Server) { $server->load('allocation');
Assert::integerish($server, 'First argument passed to getPrimaryAllocation must be instance of \Pterodactyl\Models\Server or integer, received %s.');
$instance = $this->getBuilder()->find($server, $this->getColumns());
} }
if (! $instance) { return $server;
throw new RecordNotFoundException;
}
if (! $instance->relationLoaded('allocation') || $refresh) {
$instance->load('allocation');
}
return $instance;
} }
/** /**
* {@inheritdoc} * Return all of the server variables possible and default to the variable
* default if there is no value defined for the specific server requested.
*
* @param int $id
* @param bool $returnAsObject
* @return array|object
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getVariablesWithValues($id, $returnWithObject = false) public function getVariablesWithValues(int $id, bool $returnAsObject = false)
{ {
$instance = $this->getBuilder()->with('variables', 'egg.variables') try {
->find($id, $this->getColumns()); $instance = $this->getBuilder()->with('variables', 'egg.variables')->find($id, $this->getColumns());
} catch (ModelNotFoundException $exception) {
if (! $instance) { throw new RecordNotFoundException;
throw new RecordNotFoundException();
} }
$data = []; $data = [];
$instance->egg->variables->each(function ($item) use (&$data, $instance) { $instance->getRelation('egg')->getRelation('variables')->each(function ($item) use (&$data, $instance) {
$display = $instance->variables->where('variable_id', $item->id)->pluck('variable_value')->first(); $display = $instance->getRelation('variables')->where('variable_id', $item->id)->pluck('variable_value')->first();
$data[$item->env_variable] = $display ?? $item->default_value; $data[$item->env_variable] = $display ?? $item->default_value;
}); });
if ($returnWithObject) { if ($returnAsObject) {
return (object) [ return (object) [
'data' => $data, 'data' => $data,
'server' => $instance, 'server' => $instance,
@ -171,19 +166,19 @@ class ServerRepository extends EloquentRepository implements ServerRepositoryInt
} }
/** /**
* {@inheritdoc} * Load associated databases onto the server model.
*
* @param \Pterodactyl\Models\Server $server
* @param bool $refresh
* @return \Pterodactyl\Models\Server
*/ */
public function getWithDatabases($id) public function loadDatabaseRelations(Server $server, bool $refresh = false): Server
{ {
$instance = $this->getBuilder()->with('databases.host') if (! $server->relationLoaded('databases') || $refresh) {
->where('installed', 1) $server->load('databases.host');
->find($id, $this->getColumns());
if (! $instance) {
throw new RecordNotFoundException();
} }
return $instance; return $server;
} }
/** /**
@ -212,71 +207,70 @@ class ServerRepository extends EloquentRepository implements ServerRepositoryInt
} }
/** /**
* {@inheritdoc} * Return a paginated list of servers that a user can access at a given level.
*
* @param \Pterodactyl\Models\User $user
* @param int $level
* @return \Illuminate\Pagination\LengthAwarePaginator
*/ */
public function getUserAccessServers($user) public function filterUserAccessServers(User $user, int $level): LengthAwarePaginator
{ {
Assert::numeric($user, 'First argument passed to getUserAccessServers must be numeric, received %s.'); $instance = $this->getBuilder()->with(['user']);
$subuser = $this->app->make(SubuserRepository::class);
return $this->getBuilder()->select('id')->where('owner_id', $user)->union(
$subuser->getBuilder()->select('server_id')->where('user_id', $user)
)->pluck('id')->all();
}
/**
* {@inheritdoc}
*/
public function filterUserAccessServers($user, $admin = false, $level = 'all', array $relations = [])
{
Assert::numeric($user, 'First argument passed to filterUserAccessServers must be numeric, received %s.');
Assert::boolean($admin, 'Second argument passed to filterUserAccessServers must be boolean, received %s.');
Assert::stringNotEmpty($level, 'Third argument passed to filterUserAccessServers must be a non-empty string, received %s.');
$instance = $this->getBuilder()->with($relations);
// If access level is set to owner, only display servers // If access level is set to owner, only display servers
// that the user owns. // that the user owns.
if ($level === 'owner') { if ($level === User::FILTER_LEVEL_OWNER) {
$instance->where('owner_id', $user); $instance->where('owner_id', $user->id);
} }
// If set to all, display all servers they can access, including // If set to all, display all servers they can access, including
// those they access as an admin. // those they access as an admin. If set to subuser, only return the servers they can access because
//
// If set to subuser, only return the servers they can access because
// they are owner, or marked as a subuser of the server. // they are owner, or marked as a subuser of the server.
if (($level === 'all' && ! $admin) || $level === 'subuser') { elseif (($level === User::FILTER_LEVEL_ALL && ! $user->root_admin) || $level === User::FILTER_LEVEL_SUBUSER) {
$instance->whereIn('id', $this->getUserAccessServers($user)); $instance->whereIn('id', $this->getUserAccessServers($user->id));
} }
// If set to admin, only display the servers a user can access // If set to admin, only display the servers a user can access
// as an administrator (leaves out owned and subuser of). // as an administrator (leaves out owned and subuser of).
if ($level === 'admin' && $admin) { elseif ($level === User::FILTER_LEVEL_ADMIN && $user->root_admin) {
$instance->whereIn('id', $this->getUserAccessServers($user)); $instance->whereNotIn('id', $this->getUserAccessServers($user->id));
} }
return $instance->search($this->searchTerm)->paginate( return $instance->search($this->getSearchTerm())->paginate(25);
$this->app->make('config')->get('pterodactyl.paginate.frontend.servers')
);
} }
/** /**
* {@inheritdoc} * Return a server by UUID.
*
* @param string $uuid
* @return \Pterodactyl\Models\Server
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getByUuid($uuid) public function getByUuid(string $uuid): Server
{ {
Assert::stringNotEmpty($uuid, 'First argument passed to getByUuid must be a non-empty string, received %s.'); Assert::notEmpty($uuid, 'Expected non-empty string as first argument passed to ' . __METHOD__);
$instance = $this->getBuilder()->with('nest', 'node')->where(function ($query) use ($uuid) { try {
return $this->getBuilder()->with('nest', 'node')->where(function ($query) use ($uuid) {
$query->where('uuidShort', $uuid)->orWhere('uuid', $uuid); $query->where('uuidShort', $uuid)->orWhere('uuid', $uuid);
})->first($this->getColumns()); })->firstOrFail($this->getColumns());
} catch (ModelNotFoundException $exception) {
if (! $instance) {
throw new RecordNotFoundException; throw new RecordNotFoundException;
} }
}
return $instance; /**
* Return an array of server IDs that a given user can access based
* on owner and subuser permissions.
*
* @param int $user
* @return int[]
*/
private function getUserAccessServers(int $user): array
{
return $this->getBuilder()->select('id')->where('owner_id', $user)->union(
$this->app->make(SubuserRepository::class)->getBuilder()->select('server_id')->where('user_id', $user)
)->pluck('id')->all();
} }
} }

View file

@ -1,11 +1,4 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
@ -15,7 +8,9 @@ use Pterodactyl\Contracts\Repository\ServerVariableRepositoryInterface;
class ServerVariableRepository extends EloquentRepository implements ServerVariableRepositoryInterface class ServerVariableRepository extends EloquentRepository implements ServerVariableRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {

View file

@ -1,21 +1,17 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Pterodactyl\Models\Session; use Pterodactyl\Models\Session;
use Illuminate\Support\Collection;
use Pterodactyl\Contracts\Repository\SessionRepositoryInterface; use Pterodactyl\Contracts\Repository\SessionRepositoryInterface;
class SessionRepository extends EloquentRepository implements SessionRepositoryInterface class SessionRepository extends EloquentRepository implements SessionRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -23,17 +19,24 @@ class SessionRepository extends EloquentRepository implements SessionRepositoryI
} }
/** /**
* {@inheritdoc} * Return all of the active sessions for a user.
*
* @param int $user
* @return \Illuminate\Support\Collection
*/ */
public function getUserSessions($user) public function getUserSessions(int $user): Collection
{ {
return $this->getBuilder()->where('user_id', $user)->get($this->getColumns()); return $this->getBuilder()->where('user_id', $user)->get($this->getColumns());
} }
/** /**
* {@inheritdoc} * Delete a session for a given user.
*
* @param int $user
* @param int $session
* @return null|int
*/ */
public function deleteUserSession($user, $session) public function deleteUserSession(int $user, int $session)
{ {
return $this->getBuilder()->where('user_id', $user)->where('id', $session)->delete(); return $this->getBuilder()->where('user_id', $user)->where('id', $session)->delete();
} }

View file

@ -10,16 +10,15 @@ class SettingsRepository extends EloquentRepository implements SettingsRepositor
/** /**
* @var array * @var array
*/ */
private $cache = []; private static $cache = [];
/** /**
* @var array * @var array
*/ */
private $databaseMiss = []; private static $databaseMiss = [];
/** /**
* Return an instance of the model that acts as the base for * Return the model backing this repository.
* this repository.
* *
* @return string * @return string
*/ */
@ -33,12 +32,17 @@ class SettingsRepository extends EloquentRepository implements SettingsRepositor
* *
* @param string $key * @param string $key
* @param string $value * @param string $value
*
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function set(string $key, string $value) public function set(string $key, string $value)
{ {
// Clear item from the cache. // Clear item from the cache.
$this->clearCache($key); $this->clearCache($key);
$this->withoutFresh()->updateOrCreate(['key' => $key], ['value' => $value]); $this->withoutFreshModel()->updateOrCreate(['key' => $key], ['value' => $value]);
self::$cache[$key] = $value;
} }
/** /**
@ -51,32 +55,27 @@ class SettingsRepository extends EloquentRepository implements SettingsRepositor
public function get(string $key, $default = null) public function get(string $key, $default = null)
{ {
// If item has already been requested return it from the cache. If // If item has already been requested return it from the cache. If
// we already know it is missing, immediately return the default // we already know it is missing, immediately return the default value.
// value. if (array_key_exists($key, self::$cache)) {
if (array_key_exists($key, $this->cache)) { return self::$cache[$key];
return $this->cache[$key]; } elseif (array_key_exists($key, self::$databaseMiss)) {
} elseif (array_key_exists($key, $this->databaseMiss)) { return value($default);
return $default;
} }
$instance = $this->getBuilder()->where('key', $key)->first(); $instance = $this->getBuilder()->where('key', $key)->first();
if (is_null($instance)) { if (is_null($instance)) {
$this->databaseMiss[$key] = true; self::$databaseMiss[$key] = true;
return $default; return value($default);
} }
$this->cache[$key] = $instance->value; return self::$cache[$key] = $instance->value;
return $this->cache[$key];
} }
/** /**
* Remove a key from the database cache. * Remove a key from the database cache.
* *
* @param string $key * @param string $key
* @return mixed
*/ */
public function forget(string $key) public function forget(string $key)
{ {
@ -89,8 +88,8 @@ class SettingsRepository extends EloquentRepository implements SettingsRepositor
* *
* @param string $key * @param string $key
*/ */
protected function clearCache(string $key) private function clearCache(string $key)
{ {
unset($this->cache[$key], $this->databaseMiss[$key]); unset(self::$cache[$key], self::$databaseMiss[$key]);
} }
} }

View file

@ -1,15 +1,7 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Webmozart\Assert\Assert;
use Pterodactyl\Models\Subuser; use Pterodactyl\Models\Subuser;
use Pterodactyl\Exceptions\Repository\RecordNotFoundException; use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
use Pterodactyl\Contracts\Repository\SubuserRepositoryInterface; use Pterodactyl\Contracts\Repository\SubuserRepositoryInterface;
@ -17,7 +9,9 @@ use Pterodactyl\Contracts\Repository\SubuserRepositoryInterface;
class SubuserRepository extends EloquentRepository implements SubuserRepositoryInterface class SubuserRepository extends EloquentRepository implements SubuserRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -86,39 +80,4 @@ class SubuserRepository extends EloquentRepository implements SubuserRepositoryI
return $instance; return $instance;
} }
/**
* {@inheritdoc}
*/
public function getWithServerAndPermissions($id)
{
Assert::numeric($id, 'First argument passed to getWithServerAndPermissions must be numeric, received %s.');
$instance = $this->getBuilder()->with('server', 'permission', 'user')->find($id, $this->getColumns());
if (! $instance) {
throw new RecordNotFoundException;
}
return $instance;
}
/**
* {@inheritdoc}
*/
public function getWithKey($user, $server)
{
Assert::integerish($user, 'First argument passed to getWithKey must be integer, received %s.');
Assert::integerish($server, 'Second argument passed to getWithKey must be integer, received %s.');
$instance = $this->getBuilder()->with('key')->where([
['user_id', '=', $user],
['server_id', '=', $server],
])->first();
if (is_null($instance)) {
throw new RecordNotFoundException;
}
return $instance;
}
} }

View file

@ -1,23 +1,18 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Pterodactyl\Models\Task; use Pterodactyl\Models\Task;
use Webmozart\Assert\Assert; use Illuminate\Database\Eloquent\ModelNotFoundException;
use Pterodactyl\Contracts\Repository\TaskRepositoryInterface; use Pterodactyl\Contracts\Repository\TaskRepositoryInterface;
use Pterodactyl\Exceptions\Repository\RecordNotFoundException; use Pterodactyl\Exceptions\Repository\RecordNotFoundException;
class TaskRepository extends EloquentRepository implements TaskRepositoryInterface class TaskRepository extends EloquentRepository implements TaskRepositoryInterface
{ {
/** /**
* {@inheritdoc} * Return the model backing this repository.
*
* @return string
*/ */
public function model() public function model()
{ {
@ -25,28 +20,31 @@ class TaskRepository extends EloquentRepository implements TaskRepositoryInterfa
} }
/** /**
* {@inheritdoc} * Get a task and the server relationship for that task.
*
* @param int $id
* @return \Pterodactyl\Models\Task
*
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function getTaskWithServer($id) public function getTaskForJobProcess(int $id): Task
{ {
Assert::integerish($id, 'First argument passed to getTaskWithServer must be numeric, received %s.'); try {
return $this->getBuilder()->with('server.user', 'schedule')->findOrFail($id, $this->getColumns());
$instance = $this->getBuilder()->with('server.user')->find($id, $this->getColumns()); } catch (ModelNotFoundException $exception) {
if (! $instance) {
throw new RecordNotFoundException; throw new RecordNotFoundException;
} }
return $instance;
} }
/** /**
* {@inheritdoc} * Returns the next task in a schedule.
*
* @param int $schedule
* @param int $index
* @return null|\Pterodactyl\Models\Task
*/ */
public function getNextTask($schedule, $index) public function getNextTask(int $schedule, int $index)
{ {
Assert::integerish($schedule, 'First argument passed to getNextTask must be integer, received %s.');
Assert::integerish($index, 'Second argument passed to getNextTask must be integer, received %s.');
return $this->getBuilder()->where('schedule_id', '=', $schedule) return $this->getBuilder()->where('schedule_id', '=', $schedule)
->where('sequence_id', '=', $index + 1) ->where('sequence_id', '=', $index + 1)
->first($this->getColumns()); ->first($this->getColumns());

View file

@ -1,44 +1,21 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Eloquent; namespace Pterodactyl\Repositories\Eloquent;
use Pterodactyl\Models\User; use Pterodactyl\Models\User;
use Illuminate\Foundation\Application; use Illuminate\Support\Collection;
use Pterodactyl\Repositories\Concerns\Searchable; use Pterodactyl\Repositories\Concerns\Searchable;
use Illuminate\Contracts\Pagination\LengthAwarePaginator;
use Pterodactyl\Contracts\Repository\UserRepositoryInterface; use Pterodactyl\Contracts\Repository\UserRepositoryInterface;
use Illuminate\Contracts\Config\Repository as ConfigRepository;
class UserRepository extends EloquentRepository implements UserRepositoryInterface class UserRepository extends EloquentRepository implements UserRepositoryInterface
{ {
use Searchable; use Searchable;
/** /**
* @var \Illuminate\Contracts\Config\Repository * Return the model backing this repository.
*/
protected $config;
/**
* UserRepository constructor.
* *
* @param \Illuminate\Foundation\Application $application * @return string
* @param \Illuminate\Contracts\Config\Repository $config
*/
public function __construct(Application $application, ConfigRepository $config)
{
parent::__construct($application);
$this->config = $config;
}
/**
* {@inheritdoc}
*/ */
public function model() public function model()
{ {
@ -46,28 +23,26 @@ class UserRepository extends EloquentRepository implements UserRepositoryInterfa
} }
/** /**
* {@inheritdoc} * Return all users with counts of servers and subusers of servers.
*
* @return \Illuminate\Contracts\Pagination\LengthAwarePaginator
*/ */
public function getAllUsersWithCounts() public function getAllUsersWithCounts(): LengthAwarePaginator
{ {
$users = $this->getBuilder()->withCount('servers', 'subuserOf'); return $this->getBuilder()->withCount('servers', 'subuserOf')
->search($this->getSearchTerm())
if ($this->searchTerm) { ->paginate(50, $this->getColumns());
$users->search($this->searchTerm);
}
return $users->paginate(
$this->config->get('pterodactyl.paginate.admin.users'),
$this->getColumns()
);
} }
/** /**
* {@inheritdoc} * Return all matching models for a user in a format that can be used for dropdowns.
*
* @param string $query
* @return \Illuminate\Support\Collection
*/ */
public function filterUsersByQuery($query) public function filterUsersByQuery(string $query): Collection
{ {
$this->withColumns([ $this->setColumns([
'id', 'email', 'username', 'name_first', 'name_last', 'id', 'email', 'username', 'name_first', 'name_last',
]); ]);

View file

@ -1,14 +1,8 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories; namespace Pterodactyl\Repositories;
use InvalidArgumentException;
use Illuminate\Foundation\Application; use Illuminate\Foundation\Application;
use Pterodactyl\Contracts\Repository\RepositoryInterface; use Pterodactyl\Contracts\Repository\RepositoryInterface;
@ -43,35 +37,13 @@ abstract class Repository implements RepositoryInterface
{ {
$this->app = $application; $this->app = $application;
$this->setModel($this->model()); $this->initalizeModel($this->model());
} }
/** /**
* Take the provided model and make it accessible to the rest of the repository. * Return the model backing this repository.
* *
* @param string|array $model * @return string|\Closure|object
* @return mixed
*/
protected function setModel($model)
{
if (is_array($model)) {
if (count($model) !== 2) {
throw new \InvalidArgumentException(
printf('setModel expected exactly 2 parameters, %d received.', count($model))
);
}
return $this->model = call_user_func(
$model[1],
$this->app->make($model[0])
);
}
return $this->model = $this->app->make($model);
}
/**
* @return mixed
*/ */
abstract public function model(); abstract public function model();
@ -88,10 +60,10 @@ abstract class Repository implements RepositoryInterface
/** /**
* Setup column selection functionality. * Setup column selection functionality.
* *
* @param array $columns * @param array|string $columns
* @return $this * @return $this
*/ */
public function withColumns($columns = ['*']) public function setColumns($columns = ['*'])
{ {
$clone = clone $this; $clone = clone $this;
$clone->columns = is_array($columns) ? $columns : func_get_args(); $clone->columns = is_array($columns) ? $columns : func_get_args();
@ -110,15 +82,56 @@ abstract class Repository implements RepositoryInterface
} }
/** /**
* Set repository to not return a fresh record from the DB when completed. * Stop repository update functions from returning a fresh
* model when changes are committed.
* *
* @return $this * @return $this
*/ */
public function withoutFresh() public function withoutFreshModel()
{
return $this->setFreshModel(false);
}
/**
* Return a fresh model with a repository updates a model.
*
* @return $this
*/
public function withFreshModel()
{
return $this->setFreshModel(true);
}
/**
* Set wether or not the repository should return a fresh model
* when changes are committed.
*
* @param bool $fresh
* @return $this
*/
public function setFreshModel(bool $fresh = true)
{ {
$clone = clone $this; $clone = clone $this;
$clone->withFresh = false; $clone->withFresh = $fresh;
return $clone; return $clone;
} }
/**
* Take the provided model and make it accessible to the rest of the repository.
*
* @param array $model
* @return mixed
*/
protected function initalizeModel(...$model)
{
switch (count($model)) {
case 1:
return $this->model = $this->app->make($model[0]);
case 2:
return $this->model = call_user_func([$this->app->make($model[0]), $model[1]]);
default:
throw new InvalidArgumentException('Model must be a FQCN or an array with a count of two.');
}
}
} }

View file

@ -1,146 +0,0 @@
<?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Wings;
use GuzzleHttp\Client;
use Webmozart\Assert\Assert;
use Illuminate\Foundation\Application;
use Pterodactyl\Contracts\Repository\NodeRepositoryInterface;
use Illuminate\Contracts\Config\Repository as ConfigRepository;
use Pterodactyl\Contracts\Repository\Daemon\BaseRepositoryInterface;
class BaseRepository implements BaseRepositoryInterface
{
/**
* @var \Illuminate\Foundation\Application
*/
protected $app;
/**
* @var
*/
protected $accessServer;
/**
* @var
*/
protected $accessToken;
/**
* @var
*/
protected $node;
/**
* @var \Illuminate\Contracts\Config\Repository
*/
protected $config;
/**
* @var \Pterodactyl\Contracts\Repository\NodeRepositoryInterface
*/
protected $nodeRepository;
/**
* BaseRepository constructor.
*
* @param \Illuminate\Foundation\Application $app
* @param \Illuminate\Contracts\Config\Repository $config
* @param \Pterodactyl\Contracts\Repository\NodeRepositoryInterface $nodeRepository
*/
public function __construct(
Application $app,
ConfigRepository $config,
NodeRepositoryInterface $nodeRepository
) {
$this->app = $app;
$this->config = $config;
$this->nodeRepository = $nodeRepository;
}
/**
* {@inheritdoc}
*/
public function setNode($id)
{
Assert::numeric($id, 'The first argument passed to setNode must be numeric, received %s.');
$this->node = $this->nodeRepository->find($id);
return $this;
}
/**
* {@inheritdoc}
*/
public function getNode()
{
return $this->node;
}
/**
* {@inheritdoc}
*/
public function setAccessServer($server = null)
{
Assert::nullOrString($server, 'The first argument passed to setAccessServer must be null or a string, received %s.');
$this->accessServer = $server;
return $this;
}
/**
* {@inheritdoc}
*/
public function getAccessServer()
{
return $this->accessServer;
}
/**
* {@inheritdoc}
*/
public function setAccessToken($token = null)
{
Assert::nullOrString($token, 'The first argument passed to setAccessToken must be null or a string, received %s.');
$this->accessToken = $token;
return $this;
}
/**
* {@inheritdoc}
*/
public function getAccessToken()
{
return $this->accessToken;
}
/**
* {@inheritdoc}
*/
public function getHttpClient(array $headers = [])
{
if (! is_null($this->accessToken)) {
$headers['Authorization'] = 'Bearer ' . $this->getAccessToken();
} elseif (! is_null($this->node)) {
$headers['Authorization'] = 'Bearer ' . $this->getNode()->daemonSecret;
}
return new Client([
'base_uri' => sprintf('%s://%s:%s/v1/', $this->getNode()->scheme, $this->getNode()->fqdn, $this->getNode()->daemonListen),
'timeout' => $this->config->get('pterodactyl.guzzle.timeout'),
'connect_timeout' => $this->config->get('pterodactyl.guzzle.connect_timeout'),
'headers' => $headers,
]);
}
}

View file

@ -1,30 +0,0 @@
<?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Wings;
use Webmozart\Assert\Assert;
use Pterodactyl\Contracts\Repository\Daemon\CommandRepositoryInterface;
class CommandRepository extends BaseRepository implements CommandRepositoryInterface
{
/**
* {@inheritdoc}
*/
public function send($command)
{
Assert::stringNotEmpty($command, 'First argument passed to send must be a non-empty string, received %s.');
return $this->getHttpClient()->request('POST', 'server/' . $this->getAccessServer() . '/command', [
'json' => [
'command' => $command,
],
]);
}
}

View file

@ -1,24 +0,0 @@
<?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Wings;
use Pterodactyl\Exceptions\PterodactylException;
use Pterodactyl\Contracts\Repository\Daemon\ConfigurationRepositoryInterface;
class ConfigurationRepository extends BaseRepository implements ConfigurationRepositoryInterface
{
/**
* {@inheritdoc}
*/
public function update(array $overrides = [])
{
throw new PterodactylException('This has not yet been configured.');
}
}

View file

@ -1,114 +0,0 @@
<?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Wings;
use Webmozart\Assert\Assert;
use Pterodactyl\Contracts\Repository\Daemon\FileRepositoryInterface;
class FileRepository extends BaseRepository implements FileRepositoryInterface
{
/**
* {@inheritdoc}
*/
public function getFileStat($path)
{
Assert::stringNotEmpty($path, 'First argument passed to getStat must be a non-empty string, received %s.');
$file = pathinfo($path);
$file['dirname'] = in_array($file['dirname'], ['.', './', '/']) ? null : trim($file['dirname'], '/') . '/';
$response = $this->getHttpClient()->request('GET', sprintf(
'server/' . $this->getAccessServer() . '/file/stat/%s',
rawurlencode($file['dirname'] . $file['basename'])
));
return json_decode($response->getBody());
}
/**
* {@inheritdoc}
*/
public function getContent($path)
{
Assert::stringNotEmpty($path, 'First argument passed to getContent must be a non-empty string, received %s.');
$file = pathinfo($path);
$file['dirname'] = in_array($file['dirname'], ['.', './', '/']) ? null : trim($file['dirname'], '/') . '/';
$response = $this->getHttpClient()->request('GET', sprintf(
'server/' . $this->getAccessServer() . '/file/f/%s',
rawurlencode($file['dirname'] . $file['basename'])
));
return object_get(json_decode($response->getBody()), 'content');
}
/**
* {@inheritdoc}
*/
public function putContent($path, $content)
{
Assert::stringNotEmpty($path, 'First argument passed to putContent must be a non-empty string, received %s.');
Assert::string($content, 'Second argument passed to putContent must be a string, received %s.');
$file = pathinfo($path);
$file['dirname'] = in_array($file['dirname'], ['.', './', '/']) ? null : trim($file['dirname'], '/') . '/';
return $this->getHttpClient()->request('POST', 'server/' . $this->getAccessServer() . '/file/save', [
'json' => [
'path' => rawurlencode($file['dirname'] . $file['basename']),
'content' => $content,
],
]);
}
/**
* {@inheritdoc}
*/
public function getDirectory($path)
{
Assert::string($path, 'First argument passed to getDirectory must be a string, received %s.');
$response = $this->getHttpClient()->request('GET', sprintf(
'server/' . $this->getAccessServer() . '/directory/%s',
rawurlencode($path)
));
$contents = json_decode($response->getBody());
$files = [];
$folders = [];
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' => pathinfo($value->name, PATHINFO_EXTENSION),
'size' => human_readable($value->size),
'date' => strtotime($value->modified),
'mime' => $value->mime,
]);
}
}
return [
'files' => $files,
'folders' => $folders,
];
}
}

View file

@ -1,40 +0,0 @@
<?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Wings;
use Webmozart\Assert\Assert;
use Pterodactyl\Contracts\Repository\Daemon\PowerRepositoryInterface;
use Pterodactyl\Exceptions\Repository\Daemon\InvalidPowerSignalException;
class PowerRepository extends BaseRepository implements PowerRepositoryInterface
{
/**
* {@inheritdoc}
*/
public function sendSignal($signal)
{
Assert::stringNotEmpty($signal, 'The first argument passed to sendSignal must be a non-empty string, received %s.');
switch ($signal) {
case self::SIGNAL_START:
case self::SIGNAL_STOP:
case self::SIGNAL_RESTART:
case self::SIGNAL_KILL:
return $this->getHttpClient()->request('PUT', 'server/' . $this->getAccessServer() . '/power', [
'json' => [
'action' => $signal,
],
]);
break;
default:
throw new InvalidPowerSignalException('The signal ' . $signal . ' is not defined and could not be processed.');
}
}
}

View file

@ -1,89 +0,0 @@
<?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Repositories\Wings;
use Psr\Http\Message\ResponseInterface;
use Pterodactyl\Exceptions\PterodactylException;
use Pterodactyl\Contracts\Repository\Daemon\ServerRepositoryInterface;
class ServerRepository extends BaseRepository implements ServerRepositoryInterface
{
/**
* {@inheritdoc}
*/
public function create(array $structure, array $overrides = []): ResponseInterface
{
throw new PterodactylException('This feature is not yet implemented.');
}
/**
* {@inheritdoc}
*/
public function update(array $data)
{
throw new PterodactylException('This feature is not yet implemented.');
}
/**
* {@inheritdoc}
*/
public function reinstall($data = null)
{
throw new PterodactylException('This feature is not yet implemented.');
}
/**
* {@inheritdoc}
*/
public function rebuild()
{
return $this->getHttpClient()->request('POST', 'server/' . $this->getAccessServer() . '/rebuild');
}
/**
* {@inheritdoc}
*/
public function suspend()
{
return $this->getHttpClient()->request('POST', 'server/' . $this->getAccessServer() . '/suspend');
}
/**
* {@inheritdoc}
*/
public function unsuspend()
{
return $this->getHttpClient()->request('POST', 'server/' . $this->getAccessServer() . '/unsuspend');
}
/**
* {@inheritdoc}
*/
public function delete()
{
return $this->getHttpClient()->request('DELETE', 'server/' . $this->getAccessServer());
}
/**
* {@inheritdoc}
*/
public function details()
{
return $this->getHttpClient()->request('GET', 'server/' . $this->getAccessServer());
}
/**
* {@inheritdoc}
*/
public function revokeAccessKey($key)
{
throw new PterodactylException('This feature is not yet implemented.');
}
}

View file

@ -83,11 +83,11 @@ class SetDefaultAllocationService
} }
$this->connection->beginTransaction(); $this->connection->beginTransaction();
$this->serverRepository->withoutFresh()->update($server->id, ['allocation_id' => $model->id]); $this->serverRepository->withoutFreshModel()->update($server->id, ['allocation_id' => $model->id]);
// Update on the daemon. // Update on the daemon.
try { try {
$this->daemonRepository->setAccessServer($server->uuid)->setNode($server->node_id)->update([ $this->daemonRepository->setServer($server)->update([
'build' => [ 'build' => [
'default' => [ 'default' => [
'ip' => $model->ip, 'ip' => $model->ip,

View file

@ -40,7 +40,7 @@ class PermissionService
public function create($key, $permission) public function create($key, $permission)
{ {
// @todo handle an array of permissions to do a mass assignment? // @todo handle an array of permissions to do a mass assignment?
return $this->repository->withoutFresh()->create([ return $this->repository->withoutFreshModel()->create([
'key_id' => $key, 'key_id' => $key,
'permission' => $permission, 'permission' => $permission,
]); ]);

View file

@ -75,7 +75,7 @@ class DaemonKeyCreationService
{ {
$secret = DaemonKeyRepositoryInterface::INTERNAL_KEY_IDENTIFIER . str_random(40); $secret = DaemonKeyRepositoryInterface::INTERNAL_KEY_IDENTIFIER . str_random(40);
$this->repository->withoutFresh()->create([ $this->repository->withoutFreshModel()->create([
'user_id' => $user, 'user_id' => $user,
'server_id' => $server, 'server_id' => $server,
'secret' => $secret, 'secret' => $secret,

View file

@ -108,7 +108,7 @@ class DaemonKeyDeletionService
$this->repository->delete($key->id); $this->repository->delete($key->id);
try { try {
$this->daemonRepository->setNode($server->node_id)->revokeAccessKey($key->secret); $this->daemonRepository->setServer($server)->revokeAccessKey($key->secret);
} catch (RequestException $exception) { } catch (RequestException $exception) {
$response = $exception->getResponse(); $response = $exception->getResponse();
$this->connection->rollBack(); $this->connection->rollBack();

View file

@ -78,7 +78,7 @@ class DaemonKeyUpdateService
Assert::integerish($key, 'First argument passed to handle must be an integer, received %s.'); Assert::integerish($key, 'First argument passed to handle must be an integer, received %s.');
$secret = DaemonKeyRepositoryInterface::INTERNAL_KEY_IDENTIFIER . str_random(40); $secret = DaemonKeyRepositoryInterface::INTERNAL_KEY_IDENTIFIER . str_random(40);
$this->repository->withoutFresh()->update($key, [ $this->repository->withoutFreshModel()->update($key, [
'secret' => $secret, 'secret' => $secret,
'expires_at' => $this->carbon->now()->addMinutes($this->config->get('pterodactyl.api.key_expire_time'))->toDateTimeString(), 'expires_at' => $this->carbon->now()->addMinutes($this->config->get('pterodactyl.api.key_expire_time'))->toDateTimeString(),
]); ]);

View file

@ -46,22 +46,20 @@ class RevokeMultipleDaemonKeysService
* *
* @param \Pterodactyl\Models\User $user * @param \Pterodactyl\Models\User $user
* @param bool $ignoreConnectionErrors * @param bool $ignoreConnectionErrors
*
* @throws \Pterodactyl\Exceptions\Http\Connection\DaemonConnectionException
*/ */
public function handle(User $user, bool $ignoreConnectionErrors = false) public function handle(User $user, bool $ignoreConnectionErrors = false)
{ {
$keys = $this->repository->getKeysForRevocation($user); $keys = $this->repository->getKeysForRevocation($user);
$keys->groupBy('server.node_id')->each(function ($group, $node) use ($ignoreConnectionErrors) { $keys->groupBy('node.id')->each(function ($group, $nodeId) use ($ignoreConnectionErrors) {
try { try {
$this->daemonRepository->setNode($node)->revokeAccessKey(collect($group)->pluck('secret')->toArray()); $this->daemonRepository->setNode(collect($group)->first()->getRelation('node'))->revokeAccessKey(collect($group)->pluck('secret')->toArray());
} catch (RequestException $exception) { } catch (RequestException $exception) {
if (! $ignoreConnectionErrors) { if (! $ignoreConnectionErrors) {
throw new DaemonConnectionException($exception); throw new DaemonConnectionException($exception);
} }
$this->setConnectionException($node, $exception); $this->setConnectionException($nodeId, $exception);
} }
$this->repository->deleteKeys(collect($group)->pluck('id')->toArray()); $this->repository->deleteKeys(collect($group)->pluck('id')->toArray());

View file

@ -69,7 +69,7 @@ class DatabasePasswordService
$this->dynamic->set('dynamic', $database->database_host_id); $this->dynamic->set('dynamic', $database->database_host_id);
$this->connection->beginTransaction(); $this->connection->beginTransaction();
$updated = $this->repository->withoutFresh()->update($database->id, [ $updated = $this->repository->withoutFreshModel()->update($database->id, [
'password' => $this->encrypter->encrypt($password), 'password' => $this->encrypter->encrypt($password),
]); ]);

View file

@ -57,6 +57,6 @@ class EggUpdateService
} }
} }
$this->repository->withoutFresh()->update($egg->id, $data); $this->repository->withoutFreshModel()->update($egg->id, $data);
} }
} }

View file

@ -52,7 +52,7 @@ class InstallScriptService
} }
} }
$this->repository->withoutFresh()->update($egg->id, [ $this->repository->withoutFreshModel()->update($egg->id, [
'script_install' => array_get($data, 'script_install'), 'script_install' => array_get($data, 'script_install'),
'script_is_privileged' => array_get($data, 'script_is_privileged', 1), 'script_is_privileged' => array_get($data, 'script_is_privileged', 1),
'script_entry' => array_get($data, 'script_entry'), 'script_entry' => array_get($data, 'script_entry'),

View file

@ -89,14 +89,14 @@ class EggUpdateImporterService
// Update Existing Variables // Update Existing Variables
collect($parsed->variables)->each(function ($variable) use ($egg) { collect($parsed->variables)->each(function ($variable) use ($egg) {
$this->variableRepository->withoutFresh()->updateOrCreate([ $this->variableRepository->withoutFreshModel()->updateOrCreate([
'egg_id' => $egg, 'egg_id' => $egg,
'env_variable' => $variable->env_variable, 'env_variable' => $variable->env_variable,
], collect($variable)->except(['egg_id', 'env_variable'])->toArray()); ], collect($variable)->except(['egg_id', 'env_variable'])->toArray());
}); });
$imported = collect($parsed->variables)->pluck('env_variable')->toArray(); $imported = collect($parsed->variables)->pluck('env_variable')->toArray();
$existing = $this->variableRepository->withColumns(['id', 'env_variable'])->findWhere([['egg_id', '=', $egg]]); $existing = $this->variableRepository->setColumns(['id', 'env_variable'])->findWhere([['egg_id', '=', $egg]]);
// Delete variables not present in the import. // Delete variables not present in the import.
collect($existing)->each(function ($variable) use ($egg, $imported) { collect($existing)->each(function ($variable) use ($egg, $imported) {

View file

@ -49,7 +49,7 @@ class VariableUpdateService
])); ]));
} }
$search = $this->repository->withColumns('id')->findCountWhere([ $search = $this->repository->setColumns('id')->findCountWhere([
['env_variable', '=', array_get($data, 'env_variable')], ['env_variable', '=', array_get($data, 'env_variable')],
['egg_id', '=', $variable->egg_id], ['egg_id', '=', $variable->egg_id],
['id', '!=', $variable->id], ['id', '!=', $variable->id],
@ -64,7 +64,7 @@ class VariableUpdateService
$options = array_get($data, 'options') ?? []; $options = array_get($data, 'options') ?? [];
return $this->repository->withoutFresh()->update($variable->id, array_merge($data, [ return $this->repository->withoutFreshModel()->update($variable->id, array_merge($data, [
'user_viewable' => in_array('user_viewable', $options), 'user_viewable' => in_array('user_viewable', $options),
'user_editable' => in_array('user_editable', $options), 'user_editable' => in_array('user_editable', $options),
])); ]));

View file

@ -42,6 +42,6 @@ class NestUpdateService
unset($data['author']); unset($data['author']);
} }
$this->repository->withoutFresh()->update($nest, $data); $this->repository->withoutFreshModel()->update($nest, $data);
} }
} }

View file

@ -63,7 +63,7 @@ class NodeDeletionService
$node = $node->id; $node = $node->id;
} }
$servers = $this->serverRepository->withColumns('id')->findCountWhere([['node_id', '=', $node]]); $servers = $this->serverRepository->setColumns('id')->findCountWhere([['node_id', '=', $node]]);
if ($servers > 0) { if ($servers > 0) {
throw new HasActiveServersException($this->translator->trans('exceptions.node.servers_attached')); throw new HasActiveServersException($this->translator->trans('exceptions.node.servers_attached'));
} }

View file

@ -65,11 +65,11 @@ class NodeUpdateService
if ($this->getUpdatedModel()) { if ($this->getUpdatedModel()) {
$response = $this->repository->update($node->id, $data); $response = $this->repository->update($node->id, $data);
} else { } else {
$response = $this->repository->withoutFresh()->update($node->id, $data); $response = $this->repository->withoutFreshModel()->update($node->id, $data);
} }
try { try {
$this->configRepository->setNode($node->id)->update(); $this->configRepository->setNode($node)->update();
} catch (RequestException $exception) { } catch (RequestException $exception) {
throw new DaemonConnectionException($exception); throw new DaemonConnectionException($exception);
} }

View file

@ -69,7 +69,7 @@ class PackDeletionService
public function handle($pack) public function handle($pack)
{ {
if (! $pack instanceof Pack) { if (! $pack instanceof Pack) {
$pack = $this->repository->withColumns(['id', 'uuid'])->find($pack); $pack = $this->repository->setColumns(['id', 'uuid'])->find($pack);
} }
$count = $this->serverRepository->findCountWhere([['pack_id', '=', $pack->id]]); $count = $this->serverRepository->findCountWhere([['pack_id', '=', $pack->id]]);

View file

@ -54,7 +54,7 @@ class PackUpdateService
public function handle($pack, array $data) public function handle($pack, array $data)
{ {
if (! $pack instanceof Pack) { if (! $pack instanceof Pack) {
$pack = $this->repository->withColumns(['id', 'egg_id'])->find($pack); $pack = $this->repository->setColumns(['id', 'egg_id'])->find($pack);
} }
if ((int) array_get($data, 'egg_id', $pack->egg_id) !== $pack->egg_id) { if ((int) array_get($data, 'egg_id', $pack->egg_id) !== $pack->egg_id) {
@ -70,6 +70,6 @@ class PackUpdateService
$data['visible'] = isset($data['visible']); $data['visible'] = isset($data['visible']);
$data['locked'] = isset($data['locked']); $data['locked'] = isset($data['locked']);
return $this->repository->withoutFresh()->update($pack->id, $data); return $this->repository->withoutFreshModel()->update($pack->id, $data);
} }
} }

View file

@ -1,16 +1,9 @@
<?php <?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Services\Schedules; namespace Pterodactyl\Services\Schedules;
use Carbon\Carbon;
use Cron\CronExpression; use Cron\CronExpression;
use Webmozart\Assert\Assert;
use Pterodactyl\Models\Schedule; use Pterodactyl\Models\Schedule;
use Pterodactyl\Services\Schedules\Tasks\RunTaskService; use Pterodactyl\Services\Schedules\Tasks\RunTaskService;
use Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface; use Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface;
@ -20,12 +13,17 @@ class ProcessScheduleService
/** /**
* @var \Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface * @var \Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface
*/ */
protected $repository; private $repository;
/** /**
* @var \Pterodactyl\Services\Schedules\Tasks\RunTaskService * @var \Pterodactyl\Services\Schedules\Tasks\RunTaskService
*/ */
protected $runnerService; private $runnerService;
/**
* @var \Carbon\Carbon|null
*/
private $runTimeOverride;
/** /**
* ProcessScheduleService constructor. * ProcessScheduleService constructor.
@ -39,23 +37,31 @@ class ProcessScheduleService
$this->runnerService = $runnerService; $this->runnerService = $runnerService;
} }
/**
* Set the time that this schedule should be run at. This will override the time
* defined on the schedule itself. Useful for triggering one-off task runs.
*
* @param \Carbon\Carbon $time
* @return $this
*/
public function setRunTimeOverride(Carbon $time)
{
$this->runTimeOverride = $time;
return $this;
}
/** /**
* Process a schedule and push the first task onto the queue worker. * Process a schedule and push the first task onto the queue worker.
* *
* @param int|\Pterodactyl\Models\Schedule $schedule * @param \Pterodactyl\Models\Schedule $schedule
* *
* @throws \Pterodactyl\Exceptions\Model\DataValidationException * @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/ */
public function handle($schedule) public function handle(Schedule $schedule)
{ {
Assert::true(($schedule instanceof Schedule || is_digit($schedule)), $this->repository->loadTasks($schedule);
'First argument passed to handle must be instance of \Pterodactyl\Models\Schedule or an integer, received %s.'
);
if (($schedule instanceof Schedule && ! $schedule->relationLoaded('tasks')) || ! $schedule instanceof Schedule) {
$schedule = $this->repository->getScheduleWithTasks(is_digit($schedule) ? $schedule : $schedule->id);
}
$formattedCron = sprintf('%s %s %s * %s *', $formattedCron = sprintf('%s %s %s * %s *',
$schedule->cron_minute, $schedule->cron_minute,
@ -66,10 +72,25 @@ class ProcessScheduleService
$this->repository->update($schedule->id, [ $this->repository->update($schedule->id, [
'is_processing' => true, 'is_processing' => true,
'next_run_at' => CronExpression::factory($formattedCron)->getNextRunDate(), 'next_run_at' => $this->getRunAtTime($formattedCron),
]); ]);
$task = $schedule->tasks->where('sequence_id', 1)->first(); $task = $schedule->getRelation('tasks')->where('sequence_id', 1)->first();
$this->runnerService->handle($task); $this->runnerService->handle($task);
} }
/**
* Get the timestamp to store in the database as the next_run time for a schedule.
*
* @param string $formatted
* @return \DateTime|string
*/
private function getRunAtTime(string $formatted)
{
if ($this->runTimeOverride instanceof Carbon) {
return $this->runTimeOverride->toDateTimeString();
}
return CronExpression::factory($formatted)->getNextRunDate();
}
} }

Some files were not shown because too many files have changed in this diff Show more