push site dev stuff for travel

generate.py

@@ -0,0 +1,251 @@
+import json
+import re
+
+SPLIT_CHARS = re.compile(r"[\W_-]+")
+
+def toid(name: str) -> str:
+    return ''.join(SPLIT_CHARS.split(name)).lower()
+
+pokedex: dict = json.load(open('site/data/species.json', 'r', encoding='utf-8'))
+moves: dict = json.load(open('site/data/vanilla_moves.json', 'r', encoding='utf-8'))
+items: dict = json.load(open('site/data/vanilla_items.json', 'r', encoding='utf-8'))
+abilities: dict = json.load(open('site/data/abilities.json', 'r', encoding='utf-8'))
+typechart: dict = json.load(open('site/data/typechart.json', 'r', encoding='utf-8'))
+encounters: dict = json.load(open('site/data/encounters.json', 'r', encoding='utf-8'))
+
+index = []
+index.extend(list(map(lambda s: s + ' pokemon', pokedex.keys())))
+index.extend(list(map(lambda s: s + ' move', moves.keys())))
+index.extend(list(map(lambda s: s + ' item', items.keys())))
+index.extend(list(map(lambda s: s + ' ability', abilities.keys())))
+index.extend(list(map(lambda s: s + ' type', typechart.keys())))
+index.extend(list(map(lambda s: s + ' category', ['physical', 'special', 'status'])))
+index.extend(list(map(lambda s: s + ' egggroup', ['monster', 'water1', 'bug', 'flying', 'field', 'fairy', 'grass', 'humanlike', 'water3', 'mineral', 'amorphous', 'water2', 'ditto', 'dragon', 'undiscovered'])))
+
+for key in encounters.keys():
+    if key == 'rates':
+        continue
+    index.append(key + ' location')
+
+COMMON_ABBREVIATION_INDICES: dict[str, int] = {
+    'Alakazam': 5,
+    'Arctovish': 5,
+    'Arctozolt': 5,
+    'Articuno': 5,
+    'Breloom': 3,
+    'Bronzong': 4,
+    'Celebi': 4,
+    'Charizard': 5,
+    'Donphan': 3,
+    'Dracovish': 5,
+    'Dracozolt': 5,
+    'Dragapult': 5,
+    'Dusclops': 3,
+    'Electabuzz': 6,
+    'Exeggutor': 2,
+    'Garchomp': 3,
+    'Hariyama': 4,
+    'Magearna': 2,
+    'Magnezone': 5,
+    'Mamoswine': 4,
+    'Moltres': 3,
+    'Nidoking': 4,
+    'Nidoqueen': 4,
+    'Nidorina': 4,
+    'Nidorino': 4,
+    'Regice': 3,
+    'Regidrago': 4,
+    'Regieleki': 4,
+    'Regigigas': 4,
+    'Regirock': 4,
+    'Registeel': 4,
+    'Slowbro': 4,
+    'Slowking': 4,
+    'Starmie': 4,
+    'Tyranitar': 6,
+    'Zapdos': 3,
+
+    'Acupressure': 3,
+    'Aromatherapy': 5,
+    'Boomburst': 4,
+    'Crabhammer': 4,
+    'Discharge': 3,
+    'Earthquake': 5,
+    'Extrasensory': 5,
+    'Flamethrower': 5,
+    'Headbutt': 4,
+    'Moonblast': 4,
+    'Moonlight': 4,
+    'Overheat': 4,
+    'Outrage': 3,
+    'Octazooka': 4,
+    'Payback': 3,
+    'Psyshock': 3,
+    'Psywave': 3,
+    'Rototiller': 4,
+    'Rollout': 4,
+    'Safeguard': 4,
+    'Sandstorm': 4,
+    'Smokescreen': 5,
+    'Stockpile': 5,
+    'Steamroller': 5,
+    'Superpower': 5,
+    'Supersonic': 5,
+    'Synchronoise': 7,
+    'Tailwind': 4,
+    'Telekinesis': 4,
+    'Teleport': 4,
+    'Thunderbolt': 7,
+    'Twineedle': 3,
+    'Uproar': 2,
+    'Venoshock': 4,
+    'Whirlpool': 5,
+    'Whirlwind': 5,
+}
+
+def alias(id: str, name: str, T: str) -> list[str]:
+    index = []
+
+    i = name.rfind(' ')
+    if i < 0:
+        i = name.rfind('-')
+
+    if name.endswith('-Mega-X') or name.endswith('-Mega-Y'):
+        index.append(f'mega{toid(name[:-7] + name[-1])} {T} {id} 0')
+        index.append(f'm{toid(name[:-7] + name[-1])} {T} {id} 0')
+        index.append(f'mega{toid(name[-1])} {T} {id} {len(toid(name[:-7]))}')
+        return index
+
+    if name.endswith('-Mega'):
+        index.append(f'mega{toid(name[:-5])} {T} {id} 0')
+        index.append(f'm{toid(name[:-5])} {T} {id} 0')
+        return index
+
+    if name.endswith('-Alola'):
+        index.append(f'alolan{toid(name[0:-6])} {T} {id} 0')
+        return index
+
+    if name.endswith('-Galar'):
+        index.append(f'galarian{toid(name[0:-6])} {T} {id} 0')
+        return index
+
+    if name.endswith('-Hisui'):
+        index.append(f'hisuian{toid(name[0:-6])} {T} {id} 0')
+        return index
+
+    if name.endswith('-Paldea'):
+        index.append(f'paldean{toid(name[0:-7])} {T} {id} 0')
+        return index
+
+    old_i = i
+    i = COMMON_ABBREVIATION_INDICES.get(name, i)
+    if i < 0:
+        return index
+
+    abbrev = toid(name[0] + name[i:]) if old_i < 0 else ''
+    index.append(f'{toid(name[i:])} {T} {id} {len(toid(name[:i]))}')
+    if name.startswith('Hidden Power '):
+        abbrev = f'hp{toid(name[13:])}'
+        index.append(f'{abbrev} {T} {id} {0}')
+    elif name == 'Hidden Power':
+        index.append(f'hp {T} {id} {0}')
+    elif ' ' in name:
+        abbrev = toid("".join(map(lambda s: s[1] if s[0] != 0 else s[1][0], enumerate(name.split(" ")))))
+    elif '-' in name and name[0] != '-':
+        abbrev = toid("".join(map(lambda s: s[1] if s[0] != 0 else s[1][0], enumerate(name.split("-")))))
+
+    if abbrev:
+        index.append(f'{abbrev} {T} {id} 0')
+
+    if name == 'High Jump Kick':
+        index.append(f'hjkick {T} {id} 0')
+    elif name == 'Wake-Up Slap':
+        index.append(f'wuslap {T} {id} 0')
+        index.append(f'wupslap {T} {id} 0')
+    elif name == 'Zen Headbutt':
+        index.append(f'zhbutt {T} {id} 0')
+    elif name == 'High Horsepower':
+        index.append(f'hhp {T} {id} 0')
+    elif name == 'Articuno':
+        index.append(f'cuno {T} {id} 0')
+
+    j = name.rfind(' ', i - 1)
+    if j < 0:
+        j = name.rfind('-', i - 1)
+    if name == 'Zen Headbutt':
+        j = 8
+    if j >= 0:
+        index.append(f'{toid(name[j:])} {T} {id} {len(toid(name[:j]))}')
+
+    return index
+
+[index.extend(alias(id, mon['name'], 'pokemon')) for id, mon in pokedex.items()]
+[index.extend(alias(id, move['name'], 'move')) for id, move in moves.items()]
+[index.extend(alias(id, item['name'], 'item')) for id, item in items.items()]
+[index.extend(alias(id, ability['name'], 'ability')) for id, ability in abilities.items()]
+
+index.sort()
+
+# manually rearrange some entries
+index[index.index('grass type')] = 'grass egggroup'
+index[index.index('grass egggroup')] = 'grass type'
+index[index.index('fairy type')] = 'fairy egggroup'
+index[index.index('fairy egggroup')] = 'fairy type'
+index[index.index('flying type')] = 'flying egggroup'
+index[index.index('flying egggroup')] = 'flying type'
+index[index.index('dragon type')] = 'dragon egggroup'
+index[index.index('dragon egggroup')] = 'dragon type'
+index[index.index('bug type')] = 'bug egggroup'
+index[index.index('bug egggroup')] = 'bug type'
+index[index.index('psychic type')] = 'psychic egggroup'
+index[index.index('psychic egggroup')] = 'psychic type'
+index[index.index('ditto pokemon')] = 'ditto egggroup'
+index[index.index('ditto egggroup')] = 'ditto pokemon'
+
+def index_map(s: str) -> list:
+    spl: list = list(s.split(' '))
+    if len(spl) > 3:
+        spl[3] = int(spl[3])
+        spl[2] = index.index(f'{spl[2]} {spl[1]}')
+    return spl
+
+def offset_map(entry: list):
+    id = entry[0]
+    name = ''
+    match entry[1]:
+        case 'pokemon':
+            name = pokedex[id]['name']
+        case 'move':
+            name = moves[id]['name']
+        case 'item':
+            name = items[id]['name']
+        case 'ability':
+            name = abilities[id]['name']
+        case 'location':
+            name = encounters[id]['name']
+
+    result = ''
+    i, j, non_alnum = 0, 0, 0
+    while i < len(id):
+        while not name[j].isalnum():
+            j = j + 1
+            non_alnum = non_alnum + 1
+        result = result + str(non_alnum)
+
+    if non_alnum:
+        return result
+    return ''
+
+battle_search_index = list(map(index_map, index))
+battle_offset_index = list(map(offset_map, battle_search_index))
+battle_count_index = {}
+for T in typechart.keys():
+    battle_count_index[f'{T} move'] = sum(map(lambda m: int(m['type'].lower() == T), moves.values()))
+    battle_count_index[f'{T} pokemon'] = sum(map(lambda s: int(T.upper() in s['types']), pokedex.values()))
+
+with open('site/data/search-index.js', 'w', encoding='utf-8') as outf:
+    print('// DO NOT EDIT - automatically built by porydex', file=outf, end='\n\n')
+    print(f'exports.BattleSearchIndex = {json.dumps(battle_search_index)}', file=outf, end='\n\n')
+    print(f'exports.BattleSearchIndexOffset = {json.dumps(battle_offset_index)}', file=outf, end='\n\n')
+    print(f'exports.BattleSearchCountIndex = {json.dumps(battle_count_index)}', file=outf, end='\n\n')
+

