'use strict'; const assert = require('assert'); const Tools = require('./../tools').includeFormats(); let battleNum = 1; const cache = new Map(); const RULE_FLAGS = { pokemon: 1, legality: 2, preview: 4, sleepClause: 8, cancel: 16, partialDecisions: 32, }; function capitalize(word) { return word.charAt(0).toUpperCase() + word.slice(1); } class TestTools { constructor(options) { if (!options) options = {}; const mod = options.mod || 'base'; this.baseFormat = options.baseFormat || {effectType: 'Format', mod: mod}; this.tools = Tools.mod(mod); this.modPrefix = this.baseFormat.name ? `[${this.baseFormat.name}]` : ''; if (!this.modPrefix && !this.tools.isBase) { this.modPrefix = (/^gen\d$/.test(mod) ? `[Gen ${this.tools.gen}]` : `[${mod}]`); } // Handle caches this.formats = new Map([['singles', new Map()], ['doubles', new Map()], ['triples', new Map()]]); cache.set(this.baseFormat.id || mod, this); } mod(mod) { if (cache.has(mod)) return cache.get(mod); if (Tools.dexes[mod]) return new TestTools({mod: mod}); const baseFormat = Tools.getFormat(mod); if (baseFormat.effectType === 'Format') return new TestTools({mod: baseFormat.mod, baseFormat}); throw new Error(`Mod ${mod} does not exist`); } gen(genNum) { return this.mod('gen' + genNum); } get minRollSeed() { return [0x09917, 0x06924, 0x0e1c8, 0x06af0]; } get maxRollSeed() { return [0x0967e, 0x0b79c, 0x06494, 0x068e3]; } getFormat(options) { let mask = 0; for (let property in options) { if (property === 'gameType' || !options[property]) continue; mask |= RULE_FLAGS[property]; } const gameType = Tools.getId(options.gameType || 'singles'); if (this.formats.get(gameType).has(mask)) return this.formats.get(gameType).get(mask); const gameTypePrefix = gameType === 'singles' ? '' : capitalize(gameType); const formatName = [this.modPrefix, gameTypePrefix, "Custom Game", '' + mask].filter(part => part).join(" "); const formatId = Tools.getId(formatName); const format = Object.assign(Object.assign({}, this.baseFormat), { id: formatId, name: formatName, mask: mask, gameType: options.gameType || 'singles', isCustomGameFormat: true, rated: false, }); if (!format.ruleset) format.ruleset = []; if (!format.banlist) format.banlist = []; if (options.pokemon) format.ruleset.push('Pokemon'); if (options.legality) format.banlist.push('Illegal', 'Unreleased'); if (options.preview) format.ruleset.push('Team Preview'); if (options.sleepClause) format.ruleset.push('Sleep Clause Mod'); if (options.cancel) format.ruleset.push('Cancel Mod'); // if (options.partialDecisions) format.ruleset.push('Partial Decisions'); this.tools.installFormat(formatId, format); return format; } createBattle(options, teams) { if (Array.isArray(options)) { teams = options; options = {}; } const format = this.getFormat(options || {}); const battle = BattleEngine.construct(`battle-test-${battleNum++}`, format.id); if (options && options.partialDecisions) battle.supportPartialDecisions = true; if (teams) { for (let i = 0; i < teams.length; i++) { assert(Array.isArray(teams[i]), "Team provided is not an array"); const slotNum = i + 1; battle.join('p' + slotNum, 'Guest ' + slotNum, 1, teams[i]); } } return battle; } } const common = exports = module.exports = new TestTools(); cache.set('base', common); cache.set('gen7', common);