test_level.py

import argparse
import time

from auto_navigate import navigate_to_level
from profiles import (
    DEFAULT_TARGET_LEVEL,
    WORLD_LABELS,
    WORLD_LEVEL_COUNTS,
    format_target_level_label,
)


def prompt_for_level_type():
    world_options = sorted(WORLD_LABELS.items(), key=lambda item: int(item[0]))
    print("Choose a level type:")
    for world_id, label in world_options:
        count = WORLD_LEVEL_COUNTS.get(world_id, 0)
        print(f"  {world_id}. {label.title()} ({count} levels)")

    while True:
        choice = input("Level type: ").strip().lower()
        if choice in WORLD_LABELS:
            return choice
        for world_id, label in world_options:
            if choice == label.lower():
                return world_id
        print("Enter a number like 3 or a name like speed.")


def prompt_for_level_number(world_id):
    world_label = WORLD_LABELS.get(world_id, f"mode {world_id}")
    max_level = int(WORLD_LEVEL_COUNTS.get(world_id, 1))
    print(f"Choose a {world_label} level:")
    print("  " + ", ".join(str(level) for level in range(1, max_level + 1)))

    while True:
        choice = input("Level number: ").strip()
        if choice.isdigit():
            level = int(choice)
            if 1 <= level <= max_level:
                return level
        print(f"Enter a number from 1 to {max_level}.")


def main():
    parser = argparse.ArgumentParser(
        description="Launch Fireboy & Watergirl directly into a level for inspection."
    )
    parser.add_argument("--profile", default="default", help="Named run profile to use")
    parser.add_argument("--slot", type=int, help="Window slot for multi-window placement")
    parser.add_argument(
        "--frame-rate",
        type=float,
        help="Override the active Ruffle/game FPS for this launch",
    )
    parser.add_argument(
        "--target-level",
        help="Level to launch, like 1_1, 2_4, or 3_2",
    )
    args = parser.parse_args()

    target_level = args.target_level
    if not target_level:
        world_id = prompt_for_level_type()
        level_number = prompt_for_level_number(world_id)
        target_level = f"{world_id}_{level_number}"
    else:
        target_level = target_level or DEFAULT_TARGET_LEVEL

    process = navigate_to_level(
        profile_name=args.profile,
        window_slot=args.slot,
        frame_rate=args.frame_rate,
        target_level=target_level,
    )
    if not process:
        raise SystemExit(1)

    print(
        f"Preview ready for {format_target_level_label(target_level)}. "
        "Leave this window open while you inspect the level. Press Ctrl+C to stop."
    )

    try:
        while process.poll() is None:
            time.sleep(1.0)
    except KeyboardInterrupt:
        if process.poll() is None:
            print("Closing the preview window...")
            process.terminate()
            try:
                process.wait(timeout=5)
            except Exception:
                process.kill()


if __name__ == "__main__":
    main()