@robsonbittencourt/calc 0.10.0

package/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "@robsonbittencourt/calc",
+  "version": "0.10.0",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "unpkg": "dist/production.min.js",
+  "description": "Fork from Smogon Damage Calc",
+  "repository": "github:robsonbittencourt/damage-calc",
+  "contributors": [
+    "Austin Couturier <austincouturier@icloud.com>"
+  ],
+  "license": "MIT",
+  "dependencies": {
+    "@types/node": "^18.14.2"
+  },
+  "devDependencies": {
+    "@pkmn/dex": "^0.9.19",
+    "@pkmn/eslint-config": "^2.15.0",
+    "@types/jest": "^29.4.0",
+    "@typescript-eslint/eslint-plugin": "^5.54.0",
+    "@typescript-eslint/parser": "^5.54.0",
+    "eslint": "^8.35.0",
+    "eslint-plugin-jest": "^27.2.1",
+    "jest": "^29.4.3",
+    "ts-jest": "^29.0.5",
+    "typescript": "^4.9.5"
+  },
+  "scripts": {
+    "lint": "eslint --cache src --ext ts",
+    "fix": "eslint --fix src --ext ts",
+    "compile": "tsc -p .",
+    "bundle": "node bundle",
+    "build": "npm run compile && npm run bundle",
+    "test": "jest",
+    "coverage": "jest --coverage",
+    "prepare": "npm run build",
+    "pretest": "npm run build",
+    "posttest": "npm run lint"
+  }
+}

package/src/adaptable.ts

@@ -0,0 +1,12 @@
+// By default, importing `@smogon/calc` provides a convenience wrapper that is roughly equivalent
+// to importing `@smogon/calc/adaptable` and `import Generations from '@smogon/calc/data'` and
+// using  `Generations` to populate the `Generation` param to these exports. Alternatively, an
+// application may implement a different `@smogon/calc/data/interface` and pass a `Generation` from
+// that to these exports.
+
+export {calculate} from './calc';
+export {Pokemon} from './pokemon';
+export {Move} from './move';
+export {Field, Side} from './field';
+export {Result} from './result';
+export {Stats} from './stats';

package/src/calc.ts

@@ -0,0 +1,40 @@
+import {Field} from './field';
+import type {Generation} from './data/interface';
+import type {Move} from './move';
+import type {Pokemon} from './pokemon';
+import type {Result} from './result';
+
+import {calculateRBYGSC} from './mechanics/gen12';
+import {calculateADV} from './mechanics/gen3';
+import {calculateDPP} from './mechanics/gen4';
+import {calculateBWXY} from './mechanics/gen56';
+import {calculateSMSSSV} from './mechanics/gen789';
+
+const MECHANICS = [
+  () => {},
+  calculateRBYGSC,
+  calculateRBYGSC,
+  calculateADV,
+  calculateDPP,
+  calculateBWXY,
+  calculateBWXY,
+  calculateSMSSSV,
+  calculateSMSSSV,
+  calculateSMSSSV,
+];
+
+export function calculate(
+  gen: Generation,
+  attacker: Pokemon,
+  defender: Pokemon,
+  move: Move,
+  field?: Field,
+) {
+  return MECHANICS[gen.num](
+    gen,
+    attacker.clone(),
+    defender.clone(),
+    move.clone(),
+    field ? field.clone() : new Field()
+  ) as Result;
+}

package/src/data/abilities.ts

