import re
import shutil
from pathlib import Path


__all__ = ["copy", "delete"]


_new_worlds: dict[str, str] = {}


def copy(src: str, dst: str) -> None:
    from Utils import get_file_safe_name
    from worlds.AutoWorld import AutoWorldRegister

    assert dst not in _new_worlds, "World already created"
    if '"' in dst or "\\" in dst:  # easier to reject than to escape
        raise ValueError(f"Unsupported symbols in {dst}")
    dst_folder_name = get_file_safe_name(dst.lower())
    src_cls = AutoWorldRegister.world_types[src]
    src_folder = Path(src_cls.__file__).parent
    worlds_folder = src_folder.parent
    if (not src_cls.__file__.endswith(("__init__.py", "world.py")) or not src_folder.is_dir()
            or not (worlds_folder / "generic").is_dir()):
        raise ValueError(f"Unsupported layout for copy_world from {src}")
    dst_folder = worlds_folder / dst_folder_name
    if dst_folder.is_dir():
        raise ValueError(f"Destination {dst_folder} already exists")
    shutil.copytree(src_folder, dst_folder)
    _new_worlds[dst] = str(dst_folder)

    for potential_world_class_file in ("__init__.py", "world.py"):
        with open(dst_folder / potential_world_class_file, "r", encoding="utf-8-sig") as f:
            contents = f.read()
        r_src = re.escape(src)
        contents = re.sub(r'game\s*(:\s*[a-zA-Z\[\]]+)?\s*=\s*[\'"]' + r_src + r'[\'"]', f'game = "{dst}"', contents)
        with open(dst_folder / "__init__.py", "w", encoding="utf-8") as f:
            f.write(contents)


def delete(name: str) -> None:
    assert name in _new_worlds, "World not created by this script"
    shutil.rmtree(_new_worlds[name])
    del _new_worlds[name]