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',
  // Champions exclusive
  'Dragonize',
  'Eelevate',
  'Fire Mane',
  'Mega Sol',
  'Piercing Drill',
  'Spicy Spray',
]);

const CHAMPIONS = [
  'Adaptability',
  'Aerilate',
  'Aftermath',
  'Analytic',
  'Anger Point',
  'Anticipation',
  'Armor Tail',
  'Aroma Veil',
  'Battle Armor',
  'Berserk',
  'Big Pecks',
  'Blaze',
  'Bulletproof',
  'Cheek Pouch',
  'Chlorophyll',
  'Clear Body',
  'Cloud Nine',
  'Competitive',
  'Compound Eyes',
  'Contrary',
  'Corrosion',
  'Cud Chew',
  'Curious Medicine',
  'Cursed Body',
  'Cute Charm',
  'Damp',
  'Defiant',
  'Disguise',
  'Dragonize',
  'Drizzle',
  'Drought',
  'Dry Skin',
  'Early Bird',
  'Earth Eater',
  'Eelevate',
  'Effect Spore',
  'Electric Surge',
  'Electromorphosis',
  'Fairy Aura',
  'Filter',
  'Fire Mane',
  'Flame Body',
  'Flash Fire',
  'Flower Veil',
  'Fluffy',
  'Forecast',
  'Forewarn',
  'Friend Guard',
  'Frisk',
  'Fur Coat',
  'Gale Wings',
  'Gluttony',
  'Good as Gold',
  'Gooey',
  'Guts',
  'Harvest',
  'Healer',
  'Heatproof',
  'Heavy Metal',
  'Hospitality',
  'Huge Power',
  'Hunger Switch',
  'Hustle',
  'Hydration',
  'Hyper Cutter',
  'Ice Body',
  'Illuminate',
  'Illusion',
  'Immunity',
  'Imposter',
  'Infiltrator',
  'Innards Out',
  'Inner Focus',
  'Insomnia',
  'Intimidate',
  'Iron Fist',
  'Justified',
  'Keen Eye',
  'Klutz',
  'Leaf Guard',
  'Levitate',
  'Light Metal',
  'Lightning Rod',
  'Ligtning Rod',
  'Limber',
  'Liquid Voice',
  'Long Reach',
  'Magic Bounce',
  'Magic Guard',
  'Magician',
  'Magma Armor',
  'Marvel Scale',
  'Mega Launcher',
  'Mega Sol',
  'Merciless',
  'Mimicry',
  'Minus',
  'Mirror Armor',
  'Mold Breaker',
  'Moody',
  'Motor Drive',
  'Moxie',
  'Multiscale',
  'Mummy',
  'Natural Cure',
  'No Guard',
  'Oblivious',
  'Opportunist',
  'Overcoat',
  'Overgrow',
  'Own Tempo',
  'Parental Bond',
  'Pickpocket',
  'Pickup',
  'Piercing Drill',
  'Pixilate',
  'Plus',
  'Poison Heal',
  'Poison Point',
  'Poison Touch',
  'Prankster',
  'Pressure',
  'Protean',
  'Pure Power',
  'Purifying Salt',
  'Queenly Majesty',
  'Quick Draw',
  'Quick Feet',
  'Rain Dish',
  'Receiver',
  'Reckless',
  'Refrigerate',
  'Regenerator',
  'Ripen',
  'Rivalry',
  'Rock Head',
  'Rough Skin',
  'Sand Force',
  'Sand Rush',
  'Sand Spit',
  'Sand Stream',
  'Sand Veil',
  'Sap Sipper',
  'Scrappy',
  'Screen Cleaner',
  'Shadow Tag',
  'Sharpness',
  'Shed Skin',
  'Sheer Force',
  'Shell Armor',
  'Shield Dust',
  'Skill Link',
  'Slush Rush',
  'Sniper',
  'Snow Cloak',
  'Snow Warning',
  'Solar Power',
  'Solid Rock',
  'Soundproof',
  'Speed Boost',
  'Spicy Spray',
  'Stall',
  'Stalwart',
  'Stamina',
  'Stance Change',
  'Static',
  'Steadfast',
  'Stench',
  'Sticky Hold',
  'Strong Jaw',
  'Sturdy',
  'Suction Cups',
  'Super Luck',
  'Supersweet Syrup',
  'Supreme Overlord',
  'Surge Surfer',
  'Swarm',
  'Sweet Veil',
  'Swift Swim',
  'Symbiosis',
  'Synchronize',
  'Tangled Feet',
  'Technician',
  'Telepathy',
  'Thick Fat',
  'Torrent',
  'Tough Claws',
  'Toxic Debris',
  'Trace',
  'Unaware',
  'Unburden',
  'Unnerve',
  'Unseen Fist',
  'Vital Spirit',
  'Volt Absorb',
  'Wandering Spirit',
  'Water Absorb',
  'Water Bubble',
  'Weak Armor',
  'White Smoke',
  'Zero to Hero',
];

export const ABILITIES = [CHAMPIONS, 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);
}