docs: add information about E2B

README.md

@@ -29,7 +29,9 @@ npm run dev
 
 ### Backend
 
-The backend consists of a primary Express and Socket.io server, and 3 Cloudflare Workers microservices for the D1 database, R2 storage, and Workers AI. The D1 database also contains a [service binding](https://developers.cloudflare.com/workers/runtime-apis/bindings/service-bindings/) to the R2 storage worker.
+The backend consists of a primary Express and Socket.io server, and 3 Cloudflare Workers microservices for the D1 database, R2 storage, and Workers AI. The D1 database also contains a [service binding](https://developers.cloudflare.com/workers/runtime-apis/bindings/service-bindings/) to the R2 storage worker. Each open sandbox instantiates a secure Linux sandboxes on E2B, which is used for the terminal and live preview.
+
+You will need to make an account on [E2B](https://e2b.dev/) to get an API key.
 
 #### Socket.io server
 
@@ -181,3 +183,4 @@ It should be in the form `category(scope or module): message` in your commit mes
 - [Express](https://expressjs.com/)
 - [Socket.io](https://socket.io/)
 - [Drizzle ORM](https://orm.drizzle.team/)
+- [E2B](https://e2b.dev/)

backend/database/drizzle/meta/0000_snapshot.json

@@ -1,7 +1,7 @@
 {
   "version": "5",
   "dialect": "sqlite",
-    "id": "70e3d022-aed8-4464-9063-c92113b4ebe2",
+  "id": "6570ba20-a672-400c-8147-7ba533784918",
   "prevId": "00000000-0000-0000-0000-000000000000",
   "tables": {
     "sandbox": {
@@ -35,13 +35,6 @@
           "notNull": false,
           "autoincrement": false
         },
-          "createdAt": {
-            "name": "createdAt",
-            "type": "integer",
-            "primaryKey": false,
-            "notNull": false,
-            "autoincrement": false
-          },
         "user_id": {
           "name": "user_id",
           "type": "text",
@@ -100,21 +93,6 @@
           "primaryKey": false,
           "notNull": true,
           "autoincrement": false
-          },
-          "image": {
-            "name": "image",
-            "type": "text",
-            "primaryKey": false,
-            "notNull": false,
-            "autoincrement": false
-          },
-          "generations": {
-            "name": "generations",
-            "type": "integer",
-            "primaryKey": false,
-            "notNull": false,
-            "autoincrement": false,
-            "default": 0
         }
       },
       "indexes": {
@@ -146,13 +124,6 @@
           "primaryKey": false,
           "notNull": true,
           "autoincrement": false
-          },
-          "sharedOn": {
-            "name": "sharedOn",
-            "type": "integer",
-            "primaryKey": false,
-            "notNull": false,
-            "autoincrement": false
         }
       },
       "indexes": {},
@@ -194,4 +165,4 @@
     "tables": {},
     "columns": {}
   }
-  }
+}

backend/database/drizzle/meta/0001_snapshot.json

@@ -0,0 +1,175 @@
+{
+  "version": "5",
+  "dialect": "sqlite",
+  "id": "9f64104a-4954-40c0-8155-17755ea0a243",
+  "prevId": "6570ba20-a672-400c-8147-7ba533784918",
+  "tables": {
+    "sandbox": {
+      "name": "sandbox",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "text",
+          "primaryKey": true,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "name": {
+          "name": "name",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "type": {
+          "name": "type",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "visibility": {
+          "name": "visibility",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        },
+        "user_id": {
+          "name": "user_id",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        }
+      },
+      "indexes": {
+        "sandbox_id_unique": {
+          "name": "sandbox_id_unique",
+          "columns": [
+            "id"
+          ],
+          "isUnique": true
+        }
+      },
+      "foreignKeys": {
+        "sandbox_user_id_user_id_fk": {
+          "name": "sandbox_user_id_user_id_fk",
+          "tableFrom": "sandbox",
+          "tableTo": "user",
+          "columnsFrom": [
+            "user_id"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "no action",
+          "onUpdate": "no action"
+        }
+      },
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {}
+    },
+    "user": {
+      "name": "user",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "text",
+          "primaryKey": true,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "name": {
+          "name": "name",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "email": {
+          "name": "email",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "image": {
+          "name": "image",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        }
+      },
+      "indexes": {
+        "user_id_unique": {
+          "name": "user_id_unique",
+          "columns": [
+            "id"
+          ],
+          "isUnique": true
+        }
+      },
+      "foreignKeys": {},
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {}
+    },
+    "users_to_sandboxes": {
+      "name": "users_to_sandboxes",
+      "columns": {
+        "userId": {
+          "name": "userId",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "sandboxId": {
+          "name": "sandboxId",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        }
+      },
+      "indexes": {},
+      "foreignKeys": {
+        "users_to_sandboxes_userId_user_id_fk": {
+          "name": "users_to_sandboxes_userId_user_id_fk",
+          "tableFrom": "users_to_sandboxes",
+          "tableTo": "user",
+          "columnsFrom": [
+            "userId"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "no action",
+          "onUpdate": "no action"
+        },
+        "users_to_sandboxes_sandboxId_sandbox_id_fk": {
+          "name": "users_to_sandboxes_sandboxId_sandbox_id_fk",
+          "tableFrom": "users_to_sandboxes",
+          "tableTo": "sandbox",
+          "columnsFrom": [
+            "sandboxId"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "no action",
+          "onUpdate": "no action"
+        }
+      },
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {}
+    }
+  },
+  "enums": {},
+  "_meta": {
+    "schemas": {},
+    "tables": {},
+    "columns": {}
+  }
+}

backend/database/drizzle/meta/0002_snapshot.json

@@ -0,0 +1,168 @@
+{
+  "version": "5",
+  "dialect": "sqlite",
+  "id": "5baf10d6-7697-42ba-a11a-ee4c7bd7e91e",
+  "prevId": "9f64104a-4954-40c0-8155-17755ea0a243",
+  "tables": {
+    "sandbox": {
+      "name": "sandbox",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "text",
+          "primaryKey": true,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "name": {
+          "name": "name",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "type": {
+          "name": "type",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "visibility": {
+          "name": "visibility",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        },
+        "user_id": {
+          "name": "user_id",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        }
+      },
+      "indexes": {
+        "sandbox_id_unique": {
+          "name": "sandbox_id_unique",
+          "columns": [
+            "id"
+          ],
+          "isUnique": true
+        }
+      },
+      "foreignKeys": {
+        "sandbox_user_id_user_id_fk": {
+          "name": "sandbox_user_id_user_id_fk",
+          "tableFrom": "sandbox",
+          "tableTo": "user",
+          "columnsFrom": [
+            "user_id"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "no action",
+          "onUpdate": "no action"
+        }
+      },
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {}
+    },
+    "user": {
+      "name": "user",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "text",
+          "primaryKey": true,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "name": {
+          "name": "name",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "email": {
+          "name": "email",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        }
+      },
+      "indexes": {
+        "user_id_unique": {
+          "name": "user_id_unique",
+          "columns": [
+            "id"
+          ],
+          "isUnique": true
+        }
+      },
+      "foreignKeys": {},
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {}
+    },
+    "users_to_sandboxes": {
+      "name": "users_to_sandboxes",
+      "columns": {
+        "userId": {
+          "name": "userId",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "sandboxId": {
+          "name": "sandboxId",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        }
+      },
+      "indexes": {},
+      "foreignKeys": {
+        "users_to_sandboxes_userId_user_id_fk": {
+          "name": "users_to_sandboxes_userId_user_id_fk",
+          "tableFrom": "users_to_sandboxes",
+          "tableTo": "user",
+          "columnsFrom": [
+            "userId"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "no action",
+          "onUpdate": "no action"
+        },
+        "users_to_sandboxes_sandboxId_sandbox_id_fk": {
+          "name": "users_to_sandboxes_sandboxId_sandbox_id_fk",
+          "tableFrom": "users_to_sandboxes",
+          "tableTo": "sandbox",
+          "columnsFrom": [
+            "sandboxId"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "no action",
+          "onUpdate": "no action"
+        }
+      },
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {}
+    }
+  },
+  "enums": {},
+  "_meta": {
+    "schemas": {},
+    "tables": {},
+    "columns": {}
+  }
+}

backend/database/drizzle/meta/0003_snapshot.json

@@ -0,0 +1,175 @@
+{
+  "version": "5",
+  "dialect": "sqlite",
+  "id": "37e38b82-1494-4818-8c26-b9024cce3fa9",
+  "prevId": "5baf10d6-7697-42ba-a11a-ee4c7bd7e91e",
+  "tables": {
+    "sandbox": {
+      "name": "sandbox",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "text",
+          "primaryKey": true,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "name": {
+          "name": "name",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "type": {
+          "name": "type",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "visibility": {
+          "name": "visibility",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        },
+        "user_id": {
+          "name": "user_id",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        }
+      },
+      "indexes": {
+        "sandbox_id_unique": {
+          "name": "sandbox_id_unique",
+          "columns": [
+            "id"
+          ],
+          "isUnique": true
+        }
+      },
+      "foreignKeys": {
+        "sandbox_user_id_user_id_fk": {
+          "name": "sandbox_user_id_user_id_fk",
+          "tableFrom": "sandbox",
+          "tableTo": "user",
+          "columnsFrom": [
+            "user_id"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "no action",
+          "onUpdate": "no action"
+        }
+      },
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {}
+    },
+    "user": {
+      "name": "user",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "text",
+          "primaryKey": true,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "name": {
+          "name": "name",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "email": {
+          "name": "email",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "image": {
+          "name": "image",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        }
+      },
+      "indexes": {
+        "user_id_unique": {
+          "name": "user_id_unique",
+          "columns": [
+            "id"
+          ],
+          "isUnique": true
+        }
+      },
+      "foreignKeys": {},
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {}
+    },
+    "users_to_sandboxes": {
+      "name": "users_to_sandboxes",
+      "columns": {
+        "userId": {
+          "name": "userId",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "sandboxId": {
+          "name": "sandboxId",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        }
+      },
+      "indexes": {},
+      "foreignKeys": {
+        "users_to_sandboxes_userId_user_id_fk": {
+          "name": "users_to_sandboxes_userId_user_id_fk",
+          "tableFrom": "users_to_sandboxes",
+          "tableTo": "user",
+          "columnsFrom": [
+            "userId"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "no action",
+          "onUpdate": "no action"
+        },
+        "users_to_sandboxes_sandboxId_sandbox_id_fk": {
+          "name": "users_to_sandboxes_sandboxId_sandbox_id_fk",
+          "tableFrom": "users_to_sandboxes",
+          "tableTo": "sandbox",
+          "columnsFrom": [
+            "sandboxId"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "no action",
+          "onUpdate": "no action"
+        }
+      },
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {}
+    }
+  },
+  "enums": {},
+  "_meta": {
+    "schemas": {},
+    "tables": {},
+    "columns": {}
+  }
+}

backend/database/drizzle/meta/_journal.json

@@ -5,9 +5,51 @@
     {
       "idx": 0,
       "version": "5",
-        "when": 1718032216030,
+      "when": 1714540200800,
-        "tag": "0000_melodic_the_captain",
+      "tag": "0000_big_rogue",
+      "breakpoints": true
+    },
+    {
+      "idx": 1,
+      "version": "5",
+      "when": 1714541190588,
+      "tag": "0001_empty_black_knight",
+      "breakpoints": true
+    },
+    {
+      "idx": 2,
+      "version": "5",
+      "when": 1714541209173,
+      "tag": "0002_sour_ego",
+      "breakpoints": true
+    },
+    {
+      "idx": 3,
+      "version": "5",
+      "when": 1714541233589,
+      "tag": "0003_pale_overlord",
+      "breakpoints": true
+    },
+    {
+      "idx": 4,
+      "version": "5",
+      "when": 1714565073180,
+      "tag": "0004_cuddly_wolf_cub",
+      "breakpoints": true
+    },
+    {
+      "idx": 5,
+      "version": "5",
+      "when": 1714950365718,
+      "tag": "0005_last_the_twelve",
+      "breakpoints": true
+    },
+    {
+      "idx": 6,
+      "version": "5",
+      "when": 1716432225404,
+      "tag": "0006_lively_mattie_franklin",
       "breakpoints": true
     }
   ]
-  }
+}

backend/database/src/index.ts

@@ -110,8 +110,13 @@ export default {
 				const body = await request.json()
 				const { type, name, userId, visibility } = initSchema.parse(body)
 
-				const allSandboxes = await db.select().from(sandbox).all()
+				const userSandboxes = await db
-				if (allSandboxes.length >= 8) {
+					.select()
+					.from(sandbox)
+					.where(eq(sandbox.userId, userId))
+					.all()
+
+				if (userSandboxes.length >= 8) {
 					return new Response("You reached the maximum # of sandboxes.", {
 						status: 400,
 					})

backend/database/src/schema.ts

@@ -31,7 +31,7 @@ export const sandbox = sqliteTable("sandbox", {
 	createdAt: integer("createdAt", { mode: "timestamp_ms" }),
 	userId: text("user_id")
 		.notNull()
-		.references(() => user.id, { onDelete: "cascade" }),
+		.references(() => user.id),
 });
 
 export type Sandbox = typeof sandbox.$inferSelect;

backend/server/.env.example

@@ -5,3 +5,4 @@ PORT=4000
 WORKERS_KEY=
 DATABASE_WORKER_URL=
 STORAGE_WORKER_URL=
+E2B_API_KEY=

backend/server/package-lock.json

@@ -12,8 +12,8 @@
         "concurrently": "^8.2.2",
         "cors": "^2.8.5",
         "dotenv": "^16.4.5",
+        "e2b": "^0.16.1",
         "express": "^4.19.2",
-        "node-pty": "^1.0.0",
         "rate-limiter-flexible": "^5.0.3",
         "socket.io": "^4.7.5",
         "zod": "^3.22.4"
@@ -369,6 +369,19 @@
         "node": ">=8"
       }
     },
+    "node_modules/bufferutil": {
+      "version": "4.0.8",
+      "resolved": "https://registry.npmjs.org/bufferutil/-/bufferutil-4.0.8.tgz",
+      "integrity": "sha512-4T53u4PdgsXqKaIctwF8ifXlRTTmEPJ8iEPWFdGZvcf7sbwYo6FKFEX9eNNAnzFZ7EzJAQ3CJeOtCRA4rDp7Pw==",
+      "hasInstallScript": true,
+      "optional": true,
+      "dependencies": {
+        "node-gyp-build": "^4.3.0"
+      },
+      "engines": {
+        "node": ">=6.14.2"
+      }
+    },
     "node_modules/bytes": {
       "version": "3.1.2",
       "resolved": "https://registry.npmjs.org/bytes/-/bytes-3.1.2.tgz",
@@ -662,6 +675,59 @@
         "url": "https://dotenvx.com"
       }
     },
+    "node_modules/e2b": {
+      "version": "0.16.1",
+      "resolved": "https://registry.npmjs.org/e2b/-/e2b-0.16.1.tgz",
+      "integrity": "sha512-2L1R/REEB+EezD4Q4MmcXXNATjvCYov2lv/69+PY6V95+wl1PZblIMTYAe7USxX6P6sqANxNs+kXqZr6RvXkSw==",
+      "dependencies": {
+        "isomorphic-ws": "^5.0.0",
+        "normalize-path": "^3.0.0",
+        "openapi-typescript-fetch": "^1.1.3",
+        "path-browserify": "^1.0.1",
+        "platform": "^1.3.6",
+        "ws": "^8.15.1"
+      },
+      "engines": {
+        "node": ">=18"
+      },
+      "optionalDependencies": {
+        "bufferutil": "^4.0.8",
+        "utf-8-validate": "^6.0.3"
+      }
+    },
+    "node_modules/e2b/node_modules/utf-8-validate": {
+      "version": "6.0.4",
+      "resolved": "https://registry.npmjs.org/utf-8-validate/-/utf-8-validate-6.0.4.tgz",
+      "integrity": "sha512-xu9GQDeFp+eZ6LnCywXN/zBancWvOpUMzgjLPSjy4BRHSmTelvn2E0DG0o1sTiw5hkCKBHo8rwSKncfRfv2EEQ==",
+      "hasInstallScript": true,
+      "optional": true,
+      "dependencies": {
+        "node-gyp-build": "^4.3.0"
+      },
+      "engines": {
+        "node": ">=6.14.2"
+      }
+    },
+    "node_modules/e2b/node_modules/ws": {
+      "version": "8.17.0",
+      "resolved": "https://registry.npmjs.org/ws/-/ws-8.17.0.tgz",
+      "integrity": "sha512-uJq6108EgZMAl20KagGkzCKfMEjxmKvZHG7Tlq0Z6nOky7YF7aq4mOx6xK8TJ/i1LeK4Qus7INktacctDgY8Ow==",
+      "engines": {
+        "node": ">=10.0.0"
+      },
+      "peerDependencies": {
+        "bufferutil": "^4.0.1",
+        "utf-8-validate": ">=5.0.2"
+      },
+      "peerDependenciesMeta": {
+        "bufferutil": {
+          "optional": true
+        },
+        "utf-8-validate": {
+          "optional": true
+        }
+      }
+    },
     "node_modules/ee-first": {
       "version": "1.1.1",
       "resolved": "https://registry.npmjs.org/ee-first/-/ee-first-1.1.1.tgz",
@@ -1082,6 +1148,14 @@
         "node": ">=0.12.0"
       }
     },
+    "node_modules/isomorphic-ws": {
+      "version": "5.0.0",
+      "resolved": "https://registry.npmjs.org/isomorphic-ws/-/isomorphic-ws-5.0.0.tgz",
+      "integrity": "sha512-muId7Zzn9ywDsyXgTIafTry2sV3nySZeUDe6YedVd1Hvuuep5AsIlqK+XefWpYTyJG5e503F2xIuT2lcU6rCSw==",
+      "peerDependencies": {
+        "ws": "*"
+      }
+    },
     "node_modules/lodash": {
       "version": "4.17.21",
       "resolved": "https://registry.npmjs.org/lodash/-/lodash-4.17.21.tgz",
@@ -1173,11 +1247,6 @@
       "resolved": "https://registry.npmjs.org/ms/-/ms-2.0.0.tgz",
       "integrity": "sha512-Tpp60P6IUJDTuOq/5Z8cdskzJujfwqfOTkrwIwj7IRISpnkJnT6SyJ4PCPnGMoFjC9ddhal5KVIYtAt97ix05A=="
     },
-    "node_modules/nan": {
-      "version": "2.19.0",
-      "resolved": "https://registry.npmjs.org/nan/-/nan-2.19.0.tgz",
-      "integrity": "sha512-nO1xXxfh/RWNxfd/XPfbIfFk5vgLsAxUR9y5O0cHMJu/AW9U95JLXqthYHjEp+8gQ5p96K9jUp8nbVOxCdRbtw=="
-    },
     "node_modules/negotiator": {
       "version": "0.6.3",
       "resolved": "https://registry.npmjs.org/negotiator/-/negotiator-0.6.3.tgz",
@@ -1186,13 +1255,15 @@
         "node": ">= 0.6"
       }
     },
-    "node_modules/node-pty": {
+    "node_modules/node-gyp-build": {
-      "version": "1.0.0",
+      "version": "4.8.1",
-      "resolved": "https://registry.npmjs.org/node-pty/-/node-pty-1.0.0.tgz",
+      "resolved": "https://registry.npmjs.org/node-gyp-build/-/node-gyp-build-4.8.1.tgz",
-      "integrity": "sha512-wtBMWWS7dFZm/VgqElrTvtfMq4GzJ6+edFI0Y0zyzygUSZMgZdraDUMUhCIvkjhJjme15qWmbyJbtAx4ot4uZA==",
+      "integrity": "sha512-OSs33Z9yWr148JZcbZd5WiAXhh/n9z8TxQcdMhIOlpN9AhWpLfvVFO73+m77bBABQMaY9XSvIa+qk0jlI7Gcaw==",
-      "hasInstallScript": true,
+      "optional": true,
-      "dependencies": {
+      "bin": {
-        "nan": "^2.17.0"
+        "node-gyp-build": "bin.js",
+        "node-gyp-build-optional": "optional.js",
+        "node-gyp-build-test": "build-test.js"
       }
     },
     "node_modules/nodemon": {
@@ -1265,7 +1336,6 @@
       "version": "3.0.0",
       "resolved": "https://registry.npmjs.org/normalize-path/-/normalize-path-3.0.0.tgz",
       "integrity": "sha512-6eZs5Ls3WtCisHWp9S2GUy8dqkpGi4BVSz3GaqiE6ezub0512ESztXUwUB6C6IKbQkY2Pnb/mD4WYojCRwcwLA==",
-      "dev": true,
       "engines": {
         "node": ">=0.10.0"
       }
@@ -1297,6 +1367,15 @@
         "node": ">= 0.8"
       }
     },
+    "node_modules/openapi-typescript-fetch": {
+      "version": "1.1.3",
+      "resolved": "https://registry.npmjs.org/openapi-typescript-fetch/-/openapi-typescript-fetch-1.1.3.tgz",
+      "integrity": "sha512-smLZPck4OkKMNExcw8jMgrMOGgVGx2N/s6DbKL2ftNl77g5HfoGpZGFy79RBzU/EkaO0OZpwBnslfdBfh7ZcWg==",
+      "engines": {
+        "node": ">= 12.0.0",
+        "npm": ">= 7.0.0"
+      }
+    },
     "node_modules/parseurl": {
       "version": "1.3.3",
       "resolved": "https://registry.npmjs.org/parseurl/-/parseurl-1.3.3.tgz",
@@ -1305,6 +1384,11 @@
         "node": ">= 0.8"
       }
     },
+    "node_modules/path-browserify": {
+      "version": "1.0.1",
+      "resolved": "https://registry.npmjs.org/path-browserify/-/path-browserify-1.0.1.tgz",
+      "integrity": "sha512-b7uo2UCUOYZcnF/3ID0lulOJi/bafxa1xPe7ZPsammBSpjSWQkjNxlt635YGS2MiR9GjvuXCtz2emr3jbsz98g=="
+    },
     "node_modules/path-to-regexp": {
       "version": "0.1.7",
       "resolved": "https://registry.npmjs.org/path-to-regexp/-/path-to-regexp-0.1.7.tgz",
@@ -1322,6 +1406,11 @@
         "url": "https://github.com/sponsors/jonschlinkert"
       }
     },
+    "node_modules/platform": {
+      "version": "1.3.6",
+      "resolved": "https://registry.npmjs.org/platform/-/platform-1.3.6.tgz",
+      "integrity": "sha512-fnWVljUchTro6RiCFvCXBbNhJc2NijN7oIQxbwsyL0buWJPG85v81ehlHI9fXrJsMNgTofEoWIQeClKpgxFLrg=="
+    },
     "node_modules/proxy-addr": {
       "version": "2.0.7",
       "resolved": "https://registry.npmjs.org/proxy-addr/-/proxy-addr-2.0.7.tgz",
@@ -1835,6 +1924,20 @@
         "node": ">= 0.8"
       }
     },
+    "node_modules/utf-8-validate": {
+      "version": "5.0.10",
+      "resolved": "https://registry.npmjs.org/utf-8-validate/-/utf-8-validate-5.0.10.tgz",
+      "integrity": "sha512-Z6czzLq4u8fPOyx7TU6X3dvUZVvoJmxSQ+IcrlmagKhilxlhZgxPK6C5Jqbkw1IDUmFTM+cz9QDnnLTwDz/2gQ==",
+      "hasInstallScript": true,
+      "optional": true,
+      "peer": true,
+      "dependencies": {
+        "node-gyp-build": "^4.3.0"
+      },
+      "engines": {
+        "node": ">=6.14.2"
+      }
+    },
     "node_modules/utils-merge": {
       "version": "1.0.1",
       "resolved": "https://registry.npmjs.org/utils-merge/-/utils-merge-1.0.1.tgz",

backend/server/package.json

@@ -14,8 +14,8 @@
     "concurrently": "^8.2.2",
     "cors": "^2.8.5",
     "dotenv": "^16.4.5",
+    "e2b": "^0.16.1",
     "express": "^4.19.2",
-    "node-pty": "^1.0.0",
     "rate-limiter-flexible": "^5.0.3",
     "socket.io": "^4.7.5",
     "zod": "^3.22.4"

backend/server/src/fileoperations.ts

@@ -0,0 +1,177 @@
+import * as dotenv from "dotenv";
+import {
+  R2FileBody,
+  R2Files,
+  Sandbox,
+  TFile,
+  TFileData,
+  TFolder,
+} from "./types";
+
+dotenv.config();
+
+export const getSandboxFiles = async (id: string) => {
+  const res = await fetch(
+    `${process.env.STORAGE_WORKER_URL}/api?sandboxId=${id}`,
+    {
+      headers: {
+        Authorization: `${process.env.WORKERS_KEY}`,
+      },
+    }
+  );
+  const data: R2Files = await res.json();
+
+  const paths = data.objects.map((obj) => obj.key);
+  const processedFiles = await processFiles(paths, id);
+  return processedFiles;
+};
+
+export const getFolder = async (folderId: string) => {
+  const res = await fetch(
+    `${process.env.STORAGE_WORKER_URL}/api?folderId=${folderId}`,
+    {
+      headers: {
+        Authorization: `${process.env.WORKERS_KEY}`,
+      },
+    }
+  );
+  const data: R2Files = await res.json();
+
+  return data.objects.map((obj) => obj.key);
+};
+
+const processFiles = async (paths: string[], id: string) => {
+  const root: TFolder = { id: "/", type: "folder", name: "/", children: [] };
+  const fileData: TFileData[] = [];
+
+  paths.forEach((path) => {
+    const allParts = path.split("/");
+    if (allParts[1] !== id) {
+      return;
+    }
+
+    const parts = allParts.slice(2);
+    let current: TFolder = root;
+
+    for (let i = 0; i < parts.length; i++) {
+      const part = parts[i];
+      const isFile = i === parts.length - 1 && part.includes(".");
+      const existing = current.children.find((child) => child.name === part);
+
+      if (existing) {
+        if (!isFile) {
+          current = existing as TFolder;
+        }
+      } else {
+        if (isFile) {
+          const file: TFile = { id: path, type: "file", name: part };
+          current.children.push(file);
+          fileData.push({ id: path, data: "" });
+        } else {
+          const folder: TFolder = {
+            // id: path, // todo: wrong id. for example, folder "src" ID is: projects/a7vgttfqbgy403ratp7du3ln/src/App.css
+            id: `projects/${id}/${parts.slice(0, i + 1).join("/")}`,
+            type: "folder",
+            name: part,
+            children: [],
+          };
+          current.children.push(folder);
+          current = folder;
+        }
+      }
+    }
+  });
+
+  await Promise.all(
+    fileData.map(async (file) => {
+      const data = await fetchFileContent(file.id);
+      file.data = data;
+    })
+  );
+
+  return {
+    files: root.children,
+    fileData,
+  };
+};
+
+const fetchFileContent = async (fileId: string): Promise<string> => {
+  try {
+    const fileRes = await fetch(
+      `${process.env.STORAGE_WORKER_URL}/api?fileId=${fileId}`,
+      {
+        headers: {
+          Authorization: `${process.env.WORKERS_KEY}`,
+        },
+      }
+    );
+    return await fileRes.text();
+  } catch (error) {
+    console.error("ERROR fetching file:", error);
+    return "";
+  }
+};
+
+export const createFile = async (fileId: string) => {
+  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api`, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `${process.env.WORKERS_KEY}`,
+    },
+    body: JSON.stringify({ fileId }),
+  });
+  return res.ok;
+};
+
+export const renameFile = async (
+  fileId: string,
+  newFileId: string,
+  data: string
+) => {
+  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api/rename`, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `${process.env.WORKERS_KEY}`,
+    },
+    body: JSON.stringify({ fileId, newFileId, data }),
+  });
+  return res.ok;
+};
+
+export const saveFile = async (fileId: string, data: string) => {
+  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api/save`, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `${process.env.WORKERS_KEY}`,
+    },
+    body: JSON.stringify({ fileId, data }),
+  });
+  return res.ok;
+};
+
+export const deleteFile = async (fileId: string) => {
+  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api`, {
+    method: "DELETE",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `${process.env.WORKERS_KEY}`,
+    },
+    body: JSON.stringify({ fileId }),
+  });
+  return res.ok;
+};
+
+export const getProjectSize = async (id: string) => {
+  const res = await fetch(
+    `${process.env.STORAGE_WORKER_URL}/api/size?sandboxId=${id}`,
+    {
+      headers: {
+        Authorization: `${process.env.WORKERS_KEY}`,
+      },
+    }
+  );
+  return (await res.json()).size;
+};

backend/server/src/index.ts

@@ -1,4 +1,3 @@
-import fs from "fs";
 import os from "os";
 import path from "path";
 import cors from "cors";
@@ -17,8 +16,9 @@ import {
   getSandboxFiles,
   renameFile,
   saveFile,
-} from "./utils";
+} from "./fileoperations";
-import { IDisposable, IPty, spawn } from "node-pty";
+import { LockManager } from "./utils";
+import { Sandbox, Terminal, FilesystemManager } from "e2b";
 import {
   MAX_BODY_SIZE,
   createFileRL,
@@ -43,11 +43,21 @@ const io = new Server(httpServer, {
 let inactivityTimeout: NodeJS.Timeout | null = null;
 let isOwnerConnected = false;
 
-const terminals: {
+const containers: Record<string, Sandbox> = {};
-  [id: string]: { terminal: IPty; onData: IDisposable; onExit: IDisposable };
+const connections: Record<string, number> = {};
-} = {};
+const terminals: Record<string, Terminal> = {};
 
-const dirName = path.join(__dirname, "..");
+const dirName = "/home/user";
+
+const moveFile = async (
+  filesystem: FilesystemManager,
+  filePath: string,
+  newFilePath: string
+) => {
+  const fileContents = await filesystem.readBytes(filePath);
+  await filesystem.writeBytes(newFilePath, fileContents);
+  await filesystem.remove(filePath);
+};
 
 io.use(async (socket, next) => {
   const handshakeSchema = z.object({
@@ -100,7 +110,10 @@ io.use(async (socket, next) => {
   next();
 });
 
+const lockManager = new LockManager();
+
 io.on("connection", async (socket) => {
+  try {
     if (inactivityTimeout) clearTimeout(inactivityTimeout);
 
     const data = socket.data as {
@@ -111,6 +124,7 @@ io.on("connection", async (socket) => {
 
     if (data.isOwner) {
       isOwnerConnected = true;
+      connections[data.sandboxId] = (connections[data.sandboxId] ?? 0) + 1;
     } else {
       if (!isOwnerConnected) {
         socket.emit("disableAccess", "The sandbox owner is not connected.");
@@ -118,77 +132,125 @@ io.on("connection", async (socket) => {
       }
     }
 
+    await lockManager.acquireLock(data.sandboxId, async () => {
+      try {
+        if (!containers[data.sandboxId]) {
+          containers[data.sandboxId] = await Sandbox.create();
+          console.log("Created container ", data.sandboxId);
+          io.emit(
+            "previewURL",
+            "https://" + containers[data.sandboxId].getHostname(5173)
+          );
+        }
+      } catch (e: any) {
+        console.error(`Error creating container ${data.sandboxId}:`, e);
+        io.emit("error", `Error: container creation. ${e.message ?? e}`);
+      }
+    });
+
+    // Change the owner of the project directory to user
+    const fixPermissions = async () => {
+      await containers[data.sandboxId].process.startAndWait(
+        `sudo chown -R user "${path.join(dirName, "projects", data.sandboxId)}"`
+      );
+    };
+
     const sandboxFiles = await getSandboxFiles(data.sandboxId);
-  sandboxFiles.fileData.forEach((file) => {
+    sandboxFiles.fileData.forEach(async (file) => {
       const filePath = path.join(dirName, file.id);
-    fs.mkdirSync(path.dirname(filePath), { recursive: true });
+      await containers[data.sandboxId].filesystem.makeDir(
-    fs.writeFile(filePath, file.data, function (err) {
+        path.dirname(filePath)
-      if (err) throw err;
+      );
-    });
+      await containers[data.sandboxId].filesystem.write(filePath, file.data);
     });
+    fixPermissions();
 
     socket.emit("loaded", sandboxFiles.files);
 
     socket.on("getFile", (fileId: string, callback) => {
+      console.log(fileId);
+      try {
         const file = sandboxFiles.fileData.find((f) => f.id === fileId);
         if (!file) return;
 
         callback(file.data);
+      } catch (e: any) {
+        console.error("Error getting file:", e);
+        io.emit("error", `Error: get file. ${e.message ?? e}`);
+      }
     });
 
     socket.on("getFolder", async (folderId: string, callback) => {
+      try {
         const files = await getFolder(folderId);
         callback(files);
+      } catch (e: any) {
+        console.error("Error getting folder:", e);
+        io.emit("error", `Error: get folder. ${e.message ?? e}`);
+      }
     });
 
     // todo: send diffs + debounce for efficiency
     socket.on("saveFile", async (fileId: string, body: string) => {
       try {
-      await saveFileRL.consume(data.userId, 1);
-
         if (Buffer.byteLength(body, "utf-8") > MAX_BODY_SIZE) {
           socket.emit(
-          "rateLimit",
+            "error",
-          "Rate limited: file size too large. Please reduce the file size."
+            "Error: file size too large. Please reduce the file size."
           );
           return;
         }
+        try {
+          await saveFileRL.consume(data.userId, 1);
+          await saveFile(fileId, body);
+        } catch (e) {
+          io.emit("error", "Rate limited: file saving. Please slow down.");
+          return;
+        }
 
         const file = sandboxFiles.fileData.find((f) => f.id === fileId);
         if (!file) return;
         file.data = body;
 
-      fs.writeFile(path.join(dirName, file.id), body, function (err) {
+        await containers[data.sandboxId].filesystem.write(
-        if (err) throw err;
+          path.join(dirName, file.id),
-      });
+          body
-      await saveFile(fileId, body);
+        );
-    } catch (e) {
+        fixPermissions();
-      io.emit("rateLimit", "Rate limited: file saving. Please slow down.");
+      } catch (e: any) {
+        console.error("Error saving file:", e);
+        io.emit("error", `Error: file saving. ${e.message ?? e}`);
       }
     });
 
-  socket.on("moveFile", async (fileId: string, folderId: string, callback) => {
+    socket.on(
+      "moveFile",
+      async (fileId: string, folderId: string, callback) => {
+        try {
           const file = sandboxFiles.fileData.find((f) => f.id === fileId);
           if (!file) return;
 
           const parts = fileId.split("/");
           const newFileId = folderId + "/" + parts.pop();
 
-    fs.rename(
+          await moveFile(
+            containers[data.sandboxId].filesystem,
             path.join(dirName, fileId),
-      path.join(dirName, newFileId),
+            path.join(dirName, newFileId)
-      function (err) {
-        if (err) throw err;
-      }
           );
+          fixPermissions();
 
           file.id = newFileId;
 
           await renameFile(fileId, newFileId, file.data);
           const newFiles = await getSandboxFiles(data.sandboxId);
-
           callback(newFiles.files);
-  });
+        } catch (e: any) {
+          console.error("Error moving file:", e);
+          io.emit("error", `Error: file moving. ${e.message ?? e}`);
+        }
+      }
+    );
 
     socket.on("createFile", async (name: string, callback) => {
       try {
@@ -196,19 +258,27 @@ io.on("connection", async (socket) => {
         // limit is 200mb
         if (size > 200 * 1024 * 1024) {
           io.emit(
-          "rateLimit",
+            "error",
             "Rate limited: project size exceeded. Please delete some files."
           );
           callback({ success: false });
+          return;
         }
 
+        try {
           await createFileRL.consume(data.userId, 1);
+        } catch (e) {
+          io.emit("error", "Rate limited: file creation. Please slow down.");
+          return;
+        }
 
         const id = `projects/${data.sandboxId}/${name}`;
 
-      fs.writeFile(path.join(dirName, id), "", function (err) {
+        await containers[data.sandboxId].filesystem.write(
-        if (err) throw err;
+          path.join(dirName, id),
-      });
+          ""
+        );
+        fixPermissions();
 
         sandboxFiles.files.push({
           id,
@@ -224,30 +294,42 @@ io.on("connection", async (socket) => {
         await createFile(id);
 
         callback({ success: true });
-    } catch (e) {
+      } catch (e: any) {
-      io.emit("rateLimit", "Rate limited: file creation. Please slow down.");
+        console.error("Error creating file:", e);
+        io.emit("error", `Error: file creation. ${e.message ?? e}`);
       }
     });
 
     socket.on("createFolder", async (name: string, callback) => {
+      try {
         try {
           await createFolderRL.consume(data.userId, 1);
+        } catch (e) {
+          io.emit("error", "Rate limited: folder creation. Please slow down.");
+          return;
+        }
 
         const id = `projects/${data.sandboxId}/${name}`;
 
-      fs.mkdir(path.join(dirName, id), { recursive: true }, function (err) {
+        await containers[data.sandboxId].filesystem.makeDir(
-        if (err) throw err;
+          path.join(dirName, id)
-      });
+        );
 
         callback();
-    } catch (e) {
+      } catch (e: any) {
-      io.emit("rateLimit", "Rate limited: folder creation. Please slow down.");
+        console.error("Error creating folder:", e);
+        io.emit("error", `Error: folder creation. ${e.message ?? e}`);
       }
     });
 
     socket.on("renameFile", async (fileId: string, newName: string) => {
+      try {
         try {
           await renameFileRL.consume(data.userId, 1);
+        } catch (e) {
+          io.emit("error", "Rate limited: file renaming. Please slow down.");
+          return;
+        }
 
         const file = sandboxFiles.fileData.find((f) => f.id === fileId);
         if (!file) return;
@@ -257,29 +339,33 @@ io.on("connection", async (socket) => {
         const newFileId =
           parts.slice(0, parts.length - 1).join("/") + "/" + newName;
 
-      fs.rename(
+        await moveFile(
+          containers[data.sandboxId].filesystem,
           path.join(dirName, fileId),
-        path.join(dirName, newFileId),
+          path.join(dirName, newFileId)
-        function (err) {
-          if (err) throw err;
-        }
         );
+        fixPermissions();
         await renameFile(fileId, newFileId, file.data);
-    } catch (e) {
+      } catch (e: any) {
-      io.emit("rateLimit", "Rate limited: file renaming. Please slow down.");
+        console.error("Error renaming folder:", e);
-      return;
+        io.emit("error", `Error: folder renaming. ${e.message ?? e}`);
       }
     });
 
     socket.on("deleteFile", async (fileId: string, callback) => {
+      try {
         try {
           await deleteFileRL.consume(data.userId, 1);
+        } catch (e) {
+          io.emit("error", "Rate limited: file deletion. Please slow down.");
+        }
+
         const file = sandboxFiles.fileData.find((f) => f.id === fileId);
         if (!file) return;
 
-      fs.unlink(path.join(dirName, fileId), function (err) {
+        await containers[data.sandboxId].filesystem.remove(
-        if (err) throw err;
+          path.join(dirName, fileId)
-      });
+        );
         sandboxFiles.fileData = sandboxFiles.fileData.filter(
           (f) => f.id !== fileId
         );
@@ -288,8 +374,9 @@ io.on("connection", async (socket) => {
 
         const newFiles = await getSandboxFiles(data.sandboxId);
         callback(newFiles.files);
-    } catch (e) {
+      } catch (e: any) {
-      io.emit("rateLimit", "Rate limited: file deletion. Please slow down.");
+        console.error("Error deleting file:", e);
+        io.emit("error", `Error: file deletion. ${e.message ?? e}`);
       }
     });
 
@@ -298,13 +385,14 @@ io.on("connection", async (socket) => {
     // });
 
     socket.on("deleteFolder", async (folderId: string, callback) => {
+      try {
         const files = await getFolder(folderId);
 
         await Promise.all(
           files.map(async (file) => {
-        fs.unlink(path.join(dirName, file), function (err) {
+            await containers[data.sandboxId].filesystem.remove(
-          if (err) throw err;
+              path.join(dirName, file)
-        });
+            );
 
             sandboxFiles.fileData = sandboxFiles.fileData.filter(
               (f) => f.id !== file
@@ -317,68 +405,86 @@ io.on("connection", async (socket) => {
         const newFiles = await getSandboxFiles(data.sandboxId);
 
         callback(newFiles.files);
+      } catch (e: any) {
+        console.error("Error deleting folder:", e);
+        io.emit("error", `Error: folder deletion. ${e.message ?? e}`);
+      }
     });
 
-  socket.on("createTerminal", (id: string, callback) => {
+    socket.on("createTerminal", async (id: string, callback) => {
+      try {
         if (terminals[id] || Object.keys(terminals).length >= 4) {
           return;
         }
 
-    const pty = spawn(os.platform() === "win32" ? "cmd.exe" : "bash", [], {
+        await lockManager.acquireLock(data.sandboxId, async () => {
-      name: "xterm",
+          try {
-      cols: 100,
+            terminals[id] = await containers[data.sandboxId].terminal.start({
-      cwd: path.join(dirName, "projects", data.sandboxId),
+              onData: (data: string) => {
+                io.emit("terminalResponse", { id, data });
+              },
+              size: { cols: 80, rows: 20 },
+              onExit: () => console.log("Terminal exited", id),
             });
-
+            await terminals[id].sendData(
-    const onData = pty.onData((data) => {
+              `cd "${path.join(dirName, "projects", data.sandboxId)}"\r`
-      io.emit("terminalResponse", {
+            );
-        id,
+            await terminals[id].sendData("export PS1='user> '\rclear\r");
-        data,
+            console.log("Created terminal", id);
+          } catch (e: any) {
+            console.error(`Error creating terminal ${id}:`, e);
+            io.emit("error", `Error: terminal creation. ${e.message ?? e}`);
+          }
         });
-    });
-
-    const onExit = pty.onExit((code) => console.log("exit :(", code));
-
-    pty.write("export PS1='\\u > '\r");
-    pty.write("clear\r");
-
-    terminals[id] = {
-      terminal: pty,
-      onData,
-      onExit,
-    };
 
         callback();
+      } catch (e: any) {
+        console.error(`Error creating terminal ${id}:`, e);
+        io.emit("error", `Error: terminal creation. ${e.message ?? e}`);
+      }
     });
 
-  socket.on("resizeTerminal", (dimensions: { cols: number; rows: number }) => {
+    socket.on(
+      "resizeTerminal",
+      (dimensions: { cols: number; rows: number }) => {
+        try {
           Object.values(terminals).forEach((t) => {
-      t.terminal.resize(dimensions.cols, dimensions.rows);
+            t.resize(dimensions);
-    });
           });
+        } catch (e: any) {
+          console.error("Error resizing terminal:", e);
+          io.emit("error", `Error: terminal resizing. ${e.message ?? e}`);
+        }
+      }
+    );
 
     socket.on("terminalData", (id: string, data: string) => {
+      try {
         if (!terminals[id]) {
           return;
         }
 
-    try {
+        terminals[id].sendData(data);
-      terminals[id].terminal.write(data);
+      } catch (e: any) {
-    } catch (e) {
+        console.error("Error writing to terminal:", e);
-      console.log("Error writing to terminal", e);
+        io.emit("error", `Error: writing to terminal. ${e.message ?? e}`);
       }
     });
 
-  socket.on("closeTerminal", (id: string, callback) => {
+    socket.on("closeTerminal", async (id: string, callback) => {
+      try {
         if (!terminals[id]) {
           return;
         }
 
-    terminals[id].onData.dispose();
+        await terminals[id].kill();
-    terminals[id].onExit.dispose();
         delete terminals[id];
 
         callback();
+      } catch (e: any) {
+        console.error("Error closing terminal:", e);
+        io.emit("error", `Error: closing terminal. ${e.message ?? e}`);
+      }
     });
 
     socket.on(
@@ -390,6 +496,7 @@ io.on("connection", async (socket) => {
         instructions: string,
         callback
       ) => {
+        try {
           const fetchPromise = fetch(
             `${process.env.DATABASE_WORKER_URL}/api/sandbox/generate`,
             {
@@ -423,16 +530,37 @@ io.on("connection", async (socket) => {
           const json = await generateCodeResponse.json();
 
           callback({ response: json.response, success: true });
+        } catch (e: any) {
+          console.error("Error generating code:", e);
+          io.emit("error", `Error: code generation. ${e.message ?? e}`);
+        }
       }
     );
 
     socket.on("disconnect", async () => {
+      try {
         if (data.isOwner) {
-      Object.entries(terminals).forEach((t) => {
+          connections[data.sandboxId]--;
-        const { terminal, onData, onExit } = t[1];
+        }
-        onData.dispose();
+
-        onExit.dispose();
+        if (data.isOwner && connections[data.sandboxId] <= 0) {
-        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 {
+              if (containers[data.sandboxId]) {
+                await containers[data.sandboxId].close();
+                delete containers[data.sandboxId];
+                console.log("Closed container", data.sandboxId);
+              }
+            } catch (error) {
+              console.error("Error closing container ", data.sandboxId, error);
+            }
           });
 
           socket.broadcast.emit(
@@ -457,7 +585,15 @@ io.on("connection", async (socket) => {
         // } else {
         //   console.log("number of sockets", sockets.length);
         // }
+      } catch (e: any) {
+        console.log("Error disconnecting:", e);
+        io.emit("error", `Error: disconnecting. ${e.message ?? e}`);
+      }
     });
+  } catch (e: any) {
+    console.error("Error connecting:", e);
+    io.emit("error", `Error: connection. ${e.message ?? e}`);
+  }
 });
 
 httpServer.listen(port, () => {

backend/server/src/utils.ts

@@ -1,177 +1,23 @@
-import * as dotenv from "dotenv";
+export class LockManager {
-import {
+  private locks: { [key: string]: Promise<any> };
-  R2FileBody,
-  R2Files,
-  Sandbox,
-  TFile,
-  TFileData,
-  TFolder,
-} from "./types";
 
-dotenv.config();
+  constructor() {
-
+    this.locks = {};
-export const getSandboxFiles = async (id: string) => {
-  const res = await fetch(
-    `${process.env.STORAGE_WORKER_URL}/api?sandboxId=${id}`,
-    {
-      headers: {
-        Authorization: `${process.env.WORKERS_KEY}`,
-      },
-    }
-  );
-  const data: R2Files = await res.json();
-
-  const paths = data.objects.map((obj) => obj.key);
-  const processedFiles = await processFiles(paths, id);
-  return processedFiles;
-};
-
-export const getFolder = async (folderId: string) => {
-  const res = await fetch(
-    `${process.env.STORAGE_WORKER_URL}/api?folderId=${folderId}`,
-    {
-      headers: {
-        Authorization: `${process.env.WORKERS_KEY}`,
-      },
-    }
-  );
-  const data: R2Files = await res.json();
-
-  return data.objects.map((obj) => obj.key);
-};
-
-const processFiles = async (paths: string[], id: string) => {
-  const root: TFolder = { id: "/", type: "folder", name: "/", children: [] };
-  const fileData: TFileData[] = [];
-
-  paths.forEach((path) => {
-    const allParts = path.split("/");
-    if (allParts[1] !== id) {
-      return;
   }
 
-    const parts = allParts.slice(2);
+  async acquireLock<T>(key: string, task: () => Promise<T>): Promise<T> {
-    let current: TFolder = root;
+    if (!this.locks[key]) {
-
+      this.locks[key] = new Promise<T>(async (resolve, reject) => {
-    for (let i = 0; i < parts.length; i++) {
-      const part = parts[i];
-      const isFile = i === parts.length - 1 && part.includes(".");
-      const existing = current.children.find((child) => child.name === part);
-
-      if (existing) {
-        if (!isFile) {
-          current = existing as TFolder;
-        }
-      } else {
-        if (isFile) {
-          const file: TFile = { id: path, type: "file", name: part };
-          current.children.push(file);
-          fileData.push({ id: path, data: "" });
-        } else {
-          const folder: TFolder = {
-            // id: path, // todo: wrong id. for example, folder "src" ID is: projects/a7vgttfqbgy403ratp7du3ln/src/App.css
-            id: `projects/${id}/${parts.slice(0, i + 1).join("/")}`,
-            type: "folder",
-            name: part,
-            children: [],
-          };
-          current.children.push(folder);
-          current = folder;
-        }
-      }
-    }
-  });
-
-  await Promise.all(
-    fileData.map(async (file) => {
-      const data = await fetchFileContent(file.id);
-      file.data = data;
-    })
-  );
-
-  return {
-    files: root.children,
-    fileData,
-  };
-};
-
-const fetchFileContent = async (fileId: string): Promise<string> => {
         try {
-    const fileRes = await fetch(
+          const result = await task();
-      `${process.env.STORAGE_WORKER_URL}/api?fileId=${fileId}`,
+          resolve(result);
-      {
-        headers: {
-          Authorization: `${process.env.WORKERS_KEY}`,
-        },
-      }
-    );
-    return await fileRes.text();
         } catch (error) {
-    console.error("ERROR fetching file:", error);
+          reject(error);
-    return "";
+        } finally {
+          delete this.locks[key];
         }
-};
-
-export const createFile = async (fileId: string) => {
-  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api`, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-      Authorization: `${process.env.WORKERS_KEY}`,
-    },
-    body: JSON.stringify({ fileId }),
       });
-  return res.ok;
-};
-
-export const renameFile = async (
-  fileId: string,
-  newFileId: string,
-  data: string
-) => {
-  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api/rename`, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-      Authorization: `${process.env.WORKERS_KEY}`,
-    },
-    body: JSON.stringify({ fileId, newFileId, data }),
-  });
-  return res.ok;
-};
-
-export const saveFile = async (fileId: string, data: string) => {
-  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api/save`, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-      Authorization: `${process.env.WORKERS_KEY}`,
-    },
-    body: JSON.stringify({ fileId, data }),
-  });
-  return res.ok;
-};
-
-export const deleteFile = async (fileId: string) => {
-  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api`, {
-    method: "DELETE",
-    headers: {
-      "Content-Type": "application/json",
-      Authorization: `${process.env.WORKERS_KEY}`,
-    },
-    body: JSON.stringify({ fileId }),
-  });
-  return res.ok;
-};
-
-export const getProjectSize = async (id: string) => {
-  const res = await fetch(
-    `${process.env.STORAGE_WORKER_URL}/api/size?sandboxId=${id}`,
-    {
-      headers: {
-        Authorization: `${process.env.WORKERS_KEY}`,
-      },
     }
-  );
+    return await this.locks[key];
-  return (await res.json()).size;
+  }
-};
+}

frontend/components/editor/index.tsx

@@ -46,7 +46,7 @@ export default function CodeEditor({
   // Initialize socket connection if it doesn't exist
   if (!socketRef.current) {
     socketRef.current = io(
-      `http://localhost:${process.env.NEXT_PUBLIC_SERVER_PORT}?userId=${userData.id}&sandboxId=${sandboxData.id}`,
+      `${window.location.protocol}//${window.location.hostname}:${process.env.NEXT_PUBLIC_SERVER_PORT}?userId=${userData.id}&sandboxId=${sandboxData.id}`,
       {
         timeout: 2000,
       }
@@ -94,6 +94,9 @@ export default function CodeEditor({
     }[]
   >([])
 
+  // Preview state
+  const [previewURL, setPreviewURL] = useState<string>("");
+
   const isOwner = sandboxData.userId === userData.id
   const clerk = useClerk()
 
@@ -388,7 +391,7 @@ export default function CodeEditor({
       setFiles(files)
     }
 
-    const onRateLimit = (message: string) => {
+    const onError = (message: string) => {
       toast.error(message)
     }
 
@@ -410,17 +413,19 @@ export default function CodeEditor({
     socketRef.current?.on("connect", onConnect)
     socketRef.current?.on("disconnect", onDisconnect)
     socketRef.current?.on("loaded", onLoadedEvent)
-    socketRef.current?.on("rateLimit", onRateLimit)
+    socketRef.current?.on("error", onError)
     socketRef.current?.on("terminalResponse", onTerminalResponse)
     socketRef.current?.on("disableAccess", onDisableAccess)
+    socketRef.current?.on("previewURL", setPreviewURL)
 
     return () => {
       socketRef.current?.off("connect", onConnect)
       socketRef.current?.off("disconnect", onDisconnect)
       socketRef.current?.off("loaded", onLoadedEvent)
-      socketRef.current?.off("rateLimit", onRateLimit)
+      socketRef.current?.off("error", onError)
       socketRef.current?.off("terminalResponse", onTerminalResponse)
       socketRef.current?.off("disableAccess", onDisableAccess)
+      socketRef.current?.off("previewURL", setPreviewURL)
     }
     // }, []);
   }, [terminals])
@@ -462,6 +467,7 @@ export default function CodeEditor({
         setActiveFileContent(response);
       });
     }
+
     setEditorLanguage(processFileType(tab.name));
     setActiveFileId(tab.id);
   }, [activeFileId, tabs, debouncedGetFile]);
@@ -768,6 +774,7 @@ export default function CodeEditor({
                   previewPanelRef.current?.expand()
                   setIsPreviewCollapsed(false)
                 }}
+                src={previewURL}
               />
             </ResizablePanel>
             <ResizableHandle />

frontend/components/editor/preview/index.tsx

@@ -15,9 +15,11 @@ import { toast } from "sonner"
 export default function PreviewWindow({
   collapsed,
   open,
+  src
 }: {
   collapsed: boolean
   open: () => void
+  src: string
 }) {
   const ref = useRef<HTMLIFrameElement>(null)
   const [iframeKey, setIframeKey] = useState(0)
@@ -45,7 +47,7 @@ export default function PreviewWindow({
 
                 <PreviewButton
                   onClick={() => {
-                    navigator.clipboard.writeText(`http://localhost:5173`)
+                    navigator.clipboard.writeText(src)
                     toast.info("Copied preview link to clipboard")
                   }}
                 >
@@ -73,7 +75,7 @@ export default function PreviewWindow({
             ref={ref}
             width={"100%"}
             height={"100%"}
-            src={`http://localhost:5173`}
+            src={src}
           />
         </div>
       )}