Skip to content

Commit 9a581ef

Browse files
committed
Add support for rebuilding servers from the console.
1 parent 9a803e6 commit 9a581ef

File tree

2 files changed

+123
-0
lines changed

2 files changed

+123
-0
lines changed
Lines changed: 122 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,122 @@
1+
<?php
2+
/**
3+
* Pterodactyl - Panel
4+
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
5+
*
6+
* Permission is hereby granted, free of charge, to any person obtaining a copy
7+
* of this software and associated documentation files (the "Software"), to deal
8+
* in the Software without restriction, including without limitation the rights
9+
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10+
* copies of the Software, and to permit persons to whom the Software is
11+
* furnished to do so, subject to the following conditions:
12+
*
13+
* The above copyright notice and this permission notice shall be included in all
14+
* copies or substantial portions of the Software.
15+
*
16+
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17+
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18+
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19+
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20+
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21+
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22+
* SOFTWARE.
23+
*/
24+
25+
namespace Pterodactyl\Console\Commands;
26+
27+
use Pterodactyl\Models\Node;
28+
use Pterodactyl\Models\Server;
29+
use Illuminate\Console\Command;
30+
31+
class RebuildServer extends Command
32+
{
33+
/**
34+
* The name and signature of the console command.
35+
*
36+
* @var string
37+
*/
38+
protected $signature = 'pterodactyl:rebuild
39+
{--all}
40+
{--node= : Id of node to rebuild all servers on.}
41+
{--server= : UUID of server to rebuild.}';
42+
43+
/**
44+
* The console command description.
45+
*
46+
* @var string
47+
*/
48+
protected $description = 'Rebuild docker containers for a server or multiple servers.';
49+
50+
/**
51+
* Create a new command instance.
52+
*
53+
* @return void
54+
*/
55+
public function __construct()
56+
{
57+
parent::__construct();
58+
}
59+
60+
/**
61+
* Execute the console command.
62+
*
63+
* @return mixed
64+
*/
65+
public function handle()
66+
{
67+
if ($this->option('all')) {
68+
$servers = Server::with('node')->get();
69+
} elseif ($this->option('node')) {
70+
$servers = Server::with('node')->where('node_id', $this->option('node'))->get();
71+
} elseif ($this->option('server')) {
72+
$servers = Server::with('node')->where('id', $this->option('server'))->get();
73+
} else {
74+
$this->error('You must pass a flag to determine which server(s) to rebuild.');
75+
return;
76+
}
77+
78+
$this->line('Beginning processing, do not exit this script.');
79+
$bar = $this->output->createProgressBar(count($servers));
80+
$results = collect([]);
81+
foreach($servers as $server) {
82+
try {
83+
$server->node->guzzleClient([
84+
'X-Access-Server' => $server->uuid,
85+
'X-Access-Token' => $server->node->daemonSecret,
86+
])->request('POST', '/server/rebuild');
87+
88+
$results = $results->merge([
89+
$server->uuid => [
90+
'status' => 'info',
91+
'messages' => [
92+
'[✓] Processed rebuild request for ' . $server->uuid,
93+
],
94+
],
95+
]);
96+
} catch (\Exception $ex) {
97+
$results = $results->merge([
98+
$server->uuid => [
99+
'status' => 'error',
100+
'messages' => [
101+
'[✗] Failed to process rebuild request for ' . $server->uuid,
102+
$ex->getMessage(),
103+
],
104+
],
105+
]);
106+
}
107+
108+
$bar->advance();
109+
}
110+
111+
$bar->finish();
112+
$console = $this;
113+
114+
$this->line("\n");
115+
$results->each(function ($item, $key) use ($console) {
116+
foreach($item['messages'] as $line) {
117+
$console->{$item['status']}($line);
118+
}
119+
});
120+
$this->line("\nCompleted rebuild command processing.");
121+
}
122+
}

app/Console/Kernel.php

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ class Kernel extends ConsoleKernel
2424
\Pterodactyl\Console\Commands\CleanServiceBackup::class,
2525
\Pterodactyl\Console\Commands\AddNode::class,
2626
\Pterodactyl\Console\Commands\AddLocation::class,
27+
\Pterodactyl\Console\Commands\RebuildServer::class,
2728
];
2829

2930
/**

0 commit comments

Comments
 (0)