bw-hspc-contest-env/extension/bwcontest/src/run/java.ts

import { join } from 'path';
import { exec, spawn } from 'child_process';
import * as util from 'util';
import {
	timeoutSeconds,
	type IRunner,
	type IRunnerParams,
	type IRunnerReturn,
	type RunResult
} from './types';
import kill = require('tree-kill');

const execPromise = util.promisify(exec);

interface IRunnerParamsJava extends IRunnerParams {
	srcDir: string;
	mainFile: string;
	mainClass: string;
	input: string;
	outputCallback?: (data: string) => void;
}

export const runJava: IRunner<IRunnerParamsJava> = async function (
	params: IRunnerParamsJava
): IRunnerReturn {
	console.log(`- BUILD: ${params.mainFile}`);
	const compileCommand = `javac -cp ${join(params.srcDir, 'src')} ${params.mainFile} -d ${join(params.srcDir, 'build')}`;

	try {
		await execPromise(compileCommand);
	} catch (e) {
		const buildErrorText = e?.toString() ?? 'Unknown build errors.';
		console.log('Build errors: ' + buildErrorText);
		return {
			success: false,
			runResult: { kind: 'CompileFailed', resultKindReason: buildErrorText }
		};
	}

	console.log(`- RUN: ${params.mainClass}`);
	const runCommand = `java -cp "${join(params.srcDir, 'build')}" ${params.mainClass}`;

	try {
		let outputBuffer = '';
		const child = spawn(runCommand, { shell: true });
		child.stdout.setEncoding('utf8');
		child.stdout.on('data', (data) => {
			outputBuffer += data.toString();
			params.outputCallback?.(data.toString());
		});
		child.stderr.setEncoding('utf8');
		child.stderr.on('data', (data) => {
			outputBuffer += data.toString();
			params.outputCallback?.(data.toString());
		});

		const runStartTime = performance.now();
		child.stdin.write(params.input);
		child.stdin.end();

		let timeLimitExceeded = false;
		let completedNormally = false;

		return {
			success: true,
			runResult: new Promise<RunResult>((resolve) => {
				child.on('close', () => {
					completedNormally = !timeLimitExceeded;

					const runEndTime = performance.now();
					const runtimeMilliseconds = Math.floor(runEndTime - runStartTime);

					if (completedNormally) {
						clearTimeout(timeoutHandle);
						resolve({
							kind: 'Completed',
							output: outputBuffer,
							exitCode: child.exitCode!,
							runtimeMilliseconds
						});
					} else {
						console.log(`Process terminated, total sandbox time: ${runtimeMilliseconds}ms`);
						resolve({
							kind: 'TimeLimitExceeded',
							output: outputBuffer,
							resultKindReason: `Timeout after ${timeoutSeconds} seconds`
						});
					}
				});

				const timeoutHandle = setTimeout(() => {
					if (completedNormally) {
						return;
					}

					console.log(`Run timed out after ${timeoutSeconds} seconds, killing process...`);
					timeLimitExceeded = true;

					child.stdin.end();
					child.stdin.destroy();
					child.stdout.destroy();
					child.stderr.destroy();
					child.kill('SIGKILL');
				}, timeoutSeconds * 1000);
			}),
			killFunc() {
				if (child.pid !== undefined) {
					if (!completedNormally && !timeLimitExceeded) {
						kill(child.pid);
						params.outputCallback?.('\n[Manually stopped]');
					}
				}
			}
		};
	} catch (error) {
		return { success: false, runResult: { kind: 'RunError' } };
	}
};