Spaces:
Running
Running
File size: 2,532 Bytes
5c2ed06 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
/**
* Not a chat plugin.
*
* Handles updating the level database for [Gen 9] Computer-Generated Teams.
*/
import { SQL } from "../../lib";
import { getSpeciesName } from "./randombattles/winrates";
export let addPokemon: SQL.Statement | null = null;
export let incrementWins: SQL.Statement | null = null;
export let incrementLosses: SQL.Statement | null = null;
export let dbSetupPromise: Promise<void> | null = null;
async function setupDatabase(database: SQL.DatabaseManager) {
await database.runFile('./databases/schemas/battlestats.sql');
addPokemon = await database.prepare(
'INSERT OR IGNORE INTO gen9computergeneratedteams (species_id, wins, losses, level) VALUES (?, 0, 0, ?)'
);
incrementWins = await database.prepare(
'UPDATE gen9computergeneratedteams SET wins = wins + 1 WHERE species_id = ?'
);
incrementLosses = await database.prepare(
'UPDATE gen9computergeneratedteams SET losses = losses + 1 WHERE species_id = ?'
);
}
if (Config.usesqlite && Config.usesqliteleveling) {
const database = SQL(module, {
file: './databases/battlestats.db',
});
dbSetupPromise = setupDatabase(database);
}
function getLevelSpeciesID(set: PokemonSet, format?: Format) {
if (['Basculin', 'Greninja'].includes(set.name)) return toID(set.species);
return toID(getSpeciesName(set, format || Dex.formats.get('gen9computergeneratedteams')));
}
async function updateStats(battle: RoomBattle, winner: ID) {
if (!incrementWins || !incrementLosses) await dbSetupPromise;
if (toID(battle.format) !== 'gen9computergeneratedteams') return;
// if the game is rated or part of a tournament hosted by a public room, it counts
if (battle.rated <= 1 && battle.room.parent?.game) {
let parent = battle.room.parent;
if (parent.game!.gameid === 'bestof' && parent.parent?.game) parent = parent.parent;
if (parent.game!.gameid !== 'tournament' || parent.settings.isPrivate) return;
} else if (battle.rated < 1000) {
return;
}
for (const player of battle.players) {
const team = await battle.getPlayerTeam(player);
if (!team) return;
const increment = (player.id === winner ? incrementWins : incrementLosses);
for (const set of team) {
const statsSpecies = getLevelSpeciesID(set, Dex.formats.get(battle.format));
await addPokemon?.run([statsSpecies, set.level || 100]);
await increment?.run([statsSpecies]);
}
}
}
export const handlers: Chat.Handlers = {
onBattleEnd(battle, winner) {
if (!Config.usesqlite || !Config.usesqliteleveling) return;
void updateStats(battle, winner);
},
};
|