📯 tRaNsFeR lOgS 📯

resources/scripts/api/server/getWebsocketToken.ts

@@ -5,9 +5,13 @@ interface Response {
     socket: string;
 }
 
-export default (server: string): Promise<Response> => {
+export default (server: string, transfer: boolean): Promise<Response> => {
     return new Promise((resolve, reject) => {
-        http.get(`/api/client/servers/${server}/websocket`)
+        http.get(`/api/client/servers/${server}/websocket`, {
+            params: {
+                transfer,
+            },
+        })
             .then(({ data }) => resolve({
                 token: data.data.token,
                 socket: data.data.socket,

resources/scripts/components/server/Console.tsx

@@ -84,8 +84,6 @@ export default () => {
             // Sent by the source node whenever the server was archived successfully.
             case 'archive':
                 terminal.writeln(TERMINAL_PRELUDE + 'Server has been archived successfully, attempting connection to target node..\u001b[0m');
-                // TODO: Get WebSocket information for the target node.
-                return;
         }
     };
 
@@ -167,7 +165,7 @@ export default () => {
 
     useEffect(() => {
         if (connected && instance) {
-            terminal.clear();
+            // terminal.clear();
 
             instance.addListener('status', handlePowerChangeEvent);
             instance.addListener('console output', handleConsoleOutput);

resources/scripts/components/server/TransferListener.tsx

@@ -0,0 +1,28 @@
+import useWebsocketEvent from '@/plugins/useWebsocketEvent';
+import { ServerContext } from '@/state/server';
+
+const TransferListener = () => {
+    const uuid = ServerContext.useStoreState(state => state.server.data!.uuid);
+    const getServer = ServerContext.useStoreActions(actions => actions.server.getServer);
+    const setServerFromState = ServerContext.useStoreActions(actions => actions.server.setServerFromState);
+
+    // Listen for the installation completion event and then fire off a request to fetch the updated
+    // server information. This allows the server to automatically become available to the user if they
+    // just sit on the page.
+    useWebsocketEvent('transfer status', (status: string) => {
+        if (status === 'starting') {
+            setServerFromState(s => ({ ...s, isTransferring: true }));
+            return;
+        }
+
+        if (status !== 'success') {
+            return;
+        }
+
+        getServer(uuid).catch(error => console.error(error));
+    });
+
+    return null;
+};
+
+export default TransferListener;

resources/scripts/components/server/WebsocketHandler.tsx

@@ -15,39 +15,14 @@ const reconnectErrors = [
 export default () => {
     let updatingToken = false;
     const [ error, setError ] = useState<'connecting' | string>('');
+    const [ transfer, setTransfer ] = useState<boolean>(false);
     const { connected, instance } = ServerContext.useStoreState(state => state.socket);
     const uuid = ServerContext.useStoreState(state => state.server.data?.uuid);
     const setServerStatus = ServerContext.useStoreActions(actions => actions.status.setServerStatus);
     const { setInstance, setConnectionState } = ServerContext.useStoreActions(actions => actions.socket);
 
-    const updateToken = (uuid: string, socket: Websocket) => {
-        if (updatingToken) return;
-
-        updatingToken = true;
-        getWebsocketToken(uuid)
-            .then(data => socket.setToken(data.token, true))
-            .catch(error => console.error(error))
-            .then(() => {
-                updatingToken = false;
-            });
-    };
-
-    useEffect(() => {
-        connected && setError('');
-    }, [ connected ]);
-
-    useEffect(() => {
-        return () => {
-            instance && instance.close();
-        };
-    }, [ instance ]);
-
-    useEffect(() => {
-        // If there is already an instance or there is no server, just exit out of this process
-        // since we don't need to make a new connection.
-        if (instance || !uuid) {
-            return;
-        }
+    const connect = (uuid: string, transfer = false) => {
+        setTransfer(transfer);
 
         const socket = new Websocket();
 
@@ -76,7 +51,34 @@ export default () => {
             }
         });
 
-        getWebsocketToken(uuid)
+        socket.on('transfer status', (status: string) => {
+            if (status === 'success') {
+                setTransfer(false);
+                return;
+            }
+
+            if (status === 'starting') {
+                return;
+            }
+
+            // This doesn't use the `setTransfer` hook as it doesn't want to work properly in this context,
+            // and causes all kinds of fuckery with the websocket.
+            let transfer = false;
+            if (status === 'archived') {
+                transfer = true;
+            }
+
+            // Close the current websocket connection.
+            socket.close();
+
+            setError('connecting');
+            setConnectionState(false);
+            setInstance(null);
+
+            connect(uuid, transfer);
+        });
+
+        getWebsocketToken(uuid, transfer)
             .then(data => {
                 // Connect and then set the authentication token.
                 socket.setToken(data.token).connect(data.socket);
@@ -85,6 +87,38 @@ export default () => {
                 setInstance(socket);
             })
             .catch(error => console.error(error));
+    };
+
+    const updateToken = (uuid: string, socket: Websocket) => {
+        if (updatingToken) return;
+
+        updatingToken = true;
+        getWebsocketToken(uuid, transfer)
+            .then(data => socket.setToken(data.token, true))
+            .catch(error => console.error(error))
+            .then(() => {
+                updatingToken = false;
+            });
+    };
+
+    useEffect(() => {
+        connected && setError('');
+    }, [ connected ]);
+
+    useEffect(() => {
+        return () => {
+            instance && instance.close();
+        };
+    }, [ instance ]);
+
+    useEffect(() => {
+        // If there is already an instance or there is no server, just exit out of this process
+        // since we don't need to make a new connection.
+        if (instance || !uuid) {
+            return;
+        }
+
+        connect(uuid);
     }, [ uuid ]);
 
     return (

resources/scripts/components/server/databases/CreateDatabaseButton.tsx

@@ -42,7 +42,6 @@ export default () => {
                 setVisible(false);
             })
             .catch(error => {
-                console.log(error);
                 addError({ key: 'database:create', message: httpErrorToHuman(error) });
                 setSubmitting(false);
             });

resources/scripts/routers/ServerRouter.tsx

@@ -1,3 +1,4 @@
+import TransferListener from '@/components/server/TransferListener';
 import React, { useEffect, useState } from 'react';
 import { NavLink, Route, RouteComponentProps, Switch } from 'react-router-dom';
 import NavigationBar from '@/components/NavigationBar';
@@ -128,6 +129,7 @@ const ServerRouter = ({ match, location }: RouteComponentProps<{ id: string }>)
                         </SubNavigation>
                     </CSSTransition>
                     <InstallListener/>
+                    <TransferListener/>
                     <WebsocketHandler/>
                     {((installing || transferring) && (!rootAdmin || (rootAdmin && !location.pathname.endsWith(`/server/${id}`)))) ?
                         <ScreenBlock

resources/views/admin/servers/view/manage.blade.php

@@ -72,7 +72,7 @@
                         <form action="{{ route('admin.servers.view.manage.suspension', $server->id) }}" method="POST">
                             {!! csrf_field() !!}
                             <input type="hidden" name="action" value="suspend" />
-                            <button type="submit" class="btn btn-warning">Suspend Server</button>
+                            <button type="submit" class="btn btn-warning @if($server->transfer !== null) disabled @endif">Suspend Server</button>
                         </form>
                     </div>
                 </div>