Apply node maintenance mode to servers (pterodactyl#4421)

Author: Boy132

app/Exceptions/Http/Server/ServerStateConflictException.php

@@ -17,6 +17,8 @@ public function __construct(Server $server, Throwable $previous = null)
         $message = 'This server is currently in an unsupported state, please try again later.';
         if ($server->isSuspended()) {
             $message = 'This server is currently suspended and the functionality requested is unavailable.';
+        } elseif ($server->node->isUnderMaintenance()) {
+            $message = 'The node of this server is currently under maintenance and the functionality requested is unavailable.';
         } elseif (!$server->isInstalled()) {
             $message = 'This server has not yet completed its installation process, please try again later.';
         } elseif ($server->status === Server::STATUS_RESTORING_BACKUP) {

app/Http/Middleware/Api/Client/Server/AuthenticateServerAccess.php

@@ -53,7 +53,7 @@ public function handle(Request $request, Closure $next): mixed
             // Still allow users to get information about their server if it is installing or
             // being transferred.
             if (!$request->routeIs('api:client:server.view')) {
-                if ($server->isSuspended() && !$request->routeIs('api:client:server.resources')) {
+                if (($server->isSuspended() || $server->node->isUnderMaintenance()) && !$request->routeIs('api:client:server.resources')) {
                     throw $exception;
                 }
                 if (!$user->root_admin || !$request->routeIs($this->except)) {

app/Http/Requests/Api/Application/Nodes/StoreNodeRequest.php

@@ -24,6 +24,7 @@ public function rules(array $rules = null): array
             'fqdn',
             'scheme',
             'behind_proxy',
+            'maintenance_mode',
             'memory',
             'memory_overallocate',
             'disk',

app/Models/Node.php

@@ -186,6 +186,11 @@ public function getDecryptedKey(): string
         );
     }
 
+    public function isUnderMaintenance(): bool
+    {
+        return $this->maintenance_mode;
+    }
+
     public function mounts(): HasManyThrough
     {
         return $this->hasManyThrough(Mount::class, MountNode::class, 'node_id', 'id', 'id', 'mount_id');

app/Models/Server.php

@@ -354,6 +354,7 @@ public function validateCurrentState()
     {
         if (
             $this->isSuspended() ||
+            $this->node->isUnderMaintenance() ||
             !$this->isInstalled() ||
             $this->status === self::STATUS_RESTORING_BACKUP ||
             !is_null($this->transfer)

app/Transformers/Api/Client/ServerTransformer.php

@@ -43,6 +43,7 @@ public function transform(Server $server): array
             'uuid' => $server->uuid,
             'name' => $server->name,
             'node' => $server->node->name,
+            'is_node_under_maintenance' => $server->node->isUnderMaintenance(),
             'sftp_details' => [
                 'ip' => $server->node->fqdn,
                 'port' => $server->node->daemonSFTP,

resources/scripts/api/server/getServer.ts

@@ -17,6 +17,7 @@ export interface Server {
     uuid: string;
     name: string;
     node: string;
+    isNodeUnderMaintenance: boolean;
     status: ServerStatus;
     sftpDetails: {
         ip: string;
@@ -50,6 +51,7 @@ export const rawDataToServerObject = ({ attributes: data }: FractalResponseData)
     uuid: data.uuid,
     name: data.name,
     node: data.node,
+    isNodeUnderMaintenance: data.is_node_under_maintenance,
     status: data.status,
     invocation: data.invocation,
     dockerImage: data.docker_image,

resources/scripts/components/server/ConflictStateRenderer.tsx

@@ -8,6 +8,9 @@ import ServerRestoreSvg from '@/assets/images/server_restore.svg';
 export default () => {
     const status = ServerContext.useStoreState((state) => state.server.data?.status || null);
     const isTransferring = ServerContext.useStoreState((state) => state.server.data?.isTransferring || false);
+    const isNodeUnderMaintenance = ServerContext.useStoreState(
+        (state) => state.server.data?.isNodeUnderMaintenance || false
+    );
 
     return status === 'installing' || status === 'install_failed' ? (
         <ScreenBlock
@@ -21,6 +24,12 @@ export default () => {
             image={ServerErrorSvg}
             message={'This server is suspended and cannot be accessed.'}
         />
+    ) : isNodeUnderMaintenance ? (
+        <ScreenBlock
+            title={'Node under Maintenance'}
+            image={ServerErrorSvg}
+            message={'The node of this server is currently under maintenance.'}
+        />
     ) : (
         <ScreenBlock
             title={isTransferring ? 'Transferring' : 'Restoring from Backup'}

resources/scripts/components/server/console/ServerConsoleContainer.tsx

@@ -19,12 +19,15 @@ const ServerConsoleContainer = () => {
     const isInstalling = ServerContext.useStoreState((state) => state.server.isInstalling);
     const isTransferring = ServerContext.useStoreState((state) => state.server.data!.isTransferring);
     const eggFeatures = ServerContext.useStoreState((state) => state.server.data!.eggFeatures, isEqual);
+    const isNodeUnderMaintenance = ServerContext.useStoreState((state) => state.server.data!.isNodeUnderMaintenance);
 
     return (
         <ServerContentBlock title={'Console'}>
-            {(isInstalling || isTransferring) && (
+            {(isNodeUnderMaintenance || isInstalling || isTransferring) && (
                 <Alert type={'warning'} className={'mb-4'}>
-                    {isInstalling
+                    {isNodeUnderMaintenance
+                        ? 'The node of this server is currently under maintenance and all actions are unavailable.'
+                        : isInstalling
                         ? 'This server is currently running its installation process and most actions are unavailable.'
                         : 'This server is currently being transferred to another node and all actions are unavailable.'}
                 </Alert>

resources/scripts/state/server/index.ts

@@ -30,7 +30,7 @@ const server: ServerDataStore = {
             return false;
         }
 
-        return state.data.status !== null || state.data.isTransferring;
+        return state.data.status !== null || state.data.isTransferring || state.data.isNodeUnderMaintenance;
     }),
 
     isInstalling: computed((state) => {