From 7353e88567159bd6a33fca8c7540325ef658d5a3 Mon Sep 17 00:00:00 2001
From: James Murdza <james@jamesmurdza.com>
Date: Fri, 14 Jun 2024 12:28:19 -0400
Subject: [PATCH] fix: wait until terminals are killed to close the container

---
 backend/server/src/index.ts | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/backend/server/src/index.ts b/backend/server/src/index.ts
index ebf8df4..da51a46 100644
--- a/backend/server/src/index.ts
+++ b/backend/server/src/index.ts
@@ -432,11 +432,12 @@ io.on("connection", async (socket) => {
 
   socket.on("disconnect", async () => {
     if (data.isOwner) {
-      Object.entries(terminals).forEach((t) => {
-        const terminal = t[1];
-        terminal.kill();
-        delete terminals[t[0]];
-      });
+      await Promise.all(
+        Object.entries(terminals).map(async ([key, terminal]) => {
+          await terminal.kill();
+          delete terminals[key];
+        })
+      );
 
       await lockManager.acquireLock(data.sandboxId, async () => {
         try {