#!/usr/bin/env python3
import argparse
import json
import pathlib
import subprocess

from consts.species import PokemonSpecies


argparser = argparse.ArgumentParser(
    prog='make_shinzukan_py',
    description='Packs the archive containing SinnohDex->NatDex number mapping'
)
argparser.add_argument('-k', '--knarc',
                       required=True,
                       help='Path to knarc executable')
argparser.add_argument('-s', '--source-dir',
                       required=True,
                       help='Path to the source directory (res/pokemon)')
argparser.add_argument('-p', '--private-dir',
                       required=True,
                       help='Path to the private directory (where binaries will be made)')
argparser.add_argument('-o', '--output-dir',
                       required=True,
                       help='Path to the output directory (where the NARC will be made)')
argparser.add_argument('pokedex',
                       help='List of pokemon in the Sinnoh Pokedex')
args = argparser.parse_args()

source_dir = pathlib.Path(args.source_dir)
private_dir = pathlib.Path(args.private_dir)
output_dir = pathlib.Path(args.output_dir)

private_dir.mkdir(parents=True, exist_ok=True)

NUM_SINNOH = 211

pokedex = [0 for i in range(NUM_SINNOH)]

with open(args.pokedex) as data_file:
    dex_data = json.load(data_file)
    for i, mon in enumerate(dex_data):
        pokedex[i] = PokemonSpecies[mon].value

target_fname = private_dir / f'shinzukan_0.bin'
with open(target_fname, 'wb+') as target_file:
    shinzukan = [0 for i in range(NUM_SINNOH * 2)]
    for i in range(NUM_SINNOH):
        shinzukan[i*2] = pokedex[i] & 0xff
        shinzukan[i*2 + 1] = (pokedex[i] >> 8) & 0xff
    target_file.write(bytes(shinzukan))

subprocess.run([args.knarc, '-d', private_dir, '-p', output_dir / 'shinzukan.narc'])