128 lines
4.6 KiB
TypeScript
128 lines
4.6 KiB
TypeScript
import { AIWorker } from "./AIWorker"
|
|
import { CONTAINER_TIMEOUT } from "./constants"
|
|
import { DokkuClient } from "./DokkuClient"
|
|
import { FileManager } from "./FileManager"
|
|
import { SecureGitClient } from "./SecureGitClient"
|
|
import { TerminalManager } from "./TerminalManager"
|
|
import { LockManager } from "./utils"
|
|
|
|
export interface HandlerContext {
|
|
fileManager: FileManager;
|
|
terminalManager: TerminalManager;
|
|
sandboxManager: any;
|
|
aiWorker: AIWorker;
|
|
dokkuClient: DokkuClient | null;
|
|
gitClient: SecureGitClient | null;
|
|
lockManager: LockManager
|
|
}
|
|
|
|
// Extract port number from a string
|
|
function extractPortNumber(inputString: string): number | null {
|
|
const cleanedString = inputString.replace(/\x1B\[[0-9;]*m/g, "")
|
|
const regex = /http:\/\/localhost:(\d+)/
|
|
const match = cleanedString.match(regex)
|
|
return match ? parseInt(match[1]) : null
|
|
}
|
|
|
|
// Handle heartbeat from a socket connection
|
|
export function handleHeartbeat(data: any, context: HandlerContext) {
|
|
context.sandboxManager.setTimeout(CONTAINER_TIMEOUT)
|
|
}
|
|
|
|
// Handle getting a file
|
|
export function handleGetFile(fileId: string, context: HandlerContext) {
|
|
return context.fileManager.getFile(fileId)
|
|
}
|
|
|
|
// Handle getting a folder
|
|
export function handleGetFolder(folderId: string, context: HandlerContext) {
|
|
return context.fileManager.getFolder(folderId)
|
|
}
|
|
|
|
// Handle saving a file
|
|
export function handleSaveFile(fileId: string, body: string, context: HandlerContext) {
|
|
return context.fileManager.saveFile(fileId, body)
|
|
}
|
|
|
|
// Handle moving a file
|
|
export function handleMoveFile(fileId: string, folderId: string, context: HandlerContext) {
|
|
return context.fileManager.moveFile(fileId, folderId)
|
|
}
|
|
|
|
// Handle listing apps
|
|
export async function handleListApps(context: HandlerContext) {
|
|
if (!context.dokkuClient) throw Error("Failed to retrieve apps list: No Dokku client")
|
|
return { success: true, apps: await context.dokkuClient.listApps() }
|
|
}
|
|
|
|
// Handle deploying code
|
|
export async function handleDeploy(sandboxId: string, context: HandlerContext) {
|
|
if (!context.gitClient) throw Error("Failed to retrieve apps list: No git client")
|
|
const fixedFilePaths = context.fileManager.sandboxFiles.fileData.map((file) => ({
|
|
...file,
|
|
id: file.id.split("/").slice(2).join("/"),
|
|
}))
|
|
await context.gitClient.pushFiles(fixedFilePaths, sandboxId)
|
|
return { success: true }
|
|
}
|
|
|
|
// Handle creating a file
|
|
export function handleCreateFile(name: string, context: HandlerContext) {
|
|
return context.fileManager.createFile(name)
|
|
}
|
|
|
|
// Handle creating a folder
|
|
export function handleCreateFolder(name: string, context: HandlerContext) {
|
|
return context.fileManager.createFolder(name)
|
|
}
|
|
|
|
// Handle renaming a file
|
|
export function handleRenameFile(fileId: string, newName: string, context: HandlerContext) {
|
|
return context.fileManager.renameFile(fileId, newName)
|
|
}
|
|
|
|
// Handle deleting a file
|
|
export function handleDeleteFile(fileId: string, context: HandlerContext) {
|
|
return context.fileManager.deleteFile(fileId)
|
|
}
|
|
|
|
// Handle deleting a folder
|
|
export function handleDeleteFolder(folderId: string, context: HandlerContext) {
|
|
return context.fileManager.deleteFolder(folderId)
|
|
}
|
|
|
|
// Handle creating a terminal session
|
|
export async function handleCreateTerminal(id: string, socket: any, data: any, context: HandlerContext) {
|
|
await context.lockManager.acquireLock(data.sandboxId, async () => {
|
|
await context.terminalManager.createTerminal(id, (responseString: string) => {
|
|
socket.emit("terminalResponse", { id, data: responseString })
|
|
const port = extractPortNumber(responseString)
|
|
if (port) {
|
|
socket.emit(
|
|
"previewURL",
|
|
"https://" + context.sandboxManager.getHost(port)
|
|
)
|
|
}
|
|
})
|
|
})
|
|
}
|
|
|
|
// Handle resizing a terminal
|
|
export function handleResizeTerminal(dimensions: { cols: number; rows: number }, context: HandlerContext) {
|
|
context.terminalManager.resizeTerminal(dimensions)
|
|
}
|
|
|
|
// Handle sending data to a terminal
|
|
export function handleTerminalData(id: string, data: string, context: HandlerContext) {
|
|
return context.terminalManager.sendTerminalData(id, data)
|
|
}
|
|
|
|
// Handle closing a terminal
|
|
export function handleCloseTerminal(id: string, context: HandlerContext) {
|
|
return context.terminalManager.closeTerminal(id)
|
|
}
|
|
|
|
// Handle generating code
|
|
export function handleGenerateCode(userId: string, fileName: string, code: string, line: number, instructions: string, context: HandlerContext) {
|
|
return context.aiWorker.generateCode(userId, fileName, code, line, instructions)
|
|
} |