File size: 3,425 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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
"use strict";
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
  for (var name in all)
    __defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
  if (from && typeof from === "object" || typeof from === "function") {
    for (let key of __getOwnPropNames(from))
      if (!__hasOwnProp.call(to, key) && key !== except)
        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
  }
  return to;
};
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
var cg_teams_leveling_exports = {};
__export(cg_teams_leveling_exports, {
  addPokemon: () => addPokemon,
  dbSetupPromise: () => dbSetupPromise,
  handlers: () => handlers,
  incrementLosses: () => incrementLosses,
  incrementWins: () => incrementWins
});
module.exports = __toCommonJS(cg_teams_leveling_exports);
var import_lib = require("../../lib");
var import_winrates = require("./randombattles/winrates");
let addPokemon = null;
let incrementWins = null;
let incrementLosses = null;
let dbSetupPromise = null;
async function setupDatabase(database) {
  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 = (0, import_lib.SQL)(module, {
    file: "./databases/battlestats.db"
  });
  dbSetupPromise = setupDatabase(database);
}
function getLevelSpeciesID(set, format) {
  if (["Basculin", "Greninja"].includes(set.name))
    return toID(set.species);
  return toID((0, import_winrates.getSpeciesName)(set, format || Dex.formats.get("gen9computergeneratedteams")));
}
async function updateStats(battle, winner) {
  if (!incrementWins || !incrementLosses)
    await dbSetupPromise;
  if (toID(battle.format) !== "gen9computergeneratedteams")
    return;
  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 < 1e3) {
    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]);
    }
  }
}
const handlers = {
  onBattleEnd(battle, winner) {
    if (!Config.usesqlite || !Config.usesqliteleveling)
      return;
    void updateStats(battle, winner);
  }
};
//# sourceMappingURL=cg-teams-leveling.js.map