sandbox-web-ide/backend/server/dist/index.js

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const fs_1 = __importDefault(require("fs"));
const os_1 = __importDefault(require("os"));
const path_1 = __importDefault(require("path"));
const express_1 = __importDefault(require("express"));
const dotenv_1 = __importDefault(require("dotenv"));
const http_1 = require("http");
const socket_io_1 = require("socket.io");
const zod_1 = require("zod");
const utils_1 = require("./utils");
const node_pty_1 = require("node-pty");
const ratelimit_1 = require("./ratelimit");
dotenv_1.default.config();
const app = (0, express_1.default)();
const port = process.env.PORT || 4000;
// app.use(cors())
const httpServer = (0, http_1.createServer)(app);
const io = new socket_io_1.Server(httpServer, {
    cors: {
        origin: "*",
    },
});
const terminals = {};
const dirName = path_1.default.join(__dirname, "..");
const handshakeSchema = zod_1.z.object({
    userId: zod_1.z.string(),
    sandboxId: zod_1.z.string(),
    EIO: zod_1.z.string(),
    transport: zod_1.z.string(),
});
io.use((socket, next) => __awaiter(void 0, void 0, void 0, function* () {
    const q = socket.handshake.query;
    const parseQuery = handshakeSchema.safeParse(q);
    if (!parseQuery.success) {
        console.log("Invalid request.");
        next(new Error("Invalid request."));
        return;
    }
    const { sandboxId, userId } = parseQuery.data;
    const dbUser = yield fetch(`http://localhost:8787/api/user?id=${userId}`);
    const dbUserJSON = (yield dbUser.json());
    if (!dbUserJSON) {
        console.log("DB error.");
        next(new Error("DB error."));
        return;
    }
    const sandbox = dbUserJSON.sandbox.find((s) => s.id === sandboxId);
    const sharedSandboxes = dbUserJSON.usersToSandboxes.find((uts) => uts.sandboxId === sandboxId);
    if (!sandbox && !sharedSandboxes) {
        console.log("Invalid credentials.");
        next(new Error("Invalid credentials."));
        return;
    }
    socket.data = {
        userId,
        sandboxId: sandboxId,
    };
    next();
}));
io.on("connection", (socket) => __awaiter(void 0, void 0, void 0, function* () {
    const data = socket.data;
    const sandboxFiles = yield (0, utils_1.getSandboxFiles)(data.sandboxId);
    sandboxFiles.fileData.forEach((file) => {
        const filePath = path_1.default.join(dirName, file.id);
        fs_1.default.mkdirSync(path_1.default.dirname(filePath), { recursive: true });
        fs_1.default.writeFile(filePath, file.data, function (err) {
            if (err)
                throw err;
        });
    });
    socket.emit("loaded", sandboxFiles.files);
    socket.on("getFile", (fileId, callback) => {
        const file = sandboxFiles.fileData.find((f) => f.id === fileId);
        if (!file)
            return;
        callback(file.data);
    });
    // todo: send diffs + debounce for efficiency
    socket.on("saveFile", (fileId, body) => __awaiter(void 0, void 0, void 0, function* () {
        try {
            yield ratelimit_1.saveFileRL.consume(data.userId, 1);
            if (Buffer.byteLength(body, "utf-8") > ratelimit_1.MAX_BODY_SIZE) {
                socket.emit("rateLimit", "Rate limited: file size too large. Please reduce the file size.");
                return;
            }
            const file = sandboxFiles.fileData.find((f) => f.id === fileId);
            if (!file)
                return;
            file.data = body;
            fs_1.default.writeFile(path_1.default.join(dirName, file.id), body, function (err) {
                if (err)
                    throw err;
            });
            yield (0, utils_1.saveFile)(fileId, body);
        }
        catch (e) {
            socket.emit("rateLimit", "Rate limited: file saving. Please slow down.");
        }
    }));
    socket.on("createFile", (name) => __awaiter(void 0, void 0, void 0, function* () {
        try {
            yield ratelimit_1.createFileRL.consume(data.userId, 1);
            const id = `projects/${data.sandboxId}/${name}`;
            fs_1.default.writeFile(path_1.default.join(dirName, id), "", function (err) {
                if (err)
                    throw err;
            });
            sandboxFiles.files.push({
                id,
                name,
                type: "file",
            });
            sandboxFiles.fileData.push({
                id,
                data: "",
            });
            yield (0, utils_1.createFile)(id);
        }
        catch (e) {
            socket.emit("rateLimit", "Rate limited: file creation. Please slow down.");
        }
    }));
    socket.on("renameFile", (fileId, newName) => __awaiter(void 0, void 0, void 0, function* () {
        try {
            yield ratelimit_1.renameFileRL.consume(data.userId, 1);
            const file = sandboxFiles.fileData.find((f) => f.id === fileId);
            if (!file)
                return;
            file.id = newName;
            const parts = fileId.split("/");
            const newFileId = parts.slice(0, parts.length - 1).join("/") + "/" + newName;
            fs_1.default.rename(path_1.default.join(dirName, fileId), path_1.default.join(dirName, newFileId), function (err) {
                if (err)
                    throw err;
            });
            yield (0, utils_1.renameFile)(fileId, newFileId, file.data);
        }
        catch (e) {
            socket.emit("rateLimit", "Rate limited: file renaming. Please slow down.");
            return;
        }
    }));
    socket.on("deleteFile", (fileId, callback) => __awaiter(void 0, void 0, void 0, function* () {
        try {
            yield ratelimit_1.deleteFileRL.consume(data.userId, 1);
            const file = sandboxFiles.fileData.find((f) => f.id === fileId);
            if (!file)
                return;
            fs_1.default.unlink(path_1.default.join(dirName, fileId), function (err) {
                if (err)
                    throw err;
            });
            sandboxFiles.fileData = sandboxFiles.fileData.filter((f) => f.id !== fileId);
            yield (0, utils_1.deleteFile)(fileId);
            const newFiles = yield (0, utils_1.getSandboxFiles)(data.sandboxId);
            callback(newFiles.files);
        }
        catch (e) {
            socket.emit("rateLimit", "Rate limited: file deletion. Please slow down.");
        }
    }));
    socket.on("createTerminal", ({ id }) => {
        if (terminals[id]) {
            console.log("Terminal already exists.");
            return;
        }
        if (Object.keys(terminals).length >= 4) {
            console.log("Too many terminals.");
            return;
        }
        const pty = (0, node_pty_1.spawn)(os_1.default.platform() === "win32" ? "cmd.exe" : "bash", [], {
            name: "xterm",
            cols: 100,
            cwd: path_1.default.join(dirName, "projects", data.sandboxId),
        });
        const onData = pty.onData((data) => {
            socket.emit("terminalResponse", {
                // data: Buffer.from(data, "utf-8").toString("base64"),
                data,
            });
        });
        const onExit = pty.onExit((code) => console.log("exit :(", code));
        pty.write("clear\r");
        terminals[id] = {
            terminal: pty,
            onData,
            onExit,
        };
    });
    socket.on("terminalData", (id, data) => {
        if (!terminals[id]) {
            console.log("terminals", terminals);
            return;
        }
        try {
            terminals[id].terminal.write(data);
        }
        catch (e) {
            console.log("Error writing to terminal", e);
        }
    });
    socket.on("generateCode", (fileName, code, line, instructions, callback) => __awaiter(void 0, void 0, void 0, function* () {
        const fetchPromise = fetch(`http://localhost:8787/api/sandbox/generate`, {
            method: "POST",
            headers: {
                "Content-Type": "application/json",
            },
            body: JSON.stringify({
                userId: data.userId,
            }),
        });
        const generateCodePromise = (0, utils_1.generateCode)({
            fileName,
            code,
            line,
            instructions,
        });
        const [fetchResponse, generateCodeResponse] = yield Promise.all([
            fetchPromise,
            generateCodePromise,
        ]);
        const json = yield generateCodeResponse.json();
        callback(json);
    }));
    socket.on("disconnect", () => {
        Object.entries(terminals).forEach((t) => {
            const { terminal, onData, onExit } = t[1];
            if (os_1.default.platform() !== "win32")
                terminal.kill();
            onData.dispose();
            onExit.dispose();
            delete terminals[t[0]];
        });
    });
}));
httpServer.listen(port, () => {
    console.log(`Server running on port ${port}`);
});