-
Notifications
You must be signed in to change notification settings - Fork 37
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #84 from CodinGame/debug-service
Debug services
- Loading branch information
Showing
46 changed files
with
10,944 additions
and
8,899 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,221 @@ | ||
| import express from 'express' | ||
| import { WebSocketServer } from 'ws' | ||
| import Docker from 'dockerode' | ||
| import * as http from 'http' | ||
| import * as net from 'net' | ||
| import * as fs from 'fs' | ||
| import * as stream from 'stream' | ||
|
|
||
| const docker = new Docker() | ||
| const image = 'ghcr.io/graalvm/graalvm-ce:21.2.0' | ||
|
|
||
| async function createContainer () { | ||
| const stream = await docker.pull(image) | ||
| await new Promise<void>((resolve, reject) => { | ||
| docker.modem.followProgress(stream, err => err == null ? resolve() : reject(err)) | ||
| }) | ||
| const container = await docker.createContainer({ | ||
| name: 'graalvm-debugger', | ||
| Image: image, | ||
| Entrypoint: ['sleep', 'infinity'], | ||
| HostConfig: { | ||
| NetworkMode: 'host', | ||
| Mounts: [{ | ||
| Type: 'bind', | ||
| Target: '/tmp', | ||
| Source: '/tmp' | ||
| }], | ||
| AutoRemove: true | ||
| } | ||
| }) | ||
| return container | ||
| } | ||
|
|
||
| async function prepareContainer (container: Docker.Container) { | ||
| await container.start() | ||
| // eslint-disable-next-line no-console | ||
| console.log('Installing node') | ||
| const exec = await container.exec({ | ||
| Cmd: ['gu', 'install', 'nodejs'], | ||
| AttachStdout: true, | ||
| AttachStderr: true | ||
| }) | ||
| const execStream = await exec.start({ | ||
| hijack: true | ||
| }) | ||
| execStream.pipe(process.stdout) | ||
| await new Promise(resolve => execStream.on('end', resolve)) | ||
| // eslint-disable-next-line no-console | ||
| console.log('Node installed') | ||
| } | ||
|
|
||
| // eslint-disable-next-line no-console | ||
| console.log('Pulling image/starting container...') | ||
| const containerPromise = createContainer() | ||
|
|
||
| async function exitHandler () { | ||
| // eslint-disable-next-line no-console | ||
| console.log('Exiting...') | ||
| try { | ||
| const container = await containerPromise | ||
| await container.remove({ | ||
| force: true | ||
| }) | ||
| } catch (err) { | ||
| console.error(err) | ||
| } finally { | ||
| process.exit() | ||
| } | ||
| } | ||
| process.on('exit', exitHandler.bind(null, { cleanup: true })) | ||
| process.on('SIGINT', exitHandler.bind(null, { exit: true })) | ||
| process.on('SIGUSR1', exitHandler.bind(null, { exit: true })) | ||
| process.on('SIGUSR2', exitHandler.bind(null, { exit: true })) | ||
| process.on('uncaughtException', exitHandler.bind(null, { exit: true })) | ||
|
|
||
| const container = await containerPromise | ||
| await prepareContainer(container) | ||
|
|
||
| class DAPSocket { | ||
| private socket: net.Socket | ||
| private rawData = Buffer.allocUnsafe(0) | ||
| private contentLength = -1 | ||
| constructor (private onMessage: (message: string) => void) { | ||
| this.socket = new net.Socket() | ||
| this.socket.on('data', this.onData) | ||
| } | ||
|
|
||
| private onData = (data: Buffer) => { | ||
| this.rawData = Buffer.concat([this.rawData, data]) | ||
| // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition | ||
| while (true) { | ||
| if (this.contentLength >= 0) { | ||
| if (this.rawData.length >= this.contentLength) { | ||
| const message = this.rawData.toString('utf8', 0, this.contentLength) | ||
| this.rawData = this.rawData.subarray(this.contentLength) | ||
| this.contentLength = -1 | ||
| if (message.length > 0) { | ||
| this.onMessage(message) | ||
| } | ||
| continue | ||
| } | ||
| } else { | ||
| const idx = this.rawData.indexOf(TWO_CRLF) | ||
| if (idx !== -1) { | ||
| const header = this.rawData.toString('utf8', 0, idx) | ||
| const lines = header.split(HEADER_LINESEPARATOR) | ||
| for (const h of lines) { | ||
| const kvPair = h.split(HEADER_FIELDSEPARATOR) | ||
| if (kvPair[0] === 'Content-Length') { | ||
| this.contentLength = Number(kvPair[1]) | ||
| } | ||
| } | ||
| this.rawData = this.rawData.subarray(idx + TWO_CRLF.length) | ||
| continue | ||
| } | ||
| } | ||
| break | ||
| } | ||
| } | ||
|
|
||
| public connect (port: number) { | ||
| this.socket.connect(port) | ||
| } | ||
|
|
||
| public sendMessage (message: string) { | ||
| this.socket.write(`Content-Length: ${Buffer.byteLength(message, 'utf8')}${TWO_CRLF}${message}`, 'utf8') | ||
| } | ||
| } | ||
|
|
||
| const TWO_CRLF = '\r\n\r\n' | ||
| const HEADER_LINESEPARATOR = /\r?\n/ | ||
| const HEADER_FIELDSEPARATOR = /: */ | ||
|
|
||
| const PORT = 5555 | ||
| const app = express() | ||
|
|
||
| const server = http.createServer(app) | ||
|
|
||
| const wss = new WebSocketServer({ server }) | ||
|
|
||
| async function findPortFree () { | ||
| return new Promise<number>(resolve => { | ||
| const srv = net.createServer() | ||
| srv.listen(0, () => { | ||
| const port = (srv.address() as net.AddressInfo).port | ||
| srv.close(() => resolve(port)) | ||
| }) | ||
| }) | ||
| } | ||
|
|
||
| function sequential<T, P extends unknown[]> (fn: (...params: P) => Promise<T>): (...params: P) => Promise<T> { | ||
| let promise = Promise.resolve() | ||
| return (...params: P) => { | ||
| const result = promise.then(() => { | ||
| return fn(...params) | ||
| }) | ||
|
|
||
| promise = result.then(() => {}, () => {}) | ||
| return result | ||
| } | ||
| } | ||
|
|
||
| wss.on('connection', (ws) => { | ||
| const socket = new DAPSocket(message => ws.send(message)) | ||
|
|
||
| let initialized = false | ||
|
|
||
| ws.on('message', sequential(async (message: string) => { | ||
| if (!initialized) { | ||
| try { | ||
| initialized = true | ||
| const init: { main: string, files: Record<string, string> } = JSON.parse(message) | ||
| for (const [file, content] of Object.entries(init.files)) { | ||
| fs.writeFileSync(file, content) | ||
| } | ||
| const debuggerPort = await findPortFree() | ||
| const exec = await container.exec({ | ||
| Cmd: ['node', `--dap=${debuggerPort}`, '--dap.WaitAttached', '--dap.Suspend=false', `${init.main}`], | ||
| AttachStdout: true, | ||
| AttachStderr: true | ||
| }) | ||
|
|
||
| const execStream = await exec.start({ | ||
| hijack: true | ||
| }) | ||
| const stdout = new stream.PassThrough() | ||
| const stderr = new stream.PassThrough() | ||
| container.modem.demuxStream(execStream, stdout, stderr) | ||
| function sendOutput (category: 'stdout' | 'stderr', output: Buffer) { | ||
| ws.send(JSON.stringify({ | ||
| type: 'event', | ||
| event: 'output', | ||
| body: { | ||
| category, | ||
| output: output.toString() | ||
| } | ||
| })) | ||
| } | ||
| stdout.on('data', sendOutput.bind(undefined, 'stdout')) | ||
| stderr.on('data', sendOutput.bind(undefined, 'stderr')) | ||
|
|
||
| execStream.on('end', () => { | ||
| ws.close() | ||
| }) | ||
|
|
||
| await new Promise(resolve => setTimeout(resolve, 1000)) | ||
| socket.connect(debuggerPort) | ||
|
|
||
| return | ||
| } catch (err) { | ||
| console.error('Failed to initialize', err) | ||
| } | ||
| } | ||
| socket.sendMessage(message) | ||
| })) | ||
| }) | ||
|
|
||
| server.listen(PORT, () => { | ||
| // eslint-disable-next-line no-console | ||
| console.log(`Server started on port ${PORT} :)`) | ||
| }) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.