porydex.py

@@ -26,6 +26,7 @@ def config_show(_):
     print(f'compiler command:  {str(porydex.config.compiler)}')
     print(f'path to expansion: {str(porydex.config.expansion)}')
     print(f'output directory:  {str(porydex.config.output)}')
+    print(f'output format:     {str(porydex.config.format)}')
 
 def config_set(args):
     if args.expansion:
@@ -94,21 +95,21 @@ def extract(args):
 
         with open(porydex.config.output / 'encounters.json', 'w', encoding='utf-8') as outf:
             json.dump(encounters, outf, indent=4)
-    else: # jsobj
+    else: # showdown
         with open(porydex.config.output / 'moves.js', 'w+', encoding='utf-8') as outf:
-            outf.write('BattleMovedex = ')
+            outf.write('exports.BattleMovedex = ')
             json.dump(moves, outf, indent=4)
 
         with open(porydex.config.output / 'species.js', 'w+', encoding='utf-8') as outf:
-            outf.write('BattlePokedex = ')
+            outf.write('exports.BattlePokedex = ')
             json.dump(species, outf, indent=4)
 
         with open(porydex.config.output / 'learnsets.js', 'w+', encoding='utf-8') as outf:
-            outf.write('BattleLearnsets = ')
+            outf.write('exports.BattleLearnsets = ')
             json.dump(learnsets, outf, indent=4)
 
         with open(porydex.config.output / 'encounters.js', 'w+', encoding='utf-8') as outf:
-            outf.write('Encounters = ')
+            outf.write('exports.BattleLocationdex = ')
             json.dump(encounters, outf, indent=4)
 
 def main():

porydex/config.py

@@ -5,7 +5,7 @@
 
 class OutputFormat(enum.Enum):
     json = enum.auto()
-    jsobj = enum.auto()
+    showdown = enum.auto()
 
     def __str__(self) -> str:
         return self.name.lower()

site/config/config-example.js

@@ -0,0 +1,5 @@
+exports.routes = {
+    root: 'pokemonshowdown.com',
+    client: 'play.pokemonshowdown.com',
+    replays: 'replay.pokemonshowdown.com',
+};