import json
import os
import shutil
from encryption import encrypt_file

remap = json.load(open(file="./Data_exported/Data_mods/x64/datatable/dec/remap.json", encoding="utf-8"))
os.makedirs("./Data_exported/Data_mods/x64/fumen", exist_ok=True)
os.makedirs("./Data_exported/Data_mods/x64/sound", exist_ok=True)


for entry in remap:
    if not os.path.exists("./Assets/Taiko Omnimix v8/Data/x64/fumen/" + entry["id"]) or not os.path.exists(
        "./Assets/Taiko Omnimix v8/Data/x64/sound/song_" + entry["id"] + ".nus3bank"
    ):
        print(entry["id"], "is missing!")
        exit("Make sure you exported the omnimix properly in the temp folder !")
    else:
        # region Encrypting databases
        indir = "./Assets/Taiko Omnimix v8/Data/x64/fumen/" + entry["id"]
        outdir = "./Data_exported/Data_mods/x64/fumen/" + entry["id"]
        os.system('py ./encryption.py -i "' + indir + '" -o "' + outdir + '" --enc --fumen')
        # shutil.copytree(
        #     "./Assets/Taiko Omnimix v8/Data/x64/fumen/" + entry["id"],
        #     "./Data_exported/Data_mods/x64/fumen/" + entry["id"],
        #     dirs_exist_ok=True,
        # )
        shutil.copy2(
            "./Assets/Taiko Omnimix v8/Data/x64/sound/song_" + entry["id"] + ".nus3bank",
            "./Data_exported/Data_mods/x64/sound/",
        )
        print("Copied files for ", entry["id"])