From 8069df5973cd4a0d0202d51f46586e3768fb6045 Mon Sep 17 00:00:00 2001
From: cfngc4594 <cfngc4594@gmail.com>
Date: Sat, 1 Mar 2025 17:10:19 +0800
Subject: [PATCH] feat(judge): add server-side code execution and judging
 functionality

---
 src/app/actions/judge.ts | 122 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 122 insertions(+)
 create mode 100644 src/app/actions/judge.ts

diff --git a/src/app/actions/judge.ts b/src/app/actions/judge.ts
new file mode 100644
index 0000000..dcf7b3d
--- /dev/null
+++ b/src/app/actions/judge.ts
@@ -0,0 +1,122 @@
+"use server";
+
+import tar from "tar-stream";
+import Docker from "dockerode";
+import { Readable } from "stream";
+import { LanguageConfigs } from "@/config/judge";
+
+const docker = new Docker({ socketPath: "/var/run/docker.sock" });
+
+async function prepareEnvironment(image: string, tag: string) {
+  const reference = `${image}:${tag}`;
+  const filters = { reference: [reference] };
+  const images = await docker.listImages({ filters });
+
+  if (images.length === 0) {
+    await docker.pull(reference);
+  }
+}
+
+async function createContainer(image: string, tag: string, workingDir: string) {
+  const container = await docker.createContainer({
+    Image: `${image}:${tag}`,
+    Tty: true,
+    WorkingDir: workingDir,
+  });
+
+  await container.start();
+  return container;
+}
+
+function createTarStream(filePath: string, value: string) {
+  const pack = tar.pack();
+  pack.entry({ name: filePath }, value);
+  pack.finalize();
+  return Readable.from(pack);
+}
+
+async function compileCode(container: Docker.Container, filePath: string, fileName: string) {
+  const compileExec = await container.exec({
+    Cmd: ["gcc", filePath, "-o", fileName],
+    AttachStdout: true,
+    AttachStderr: true,
+  });
+
+  return new Promise<string>((resolve, reject) => {
+    compileExec.start({}, (error, stream) => {
+      if (error) {
+        return reject(error);
+      }
+      if (!stream) {
+        return reject(new Error("Stream is undefined"));
+      }
+
+      let data = "";
+      stream.on("data", (chunk) => (data += chunk.toString()));
+      stream.on("end", () => resolve(data));
+      stream.on("error", (error) => reject(error));
+    });
+  });
+}
+
+async function runCode(container: Docker.Container, fileName: string) {
+  const runExec = await container.exec({
+    Cmd: [`./${fileName}`],
+    AttachStdout: true,
+    AttachStderr: true,
+  });
+
+  return new Promise<string>((resolve, reject) => {
+    runExec.start({}, (error, stream) => {
+      if (error) {
+        return reject(error);
+      }
+      if (!stream) {
+        return reject(new Error("Stream is undefined"));
+      }
+
+      let data = "";
+      stream.on("data", (chunk) => (data += chunk.toString()));
+      stream.on("end", () => resolve(data));
+      stream.on("error", (error) => reject(error));
+    });
+  });
+}
+
+async function cleanupContainer(container: Docker.Container) {
+  try {
+    await container.stop({ t: 0 });
+    await container.remove();
+  } catch (error) {
+    console.error("Container cleanup failed:", error);
+  }
+}
+
+export async function judge(language: string, value: string) {
+  const { image, tag, fileName, extension, workingDir } = LanguageConfigs[language];
+  const filePath = `${fileName}.${extension}`;
+  let container: Docker.Container | undefined;
+
+  try {
+    await prepareEnvironment(image, tag);
+    container = await createContainer(image, tag, workingDir);
+
+    const tarStream = createTarStream(filePath, value);
+    await container.putArchive(tarStream, { path: workingDir });
+
+    const compileOutput = await compileCode(container, filePath, fileName);
+    if (compileOutput) {
+      return compileOutput;
+    }
+
+    const runOutput = await runCode(container, fileName);
+    return runOutput;
+  } catch (error) {
+    console.error("Error during judging:", error);
+    throw error;
+  } finally {
+    if (container) {
+      cleanupContainer(container).catch(() => {});
+    }
+  }
+}