237 lines
8.2 KiB
TypeScript
237 lines
8.2 KiB
TypeScript
|
|
import fs from 'fs';
|
|
import path from 'path';
|
|
import { spawn } from 'child_process';
|
|
import { CodeListingMap, WorkerBuildStep, WorkerError, WorkerErrorResult, WorkerFileUpdate, WorkerResult, isOutputResult } from '../../common/workertypes';
|
|
import { getBasePlatform, getRootBasePlatform, replaceAll } from '../../common/util';
|
|
import { BuildStep, makeErrorMatcher } from '../workermain';
|
|
import { parseObjDump } from './clang';
|
|
|
|
|
|
const LLVM_MOS_TOOL: ServerBuildTool = {
|
|
name: 'llvm-mos',
|
|
version: '',
|
|
extensions: ['.c', '.cpp', '.s', '.S', '.C'],
|
|
archs: ['6502'],
|
|
platforms: ['atari8', 'c64', 'nes', 'pce', 'vcs'],
|
|
platform_configs: {
|
|
default: {
|
|
binpath: 'llvm-mos/bin',
|
|
command: 'mos-clang',
|
|
args: ['-Os', '-g', '-o', '$OUTFILE', '$INFILES'],
|
|
},
|
|
debug: { // TODO
|
|
binpath: 'llvm-mos/bin',
|
|
command: 'llvm-objdump',
|
|
args: ['-l', '-t', '$WORKDIR/a.out.elf', '>$WORKDIR/debug.out']
|
|
},
|
|
c64: {
|
|
command: 'mos-c64-clang',
|
|
},
|
|
atari8: {
|
|
command: 'mos-atari8-clang',
|
|
},
|
|
nes: {
|
|
command: 'mos-nes-nrom-clang', // TODO
|
|
libargs: ['-lneslib', '-lfamitone2']
|
|
},
|
|
pce: {
|
|
command: 'mos-pce-clang', // TODO
|
|
},
|
|
vcs: {
|
|
command: 'mos-atari2600-3e-clang', // TODO
|
|
},
|
|
}
|
|
}
|
|
|
|
export function findBestTool(step: BuildStep) {
|
|
if (!step?.tool) throw new Error('No tool specified');
|
|
const [name, version] = step.tool.split('@');
|
|
for (let tool of TOOLS) {
|
|
if (tool.name === name && (!version || version === 'latest' || tool.version === version)) {
|
|
return tool;
|
|
}
|
|
}
|
|
throw new Error(`Tool not found: ${step.tool}`);
|
|
}
|
|
|
|
export const TOOLS: ServerBuildTool[] = [
|
|
Object.assign({}, LLVM_MOS_TOOL, { version: 'latest' }),
|
|
];
|
|
|
|
interface ServerBuildTool {
|
|
name: string;
|
|
version: string;
|
|
extensions: string[];
|
|
archs: string[];
|
|
platforms: string[];
|
|
platform_configs: { [platform: string]: ServerBuildToolPlatformConfig };
|
|
}
|
|
|
|
interface ServerBuildToolPlatformConfig {
|
|
binpath?: string;
|
|
command?: string;
|
|
args?: string[];
|
|
libargs?: string[];
|
|
}
|
|
|
|
|
|
|
|
export class ServerBuildEnv {
|
|
|
|
rootdir: string;
|
|
sessionID: string;
|
|
tool: ServerBuildTool;
|
|
sessionDir: string;
|
|
|
|
constructor(rootdir: string, sessionID: string, tool: ServerBuildTool) {
|
|
this.rootdir = path.resolve(rootdir);
|
|
this.sessionID = sessionID;
|
|
this.tool = tool;
|
|
// make sure sessionID is well-formed
|
|
if (!sessionID.match(/^[a-zA-Z0-9_-]+$/)) {
|
|
throw new Error(`Invalid sessionID: ${sessionID}`);
|
|
}
|
|
// create sessionID directory if it doesn't exist
|
|
this.sessionDir = path.join(this.rootdir, 'sessions', sessionID);
|
|
if (!fs.existsSync(this.sessionDir)) {
|
|
fs.mkdirSync(this.sessionDir);
|
|
}
|
|
}
|
|
|
|
async addFileUpdate(file: WorkerFileUpdate) {
|
|
// make sure file.path contains no path components
|
|
if (file.path.match(/[\\\/]/)) {
|
|
throw new Error(`Invalid file path: ${file.path}`);
|
|
}
|
|
await fs.promises.writeFile(path.join(this.sessionDir, file.path), file.data);
|
|
}
|
|
|
|
async build(step: WorkerBuildStep, platform?: string): Promise<WorkerResult> {
|
|
// build config
|
|
let platformID = platform || getRootBasePlatform(step.platform);
|
|
let config = this.tool.platform_configs[platformID];
|
|
if (!config) {
|
|
throw new Error(`No config for platform ${platformID}`);
|
|
}
|
|
let defaultConfig = this.tool.platform_configs.default;
|
|
if (!defaultConfig) {
|
|
throw new Error(`No default config for tool ${this.tool.name}`);
|
|
}
|
|
config = Object.assign({}, defaultConfig, config); // combine configs
|
|
// copy args
|
|
let args = config.args.slice(0); //copy array
|
|
let command = config.command;
|
|
// replace $OUTFILE
|
|
let outfile = path.join(this.sessionDir, 'a.out'); // TODO? a.out
|
|
for (let i = 0; i < args.length; i++) {
|
|
args[i] = args[i].replace(/\$OUTFILE/g, outfile);
|
|
args[i] = args[i].replace(/\$WORKDIR/g, this.sessionDir);
|
|
}
|
|
// replace $INFILES with the list of input files
|
|
// TODO
|
|
let infiles = [];
|
|
for (let i = 0; i < step.files.length; i++) {
|
|
let f = step.files[i];
|
|
for (let ext of this.tool.extensions) {
|
|
if (f.endsWith(ext)) {
|
|
infiles.push(path.join(this.sessionDir, f));
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
for (let i = 0; i < args.length; i++) {
|
|
if (args[i] === '$INFILES') {
|
|
args = args.slice(0, i).concat(infiles).concat(args.slice(i + 1));
|
|
break;
|
|
}
|
|
}
|
|
if (config.libargs) {
|
|
args = args.concat(config.libargs);
|
|
}
|
|
console.log(`Running: ${command} ${args.join(' ')}`);
|
|
// spawn after setting PATH env var
|
|
// TODO
|
|
let childProcess = spawn(command, args, {
|
|
shell: true,
|
|
cwd: this.rootdir,
|
|
env: { PATH: path.join(this.rootdir, config.binpath)
|
|
} });
|
|
let outputData = '';
|
|
let errorData = '';
|
|
// TODO?
|
|
childProcess.stdout.on('data', (data) => {
|
|
outputData += data.toString();
|
|
});
|
|
childProcess.stderr.on('data', (data) => {
|
|
errorData += data.toString();
|
|
});
|
|
return new Promise((resolve, reject) => {
|
|
childProcess.on('close', async (code) => {
|
|
if (code === 0) {
|
|
if (platform === 'debug') {
|
|
resolve(this.processDebugInfo(step));
|
|
} else {
|
|
resolve(this.processOutput(step));
|
|
}
|
|
} else {
|
|
errorData = replaceAll(errorData, this.sessionDir, '');
|
|
errorData = replaceAll(errorData, this.rootdir, '');
|
|
// remove folder paths
|
|
errorData = errorData.replace(/(\/var\/folders\/.+?\/).+?:/g, '');
|
|
let errorResult = await this.processErrors(step, errorData);
|
|
if (errorResult.errors.length === 0) {
|
|
errorResult.errors.push({ line: 0, msg: `Build failed.\n\n${errorData}` });
|
|
}
|
|
resolve(errorResult);
|
|
}
|
|
});
|
|
});
|
|
}
|
|
|
|
async processErrors(step: WorkerBuildStep, errorData: string): Promise<WorkerErrorResult> {
|
|
let errors = [];
|
|
// split errorData into lines
|
|
let errorMatcher = makeErrorMatcher(errors, /([^:/]+):(\d+):(\d+):\s*(.+)/, 2, 4, step.path, 1);
|
|
for (let line of errorData.split('\n')) {
|
|
errorMatcher(line);
|
|
}
|
|
return { errors };
|
|
}
|
|
|
|
async processOutput(step: WorkerBuildStep): Promise<WorkerResult> {
|
|
let outfile = path.join(this.sessionDir, 'a.out');
|
|
let output = await fs.promises.readFile(outfile, { encoding: 'base64' });
|
|
return { output };
|
|
}
|
|
|
|
async processDebugInfo(step: WorkerBuildStep): Promise<WorkerResult> {
|
|
let dbgfile = path.join(this.sessionDir, 'debug.out');
|
|
let dbglist = (await fs.promises.readFile(dbgfile)).toString();
|
|
let { listings, symbolmap, segments } = parseObjDump(dbglist);
|
|
return { output: [], listings, symbolmap, segments };
|
|
}
|
|
|
|
async compileAndLink(step: WorkerBuildStep, updates: WorkerFileUpdate[]): Promise<WorkerResult> {
|
|
for (let file of updates) {
|
|
await this.addFileUpdate(file);
|
|
}
|
|
try {
|
|
let result = await this.build(step);
|
|
// did we succeed?
|
|
if (isOutputResult(result)) {
|
|
// do the debug info
|
|
const debugInfo = await this.build(step, 'debug');
|
|
if (isOutputResult(debugInfo)) {
|
|
result.listings = debugInfo.listings;
|
|
result.symbolmap = debugInfo.symbolmap;
|
|
}
|
|
}
|
|
return result;
|
|
} catch (err) {
|
|
return { errors: [{line:0, msg: err.toString()}] };
|
|
}
|
|
}
|
|
}
|
|
|