#!/usr/bin/env python3
import itertools
import json
import pathlib
import sys

from consts import species


def as_species(s: str) -> bytes:
    return species.PokemonSpecies[s].value.to_bytes(4, 'little')

def convert_land(encs: list) -> bytes:
    return b''.join(itertools.chain.from_iterable([
        (
            int(encs[i]['level']).to_bytes(4, 'little'),
            as_species(encs[i]['species']),
        )
        for i in range(12)
    ]))

def convert_water(encs: list) -> bytes:
    return b''.join(itertools.chain.from_iterable([
        (
            int(encs[i]['level_max']).to_bytes(1, 'little'),
            int(encs[i]['level_min']).to_bytes(1, 'little'),
            (0).to_bytes(2, 'little'),
            as_species(encs[i]['species']),
        )
        for i in range(5)
    ]))


input_path = pathlib.Path(sys.argv[1])
output_path = pathlib.Path(sys.argv[2])

data = {}
with open(input_path, 'r', encoding='utf-8') as input_file:
    data = json.load(input_file)

packables = bytearray([])
packables.extend(int(data['land_rate']).to_bytes(4, 'little'))
packables.extend(convert_land(data['land_encounters']))

for enc_type, i in itertools.product(['swarms', 'morning', 'night'], range(2)):
    packables.extend(as_species(data[enc_type][i]))

for i in range(4):
    packables.extend(as_species(data['radar'][i]))

for key in ['rate_form0', 'rate_form1', 'rate_form2', 'rate_form3', 'rate_form4', 'unown_table']:
    packables.extend(int(data[key]).to_bytes(4, 'little'))

for version, i in itertools.product(['ruby', 'sapphire', 'emerald', 'firered', 'leafgreen'], range(2)):
    packables.extend(as_species(data[version][i]))

packables.extend(data['surf_rate'].to_bytes(4, 'little'))
packables.extend(convert_water(data['surf_encounters']))
packables.extend((0).to_bytes(44, 'little'))

for rod in ['old', 'good', 'super']:
    packables.extend(data[f'{rod}_rod_rate'].to_bytes(4, 'little'))
    packables.extend(convert_water(data[f'{rod}_rod_encounters']))

with open(output_path, 'wb') as output_file:
    output_file.write(packables)