mirror of
https://github.com/pagefaultgames/pokerogue.git
synced 2024-12-04 20:56:09 +00:00
657 lines
26 KiB
TypeScript
657 lines
26 KiB
TypeScript
import { MainClient, NamedAPIResource } from 'pokenode-ts';
|
||
import { MoveTarget, Moves, allMoves } from './move';
|
||
import * as Utils from '../utils';
|
||
import fs from 'vite-plugin-fs/browser';
|
||
import PokemonSpecies, { PokemonForm, SpeciesFormKey, allSpecies } from './pokemon-species';
|
||
import { GrowthRate } from './exp';
|
||
import { Type } from './type';
|
||
import { Abilities, allAbilities } from './ability';
|
||
import { Species } from './species';
|
||
import { pokemonFormLevelMoves } from './pokemon-level-moves';
|
||
import { tmSpecies } from './tms';
|
||
|
||
const targetMap = {
|
||
'specific-move': MoveTarget.ATTACKER,
|
||
'selected-pokemon-me-first': MoveTarget.NEAR_ENEMY,
|
||
'ally': MoveTarget.NEAR_ALLY,
|
||
'users-field': MoveTarget.USER_SIDE,
|
||
'user-or-ally': MoveTarget.USER_OR_NEAR_ALLY,
|
||
'opponents-field': MoveTarget.ENEMY_SIDE,
|
||
'user': MoveTarget.USER,
|
||
'random-opponent': MoveTarget.RANDOM_NEAR_ENEMY,
|
||
'all-other-pokemon': MoveTarget.ALL_NEAR_OTHERS,
|
||
'selected-pokemon': MoveTarget.NEAR_OTHER,
|
||
'all-opponents': MoveTarget.ALL_NEAR_ENEMIES,
|
||
'entire-field': MoveTarget.BOTH_SIDES,
|
||
'user-and-allies': MoveTarget.USER_AND_ALLIES,
|
||
'all-pokemon': MoveTarget.ALL,
|
||
'all-allies': MoveTarget.NEAR_ALLY,
|
||
'fainting-pokemon': MoveTarget.NEAR_OTHER
|
||
};
|
||
|
||
const generationMap = {
|
||
'generation-i': 1,
|
||
'generation-ii': 2,
|
||
'generation-iii': 3,
|
||
'generation-iv': 4,
|
||
'generation-v': 5,
|
||
'generation-vi': 6,
|
||
'generation-vii': 7,
|
||
'generation-viii': 8,
|
||
'generation-ix': 9
|
||
};
|
||
|
||
const growthRateMap = {
|
||
'slow-then-very-fast': GrowthRate.ERRATIC,
|
||
'fast': GrowthRate.FAST,
|
||
'medium': GrowthRate.MEDIUM_FAST,
|
||
'medium-slow': GrowthRate.MEDIUM_SLOW,
|
||
'slow': GrowthRate.SLOW,
|
||
'fast-then-very-slow': GrowthRate.FLUCTUATING
|
||
};
|
||
|
||
const regionalForms = [ 'alola', 'galar', 'hisui', 'paldea' ];
|
||
|
||
const ignoredForms = [ 'gmax', 'totem', 'cap', 'starter' ];
|
||
|
||
const generationDexNumbers = {
|
||
1: 151,
|
||
2: 251,
|
||
3: 386,
|
||
4: 494,
|
||
5: 649,
|
||
6: 721,
|
||
7: 809,
|
||
8: 905,
|
||
9: 1010
|
||
};
|
||
|
||
const versions = [ 'scarlet-violet', 'sword-shield', 'sun-moon' ];
|
||
|
||
type LevelMove = [level: integer, moveId: integer];
|
||
|
||
interface SpeciesLevelMoves {
|
||
[key: string]: LevelMove[]
|
||
}
|
||
|
||
interface FormLevelMoves {
|
||
[key: integer]: LevelMove[]
|
||
}
|
||
|
||
interface SpeciesFormLevelMoves {
|
||
[key: string]: FormLevelMoves
|
||
}
|
||
|
||
interface TmSpecies {
|
||
[key: string]: Array<string | string[]>
|
||
}
|
||
|
||
export async function printPokemon() {
|
||
const api = new MainClient();
|
||
|
||
const useExistingTmList = true;
|
||
|
||
let enumStr = `export enum Species {\n`;
|
||
let pokemonSpeciesStr = `\tallSpecies.push(\n`;
|
||
const speciesLevelMoves: SpeciesLevelMoves = {};
|
||
const speciesFormLevelMoves: SpeciesFormLevelMoves = {};
|
||
const moveTmSpecies: TmSpecies = {};
|
||
|
||
let pokemonArr: NamedAPIResource[] = [];
|
||
|
||
let offset = 0;
|
||
let pokemonResponse = await api.pokemon.listPokemons(offset, 2000)
|
||
|
||
pokemonArr = pokemonResponse.results;
|
||
|
||
const types = Utils.getEnumKeys(Type).map(t => t.toLowerCase());
|
||
const abilities = Utils.getEnumKeys(Abilities).map(a => a.toLowerCase().replace(/\_/g, '-'));
|
||
|
||
const pokemonSpeciesList: PokemonSpecies[] = [];
|
||
|
||
for (let p of pokemonArr) {
|
||
const pokemon = await api.pokemon.getPokemonByName(p.name);
|
||
|
||
let region: string = '';
|
||
|
||
if (pokemon.id > 10000) {
|
||
const dexIdMatch = /\/(\d+)\//.exec(pokemon.species.url);
|
||
if (!dexIdMatch)
|
||
continue;
|
||
|
||
const matchingSpecies = pokemonSpeciesList[parseInt(dexIdMatch[1]) - 1];
|
||
|
||
if (!matchingSpecies)
|
||
continue;
|
||
|
||
const speciesKey = (matchingSpecies as any).key as string;
|
||
|
||
const formName = pokemon.name.slice(speciesKey.length + 1);
|
||
|
||
if (ignoredForms.filter(f => formName.indexOf(f) > -1).length)
|
||
continue;
|
||
|
||
let shortFormName = formName.indexOf('-') > -1
|
||
? formName.slice(0, formName.indexOf('-'))
|
||
: formName;
|
||
|
||
if (regionalForms.indexOf(shortFormName) > -1)
|
||
region = shortFormName.toUpperCase();
|
||
else {
|
||
const formBaseStats: integer[] = [];
|
||
let formBaseTotal = 0;
|
||
// Assume correct stat order in API result
|
||
for (let stat of pokemon.stats) {
|
||
formBaseStats.push(stat.base_stat);
|
||
formBaseTotal += stat.base_stat;
|
||
}
|
||
|
||
const [ formType1, formType2 ] = [ types.indexOf(pokemon.types.find(t => t.slot === 1).type.name), types.indexOf(pokemon.types.find(t => t.slot === 2)?.type.name) ];
|
||
const [ formAbility1, formAbility2, formAbilityHidden ] = [
|
||
Math.max(abilities.indexOf(pokemon.abilities.find(a => a.slot === 1)?.ability.name), 0),
|
||
Math.max(abilities.indexOf(pokemon.abilities.find(a => a.slot === 2)?.ability.name), 0),
|
||
Math.max(abilities.indexOf(pokemon.abilities.find(a => a.slot === 3)?.ability.name), 0)
|
||
];
|
||
|
||
const pokemonForm = new PokemonForm(formName, formName, formType1 as Type, formType2 > -1 ? formType2 as Type : null, pokemon.height / 10, pokemon.weight / 10,
|
||
formAbility1 as Abilities, formAbility2 as Abilities, formAbilityHidden as Abilities, formBaseTotal, formBaseStats[0], formBaseStats[1], formBaseStats[2], formBaseStats[3], formBaseStats[4], formBaseStats[5],
|
||
matchingSpecies.catchRate, matchingSpecies.baseFriendship, matchingSpecies.baseExp, matchingSpecies.genderDiffs);
|
||
pokemonForm.speciesId = matchingSpecies.speciesId;
|
||
pokemonForm.formIndex = matchingSpecies.forms.length;
|
||
pokemonForm.generation = matchingSpecies.generation;
|
||
|
||
let moveVer: string;
|
||
|
||
if (!speciesFormLevelMoves.hasOwnProperty(speciesKey))
|
||
speciesFormLevelMoves[speciesKey] = [];
|
||
speciesFormLevelMoves[speciesKey][pokemonForm.formIndex] = [];
|
||
|
||
for (let version of versions) {
|
||
if (pokemon.moves.find(m => m.version_group_details.find(v => v.version_group.name === version && v.move_learn_method.name === 'level-up'))) {
|
||
moveVer = version;
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (moveVer) {
|
||
pokemon.moves.forEach(moveData => {
|
||
moveData.version_group_details.filter(v => versions.indexOf(v.version_group.name) > -1).forEach(verData => {
|
||
const isMoveVer = verData.version_group.name === moveVer;
|
||
|
||
const moveName = moveData.move.name.toUpperCase().replace(/\_/g, '').replace(/\-/g, '_');
|
||
const moveId = Math.max(Utils.getEnumKeys(Moves).indexOf(moveName), 0);
|
||
|
||
const learnMethod = verData.move_learn_method.name;
|
||
|
||
if (isMoveVer && learnMethod === 'level-up')
|
||
speciesFormLevelMoves[speciesKey][pokemonForm.formIndex].push([ verData.level_learned_at, moveId ]);
|
||
|
||
if ([ 'machine', 'tutor' ].indexOf(learnMethod) > -1 || (useExistingTmList && tmSpecies.hasOwnProperty(moveId as Moves) && learnMethod === 'level-up')) {
|
||
if (!moveTmSpecies.hasOwnProperty(moveId))
|
||
moveTmSpecies[moveId] = [];
|
||
const speciesIndex = moveTmSpecies[moveId].findIndex(s => s[0] === speciesKey);
|
||
if (speciesIndex === -1)
|
||
moveTmSpecies[moveId].push([ speciesKey, formName ]);
|
||
else
|
||
(moveTmSpecies[moveId][speciesIndex] as string[]).push(formName);
|
||
}
|
||
});
|
||
});
|
||
|
||
if (JSON.stringify(speciesLevelMoves[speciesKey]) === JSON.stringify(speciesFormLevelMoves[speciesKey][pokemonForm.formIndex])) {
|
||
delete speciesFormLevelMoves[speciesKey][pokemonForm.formIndex];
|
||
if (!Object.keys(speciesFormLevelMoves[speciesKey]).length)
|
||
delete speciesFormLevelMoves[speciesKey];
|
||
}
|
||
}
|
||
|
||
matchingSpecies.forms.push(pokemonForm);
|
||
continue;
|
||
}
|
||
}
|
||
|
||
const species = await api.pokemon.getPokemonSpeciesByName(pokemon.species.name);
|
||
|
||
let speciesKey = species.name.toUpperCase().replace(/\-/g, '_');
|
||
|
||
const matchingExistingSpecies = allSpecies.find(s => Species[s.speciesId] === speciesKey);
|
||
|
||
let dexId = species.id;
|
||
|
||
if (region) {
|
||
dexId += (regionalForms.indexOf(region.toLowerCase()) + 1) * 2000;
|
||
speciesKey = `${region}_${speciesKey}`;
|
||
}
|
||
|
||
let generationIndex = 0;
|
||
|
||
if (!region)
|
||
while (++generationIndex < 9 && dexId > generationDexNumbers[generationIndex]);
|
||
else
|
||
generationIndex = regionalForms.indexOf(region.toLowerCase()) + 6;
|
||
|
||
const baseStats: integer[] = [];
|
||
let baseTotal = 0;
|
||
// Assume correct stat order in API result
|
||
for (let stat of pokemon.stats) {
|
||
baseStats.push(stat.base_stat);
|
||
baseTotal += stat.base_stat;
|
||
}
|
||
|
||
console.log(pokemon);
|
||
|
||
const [ type1, type2 ] = [ types.indexOf(pokemon.types.find(t => t.slot === 1).type.name), types.indexOf(pokemon.types.find(t => t.slot === 2)?.type.name) ];
|
||
const [ ability1, ability2, abilityHidden ] = [
|
||
Math.max(abilities.indexOf(pokemon.abilities.find(a => a.slot === 1)?.ability.name), 0),
|
||
Math.max(abilities.indexOf(pokemon.abilities.find(a => a.slot === 2)?.ability.name), 0),
|
||
Math.max(abilities.indexOf(pokemon.abilities.find(a => a.slot === 3)?.ability.name), 0)
|
||
];
|
||
|
||
const pokemonSpecies = new PokemonSpecies(dexId, species.names.find(n => n.language.name === 'en').name, generationIndex, species.is_legendary && baseTotal < 660, species.is_legendary && baseTotal >= 660, species.is_mythical,
|
||
species.genera.find(g => g.language.name === 'en')?.genus, type1 as Type, type2 > -1 ? type2 as Type : null, pokemon.height / 10, pokemon.weight / 10, ability1 as Abilities, ability2 as Abilities, abilityHidden as Abilities,
|
||
baseTotal, baseStats[0], baseStats[1], baseStats[2], baseStats[3], baseStats[4], baseStats[5], species.capture_rate, species.base_happiness, pokemon.base_experience, growthRateMap[species.growth_rate.name],
|
||
species.gender_rate < 9 ? 100 - (species.gender_rate * 12.5) : null, species.has_gender_differences, species.forms_switchable);
|
||
|
||
(pokemonSpecies as any).key = speciesKey;
|
||
|
||
pokemonSpeciesList.push(pokemonSpecies);
|
||
|
||
let moveVer: string;
|
||
|
||
speciesLevelMoves[speciesKey] = [];
|
||
|
||
for (let version of versions) {
|
||
if (pokemon.moves.find(m => m.version_group_details.find(v => v.version_group.name === version && v.move_learn_method.name === 'level-up'))) {
|
||
moveVer = version;
|
||
break;
|
||
}
|
||
}
|
||
|
||
const speciesTmMoves: integer[] = [];
|
||
|
||
if (moveVer) {
|
||
pokemon.moves.forEach(moveData => {
|
||
const verData = moveData.version_group_details.find(v => v.version_group.name === moveVer);
|
||
if (!verData)
|
||
return;
|
||
|
||
const moveName = moveData.move.name.toUpperCase().replace(/\_/g, '').replace(/\-/g, '_');
|
||
const moveId = Math.max(Utils.getEnumKeys(Moves).indexOf(moveName), 0);
|
||
|
||
switch (verData.move_learn_method.name) {
|
||
case 'level-up':
|
||
speciesLevelMoves[speciesKey].push([ verData.level_learned_at, moveId ]);
|
||
break;
|
||
case 'machine':
|
||
case 'tutor':
|
||
if (moveId > 0) {
|
||
if (!moveTmSpecies.hasOwnProperty(moveId))
|
||
moveTmSpecies[moveId] = [];
|
||
if (moveTmSpecies[moveId].indexOf(speciesKey) === -1)
|
||
moveTmSpecies[moveId].push(speciesKey);
|
||
speciesTmMoves.push(moveId);
|
||
}
|
||
break;
|
||
}
|
||
});
|
||
}
|
||
|
||
for (let f of pokemon.forms) {
|
||
const form = await api.pokemon.getPokemonFormByName(f.name);
|
||
const formIndex = pokemonSpecies.forms.length;
|
||
|
||
const matchingForm = matchingExistingSpecies && matchingExistingSpecies.forms.length > formIndex
|
||
? matchingExistingSpecies.forms.find(f2 => f2.formKey === form.form_name || f2.formName === form.form_name) || matchingExistingSpecies.forms[formIndex]
|
||
: null;
|
||
const formName = matchingForm
|
||
? matchingForm.formName
|
||
: form.form_names.find(fn => fn.language.name === 'en')?.name || form.form_name;
|
||
const formKey = matchingForm
|
||
? matchingForm.formKey
|
||
: form.form_name;
|
||
|
||
const [ formType1, formType2 ] = [ types.indexOf(form.types.find(t => t.slot === 1).type.name), types.indexOf(form.types.find(t => t.slot === 2)?.type.name) ];
|
||
const pokemonForm = new PokemonForm(formName, formKey, formType1 as Type, formType2 > -1 ? formType2 as Type : null,
|
||
pokemonSpecies.height, pokemonSpecies.weight, pokemonSpecies.ability1, pokemonSpecies.ability2, pokemonSpecies.abilityHidden, baseTotal, baseStats[0], baseStats[1], baseStats[2], baseStats[3], baseStats[4], baseStats[5],
|
||
pokemonSpecies.catchRate, pokemonSpecies.baseFriendship, pokemonSpecies.baseExp, pokemonSpecies.genderDiffs);
|
||
pokemonForm.speciesId = pokemonSpecies.speciesId;
|
||
pokemonForm.formIndex = formIndex;
|
||
pokemonForm.generation = pokemonSpecies.generation;
|
||
|
||
if (!pokemonForm.formIndex && speciesTmMoves.length) {
|
||
for (let moveId of speciesTmMoves) {
|
||
const speciesIndex = moveTmSpecies[moveId].findIndex(s => s === speciesKey);
|
||
moveTmSpecies[moveId][speciesIndex] = [
|
||
speciesKey,
|
||
formKey
|
||
];
|
||
}
|
||
}
|
||
|
||
pokemonSpecies.forms.push(pokemonForm);
|
||
}
|
||
|
||
console.log(pokemonSpecies.name, pokemonSpecies);
|
||
}
|
||
|
||
for (let pokemonSpecies of pokemonSpeciesList) {
|
||
const speciesKey = (pokemonSpecies as any).key as string;
|
||
|
||
enumStr += ` ${speciesKey}${pokemonSpecies.speciesId >= 2000 ? ` = ${pokemonSpecies.speciesId}` : ''},\n`;
|
||
pokemonSpeciesStr += ` new PokemonSpecies(Species.${speciesKey}, "${pokemonSpecies.name}", ${pokemonSpecies.generation}, ${pokemonSpecies.pseudoLegendary}, ${pokemonSpecies.legendary}, ${pokemonSpecies.mythical}, "${pokemonSpecies.species}", Type.${Type[pokemonSpecies.type1]}, ${pokemonSpecies.type2 ? `Type.${Type[pokemonSpecies.type2]}` : 'null'}, ${pokemonSpecies.height}, ${pokemonSpecies.weight}, Abilities.${Abilities[pokemonSpecies.ability1]}, Abilities.${Abilities[pokemonSpecies.ability2]}, Abilities.${Abilities[pokemonSpecies.abilityHidden]}, ${pokemonSpecies.baseTotal}, ${pokemonSpecies.baseStats[0]}, ${pokemonSpecies.baseStats[1]}, ${pokemonSpecies.baseStats[2]}, ${pokemonSpecies.baseStats[3]}, ${pokemonSpecies.baseStats[4]}, ${pokemonSpecies.baseStats[5]}, ${pokemonSpecies.catchRate}, ${pokemonSpecies.baseFriendship}, ${pokemonSpecies.baseExp}, GrowthRate.${GrowthRate[pokemonSpecies.growthRate]}, ${pokemonSpecies.malePercent}, ${pokemonSpecies.genderDiffs}`;
|
||
if (pokemonSpecies.forms.length > 1) {
|
||
pokemonSpeciesStr += `, ${pokemonSpecies.canChangeForm},`;
|
||
for (let form of pokemonSpecies.forms)
|
||
pokemonSpeciesStr += `\n new PokemonForm("${form.formName}", "${form.formName}", Type.${Type[form.type1]}, ${form.type2 ? `Type.${Type[form.type2]}` : 'null'}, ${form.height}, ${form.weight}, Abilities.${Abilities[form.ability1]}, Abilities.${Abilities[form.ability2]}, Abilities.${Abilities[form.abilityHidden]}, ${form.baseTotal}, ${form.baseStats[0]}, ${form.baseStats[1]}, ${form.baseStats[2]}, ${form.baseStats[3]}, ${form.baseStats[4]}, ${form.baseStats[5]}, ${form.catchRate}, ${form.baseFriendship}, ${form.baseExp}${form.genderDiffs ? ', true' : ''}),`;
|
||
pokemonSpeciesStr += '\n ';
|
||
}
|
||
pokemonSpeciesStr += `),\n`;
|
||
}
|
||
|
||
let speciesLevelMovesStr = `export const pokemonSpeciesLevelMoves: PokemonSpeciesLevelMoves = {\n`;
|
||
let speciesFormLevelMovesStr = `export const pokemonFormLevelMoves: PokemonSpeciesFormLevelMoves = {\n`;
|
||
let tmSpeciesStr = `export const tmSpecies: TmSpecies = {\n`;
|
||
|
||
for (let species of Object.keys(speciesLevelMoves)) {
|
||
speciesLevelMovesStr += ` [Species.${species}]: [\n`;
|
||
|
||
const orderedLevelMoves = speciesLevelMoves[species].sort((a: LevelMove, b: LevelMove) => {
|
||
if (a[0] !== b[0])
|
||
return a[0] < b[0] ? -1 : 1;
|
||
return a[1] < b[1] ? -1 : 1;
|
||
});
|
||
|
||
for (let lm of orderedLevelMoves)
|
||
speciesLevelMovesStr += ` [ ${lm[0]}, Moves.${Moves[lm[1]]} ],\n`;
|
||
|
||
speciesLevelMovesStr += ` ],\n`;
|
||
}
|
||
|
||
for (let species of Object.keys(speciesFormLevelMoves)) {
|
||
speciesFormLevelMovesStr += ` [Species.${species}]: {\n`;
|
||
|
||
for (let f of Object.keys(speciesFormLevelMoves[species])) {
|
||
speciesFormLevelMovesStr += ` ${f}: [\n`;
|
||
|
||
const orderedLevelMoves = speciesFormLevelMoves[species][f].sort((a: LevelMove, b: LevelMove) => {
|
||
if (a[0] !== b[0])
|
||
return a[0] < b[0] ? -1 : 1;
|
||
return a[1] < b[1] ? -1 : 1;
|
||
});
|
||
|
||
for (let lm of orderedLevelMoves)
|
||
speciesFormLevelMovesStr += ` [ ${lm[0]}, Moves.${Moves[lm[1]]} ],\n`;
|
||
|
||
speciesFormLevelMovesStr += ` ],\n`;
|
||
}
|
||
|
||
speciesFormLevelMovesStr += ` },\n`;
|
||
}
|
||
|
||
for (let moveId of Object.keys(moveTmSpecies)) {
|
||
tmSpeciesStr += ` [Moves.${Moves[parseInt(moveId)]}]: [\n`;
|
||
for (let species of moveTmSpecies[moveId]) {
|
||
if (typeof species === 'string')
|
||
tmSpeciesStr += ` Species.${species},\n`;
|
||
else {
|
||
const matchingExistingSpecies = allSpecies.find(s => Species[s.speciesId] === species[0]);
|
||
const forms = (species as string[]).slice(1);
|
||
if (matchingExistingSpecies && (!pokemonFormLevelMoves.hasOwnProperty(matchingExistingSpecies.speciesId) || matchingExistingSpecies.forms.length <= 1 || (matchingExistingSpecies.forms.length === 2 && matchingExistingSpecies.forms[1].formKey.indexOf(SpeciesFormKey.MEGA) > -1) || matchingExistingSpecies.forms.length === forms.length))
|
||
tmSpeciesStr += ` Species.${species[0]},\n`;
|
||
else {
|
||
tmSpeciesStr += ` [\n Species.${species[0]},\n`;
|
||
for (let form of forms)
|
||
tmSpeciesStr += ` '${form}',\n`;
|
||
tmSpeciesStr += ` ],\n`;
|
||
}
|
||
}
|
||
}
|
||
tmSpeciesStr += ` ],\n`;
|
||
}
|
||
|
||
enumStr += `\n};`;
|
||
pokemonSpeciesStr += ` );`;
|
||
speciesLevelMovesStr += `\n};`;
|
||
speciesFormLevelMovesStr += `\n};`;
|
||
tmSpeciesStr += `\n};`;
|
||
|
||
console.log(enumStr);
|
||
console.log(pokemonSpeciesStr);
|
||
console.log(speciesLevelMovesStr);
|
||
console.log(speciesFormLevelMovesStr);
|
||
console.log(tmSpeciesStr);
|
||
|
||
console.log(moveTmSpecies)
|
||
}
|
||
|
||
export async function printAbilities() {
|
||
const replaceText = true;
|
||
|
||
let abilityContent: string = await fs.readFile('./src/data/ability.ts');
|
||
|
||
const api = new MainClient();
|
||
|
||
let enumStr = `export enum Abilities {\n NONE,`;
|
||
let abilityStr = ' allAbilities.push(';
|
||
|
||
abilityContent = abilityContent.slice(abilityContent.indexOf(abilityStr));
|
||
|
||
let abilities: NamedAPIResource[] = [];
|
||
let offset = 0;
|
||
let abilitiesResponse = await api.pokemon.listAbilities(offset, 2000);
|
||
abilities = abilitiesResponse.results;
|
||
|
||
for (let a of abilities) {
|
||
const ability = await api.pokemon.getAbilityByName(a.name);
|
||
const abilityEnumName = ability.name.toUpperCase().replace(/\_/g, '').replace(/\-/g, '_');
|
||
enumStr += `\n ${abilityEnumName},`;
|
||
console.log(ability.name, ability);
|
||
|
||
const matchingLineIndex = abilityContent.search(new RegExp(`new Ability\\\(Abilities.${abilityEnumName},`));
|
||
let matchingLine = matchingLineIndex > -1 ? abilityContent.slice(matchingLineIndex) : null;
|
||
if (matchingLine)
|
||
matchingLine = matchingLine.slice(0, matchingLine.search(/,(?: \/\/.*?)?(?:\r)?\n[ \t]+(?:new|\);)/));
|
||
|
||
let abilityName = ability.names.find(ln => ln.language.name === 'en').name;
|
||
[ 'N', 'P' ].every(s => {
|
||
if (!matchingLine || matchingLine.indexOf(` (${s})`) > -1) {
|
||
abilityName += ` (${s})`;
|
||
return false;
|
||
}
|
||
return true;
|
||
});
|
||
|
||
let flavorText: string;
|
||
if (!matchingLine || replaceText) {
|
||
for (let version of versions) {
|
||
if ((flavorText = ability.flavor_text_entries.find(fte => fte.language.name === 'en' && fte.version_group.name === version)?.flavor_text) || '') {
|
||
if (flavorText.indexOf('forgotten') > -1)
|
||
continue;
|
||
break;
|
||
}
|
||
}
|
||
} else if (matchingLine)
|
||
flavorText = allAbilities[ability.id].description;
|
||
abilityStr += `\n new Ability(Abilities.${abilityEnumName}, "${abilityName}", "${flavorText?.replace(/\n/g, '\\n').replace(/ /g, ' ').replace(/’/g, '\'') || ''}", ${generationMap[ability.generation.name]})`;
|
||
if (matchingLine && matchingLine.length > 1) {
|
||
const newLineIndex = matchingLine.indexOf('\n');
|
||
if (newLineIndex > -1)
|
||
abilityStr += matchingLine.slice(newLineIndex);
|
||
}
|
||
abilityStr += ',';
|
||
}
|
||
|
||
enumStr += `\n};`;
|
||
abilityStr += `\n);`;
|
||
|
||
console.log(enumStr);
|
||
console.log(abilityStr);
|
||
}
|
||
|
||
export async function printMoves() {
|
||
const replaceText = true;
|
||
|
||
let moveContent: string = await fs.readFile('./src/data/move.ts');
|
||
|
||
const api = new MainClient();
|
||
|
||
let enumStr = `export enum Moves {\n NONE,`;
|
||
let moveStr = ' allMoves.push(';
|
||
|
||
moveContent = moveContent.slice(moveContent.indexOf(moveStr));
|
||
|
||
let moves: NamedAPIResource[] = [];
|
||
let offset = 0;
|
||
let movesResponse = await api.move.listMoves(offset, 2000);
|
||
moves = movesResponse.results;
|
||
|
||
console.log(moves);
|
||
|
||
for (let m of moves) {
|
||
const move = await api.move.getMoveByName(m.name);
|
||
const moveEnumName = move.name.toUpperCase().replace(/\_/g, '').replace(/\-/g, '_');
|
||
enumStr += `\n ${moveEnumName},`;
|
||
console.log(move.name, move);
|
||
|
||
const matchingLineIndex = moveContent.search(new RegExp(`new (?:Attack|(?:Self)?Status)Move\\\(Moves.${Moves[move.id]},`));
|
||
let matchingLine = matchingLineIndex > -1 ? moveContent.slice(matchingLineIndex) : null;
|
||
if (matchingLine)
|
||
matchingLine = matchingLine.slice(0, matchingLine.search(/,(?: \/\/.*?)?(?:\r)?\n[ \t]+(?:new|\);)/));
|
||
|
||
let moveName = move.names.find(ln => ln.language.name === 'en').name;
|
||
[ 'N', 'P' ].every(s => {
|
||
if (!matchingLine || matchingLine.indexOf(` (${s})`) > -1) {
|
||
moveName += ` (${s})`;
|
||
return false;
|
||
}
|
||
return true;
|
||
});
|
||
|
||
let flavorText: string;
|
||
if (!matchingLine || replaceText) {
|
||
for (let version of versions) {
|
||
if ((flavorText = move.flavor_text_entries.find(fte => fte.language.name === 'en' && fte.version_group.name === version)?.flavor_text) || '') {
|
||
if (flavorText.indexOf('forgotten') > -1)
|
||
continue;
|
||
break;
|
||
}
|
||
}
|
||
} else if (matchingLine)
|
||
flavorText = allMoves[move.id].effect;
|
||
const moveTarget = targetMap[move.target.name];
|
||
const moveTm = move.id < allMoves.length ? allMoves[move.id].tm : -1;
|
||
moveStr += `\n new ${move.damage_class.name !== 'status' ? 'Attack' : (moveTarget === MoveTarget.USER ? 'Self' : '') + 'Status'}Move(Moves.${moveEnumName}, "${moveName}", Type.${move.type.name.toUpperCase()}${move.damage_class.name !== 'status' ? `, MoveCategory.${move.damage_class.name.toUpperCase()}` : ''}${move.damage_class.name !== 'status' ? `, ${move.power || -1}` : ''}, ${move.accuracy || -1}, ${move.pp}, ${moveTm}, "${flavorText?.replace(/\n/g, '\\n').replace(/ /g, ' ').replace(/’/g, '\'') || ''}", ${move.effect_chance || -1}, ${move.priority}, ${generationMap[move.generation.name]})`;
|
||
const expectedTarget = move.damage_class.name !== 'status' || moveTarget !== MoveTarget.USER ? MoveTarget.NEAR_OTHER : MoveTarget.USER;
|
||
if (matchingLine && matchingLine.length > 1) {
|
||
const newLineIndex = matchingLine.indexOf('\n');
|
||
if (newLineIndex > -1) {
|
||
console.log(matchingLine.slice(newLineIndex).replace(/(?:\r)?\n[ \t]+.target\(.*?\)/g, ''), newLineIndex)
|
||
moveStr += matchingLine.slice(newLineIndex).replace(/(?:\r)?\n[ \t]+.target\(.*?\)/g, '');
|
||
}
|
||
}
|
||
if (moveTarget !== expectedTarget)
|
||
moveStr += `\n .target(MoveTarget.${MoveTarget[moveTarget]})`;
|
||
moveStr += ',';
|
||
}
|
||
|
||
enumStr += `\n};`;
|
||
moveStr += `\n);`;
|
||
|
||
console.log(enumStr);
|
||
console.log(moveStr);
|
||
}
|
||
|
||
export async function printTmSpecies() {
|
||
const moveTmSpecies: TmSpecies = {};
|
||
|
||
const api = new MainClient();
|
||
|
||
const moveIds = Object.keys(tmSpecies).map(k => parseInt(k) as Moves);
|
||
|
||
for (let moveId of moveIds) {
|
||
const move = await api.move.getMoveById(moveId);
|
||
|
||
moveTmSpecies[moveId] = [];
|
||
|
||
for (let species of move.learned_by_pokemon) {
|
||
const dexIdMatch = /\/(\d+)\//.exec(species.url);
|
||
if (!dexIdMatch)
|
||
continue;
|
||
|
||
let dexId = parseInt(dexIdMatch[1]);
|
||
|
||
let matchingSpecies: PokemonSpecies;
|
||
let formKey = '';
|
||
|
||
console.log(species.name);
|
||
|
||
if (dexId < 10000)
|
||
matchingSpecies = allSpecies[dexId - 1];
|
||
else {
|
||
const pokemon = await api.pokemon.getPokemonById(dexId);
|
||
|
||
const speciesDexIdMatch = /\/(\d+)\//.exec(pokemon.species.url);
|
||
if (!speciesDexIdMatch)
|
||
continue;
|
||
|
||
const speciesDexId = parseInt(speciesDexIdMatch[1]);
|
||
|
||
const speciesKey = Species[allSpecies[speciesDexId - 1].speciesId];
|
||
|
||
formKey = species.name.slice(speciesKey.length + 1);
|
||
|
||
const regionKey = regionalForms.find(r => formKey.indexOf(r) > -1);
|
||
|
||
if (regionKey) {
|
||
formKey = formKey.slice(regionKey.length + 1);
|
||
matchingSpecies = allSpecies.find(s => Species[s.speciesId] === `${regionKey.toUpperCase()}_${speciesKey}`);
|
||
} else
|
||
matchingSpecies = allSpecies[speciesDexId - 1];
|
||
}
|
||
|
||
if (!matchingSpecies) {
|
||
console.log('NO MATCH', species.name);
|
||
continue;
|
||
}
|
||
|
||
const speciesKey = Species[matchingSpecies.speciesId];
|
||
|
||
const matchingIndex = moveTmSpecies[moveId].findIndex(s => Array.isArray(s) ? s[0] === speciesKey : s === speciesKey);
|
||
|
||
if (matchingIndex === -1)
|
||
moveTmSpecies[moveId].push(!formKey ? speciesKey : [ speciesKey, formKey ]);
|
||
else {
|
||
if (!Array.isArray(moveTmSpecies[moveId][matchingIndex]))
|
||
moveTmSpecies[moveId][matchingIndex] = [ moveTmSpecies[moveId][matchingIndex] as string, '' ];
|
||
(moveTmSpecies[moveId][matchingIndex] as string[]).push(formKey);
|
||
}
|
||
}
|
||
}
|
||
|
||
let tmSpeciesStr = `export const tmSpecies: TmSpecies = {\n`;
|
||
|
||
for (let moveId of Object.keys(moveTmSpecies)) {
|
||
tmSpeciesStr += ` [Moves.${Moves[parseInt(moveId)]}]: [\n`;
|
||
for (let species of moveTmSpecies[moveId]) {
|
||
if (typeof species === 'string')
|
||
tmSpeciesStr += ` Species.${species},\n`;
|
||
else {
|
||
const matchingExistingSpecies = allSpecies.find(s => Species[s.speciesId] === species[0]);
|
||
const forms = (species as string[]).slice(1);
|
||
if (matchingExistingSpecies && (!pokemonFormLevelMoves.hasOwnProperty(matchingExistingSpecies.speciesId) || matchingExistingSpecies.forms.length <= 1 || (matchingExistingSpecies.forms.length === 2 && matchingExistingSpecies.forms[1].formKey.indexOf(SpeciesFormKey.MEGA) > -1) || matchingExistingSpecies.forms.length === forms.length))
|
||
tmSpeciesStr += ` Species.${species[0]},\n`;
|
||
else {
|
||
tmSpeciesStr += ` [\n Species.${species[0]},\n`;
|
||
for (let form of forms)
|
||
tmSpeciesStr += ` '${form}',\n`;
|
||
tmSpeciesStr += ` ],\n`;
|
||
}
|
||
}
|
||
}
|
||
tmSpeciesStr += ` ],\n`;
|
||
}
|
||
|
||
tmSpeciesStr += `\n};`;
|
||
|
||
console.log(tmSpeciesStr);
|
||
} |