From 5fb98c039596394285c5a89dabd82423c6f21fa7 Mon Sep 17 00:00:00 2001
From: clfreville2 <clement.freville2@etu.uca.fr>
Date: Mon, 15 Jan 2024 09:27:40 +0100
Subject: [PATCH] Rewrite using Fastify

---
 package.json  |   7 ++--
 src/runner.ts |  13 +++++++
 src/server.ts | 102 ++++++++++++++++++++------------------------------
 3 files changed, 58 insertions(+), 64 deletions(-)
 create mode 100644 src/runner.ts

diff --git a/package.json b/package.json
index b473132..93e07f6 100644
--- a/package.json
+++ b/package.json
@@ -11,10 +11,11 @@
     "tsx": "^4.7.0",
     "typescript": "^5.3.3"
   },
-  "peerDependencies": {
-    "typescript": "^5.0.0"
-  },
   "dependencies": {
+    "@fastify/cors": "^8.5.0",
+    "@fastify/type-provider-typebox": "^4.0.0",
+    "@sinclair/typebox": "^0.32.9",
+    "fastify": "^4.25.2",
     "nanoid": "^5.0.2",
     "zeromq": "6.0.0-beta.17"
   }
diff --git a/src/runner.ts b/src/runner.ts
new file mode 100644
index 0000000..2c886bb
--- /dev/null
+++ b/src/runner.ts
@@ -0,0 +1,13 @@
+export const IMAGES = {
+  moshell: 'ghcr.io/moshell-lang/moshell:master',
+};
+
+export function allocateBuffer(jobId: string, code: string, image: string): Buffer {
+  const buffer = Buffer.allocUnsafe(jobId.length + image.length + code.length + 8);
+  buffer.write(jobId, 0);
+  buffer.writeUInt32BE(image.length, jobId.length);
+  buffer.writeUInt32BE(code.length, jobId.length + 4);
+  buffer.write(image, jobId.length + 8);
+  buffer.write(code, jobId.length + 8 + image.length);
+  return buffer;
+}
diff --git a/src/server.ts b/src/server.ts
index cd312ed..cde8dbd 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -1,76 +1,56 @@
-import http from 'http';
+import cors from '@fastify/cors';
+import { Type, TypeBoxTypeProvider } from '@fastify/type-provider-typebox';
+import Fastify, { FastifyReply } from 'fastify';
 import { nanoid } from 'nanoid';
+import { allocateBuffer, IMAGES } from 'runner';
 import { Pull, Push } from 'zeromq';
 
-const host = 'localhost';
-const port = 3000;
-
 const sender = new Push();
 await sender.bind(`tcp://127.0.0.1:5557`);
 const receiver = new Pull();
 await receiver.bind(`tcp://127.0.0.1:5558`);
 
+const clients: Record<string, FastifyReply> = {};
 const generateId = () => nanoid(32);
 
-const clients: Record<string, http.ServerResponse> = {};
-
-const CORS = {
-  'Access-Control-Allow-Methods': '*',
-  'Access-Control-Allow-Headers': '*',
-  'Access-Control-Allow-Origin': process.env.ALLOW_ORIGIN || '*',
-};
-
-const server = http.createServer((req, res) => {
-  if (req.method === 'OPTIONS') {
-    res.writeHead(200, CORS);
-    res.end();
-    return;
-  }
-  switch (req.url) {
-    case '/run':
-      const jobId = generateId();
-      const code = 'echo a';
-      const image = 'ghcr.io/moshell-lang/moshell:master';
-      const buffer = Buffer.allocUnsafe(jobId.length + image.length + code.length + 8);
-      buffer.write(jobId, 0);
-      buffer.writeUInt32BE(image.length, jobId.length);
-      buffer.writeUInt32BE(code.length, jobId.length + 4);
-      buffer.write(image, jobId.length + 8);
-      buffer.write(code, jobId.length + 8 + image.length);
-      res.writeHead(200, {
-        'Content-Type': 'text/event-stream',
-        Connection: 'keep-alive',
-        'Cache-Control': 'no-cache',
-        ...CORS,
-      });
-      sender.send(buffer).then(() => {
-        res.write('event: connected\n');
-        res.write(`data: ${jobId}\n`);
-        res.write('id: 0\n\n');
-      });
-      req.on('close', () => {
-        res.end('OK');
-        delete clients[jobId];
-      });
-      clients[jobId] = res;
-      break;
-    default:
-      res.writeHead(404, CORS);
-      res.end('404!');
-  }
+const fastify = Fastify({
+  logger: true,
+}).withTypeProvider<TypeBoxTypeProvider>();
+await fastify.register(cors, {
+  origin: process.env.ALLOW_ORIGIN || '*',
 });
-server.listen(port, () => {
-  console.log(`Server is running on http://${host}:${port}`);
+
+fastify.post('/run', {
+  schema: {
+    body: Type.Object({
+      code: Type.String(),
+      language: Type.String(),
+    }),
+  },
+}, (req, reply) => {
+  const { code, language } = req.body;
+  const jobId = generateId();
+  const buffer = allocateBuffer(jobId, code, IMAGES.moshell);
+  sender.send(buffer).then(() => {
+    reply.raw.write('event: connected\n');
+    reply.raw.write(`data: ${jobId}\n`);
+    reply.raw.write('id: 0\n\n');
+  });
+  clients[jobId] = reply;
 });
 
-for await (const [buff] of receiver) {
-  const jobId = buff.subarray(0, 32).toString();
-  console.log(`Received ${jobId}`);
-  const res = clients[jobId];
-  if (!res) {
-    continue;
+async function forwardOutput() {
+  for await (const [buff] of receiver) {
+    const jobId = buff.subarray(0, 32).toString();
+    console.log(`Received ${jobId}`);
+    const reply = clients[jobId];
+    if (!reply) {
+      continue;
+    }
+    reply.raw.write('event: message\n');
+    reply.raw.write(`data: ${encodeURIComponent(buff.subarray(32).toString())}\n`);
+    reply.raw.write('id: 1\n\n');
   }
-  res.write('event: message\n');
-  res.write(`data: ${encodeURIComponent(buff.subarray(32).toString())}\n`);
-  res.write('id: 1\n\n');
 }
+
+await Promise.all([fastify.listen({ port: 3000 }), forwardOutput()]);