import unreal


SOURCE_CHARACTER = "/Game/ThirdPerson/Blueprints/BP_ThirdPersonCharacter"
SOURCE_CONTROLLER = "/Game/ThirdPerson/Blueprints/BP_ThirdPersonPlayerController"
SOURCE_GAME_MODE = "/Game/ThirdPerson/Blueprints/BP_ThirdPersonGameMode"

DEST_ROOT = "/Game/Agrarian/Blueprints/Characters"
DEST_CHARACTER = f"{DEST_ROOT}/BP_AgrarianPlayerCharacter"
DEST_CONTROLLER = f"{DEST_ROOT}/BP_AgrarianPlayerController"
DEST_GAME_MODE = f"{DEST_ROOT}/BP_AgrarianGameMode"


def ensure_directory(path):
    if not unreal.EditorAssetLibrary.does_directory_exist(path):
        if not unreal.EditorAssetLibrary.make_directory(path):
            raise RuntimeError(f"Could not create content directory: {path}")


def duplicate_if_missing(source_path, destination_path):
    if unreal.EditorAssetLibrary.does_asset_exist(destination_path):
        unreal.log(f"Keeping existing asset: {destination_path}")
        return unreal.EditorAssetLibrary.load_asset(destination_path)

    if not unreal.EditorAssetLibrary.does_asset_exist(source_path):
        raise RuntimeError(f"Source asset missing: {source_path}")

    duplicated = unreal.EditorAssetLibrary.duplicate_asset(source_path, destination_path)
    if not duplicated:
        raise RuntimeError(f"Could not duplicate {source_path} to {destination_path}")

    unreal.log(f"Created {destination_path} from {source_path}")
    return duplicated


def load_blueprint_class(path):
    generated_class = unreal.EditorAssetLibrary.load_blueprint_class(path)
    if not generated_class:
        raise RuntimeError(f"Could not load generated Blueprint class: {path}")
    return generated_class


def main():
    ensure_directory(DEST_ROOT)

    character_bp = duplicate_if_missing(SOURCE_CHARACTER, DEST_CHARACTER)
    controller_bp = duplicate_if_missing(SOURCE_CONTROLLER, DEST_CONTROLLER)
    game_mode_bp = duplicate_if_missing(SOURCE_GAME_MODE, DEST_GAME_MODE)

    character_class = load_blueprint_class(DEST_CHARACTER)
    controller_class = load_blueprint_class(DEST_CONTROLLER)
    game_mode_cdo = unreal.get_default_object(game_mode_bp.generated_class())

    game_mode_cdo.set_editor_property("default_pawn_class", character_class)
    game_mode_cdo.set_editor_property("player_controller_class", controller_class)

    unreal.EditorAssetLibrary.save_loaded_asset(character_bp)
    unreal.EditorAssetLibrary.save_loaded_asset(controller_bp)
    unreal.EditorAssetLibrary.save_loaded_asset(game_mode_bp)

    unreal.log(
        "Agrarian player Blueprint assets are ready: "
        f"{DEST_CHARACTER}, {DEST_CONTROLLER}, {DEST_GAME_MODE}"
    )


main()