@@ -0,0 +1,383 @@
+import type * as I from './interface';
+import {toID} from '../util';
+
+const RBY: string[] = [];
+
+const GSC: string[] = [];
+
+const ADV = [
+  'Air Lock',
+  'Arena Trap',
+  'Battle Armor',
+  'Blaze',
+  'Chlorophyll',
+  'Clear Body',
+  'Cloud Nine',
+  'Color Change',
+  'Compound Eyes',
+  'Cute Charm',
+  'Drizzle',
+  'Damp',
+  'Drought',
+  'Early Bird',
+  'Effect Spore',
+  'Flame Body',
+  'Flash Fire',
+  'Forecast',
+  'Guts',
+  'Huge Power',
+  'Hustle',
+  'Hyper Cutter',
+  'Illuminate',
+  'Immunity',
+  'Inner Focus',
+  'Insomnia',
+  'Intimidate',
+  'Keen Eye',
+  'Levitate',
+  'Lightning Rod',
+  'Limber',
+  'Liquid Ooze',
+  'Magma Armor',
+  'Magnet Pull',
+  'Marvel Scale',
+  'Minus',
+  'Natural Cure',
+  'Oblivious',
+  'Overgrow',
+  'Own Tempo',
+  'Pickup',
+  'Plus',
+  'Poison Point',
+  'Pressure',
+  'Pure Power',
+  'Rain Dish',
+  'Rock Head',
+  'Rough Skin',
+  'Run Away',
+  'Sand Stream',
+  'Sand Veil',
+  'Serene Grace',
+  'Shadow Tag',
+  'Shed Skin',
+  'Shell Armor',
+  'Shield Dust',
+  'Soundproof',
+  'Speed Boost',
+  'Static',
+  'Stench',
+  'Sticky Hold',
+  'Sturdy',
+  'Suction Cups',
+  'Swarm',
+  'Swift Swim',
+  'Synchronize',
+  'Thick Fat',
+  'Torrent',
+  'Trace',
+  'Truant',
+  'Vital Spirit',
+  'Volt Absorb',
+  'Water Absorb',
+  'Water Veil',
+  'White Smoke',
+  'Wonder Guard',
+];
+
+const DPP = ADV.concat([
+  'Adaptability',
+  'Aftermath',
+  'Anger Point',
+  'Anticipation',
+  'Bad Dreams',
+  'Download',
+  'Dry Skin',
+  'Filter',
+  'Flower Gift',
+  'Forewarn',
+  'Frisk',
+  'Gluttony',
+  'Heatproof',
+  'Honey Gather',
+  'Hydration',
+  'Ice Body',
+  'Iron Fist',
+  'Klutz',
+  'Leaf Guard',
+  'Magic Guard',
+  'Mold Breaker',
+  'Motor Drive',
+  'Mountaineer',
+  'Multitype',
+  'No Guard',
+  'Normalize',
+  'Persistent',
+  'Poison Heal',
+  'Quick Feet',
+  'Rebound',
+  'Reckless',
+  'Rivalry',
+  'Scrappy',
+  'Simple',
+  'Skill Link',
+  'Slow Start',
+  'Sniper',
+  'Snow Cloak',
+  'Snow Warning',
+  'Solar Power',
+  'Solid Rock',
+  'Stall',
+  'Steadfast',
+  'Storm Drain',
+  'Super Luck',
+  'Tangled Feet',
+  'Technician',
+  'Tinted Lens',
+  'Unaware',
+  'Unburden',
+]);
+
+const BW = DPP.concat([
+  'Analytic',
+  'Big Pecks',
+  'Contrary',
+  'Cursed Body',
+  'Defeatist',
+  'Defiant',
+  'Flare Boost',
+  'Friend Guard',
+  'Harvest',
+  'Healer',
+  'Heavy Metal',
+  'Illusion',
+  'Imposter',
+  'Infiltrator',
+  'Iron Barbs',
+  'Light Metal',
+  'Justified',
+  'Magic Bounce',
+  'Moody',
+  'Moxie',
+  'Multiscale',
+  'Mummy',
+  'Overcoat',
+  'Pickpocket',
+  'Poison Touch',
+  'Prankster',
+  'Rattled',
+  'Regenerator',
+  'Sand Force',
+  'Sand Rush',
+  'Sap Sipper',
+  'Sheer Force',
+  'Telepathy',
+  'Teravolt',
+  'Toxic Boost',
+  'Turboblaze',
+  'Unnerve',
+  'Victory Star',
+  'Weak Armor',
+  'Wonder Skin',
+  'Zen Mode',
+]);
+
+const XY = BW.concat([
+  'Aerilate',
+  'Aura Break',
+  'Aroma Veil',
+  'Bulletproof',
+  'Cheek Pouch',
+  'Competitive',
+  'Dark Aura',
+  'Delta Stream',
+  'Desolate Land',
+  'Fairy Aura',
+  'Flower Veil',
+  'Fur Coat',
+  'Gale Wings',
+  'Gooey',
+  'Grass Pelt',
+  'Magician',
+  'Mega Launcher',
+  'Parental Bond',
+  'Pixilate',
+  'Primordial Sea',
+  'Protean',
+  'Refrigerate',
+  'Stance Change',
+  'Strong Jaw',
+  'Sweet Veil',
+  'Symbiosis',
+  'Tough Claws',
+]);
+
+const SM = XY.concat([
+  'Battery',
+  'Battle Bond',
+  'Beast Boost',
+  'Berserk',
+  'Comatose',
+  'Corrosion',
+  'Dancer',
+  'Dazzling',
+  'Disguise',
+  'Electric Surge',
+  'Emergency Exit',
+  'Fluffy',
+  'Full Metal Body',
+  'Galvanize',
+  'Grassy Surge',
+  'Innards Out',
+  'Liquid Voice',
+  'Long Reach',
+  'Merciless',
+  'Misty Surge',
+  'Neuroforce',
+  'Power Construct',
+  'Power of Alchemy',
+  'Prism Armor',
+  'Psychic Surge',
+  'Queenly Majesty',
+  'RKS System',
+  'Receiver',
+  'Schooling',
+  'Shadow Shield',
+  'Shields Down',
+  'Slush Rush',
+  'Stamina',
+  'Stakeout',
+  'Steelworker',
+  'Soul-Heart',
+  'Surge Surfer',
+  'Tangling Hair',
+  'Triage',
+  'Water Bubble',
+  'Water Compaction',
+  'Wimp Out',
+]);
+
+const SS = SM.concat([
+  'As One (Glastrier)',
+  'As One (Spectrier)',
+  'Ball Fetch',
+  'Chilling Neigh',
+  'Cotton Down',
+  'Curious Medicine',
+  'Dauntless Shield',
+  'Dragon\'s Maw',
+  'Gorilla Tactics',
+  'Grim Neigh',
+  'Gulp Missile',
+  'Hunger Switch',
+  'Ice Face',
+  'Ice Scales',
+  'Intrepid Sword',
+  'Libero',
+  'Mimicry',
+  'Mirror Armor',
+  'Neutralizing Gas',
+  'Pastel Veil',
+  'Perish Body',
+  'Power Spot',
+  'Propeller Tail',
+  'Punk Rock',
+  'Quick Draw',
+  'Ripen',
+  'Sand Spit',
+  'Screen Cleaner',
+  'Stalwart',
+  'Steam Engine',
+  'Steely Spirit',
+  'Transistor',
+  'Unseen Fist',
+  'Wandering Spirit',
+]);
+
+const SV = SS.concat([
+  'Anger Shell',
+  'Armor Tail',
+  'Beads of Ruin',
+  'Commander',
+  'Costar',
+  'Cud Chew',
+  'Earth Eater',
+  'Electromorphosis',
+  'Embody Aspect (Cornerstone)',
+  'Embody Aspect (Hearthflame)',
+  'Embody Aspect (Teal)',
+  'Embody Aspect (Wellspring)',
+  'Good as Gold',
+  'Guard Dog',
+  'Hadron Engine',
+  'Hospitality',
+  'Lingering Aroma',
+  'Mind\'s Eye',
+  'Mycelium Might',
+  'Opportunist',
+  'Orichalcum Pulse',
+  'Poison Puppeteer',
+  'Protosynthesis',
+  'Purifying Salt',
+  'Quark Drive',
+  'Rocky Payload',
+  'Seed Sower',
+  'Sharpness',
+  'Supersweet Syrup',
+  'Supreme Overlord',
+  'Sword of Ruin',
+  'Tablets of Ruin',
+  'Tera Shell',
+  'Tera Shift',
+  'Teraform Zero',
+  'Thermal Exchange',
+  'Toxic Chain',
+  'Toxic Debris',
+  'Vessel of Ruin',
+  'Well-Baked Body',
+  'Wind Power',
+  'Wind Rider',
+  'Zero to Hero',
+]);
+
+export const ABILITIES = [[], RBY, GSC, ADV, DPP, BW, XY, SM, SS, SV];
+
+export class Abilities implements I.Abilities {
+  private readonly gen: I.GenerationNum;
+
+  constructor(gen: I.GenerationNum) {
+    this.gen = gen;
+  }
+
+  get(id: I.ID) {
+    return ABILITIES_BY_ID[this.gen][id];
+  }
+
+  *[Symbol.iterator]() {
+    for (const id in ABILITIES_BY_ID[this.gen]) {
+      yield this.get(id as I.ID)!;
+    }
+  }
+}
+
+class Ability implements I.Ability {
+  readonly kind: 'Ability';
+  readonly id: I.ID;
+  readonly name: I.AbilityName;
+
+  constructor(name: string) {
+    this.kind = 'Ability';
+    this.id = toID(name);
+    this.name = name as I.AbilityName;
+  }
+}
+
+const ABILITIES_BY_ID: Array<{[id: string]: Ability}> = [];
+
+for (const abilities of ABILITIES) {
+  const map: {[id: string]: Ability} = {};
+  for (const ability of abilities) {
+    const a = new Ability(ability);
+    map[a.id] = a;
+  }
+  ABILITIES_BY_ID.push(map);
+}

