mirror of
https://github.com/massbug/judge4c.git
synced 2025-05-18 07:16:34 +00:00
refactor(actions): move actions from @/app/actions to @/actions
This commit is contained in:
parent
901e2aa487
commit
30fd6fd386
69
src/actions/auth.ts
Normal file
69
src/actions/auth.ts
Normal file
@ -0,0 +1,69 @@
|
|||||||
|
"use server";
|
||||||
|
|
||||||
|
import bcrypt from "bcrypt";
|
||||||
|
import prisma from "@/lib/prisma";
|
||||||
|
import { signIn } from "@/lib/auth";
|
||||||
|
import { authSchema } from "@/lib/zod";
|
||||||
|
import { CredentialsSignInFormValues } from "@/components/credentials-sign-in-form";
|
||||||
|
import { CredentialsSignUpFormValues } from "@/components/credentials-sign-up-form";
|
||||||
|
|
||||||
|
const saltRounds = 10;
|
||||||
|
|
||||||
|
export async function signInWithCredentials(formData: CredentialsSignInFormValues) {
|
||||||
|
try {
|
||||||
|
// Parse credentials using authSchema for validation
|
||||||
|
const { email, password } = await authSchema.parseAsync(formData);
|
||||||
|
|
||||||
|
// Find user by email
|
||||||
|
const user = await prisma.user.findUnique({ where: { email } });
|
||||||
|
|
||||||
|
// Check if the user exists
|
||||||
|
if (!user) {
|
||||||
|
throw new Error("User not found.");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if the user has a password
|
||||||
|
if (!user.password) {
|
||||||
|
throw new Error("Invalid credentials.");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if the password matches
|
||||||
|
const passwordMatch = await bcrypt.compare(password, user.password);
|
||||||
|
if (!passwordMatch) {
|
||||||
|
throw new Error("Incorrect password.");
|
||||||
|
}
|
||||||
|
|
||||||
|
await signIn("credentials", { ...formData, redirect: false });
|
||||||
|
return { success: true };
|
||||||
|
} catch (error) {
|
||||||
|
return { error: error instanceof Error ? error.message : "Failed to sign in. Please try again." };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function signUpWithCredentials(formData: CredentialsSignUpFormValues) {
|
||||||
|
try {
|
||||||
|
const validatedData = await authSchema.parseAsync(formData);
|
||||||
|
|
||||||
|
// Check if user already exists
|
||||||
|
const existingUser = await prisma.user.findUnique({ where: { email: validatedData.email } });
|
||||||
|
if (existingUser) {
|
||||||
|
throw new Error("User already exists");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Hash password and create user
|
||||||
|
const pwHash = await bcrypt.hash(validatedData.password, saltRounds);
|
||||||
|
const user = await prisma.user.create({
|
||||||
|
data: { email: validatedData.email, password: pwHash },
|
||||||
|
});
|
||||||
|
|
||||||
|
// Assign admin role if first user
|
||||||
|
const userCount = await prisma.user.count();
|
||||||
|
if (userCount === 1) {
|
||||||
|
await prisma.user.update({ where: { id: user.id }, data: { role: "ADMIN" } });
|
||||||
|
}
|
||||||
|
|
||||||
|
return { success: true };
|
||||||
|
} catch (error) {
|
||||||
|
return { error: error instanceof Error ? error.message : "Registration failed. Please try again." };
|
||||||
|
}
|
||||||
|
}
|
346
src/actions/judge.ts
Normal file
346
src/actions/judge.ts
Normal file
@ -0,0 +1,346 @@
|
|||||||
|
"use server";
|
||||||
|
|
||||||
|
import fs from "fs";
|
||||||
|
import tar from "tar-stream";
|
||||||
|
import Docker from "dockerode";
|
||||||
|
import prisma from "@/lib/prisma";
|
||||||
|
import { v4 as uuid } from "uuid";
|
||||||
|
import { auth } from "@/lib/auth";
|
||||||
|
import { redirect } from "next/navigation";
|
||||||
|
import { Readable, Writable } from "stream";
|
||||||
|
import { ExitCode, EditorLanguage, JudgeResult } from "@/generated/client";
|
||||||
|
|
||||||
|
const isRemote = process.env.DOCKER_HOST_MODE === "remote";
|
||||||
|
|
||||||
|
// Docker client initialization
|
||||||
|
const docker = isRemote
|
||||||
|
? new Docker({
|
||||||
|
protocol: process.env.DOCKER_REMOTE_PROTOCOL as "https" | "http" | "ssh" | undefined,
|
||||||
|
host: process.env.DOCKER_REMOTE_HOST,
|
||||||
|
port: process.env.DOCKER_REMOTE_PORT,
|
||||||
|
ca: fs.readFileSync(process.env.DOCKER_REMOTE_CA_PATH || "/certs/ca.pem"),
|
||||||
|
cert: fs.readFileSync(process.env.DOCKER_REMOTE_CERT_PATH || "/certs/cert.pem"),
|
||||||
|
key: fs.readFileSync(process.env.DOCKER_REMOTE_KEY_PATH || "/certs/key.pem"),
|
||||||
|
})
|
||||||
|
: new Docker({ socketPath: "/var/run/docker.sock" });
|
||||||
|
|
||||||
|
// Prepare Docker image environment
|
||||||
|
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);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create Docker container with keep-alive
|
||||||
|
async function createContainer(
|
||||||
|
image: string,
|
||||||
|
tag: string,
|
||||||
|
workingDir: string,
|
||||||
|
memoryLimit?: number
|
||||||
|
) {
|
||||||
|
const container = await docker.createContainer({
|
||||||
|
Image: `${image}:${tag}`,
|
||||||
|
Cmd: ["tail", "-f", "/dev/null"], // Keep container alive
|
||||||
|
WorkingDir: workingDir,
|
||||||
|
HostConfig: {
|
||||||
|
Memory: memoryLimit ? memoryLimit * 1024 * 1024 : undefined,
|
||||||
|
MemorySwap: memoryLimit ? memoryLimit * 1024 * 1024 : undefined,
|
||||||
|
},
|
||||||
|
NetworkDisabled: true,
|
||||||
|
});
|
||||||
|
|
||||||
|
await container.start();
|
||||||
|
return container;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create tar stream for code submission
|
||||||
|
function createTarStream(file: string, value: string) {
|
||||||
|
const pack = tar.pack();
|
||||||
|
pack.entry({ name: file }, value);
|
||||||
|
pack.finalize();
|
||||||
|
return Readable.from(pack);
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function judge(
|
||||||
|
language: EditorLanguage,
|
||||||
|
value: string,
|
||||||
|
): Promise<JudgeResult> {
|
||||||
|
const session = await auth();
|
||||||
|
if (!session) redirect("/sign-in");
|
||||||
|
|
||||||
|
let container: Docker.Container | null = null;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const config = await prisma.editorLanguageConfig.findUnique({
|
||||||
|
where: { language },
|
||||||
|
include: {
|
||||||
|
dockerConfig: true,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!config || !config.dockerConfig) {
|
||||||
|
return {
|
||||||
|
id: uuid(),
|
||||||
|
output: "Configuration Error: Missing editor or docker configuration",
|
||||||
|
exitCode: ExitCode.SE,
|
||||||
|
executionTime: null,
|
||||||
|
memoryUsage: null,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const {
|
||||||
|
image,
|
||||||
|
tag,
|
||||||
|
workingDir,
|
||||||
|
memoryLimit,
|
||||||
|
timeLimit,
|
||||||
|
compileOutputLimit,
|
||||||
|
runOutputLimit,
|
||||||
|
} = config.dockerConfig;
|
||||||
|
const { fileName, fileExtension } = config;
|
||||||
|
const file = `${fileName}.${fileExtension}`;
|
||||||
|
|
||||||
|
// Prepare the environment and create a container
|
||||||
|
await prepareEnvironment(image, tag);
|
||||||
|
container = await createContainer(image, tag, workingDir, memoryLimit);
|
||||||
|
|
||||||
|
// Upload code to the container
|
||||||
|
const tarStream = createTarStream(file, value);
|
||||||
|
await container.putArchive(tarStream, { path: workingDir });
|
||||||
|
|
||||||
|
// Compile the code
|
||||||
|
const compileResult = await compile(container, file, fileName, compileOutputLimit);
|
||||||
|
if (compileResult.exitCode === ExitCode.CE) {
|
||||||
|
return compileResult;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Run the code
|
||||||
|
const runResult = await run(container, fileName, timeLimit, runOutputLimit);
|
||||||
|
return runResult;
|
||||||
|
} catch (error) {
|
||||||
|
console.error(error);
|
||||||
|
return {
|
||||||
|
id: uuid(),
|
||||||
|
output: "System Error",
|
||||||
|
exitCode: ExitCode.SE,
|
||||||
|
executionTime: null,
|
||||||
|
memoryUsage: null,
|
||||||
|
};
|
||||||
|
} finally {
|
||||||
|
if (container) {
|
||||||
|
await container.kill();
|
||||||
|
await container.remove();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function compile(
|
||||||
|
container: Docker.Container,
|
||||||
|
file: string,
|
||||||
|
fileName: string,
|
||||||
|
maxOutput: number = 1 * 1024 * 1024
|
||||||
|
): Promise<JudgeResult> {
|
||||||
|
const compileExec = await container.exec({
|
||||||
|
Cmd: ["gcc", "-O2", file, "-o", fileName],
|
||||||
|
AttachStdout: true,
|
||||||
|
AttachStderr: true,
|
||||||
|
});
|
||||||
|
|
||||||
|
return new Promise<JudgeResult>((resolve, reject) => {
|
||||||
|
compileExec.start({}, (error, stream) => {
|
||||||
|
if (error || !stream) {
|
||||||
|
return reject({ output: "System Error", exitCode: ExitCode.SE });
|
||||||
|
}
|
||||||
|
|
||||||
|
const stdoutChunks: string[] = [];
|
||||||
|
let stdoutLength = 0;
|
||||||
|
const stdoutStream = new Writable({
|
||||||
|
write(chunk, _encoding, callback) {
|
||||||
|
let text = chunk.toString();
|
||||||
|
if (stdoutLength + text.length > maxOutput) {
|
||||||
|
text = text.substring(0, maxOutput - stdoutLength);
|
||||||
|
stdoutChunks.push(text);
|
||||||
|
stdoutLength = maxOutput;
|
||||||
|
callback();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
stdoutChunks.push(text);
|
||||||
|
stdoutLength += text.length;
|
||||||
|
callback();
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const stderrChunks: string[] = [];
|
||||||
|
let stderrLength = 0;
|
||||||
|
const stderrStream = new Writable({
|
||||||
|
write(chunk, _encoding, callback) {
|
||||||
|
let text = chunk.toString();
|
||||||
|
if (stderrLength + text.length > maxOutput) {
|
||||||
|
text = text.substring(0, maxOutput - stderrLength);
|
||||||
|
stderrChunks.push(text);
|
||||||
|
stderrLength = maxOutput;
|
||||||
|
callback();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
stderrChunks.push(text);
|
||||||
|
stderrLength += text.length;
|
||||||
|
callback();
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
docker.modem.demuxStream(stream, stdoutStream, stderrStream);
|
||||||
|
|
||||||
|
stream.on("end", async () => {
|
||||||
|
const stdout = stdoutChunks.join("");
|
||||||
|
const stderr = stderrChunks.join("");
|
||||||
|
const exitCode = (await compileExec.inspect()).ExitCode;
|
||||||
|
|
||||||
|
let result: JudgeResult;
|
||||||
|
|
||||||
|
if (exitCode !== 0 || stderr) {
|
||||||
|
result = {
|
||||||
|
id: uuid(),
|
||||||
|
output: stderr || "Compilation Error",
|
||||||
|
exitCode: ExitCode.CE,
|
||||||
|
executionTime: null,
|
||||||
|
memoryUsage: null,
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
result = {
|
||||||
|
id: uuid(),
|
||||||
|
output: stdout,
|
||||||
|
exitCode: ExitCode.CS,
|
||||||
|
executionTime: null,
|
||||||
|
memoryUsage: null,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
resolve(result);
|
||||||
|
});
|
||||||
|
|
||||||
|
stream.on("error", () => {
|
||||||
|
reject({ output: "System Error", exitCode: ExitCode.SE });
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Run code and implement timeout
|
||||||
|
async function run(
|
||||||
|
container: Docker.Container,
|
||||||
|
fileName: string,
|
||||||
|
timeLimit: number = 1000,
|
||||||
|
maxOutput: number = 1 * 1024 * 1024,
|
||||||
|
): Promise<JudgeResult> {
|
||||||
|
const runExec = await container.exec({
|
||||||
|
Cmd: [`./${fileName}`],
|
||||||
|
AttachStdout: true,
|
||||||
|
AttachStderr: true,
|
||||||
|
AttachStdin: true,
|
||||||
|
});
|
||||||
|
|
||||||
|
return new Promise<JudgeResult>((resolve, reject) => {
|
||||||
|
const stdoutChunks: string[] = [];
|
||||||
|
let stdoutLength = 0;
|
||||||
|
const stdoutStream = new Writable({
|
||||||
|
write(chunk, _encoding, callback) {
|
||||||
|
let text = chunk.toString();
|
||||||
|
if (stdoutLength + text.length > maxOutput) {
|
||||||
|
text = text.substring(0, maxOutput - stdoutLength);
|
||||||
|
stdoutChunks.push(text);
|
||||||
|
stdoutLength = maxOutput;
|
||||||
|
callback();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
stdoutChunks.push(text);
|
||||||
|
stdoutLength += text.length;
|
||||||
|
callback();
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
const stderrChunks: string[] = [];
|
||||||
|
let stderrLength = 0;
|
||||||
|
const stderrStream = new Writable({
|
||||||
|
write(chunk, _encoding, callback) {
|
||||||
|
let text = chunk.toString();
|
||||||
|
if (stderrLength + text.length > maxOutput) {
|
||||||
|
text = text.substring(0, maxOutput - stderrLength);
|
||||||
|
stderrChunks.push(text);
|
||||||
|
stderrLength = maxOutput;
|
||||||
|
callback();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
stderrChunks.push(text);
|
||||||
|
stderrLength += text.length;
|
||||||
|
callback();
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
// Start the exec stream
|
||||||
|
runExec.start({ hijack: true }, (error, stream) => {
|
||||||
|
if (error || !stream) {
|
||||||
|
return reject({ output: "System Error", exitCode: ExitCode.SE });
|
||||||
|
}
|
||||||
|
|
||||||
|
stream.write("[2,7,11,15]\n9\n[3,2,4]\n6\n[3,3]\n6");
|
||||||
|
stream.end();
|
||||||
|
|
||||||
|
docker.modem.demuxStream(stream, stdoutStream, stderrStream);
|
||||||
|
|
||||||
|
// Timeout mechanism
|
||||||
|
const timeoutId = setTimeout(async () => {
|
||||||
|
resolve({
|
||||||
|
id: uuid(),
|
||||||
|
output: "Time Limit Exceeded",
|
||||||
|
exitCode: ExitCode.TLE,
|
||||||
|
executionTime: null,
|
||||||
|
memoryUsage: null,
|
||||||
|
});
|
||||||
|
}, timeLimit);
|
||||||
|
|
||||||
|
stream.on("end", async () => {
|
||||||
|
clearTimeout(timeoutId); // Clear the timeout if the program finishes before the time limit
|
||||||
|
const stdout = stdoutChunks.join("");
|
||||||
|
const stderr = stderrChunks.join("");
|
||||||
|
const exitCode = (await runExec.inspect()).ExitCode;
|
||||||
|
|
||||||
|
let result: JudgeResult;
|
||||||
|
|
||||||
|
// Exit code 0 means successful execution
|
||||||
|
if (exitCode === 0) {
|
||||||
|
result = {
|
||||||
|
id: uuid(),
|
||||||
|
output: stdout,
|
||||||
|
exitCode: ExitCode.AC,
|
||||||
|
executionTime: null,
|
||||||
|
memoryUsage: null,
|
||||||
|
};
|
||||||
|
} else if (exitCode === 137) {
|
||||||
|
result = {
|
||||||
|
id: uuid(),
|
||||||
|
output: stderr || "Memory Limit Exceeded",
|
||||||
|
exitCode: ExitCode.MLE,
|
||||||
|
executionTime: null,
|
||||||
|
memoryUsage: null,
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
result = {
|
||||||
|
id: uuid(),
|
||||||
|
output: stderr || "Runtime Error",
|
||||||
|
exitCode: ExitCode.RE,
|
||||||
|
executionTime: null,
|
||||||
|
memoryUsage: null,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
resolve(result);
|
||||||
|
});
|
||||||
|
|
||||||
|
stream.on("error", () => {
|
||||||
|
clearTimeout(timeoutId); // Clear timeout in case of error
|
||||||
|
reject({ output: "System Error", exitCode: ExitCode.SE });
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
29
src/actions/language-server.ts
Normal file
29
src/actions/language-server.ts
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
"use server";
|
||||||
|
|
||||||
|
import prisma from "@/lib/prisma";
|
||||||
|
import { EditorLanguage } from "@/generated/client";
|
||||||
|
import { SettingsLanguageServerFormValues } from "@/app/(app)/dashboard/@admin/settings/language-server/form";
|
||||||
|
|
||||||
|
export const getLanguageServerConfig = async (language: EditorLanguage) => {
|
||||||
|
return await prisma.languageServerConfig.findUnique({
|
||||||
|
where: { language },
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
export const handleLanguageServerConfigSubmit = async (
|
||||||
|
language: EditorLanguage,
|
||||||
|
data: SettingsLanguageServerFormValues
|
||||||
|
) => {
|
||||||
|
const existing = await getLanguageServerConfig(language);
|
||||||
|
|
||||||
|
if (existing) {
|
||||||
|
await prisma.languageServerConfig.update({
|
||||||
|
where: { language },
|
||||||
|
data,
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
await prisma.languageServerConfig.create({
|
||||||
|
data: { ...data, language },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
};
|
@ -23,7 +23,7 @@ import { Button } from "@/components/ui/button";
|
|||||||
import { Separator } from "@/components/ui/separator";
|
import { Separator } from "@/components/ui/separator";
|
||||||
import { zodResolver } from "@hookform/resolvers/zod";
|
import { zodResolver } from "@hookform/resolvers/zod";
|
||||||
import { EditorLanguage, LanguageServerProtocol } from "@/generated/client";
|
import { EditorLanguage, LanguageServerProtocol } from "@/generated/client";
|
||||||
import { handleLanguageServerConfigSubmit } from "@/app/actions/language-server";
|
import { handleLanguageServerConfigSubmit } from "@/actions/language-server";
|
||||||
|
|
||||||
const settingsLanguageServerFormSchema = z.object({
|
const settingsLanguageServerFormSchema = z.object({
|
||||||
protocol: z.nativeEnum(LanguageServerProtocol),
|
protocol: z.nativeEnum(LanguageServerProtocol),
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
import { EditorLanguage } from "@/generated/client";
|
import { EditorLanguage } from "@/generated/client";
|
||||||
import { getLanguageServerConfig } from "@/app/actions/language-server";
|
import { getLanguageServerConfig } from "@/actions/language-server";
|
||||||
import { LanguageServerAccordion } from "@/app/(app)/dashboard/@admin/settings/language-server/accordion";
|
import { LanguageServerAccordion } from "@/app/(app)/dashboard/@admin/settings/language-server/accordion";
|
||||||
|
|
||||||
export default async function SettingsLanguageServerPage() {
|
export default async function SettingsLanguageServerPage() {
|
||||||
|
@ -17,7 +17,7 @@ import { Input } from "@/components/ui/input";
|
|||||||
import { Button } from "@/components/ui/button";
|
import { Button } from "@/components/ui/button";
|
||||||
import { useState, useTransition } from "react";
|
import { useState, useTransition } from "react";
|
||||||
import { zodResolver } from "@hookform/resolvers/zod";
|
import { zodResolver } from "@hookform/resolvers/zod";
|
||||||
import { signInWithCredentials } from "@/app/actions/auth";
|
import { signInWithCredentials } from "@/actions/auth";
|
||||||
import { EyeIcon, EyeOffIcon, MailIcon } from "lucide-react";
|
import { EyeIcon, EyeOffIcon, MailIcon } from "lucide-react";
|
||||||
|
|
||||||
export type CredentialsSignInFormValues = z.infer<typeof authSchema>;
|
export type CredentialsSignInFormValues = z.infer<typeof authSchema>;
|
||||||
|
@ -17,7 +17,7 @@ import { Input } from "@/components/ui/input";
|
|||||||
import { Button } from "@/components/ui/button";
|
import { Button } from "@/components/ui/button";
|
||||||
import { useState, useTransition } from "react";
|
import { useState, useTransition } from "react";
|
||||||
import { zodResolver } from "@hookform/resolvers/zod";
|
import { zodResolver } from "@hookform/resolvers/zod";
|
||||||
import { signUpWithCredentials } from "@/app/actions/auth";
|
import { signUpWithCredentials } from "@/actions/auth";
|
||||||
import { EyeIcon, EyeOffIcon, MailIcon } from "lucide-react";
|
import { EyeIcon, EyeOffIcon, MailIcon } from "lucide-react";
|
||||||
|
|
||||||
export type CredentialsSignUpFormValues = z.infer<typeof authSchema>;
|
export type CredentialsSignUpFormValues = z.infer<typeof authSchema>;
|
||||||
|
@ -8,7 +8,7 @@ import {
|
|||||||
} from "@/components/ui/tooltip";
|
} from "@/components/ui/tooltip";
|
||||||
import { cn } from "@/lib/utils";
|
import { cn } from "@/lib/utils";
|
||||||
import { useState } from "react";
|
import { useState } from "react";
|
||||||
import { judge } from "@/app/actions/judge";
|
import { judge } from "@/actions/judge";
|
||||||
import { Button } from "@/components/ui/button";
|
import { Button } from "@/components/ui/button";
|
||||||
import { useProblem } from "@/hooks/use-problem";
|
import { useProblem } from "@/hooks/use-problem";
|
||||||
import { LoaderCircleIcon, PlayIcon } from "lucide-react";
|
import { LoaderCircleIcon, PlayIcon } from "lucide-react";
|
||||||
|
Loading…
Reference in New Issue
Block a user