Function main

scripts/freeze_viewer.py:107–197

Main function to generate static site

()

def main():
    """Main function to generate static site"""
    parser = argparse.ArgumentParser(description="Generate static version of CodeClash Trajectory Viewer")
    parser.add_argument("--logs-dir", type=str, default="logs", help="Directory containing game logs (default: logs)")
    parser.add_argument(
        "--output-dir", type=str, default="build", help="Output directory for static files (default: build)"
    )
    parser.add_argument("--clean", action="store_true", help="Clean output directory before building")

    args = parser.parse_args()

    # Set up paths
    logs_dir = Path(args.logs_dir).resolve()
    output_dir = Path(args.output_dir).resolve()

    assert logs_dir.exists()

    # Set the logs directory for the app
    set_log_base_directory(logs_dir)

    # Clean output directory if requested
    if args.clean and output_dir.exists():
        print(f"Cleaning output directory: {output_dir}")
        shutil.rmtree(output_dir)

    # Create output directory
    output_dir.mkdir(parents=True, exist_ok=True)

    print(f"Generating static site from logs in: {logs_dir}")
    print(f"Output directory: {output_dir}")

    # Set up freezer
    freezer = setup_freezer(str(output_dir))

    # Generate static site
    print("Freezing Flask application...")

    # Try to freeze - it will likely fail on API endpoints, but we'll continue
    try:
        urls = freezer.freeze()
        print(f"Successfully froze {len(urls)} URL(s)")
    except Exception as e:
        print(f"Initial freeze had errors (expected): {type(e).__name__}")

    # Now manually generate the game view pages
    print("Generating game view pages...")
    from codeclash.viewer.app import LOG_BASE_DIR, game_view

    game_folders = find_all_game_folders(LOG_BASE_DIR)
    print(f"Found {len(game_folders)} game folder(s) to process")

    for i, game_folder in enumerate(game_folders, 1):
        folder_path = game_folder["name"]
        print(f"  [{i}/{len(game_folders)}] Generating page for: {folder_path}")

        # Create the output directory structure if needed (for nested paths)
        game_dir = output_dir / "game"
        game_dir.mkdir(parents=True, exist_ok=True)