draft update protocl

1 files changed, 5 insertions, 4 deletions

diff --git a/src/server/session_manager/session_manager.ts b/src/server/session_manager/session_manager.ts
index d5ef94dc7..35cc833f2 100644
--- a/src/server/session_manager/session_manager.ts
+++ b/src/server/session_manager/session_manager.ts
@@ -23,13 +23,15 @@ registerCommand("restart", [], async () => {
     await endPrevious();
 });
 
-registerCommand("exit", [], async () => {
+registerCommand("exit", [], exit);
+
+async function exit() {
     set(SessionState.EXITING);
     identifiedLog(cyan("Initializing session end"));
     await endPrevious();
     identifiedLog("Cleanup complete. Exiting session...\n");
     execSync(killAllCommand());
-});
+}
 
 registerCommand("update", [], async () => {
     set(SessionState.UPDATING);
@@ -43,8 +45,7 @@ registerCommand("update", [], async () => {
             resolve();
         });
     });
-    set(SessionState.MANUALLY_RESTARTING);
-    identifiedLog("Update complete. Initializing manual restart...\n");
+    await exit();
 });
 
 registerCommand("state", [], () => identifiedLog(state));