46 lines
1.1 KiB
JavaScript
46 lines
1.1 KiB
JavaScript
|
const { VM } = require('vm2')
|
||
|
|
const { Server } = require('socket.io')
|
||
|
|
const util = require('util')
|
||
|
|
const { stylize } = require('./colors')
|
||
|
|
|
||
|
|
const BRIDGE_PREFIX = 'function:'
|
||
|
|
|
||
|
|
const io = new Server(3069)
|
||
|
|
|
||
|
|
function checkString (string) {
|
||
|
|
if (typeof(string) !== 'string') throw new Error('Must be a string')
|
||
|
|
}
|
||
|
|
|
||
|
|
io.on('connection', (socket) => {
|
||
|
|
let vm
|
||
|
|
function resetVM () {
|
||
|
|
vm = new VM({
|
||
|
|
timeout: 3000,
|
||
|
|
sandbox: {
|
||
|
|
chat: (message) => {
|
||
|
|
checkString(message)
|
||
|
|
socket.emit(BRIDGE_PREFIX + 'chat', message)
|
||
|
|
},
|
||
|
|
core: (command) => {
|
||
|
|
checkString(command)
|
||
|
|
socket.emit(BRIDGE_PREFIX + 'core', command)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
resetVM()
|
||
|
|
|
||
|
|
socket.on('runCode', (transactionId, code) => {
|
||
|
|
try {
|
||
|
|
const output = vm.run(code)
|
||
|
|
|
||
|
|
socket.emit('codeOutput', transactionId, false, util.inspect(output, { stylize }))
|
||
|
|
} catch (e) {
|
||
|
|
socket.emit('codeOutput', transactionId, true, e.toString())
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
socket.on('reset', resetVM)
|
||
|
|
})
|