[extension] Add cpp support
This commit is contained in:
parent
245b987db9
commit
17d9a6e0a4
3
extension/bwcontest/.vscode/settings.json
vendored
3
extension/bwcontest/.vscode/settings.json
vendored
@ -5,5 +5,6 @@
|
|||||||
"search.exclude": {
|
"search.exclude": {
|
||||||
"out": true
|
"out": true
|
||||||
},
|
},
|
||||||
"typescript.tsc.autoDetect": "off"
|
"typescript.tsc.autoDetect": "off",
|
||||||
|
"cSpell.enabled": false
|
||||||
}
|
}
|
||||||
|
@ -4,7 +4,7 @@ import { cloneAndOpenRepo } from './extension';
|
|||||||
import { BWPanel } from './problemPanel';
|
import { BWPanel } from './problemPanel';
|
||||||
import urlJoin from 'url-join';
|
import urlJoin from 'url-join';
|
||||||
|
|
||||||
export type ContestLanguage = 'Java' | 'CSharp';
|
export type ContestLanguage = 'Java' | 'CSharp' | 'CPP';
|
||||||
|
|
||||||
export type TeamData = {
|
export type TeamData = {
|
||||||
teamId: number;
|
teamId: number;
|
||||||
|
@ -7,6 +7,7 @@ import { join } from 'path';
|
|||||||
import { TeamData } from './SidebarProvider';
|
import { TeamData } from './SidebarProvider';
|
||||||
import { submitProblem } from './submit';
|
import { submitProblem } from './submit';
|
||||||
import { runCSharp } from './run/csharp';
|
import { runCSharp } from './run/csharp';
|
||||||
|
import { runCpp } from './run/cpp';
|
||||||
|
|
||||||
export type ProblemData = {
|
export type ProblemData = {
|
||||||
id: number;
|
id: number;
|
||||||
@ -202,6 +203,21 @@ export class BWPanel {
|
|||||||
this.webviewPostMessage({ msg: 'onRunningDone' });
|
this.webviewPostMessage({ msg: 'onRunningDone' });
|
||||||
}
|
}
|
||||||
);
|
);
|
||||||
|
} else if (teamData.language === 'CPP') {
|
||||||
|
killFunc = await runCpp(
|
||||||
|
join(repoDir, 'BWContest', teamData.contestId.toString(), teamData.teamId.toString()),
|
||||||
|
problem.pascalName,
|
||||||
|
input,
|
||||||
|
process.platform === 'win32' ? 'VisualStudio' : 'GCC',
|
||||||
|
(data: string) => {
|
||||||
|
outputBuffer.push(data);
|
||||||
|
this.webviewPostMessage({ msg: 'onRunningOutput', data: outputBuffer.join('') });
|
||||||
|
},
|
||||||
|
() => {
|
||||||
|
this.runningProgram = undefined;
|
||||||
|
this.webviewPostMessage({ msg: 'onRunningDone' });
|
||||||
|
}
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (killFunc !== undefined) {
|
if (killFunc !== undefined) {
|
||||||
|
93
extension/bwcontest/src/run/cpp.ts
Normal file
93
extension/bwcontest/src/run/cpp.ts
Normal file
@ -0,0 +1,93 @@
|
|||||||
|
import * as fs from 'fs-extra';
|
||||||
|
import { join } from 'path';
|
||||||
|
import os = require('os');
|
||||||
|
import { exec, spawn } from 'child_process';
|
||||||
|
import util = require('node:util');
|
||||||
|
import kill = require('tree-kill');
|
||||||
|
|
||||||
|
const execPromise = util.promisify(exec);
|
||||||
|
|
||||||
|
export type CppPlatform = 'VisualStudio' | 'GCC';
|
||||||
|
|
||||||
|
export async function runCpp(
|
||||||
|
srcDir: string,
|
||||||
|
problemName: string,
|
||||||
|
input: string,
|
||||||
|
cppPlatform: CppPlatform,
|
||||||
|
outputCallback: (data: string) => void,
|
||||||
|
doneCallback: () => void
|
||||||
|
): Promise<(() => void) | undefined> {
|
||||||
|
const tempDir = os.tmpdir();
|
||||||
|
const buildDir = join(tempDir, 'bwcontest_cpp');
|
||||||
|
if (await fs.exists(buildDir)) {
|
||||||
|
await fs.remove(buildDir);
|
||||||
|
}
|
||||||
|
await fs.mkdir(buildDir);
|
||||||
|
|
||||||
|
const configureCommand = `cmake -S ${srcDir} -B ${buildDir}`;
|
||||||
|
try {
|
||||||
|
await execPromise(configureCommand);
|
||||||
|
} catch (error) {
|
||||||
|
outputCallback('[Configure Error]\n\n');
|
||||||
|
outputCallback(String(error));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const compileCommand = `cmake --build ${buildDir} --target ${problemName}`;
|
||||||
|
try {
|
||||||
|
await execPromise(compileCommand);
|
||||||
|
} catch (error) {
|
||||||
|
outputCallback('[Compile Error]\n\n');
|
||||||
|
outputCallback(String(error));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let runCommand: string = '';
|
||||||
|
if (cppPlatform === 'VisualStudio') {
|
||||||
|
runCommand = `${join(buildDir, 'Debug', `${problemName}.exe`)}`;
|
||||||
|
} else if (cppPlatform === 'GCC') {
|
||||||
|
runCommand = `${(join(buildDir), problemName)}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
const child = spawn(runCommand, { shell: true });
|
||||||
|
child.stdout.setEncoding('utf8');
|
||||||
|
child.stdout.on('data', (data) => {
|
||||||
|
outputCallback(data.toString());
|
||||||
|
});
|
||||||
|
child.stderr.setEncoding('utf8');
|
||||||
|
child.stderr.on('data', (data) => {
|
||||||
|
outputCallback(data.toString());
|
||||||
|
});
|
||||||
|
child.stdin.write(input);
|
||||||
|
child.stdin.end();
|
||||||
|
|
||||||
|
let done = false;
|
||||||
|
|
||||||
|
child.on('close', () => {
|
||||||
|
if (done === false) {
|
||||||
|
done = true;
|
||||||
|
doneCallback();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
setTimeout(() => {
|
||||||
|
if (done === false) {
|
||||||
|
console.log('\n[30 seconds reached, killing process...]');
|
||||||
|
done = true;
|
||||||
|
if (child.pid) {
|
||||||
|
kill(child.pid);
|
||||||
|
}
|
||||||
|
outputCallback('\n[Timeout after 30 seconds]');
|
||||||
|
doneCallback();
|
||||||
|
}
|
||||||
|
}, 30000);
|
||||||
|
|
||||||
|
return () => {
|
||||||
|
if (child.pid) {
|
||||||
|
done = true;
|
||||||
|
kill(child.pid);
|
||||||
|
outputCallback('\n[Manually stopped]');
|
||||||
|
doneCallback();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user