run_desktop_release.py

import argparse
import subprocess
import sys
from pathlib import Path

import sts2_recovery as recover


def parse_args() -> argparse.Namespace:
    script_dir = Path(__file__).resolve().parent
    parser = argparse.ArgumentParser(description="Run the exported Windows Desktop Release package.")
    parser.add_argument(
        "--output-dir",
        "-OutputDir",
        type=Path,
        default=script_dir / "build" / "recovered_project",
        help="existing recovered project directory",
    )
    parser.add_argument(
        "--renderer",
        default=None,
        help="optional rendering driver override, e.g. d3d12 / vulkan / opengl3",
    )
    parser.add_argument(
        "--quit-after",
        type=int,
        default=0,
        help="quit automatically after N seconds",
    )
    parser.add_argument(
        "--log-file",
        type=Path,
        default=None,
        help="optional exported runtime log file path",
    )
    parser.add_argument(
        "extra_args",
        nargs=argparse.REMAINDER,
        help="extra arguments forwarded to the exported executable (prefix with --)",
    )
    return parser.parse_args()


def main() -> int:
    args = parse_args()
    args.output_dir = args.output_dir.resolve()
    export_exe = (args.output_dir / recover.WINDOWS_RELEASE_EXPORT_RELATIVE_PATH).resolve()

    if not export_exe.exists():
        print(
            "未发现桌面 Release 导出包，请先运行 "
            "`python .\\export_desktop_release.py`。",
            file=sys.stderr,
        )
        return 1

    cmd = [str(export_exe)]
    if args.renderer:
        cmd.extend(["--rendering-driver", args.renderer])
    if args.quit_after > 0:
        cmd.extend(["--quit-after", str(args.quit_after)])
    if args.log_file is not None:
        cmd.extend(["--log-file", str(args.log_file.resolve())])
    extra_args = list(args.extra_args)
    if extra_args and extra_args[0] == "--":
        extra_args = extra_args[1:]
    cmd.extend(extra_args)

    recover.log("$ " + subprocess.list2cmdline(cmd))
    completed = subprocess.run(cmd, cwd=str(export_exe.parent), check=False)
    return completed.returncode


if __name__ == "__main__":
    try:
        raise SystemExit(main())
    except Exception as exc:
        print(f"ERROR: {exc}", file=sys.stderr)
        raise