import subprocess
import shutil
import os
import sys
import argparse
from concurrent.futures import ThreadPoolExecutor

output_folder_path = "../pack"
IGNORE_FOLDERS = {
	"zz_ignore_old"
}


def pack_folder(folder_path):
    folder_name = os.path.basename(folder_path)
    if not os.path.exists(folder_name):
        print(f"Error: Folder \"{folder_name}\" doesn't exist")
        return

    try:
        result = subprocess.run(["PackMakerLite.exe", "--nolog", "--parallel", "-p", folder_name], check=True)
        # print(f"Packing completed for: {folder_name}")
    except subprocess.CalledProcessError as e:
        print(f"Error occurred while packing {folder_name}: {e}")
        return

    eix_file = f"{folder_name}.eix"
    epk_file = f"{folder_name}.epk"

    if not os.path.exists(eix_file) or not os.path.exists(epk_file):
        print(f"Packing failed: {eix_file} or {epk_file} not found.")
        return

    try:
        shutil.move(eix_file, os.path.join(output_folder_path, eix_file))
        shutil.move(epk_file, os.path.join(output_folder_path, epk_file))
        print(f"Moved {eix_file} and {epk_file} to {output_folder_path}")
    except Exception as e:
        print(f"Error occurred while moving files: {e}")

def pack_all_folders():
    all_folders = [f for f in os.listdir() if os.path.isdir(f) and f not in IGNORE_FOLDERS]

    with ThreadPoolExecutor() as executor:
        executor.map(pack_folder, all_folders)

def main():
    parser = argparse.ArgumentParser(description="Pack folders for the game.")
    parser.add_argument("folder_name", nargs="?", help="The name of the folder to pack")
    parser.add_argument("--all", action="store_true", help="Pack all folders")

    args = parser.parse_args()

    if not os.path.exists(output_folder_path):
        os.makedirs(output_folder_path)

    if args.all:
        pack_all_folders()
    elif args.folder_name:
        folder_path = os.path.abspath(args.folder_name)
        pack_folder(folder_path)
    else:
        print("Please provide a folder name or use the --all option to pack all folders.")

if __name__ == "__main__":
    main()