refactor: separate socket event handlers into functions
This commit is contained in:
James Murdza
parent
6ec17fad7e
commit
cec6b0c8c5
@ -169,8 +169,127 @@ const aiWorker = new AIWorker(
|
||||
process.env.DATABASE_WORKER_URL!,
|
||||
process.env.WORKERS_KEY!
|
||||
)
|
||||
// Handle heartbeat from a socket connection
|
||||
function handleHeartbeat(socket: any, data: any, containers: any) {
|
||||
containers[data.sandboxId].setTimeout(CONTAINER_TIMEOUT)
|
||||
}
|
||||
|
||||
// Handle socket connections
|
||||
// Handle getting a file
|
||||
function handleGetFile(fileManager: FileManager, fileId: string) {
|
||||
return fileManager.getFile(fileId)
|
||||
}
|
||||
|
||||
// Handle getting a folder
|
||||
function handleGetFolder(fileManager: FileManager, folderId: string) {
|
||||
return fileManager.getFolder(folderId)
|
||||
}
|
||||
|
||||
// Handle saving a file
|
||||
function handleSaveFile(fileManager: FileManager, fileId: string, body: string) {
|
||||
return fileManager.saveFile(fileId, body)
|
||||
}
|
||||
|
||||
// Handle moving a file
|
||||
function handleMoveFile(fileManager: FileManager, fileId: string, folderId: string) {
|
||||
return fileManager.moveFile(fileId, folderId)
|
||||
}
|
||||
|
||||
// Handle listing apps
|
||||
async function handleListApps(client: DokkuClient | null) {
|
||||
if (!client) throw Error("Failed to retrieve apps list: No Dokku client")
|
||||
return { success: true, apps: await client.listApps() }
|
||||
}
|
||||
|
||||
// Handle deploying code
|
||||
async function handleDeploy(git: SecureGitClient | null, fileManager: FileManager, sandboxId: string) {
|
||||
if (!git) throw Error("Failed to retrieve apps list: No git client")
|
||||
const fixedFilePaths = fileManager.sandboxFiles.fileData.map((file) => ({
|
||||
...file,
|
||||
id: file.id.split("/").slice(2).join("/"),
|
||||
}))
|
||||
await git.pushFiles(fixedFilePaths, sandboxId)
|
||||
return { success: true }
|
||||
}
|
||||
|
||||
// Handle creating a file
|
||||
function handleCreateFile(fileManager: FileManager, name: string) {
|
||||
return fileManager.createFile(name)
|
||||
}
|
||||
|
||||
// Handle creating a folder
|
||||
function handleCreateFolder(fileManager: FileManager, name: string) {
|
||||
return fileManager.createFolder(name)
|
||||
}
|
||||
|
||||
// Handle renaming a file
|
||||
function handleRenameFile(fileManager: FileManager, fileId: string, newName: string) {
|
||||
return fileManager.renameFile(fileId, newName)
|
||||
}
|
||||
|
||||
// Handle deleting a file
|
||||
function handleDeleteFile(fileManager: FileManager, fileId: string) {
|
||||
return fileManager.deleteFile(fileId)
|
||||
}
|
||||
|
||||
// Handle deleting a folder
|
||||
function handleDeleteFolder(fileManager: FileManager, folderId: string) {
|
||||
return fileManager.deleteFolder(folderId)
|
||||
}
|
||||
|
||||
// Handle creating a terminal session
|
||||
async function handleCreateTerminal(terminalManager: TerminalManager, id: string, socket: any, containers: any, data: any) {
|
||||
await lockManager.acquireLock(data.sandboxId, async () => {
|
||||
await terminalManager.createTerminal(id, (responseString: string) => {
|
||||
socket.emit("terminalResponse", { id, data: responseString })
|
||||
const port = extractPortNumber(responseString)
|
||||
if (port) {
|
||||
socket.emit(
|
||||
"previewURL",
|
||||
"https://" + containers[data.sandboxId].getHost(port)
|
||||
)
|
||||
}
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
// Handle resizing a terminal
|
||||
function handleResizeTerminal(terminalManager: TerminalManager, dimensions: { cols: number; rows: number }) {
|
||||
terminalManager.resizeTerminal(dimensions)
|
||||
}
|
||||
|
||||
// Handle sending data to a terminal
|
||||
function handleTerminalData(terminalManager: TerminalManager, id: string, data: string) {
|
||||
return terminalManager.sendTerminalData(id, data)
|
||||
}
|
||||
|
||||
// Handle closing a terminal
|
||||
function handleCloseTerminal(terminalManager: TerminalManager, id: string) {
|
||||
return terminalManager.closeTerminal(id)
|
||||
}
|
||||
|
||||
// Handle generating code
|
||||
function handleGenerateCode(aiWorker: AIWorker, userId: string, fileName: string, code: string, line: number, instructions: string) {
|
||||
return aiWorker.generateCode(userId, fileName, code, line, instructions)
|
||||
}
|
||||
|
||||
// Handle a client disconnecting from the server
|
||||
async function handleDisconnect(data: any, connections: any, terminalManager: TerminalManager, fileManager: FileManager, socket: any) {
|
||||
if (data.isOwner) {
|
||||
connections[data.sandboxId]--
|
||||
}
|
||||
|
||||
await terminalManager.closeAllTerminals()
|
||||
await fileManager.closeWatchers()
|
||||
|
||||
if (data.isOwner && connections[data.sandboxId] <= 0) {
|
||||
socket.broadcast.emit(
|
||||
"disableAccess",
|
||||
"The sandbox owner has disconnected."
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
// Handle a client connecting to the server
|
||||
io.on("connection", async (socket) => {
|
||||
try {
|
||||
const data = socket.data as {
|
||||
@ -240,72 +359,54 @@ io.on("connection", async (socket) => {
|
||||
// Handle various socket events (heartbeat, file operations, terminal operations, etc.)
|
||||
socket.on("heartbeat", async () => {
|
||||
try {
|
||||
// This keeps the container alive for another CONTAINER_TIMEOUT seconds.
|
||||
// The E2B docs are unclear, but the timeout is relative to the time of this method call.
|
||||
await containers[data.sandboxId].setTimeout(CONTAINER_TIMEOUT)
|
||||
handleHeartbeat(socket, data, containers)
|
||||
} catch (e: any) {
|
||||
console.error("Error setting timeout:", e)
|
||||
socket.emit("error", `Error: set timeout. ${e.message ?? e}`)
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to get file content
|
||||
socket.on("getFile", async (fileId: string, callback) => {
|
||||
try {
|
||||
const fileContent = await fileManager.getFile(fileId)
|
||||
callback(fileContent)
|
||||
callback(await handleGetFile(fileManager, fileId))
|
||||
} catch (e: any) {
|
||||
console.error("Error getting file:", e)
|
||||
socket.emit("error", `Error: get file. ${e.message ?? e}`)
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to get folder contents
|
||||
socket.on("getFolder", async (folderId: string, callback) => {
|
||||
try {
|
||||
const files = await fileManager.getFolder(folderId)
|
||||
callback(files)
|
||||
callback(await handleGetFolder(fileManager, folderId))
|
||||
} catch (e: any) {
|
||||
console.error("Error getting folder:", e)
|
||||
socket.emit("error", `Error: get folder. ${e.message ?? e}`)
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to save file
|
||||
socket.on("saveFile", async (fileId: string, body: string) => {
|
||||
try {
|
||||
await saveFileRL.consume(data.userId, 1)
|
||||
await fileManager.saveFile(fileId, body)
|
||||
await handleSaveFile(fileManager, fileId, body)
|
||||
} catch (e: any) {
|
||||
console.error("Error saving file:", e)
|
||||
socket.emit("error", `Error: file saving. ${e.message ?? e}`)
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to move file
|
||||
socket.on(
|
||||
"moveFile",
|
||||
async (fileId: string, folderId: string, callback) => {
|
||||
socket.on("moveFile", async (fileId: string, folderId: string, callback) => {
|
||||
try {
|
||||
const newFiles = await fileManager.moveFile(fileId, folderId)
|
||||
callback(newFiles)
|
||||
callback(await handleMoveFile(fileManager, fileId, folderId))
|
||||
} catch (e: any) {
|
||||
console.error("Error moving file:", e)
|
||||
socket.emit("error", `Error: file moving. ${e.message ?? e}`)
|
||||
}
|
||||
}
|
||||
)
|
||||
})
|
||||
|
||||
// Handle request to list apps
|
||||
socket.on("list", async (callback: (response: DokkuResponse) => void) => {
|
||||
console.log("Retrieving apps list...")
|
||||
try {
|
||||
if (!client)
|
||||
throw Error("Failed to retrieve apps list: No Dokku client")
|
||||
callback({
|
||||
success: true,
|
||||
apps: await client.listApps(),
|
||||
})
|
||||
callback(await handleListApps(client))
|
||||
} catch (error) {
|
||||
callback({
|
||||
success: false,
|
||||
@ -314,24 +415,10 @@ io.on("connection", async (socket) => {
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to deploy project
|
||||
socket.on("deploy", async (callback: (response: DokkuResponse) => void) => {
|
||||
try {
|
||||
// Push the project files to the Dokku server
|
||||
console.log("Deploying project ${data.sandboxId}...")
|
||||
if (!git) throw Error("Failed to retrieve apps list: No git client")
|
||||
// Remove the /project/[id]/ component of each file path:
|
||||
const fixedFilePaths = fileManager.sandboxFiles.fileData.map((file) => {
|
||||
return {
|
||||
...file,
|
||||
id: file.id.split("/").slice(2).join("/"),
|
||||
}
|
||||
})
|
||||
// Push all files to Dokku.
|
||||
await git.pushFiles(fixedFilePaths, data.sandboxId)
|
||||
callback({
|
||||
success: true,
|
||||
})
|
||||
callback(await handleDeploy(git, fileManager, data.sandboxId))
|
||||
} catch (error) {
|
||||
callback({
|
||||
success: false,
|
||||
@ -340,23 +427,20 @@ io.on("connection", async (socket) => {
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to create a new file
|
||||
socket.on("createFile", async (name: string, callback) => {
|
||||
try {
|
||||
await createFileRL.consume(data.userId, 1)
|
||||
const success = await fileManager.createFile(name)
|
||||
callback({ success })
|
||||
callback({ success: await handleCreateFile(fileManager, name) })
|
||||
} catch (e: any) {
|
||||
console.error("Error creating file:", e)
|
||||
socket.emit("error", `Error: file creation. ${e.message ?? e}`)
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to create a new folder
|
||||
socket.on("createFolder", async (name: string, callback) => {
|
||||
try {
|
||||
await createFolderRL.consume(data.userId, 1)
|
||||
await fileManager.createFolder(name)
|
||||
await handleCreateFolder(fileManager, name)
|
||||
callback()
|
||||
} catch (e: any) {
|
||||
console.error("Error creating folder:", e)
|
||||
@ -364,61 +448,38 @@ io.on("connection", async (socket) => {
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to rename a file
|
||||
socket.on("renameFile", async (fileId: string, newName: string) => {
|
||||
try {
|
||||
await renameFileRL.consume(data.userId, 1)
|
||||
await fileManager.renameFile(fileId, newName)
|
||||
await handleRenameFile(fileManager, fileId, newName)
|
||||
} catch (e: any) {
|
||||
console.error("Error renaming file:", e)
|
||||
socket.emit("error", `Error: file renaming. ${e.message ?? e}`)
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to delete a file
|
||||
socket.on("deleteFile", async (fileId: string, callback) => {
|
||||
try {
|
||||
await deleteFileRL.consume(data.userId, 1)
|
||||
const newFiles = await fileManager.deleteFile(fileId)
|
||||
callback(newFiles)
|
||||
callback(await handleDeleteFile(fileManager, fileId))
|
||||
} catch (e: any) {
|
||||
console.error("Error deleting file:", e)
|
||||
socket.emit("error", `Error: file deletion. ${e.message ?? e}`)
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to delete a folder
|
||||
socket.on("deleteFolder", async (folderId: string, callback) => {
|
||||
try {
|
||||
const newFiles = await fileManager.deleteFolder(folderId)
|
||||
callback(newFiles)
|
||||
callback(await handleDeleteFolder(fileManager, folderId))
|
||||
} catch (e: any) {
|
||||
console.error("Error deleting folder:", e)
|
||||
socket.emit("error", `Error: folder deletion. ${e.message ?? e}`)
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to create a new terminal
|
||||
socket.on("createTerminal", async (id: string, callback) => {
|
||||
try {
|
||||
await lockManager.acquireLock(data.sandboxId, async () => {
|
||||
let terminalManager = terminalManagers[data.sandboxId]
|
||||
if (!terminalManager) {
|
||||
terminalManager = terminalManagers[data.sandboxId] =
|
||||
new TerminalManager(containers[data.sandboxId])
|
||||
}
|
||||
|
||||
await terminalManager.createTerminal(id, (responseString: string) => {
|
||||
socket.emit("terminalResponse", { id, data: responseString })
|
||||
const port = extractPortNumber(responseString)
|
||||
if (port) {
|
||||
socket.emit(
|
||||
"previewURL",
|
||||
"https://" + containers[data.sandboxId].getHost(port)
|
||||
)
|
||||
}
|
||||
})
|
||||
})
|
||||
await handleCreateTerminal(terminalManager, id, socket, containers, data)
|
||||
callback()
|
||||
} catch (e: any) {
|
||||
console.error(`Error creating terminal ${id}:`, e)
|
||||
@ -426,33 +487,27 @@ io.on("connection", async (socket) => {
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to resize terminal
|
||||
socket.on(
|
||||
"resizeTerminal",
|
||||
(dimensions: { cols: number; rows: number }) => {
|
||||
socket.on("resizeTerminal", (dimensions: { cols: number; rows: number }) => {
|
||||
try {
|
||||
terminalManager.resizeTerminal(dimensions)
|
||||
handleResizeTerminal(terminalManager, dimensions)
|
||||
} catch (e: any) {
|
||||
console.error("Error resizing terminal:", e)
|
||||
socket.emit("error", `Error: terminal resizing. ${e.message ?? e}`)
|
||||
}
|
||||
}
|
||||
)
|
||||
})
|
||||
|
||||
// Handle terminal input data
|
||||
socket.on("terminalData", async (id: string, data: string) => {
|
||||
try {
|
||||
await terminalManager.sendTerminalData(id, data)
|
||||
await handleTerminalData(terminalManager, id, data)
|
||||
} catch (e: any) {
|
||||
console.error("Error writing to terminal:", e)
|
||||
socket.emit("error", `Error: writing to terminal. ${e.message ?? e}`)
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to close terminal
|
||||
socket.on("closeTerminal", async (id: string, callback) => {
|
||||
try {
|
||||
await terminalManager.closeTerminal(id)
|
||||
await handleCloseTerminal(terminalManager, id)
|
||||
callback()
|
||||
} catch (e: any) {
|
||||
console.error("Error closing terminal:", e)
|
||||
@ -460,48 +515,18 @@ io.on("connection", async (socket) => {
|
||||
}
|
||||
})
|
||||
|
||||
// Handle request to generate code
|
||||
socket.on(
|
||||
"generateCode",
|
||||
async (
|
||||
fileName: string,
|
||||
code: string,
|
||||
line: number,
|
||||
instructions: string,
|
||||
callback
|
||||
) => {
|
||||
socket.on("generateCode", async (fileName: string, code: string, line: number, instructions: string, callback) => {
|
||||
try {
|
||||
const result = await aiWorker.generateCode(
|
||||
data.userId,
|
||||
fileName,
|
||||
code,
|
||||
line,
|
||||
instructions
|
||||
)
|
||||
callback(result)
|
||||
callback(await handleGenerateCode(aiWorker, data.userId, fileName, code, line, instructions))
|
||||
} catch (e: any) {
|
||||
console.error("Error generating code:", e)
|
||||
socket.emit("error", `Error: code generation. ${e.message ?? e}`)
|
||||
}
|
||||
}
|
||||
)
|
||||
})
|
||||
|
||||
// Handle socket disconnection
|
||||
socket.on("disconnect", async () => {
|
||||
try {
|
||||
if (data.isOwner) {
|
||||
connections[data.sandboxId]--
|
||||
}
|
||||
|
||||
await terminalManager.closeAllTerminals()
|
||||
await fileManager.closeWatchers()
|
||||
|
||||
if (data.isOwner && connections[data.sandboxId] <= 0) {
|
||||
socket.broadcast.emit(
|
||||
"disableAccess",
|
||||
"The sandbox owner has disconnected."
|
||||
)
|
||||
}
|
||||
await handleDisconnect(data, connections, terminalManager, fileManager, socket)
|
||||
} catch (e: any) {
|
||||
console.log("Error disconnecting:", e)
|
||||
socket.emit("error", `Error: disconnecting. ${e.message ?? e}`)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user