mirror of
https://github.com/smogon/pokemon-showdown.git
synced 2026-05-18 11:14:39 -05:00
e8451f8104
A lot of our code for child processes doesn't really follow our original standards. This refactors them back to work the way they were intended to.
104 lines
2.8 KiB
TypeScript
104 lines
2.8 KiB
TypeScript
/**
|
|
* Team Validator
|
|
* Pokemon Showdown - http://pokemonshowdown.com/
|
|
*
|
|
* Spawns a child process to validate teams.
|
|
*
|
|
* @license MIT
|
|
*/
|
|
|
|
import {crashlogger} from '../lib/crashlogger';
|
|
|
|
export class TeamValidatorAsync {
|
|
format: Format;
|
|
|
|
constructor(format: string) {
|
|
this.format = Dex.getFormat(format);
|
|
}
|
|
|
|
validateTeam(team: string, options?: {removeNicknames?: boolean}) {
|
|
let formatid = this.format.id;
|
|
if (this.format.customRules) formatid += '@@@' + this.format.customRules.join(',');
|
|
return PM.query({formatid, options, team});
|
|
}
|
|
|
|
static get(format: string) {
|
|
return new TeamValidatorAsync(format);
|
|
}
|
|
}
|
|
|
|
export const get = TeamValidatorAsync.get;
|
|
|
|
/*********************************************************
|
|
* Process manager
|
|
*********************************************************/
|
|
|
|
import {QueryProcessManager} from '../lib/process-manager';
|
|
|
|
export const PM = new QueryProcessManager<{
|
|
formatid: string, options?: {removeNicknames?: boolean}, team: string,
|
|
}>(module, message => {
|
|
const {formatid, options, team} = message;
|
|
const parsedTeam = Dex.fastUnpackTeam(team);
|
|
|
|
if (Config.debugvalidatorprocesses && process.send) {
|
|
process.send('DEBUG\n' + JSON.stringify(message));
|
|
}
|
|
|
|
let problems;
|
|
try {
|
|
problems = TeamValidator.get(formatid).validateTeam(parsedTeam, options);
|
|
} catch (err) {
|
|
crashlogger(err, 'A team validation', {
|
|
formatid,
|
|
team,
|
|
});
|
|
problems = [
|
|
`Your team crashed the validator. We'll fix this crash within a few minutes (we're automatically notified),` +
|
|
` but if you don't want to wait, just use a different team for now.`,
|
|
];
|
|
}
|
|
|
|
if (problems?.length) {
|
|
return '0' + problems.join('\n');
|
|
}
|
|
const packedTeam = Dex.packTeam(parsedTeam);
|
|
// console.log('FROM: ' + message.substr(pipeIndex2 + 1));
|
|
// console.log('TO: ' + packedTeam);
|
|
return '1' + packedTeam;
|
|
});
|
|
|
|
import {Repl} from '../lib/repl';
|
|
import {TeamValidator} from '../sim/team-validator';
|
|
import {Config} from './config-loader';
|
|
|
|
if (!PM.isParentProcess) {
|
|
// This is a child process!
|
|
global.Config = Config;
|
|
|
|
global.TeamValidator = TeamValidator;
|
|
|
|
global.Monitor = {
|
|
crashlog(error: Error, source = 'A team validator process', details: AnyObject | null = null) {
|
|
const repr = JSON.stringify([error.name, error.message, source, details]);
|
|
process.send!(`THROW\n@!!@${repr}\n${error.stack}`);
|
|
},
|
|
};
|
|
|
|
if (Config.crashguard) {
|
|
process.on('uncaughtException', (err: Error) => {
|
|
Monitor.crashlog(err, `A team validator process`);
|
|
});
|
|
process.on('unhandledRejection', err => {
|
|
Monitor.crashlog(err as any || {}, 'A team validator process Promise');
|
|
});
|
|
}
|
|
|
|
global.Dex = require('../sim/dex').Dex.includeData();
|
|
|
|
// eslint-disable-next-line no-eval
|
|
Repl.start(`team-validator-${process.pid}`, cmd => eval(cmd));
|
|
} else {
|
|
PM.spawn(global.Config ? Config.validatorprocesses : 1);
|
|
}
|