package/src/data/index.ts

@@ -0,0 +1,36 @@
+import type * as I from './interface';
+
+import {Abilities} from './abilities';
+import {Items} from './items';
+import {Moves} from './moves';
+import {Species} from './species';
+import {Types} from './types';
+import {Natures} from './natures';
+
+export const Generations: I.Generations = new (class {
+  get(gen: I.GenerationNum) {
+    return new Generation(gen);
+  }
+})();
+
+class Generation implements I.Generation {
+  num: I.GenerationNum;
+
+  abilities: Abilities;
+  items: Items;
+  moves: Moves;
+  species: Species;
+  types: Types;
+  natures: Natures;
+
+  constructor(num: I.GenerationNum) {
+    this.num = num;
+
+    this.abilities = new Abilities(num);
+    this.items = new Items(num);
+    this.moves = new Moves(num);
+    this.species = new Species(num);
+    this.types = new Types(num);
+    this.natures = new Natures();
+  }
+}

package/src/data/interface.ts

@@ -0,0 +1,176 @@
+export interface As<T> {__brand: T}
+export type ID = (string & As<'ID'>) | (string & { __isID: true }) | '';
+export type GenerationNum = 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9;
+export type GenderName = 'M' | 'F' | 'N';
+export type StatID = 'hp' | StatIDExceptHP;
+export type StatIDExceptHP = 'atk' | 'def' | 'spa' | 'spd' | 'spe';
+export type StatsTable<T = number> = {[stat in StatID]: T};
+
+export type AbilityName = string & As<'AbilityName'>;
+export type ItemName = string & As<'ItemName'>;
+export type MoveName = string & As<'MoveName'>;
+export type SpeciesName = string & As<'SpeciesName'>;
+
+export type StatusName = 'slp' | 'psn' | 'brn' | 'frz' | 'par' | 'tox';
+
+export type GameType = 'Singles' | 'Doubles';
+export type Terrain = 'Electric' | 'Grassy' | 'Psychic' | 'Misty';
+export type Weather =
+  | 'Sand' | 'Sun' | 'Rain' | 'Hail' | 'Snow' | 'Harsh Sunshine' | 'Heavy Rain' | 'Strong Winds';
+
+export type NatureName =
+  'Adamant' | 'Bashful' | 'Bold' | 'Brave' | 'Calm' |
+  'Careful' | 'Docile' | 'Gentle' | 'Hardy' | 'Hasty' |
+  'Impish' | 'Jolly' | 'Lax' | 'Lonely' | 'Mild' |
+  'Modest' | 'Naive' | 'Naughty' | 'Quiet' | 'Quirky' |
+  'Rash' | 'Relaxed' | 'Sassy' | 'Serious' | 'Timid';
+
+export type TypeName =
+  'Normal' | 'Fighting' | 'Flying' | 'Poison' | 'Ground' | 'Rock' | 'Bug' | 'Ghost' | 'Steel' |
+  'Fire' | 'Water' | 'Grass' | 'Electric' | 'Psychic' | 'Ice' | 'Dragon' | 'Dark' | 'Fairy' |
+  'Stellar' | '???';
+
+export type MoveCategory = 'Physical' | 'Special' | 'Status';
+
+export type MoveTarget =
+  'adjacentAlly' | 'adjacentAllyOrSelf' | 'adjacentFoe' | 'all' |
+  'allAdjacent' | 'allAdjacentFoes' | 'allies' | 'allySide' | 'allyTeam' |
+  'any' | 'foeSide' | 'normal' | 'randomNormal' | 'scripted' | 'self';
+
+export interface Generations {
+  get(gen: GenerationNum): Generation;
+}
+
+export interface Generation {
+  readonly num: GenerationNum;
+  readonly abilities: Abilities;
+  readonly items: Items;
+  readonly moves: Moves;
+  readonly species: Species;
+  readonly types: Types;
+  readonly natures: Natures;
+}
+
+export type DataKind = 'Ability' | 'Item' | 'Move' | 'Species' | 'Type' | 'Nature';
+
+export interface Data<NameT> {
+  readonly id: ID;
+  readonly name: NameT;
+  readonly kind: DataKind;
+}
+
+export interface Abilities {
+  get(id: ID): Ability | undefined;
+  [Symbol.iterator](): IterableIterator<Ability>;
+}
+
+export interface Ability extends Data<AbilityName> {
+  readonly kind: 'Ability';
+}
+
+export interface Items {
+  get(id: ID): Item | undefined;
+  [Symbol.iterator](): IterableIterator<Item>;
+}
+
+export interface Item extends Data<ItemName> {
+  readonly kind: 'Item';
+  readonly megaEvolves?: SpeciesName;
+  readonly isBerry?: boolean;
+  readonly naturalGift?: Readonly<{basePower: number; type: TypeName}>;
+}
+
+export interface Moves {
+  get(id: ID): Move | undefined;
+  [Symbol.iterator](): IterableIterator<Move>;
+}
+
+export interface MoveFlags {
+  contact?: 1 | 0;
+  bite?: 1 | 0;
+  sound?: 1 | 0;
+  // TODO: heal?: 1 | 0;
+  punch?: 1 | 0;
+  bullet?: 1 | 0;
+  pulse?: 1 | 0;
+  slicing?: 1 | 0;
+  wind?: 1| 0;
+}
+
+export interface SelfOrSecondaryEffect {
+  boosts?: Partial<StatsTable>;
+}
+
+export interface Move extends Data<MoveName> {
+  readonly kind: 'Move';
+  readonly basePower: number;
+  readonly type: TypeName;
+  readonly category?: MoveCategory;
+  readonly flags: MoveFlags;
+  readonly secondaries?: any;
+  readonly target?: MoveTarget;
+  readonly recoil?: [number, number];
+  readonly hasCrashDamage?: boolean;
+  readonly mindBlownRecoil?: boolean;
+  readonly struggleRecoil?: boolean;
+  readonly willCrit?: boolean;
+  readonly drain?: [number, number];
+  readonly priority?: number;
+  readonly self?: SelfOrSecondaryEffect | null;
+  readonly ignoreDefensive?: boolean;
+  readonly overrideOffensiveStat?: StatIDExceptHP;
+  readonly overrideDefensiveStat?: StatIDExceptHP;
+  readonly overrideOffensivePokemon?: 'target' | 'source';
+  readonly overrideDefensivePokemon?: 'target' | 'source';
+  readonly breaksProtect?: boolean;
+  readonly isZ?: boolean | string;
+  readonly zMove?: {
+    basePower?: number;
+  };
+  readonly isMax?: boolean | string;
+  readonly maxMove?: {
+    basePower: number;
+  };
+  readonly multihit?: number | number[];
+  readonly multiaccuracy?: boolean;
+}
+
+export interface Species {
+  get(id: ID): Specie | undefined;
+  [Symbol.iterator](): IterableIterator<Specie>;
+}
+
+export interface Specie extends Data<SpeciesName> {
+  readonly kind: 'Species';
+  readonly types: [TypeName] | [TypeName, TypeName];
+  readonly baseStats: Readonly<StatsTable>;
+  readonly weightkg: number;
+  readonly nfe?: boolean;
+  readonly gender?: GenderName;
+  readonly otherFormes?: SpeciesName[];
+  readonly baseSpecies?: SpeciesName;
+  readonly abilities?: {0: AbilityName | ''};
+}
+
+export interface Types {
+  get(id: ID): Type | undefined;
+  [Symbol.iterator](): IterableIterator<Type>;
+}
+
+export type TypeEffectiveness = 0 | 0.5 | 1 | 2;
+
+export interface Type extends Data<TypeName> {
+  readonly kind: 'Type';
+  readonly effectiveness: Readonly<{[type in TypeName]?: TypeEffectiveness}>;
+}
+
+export interface Natures {
+  get(id: ID): Nature | undefined;
+  [Symbol.iterator](): IterableIterator<Nature>;
+}
+
+export interface Nature extends Data<NatureName> {
+  readonly kind: 'Nature';
+  readonly plus?: StatID;
+  readonly minus?: StatID;
+}