build_exe.py

from __future__ import annotations

import argparse
import os
import subprocess
import sys
from contextlib import contextmanager
from dataclasses import dataclass
from pathlib import Path

from printer_tool.packaging import (
    APP_NAME,
    MAINTENANCE_VISIBLE_APP_NAME,
    build_pyinstaller_command,
    map_path_to_alias_root,
    path_needs_ascii_alias,
)


ASCII_ALIAS_ENV = "PRINTER_TOOL_ASCII_ALIAS_ACTIVE"


@dataclass(frozen=True)
class BuildVariant:
    app_name: str
    show_maintenance_ui: bool


def _pick_drive_letter() -> str:
    for letter in "PQRSTUVWXYZ":
        if not Path(f"{letter}:\\").exists():
            return f"{letter}:"
    raise RuntimeError("未找到可用的临时盘符，无法创建 ASCII 打包路径。")


@contextmanager
def temporary_ascii_alias(project_root: Path):
    drive = _pick_drive_letter()
    subprocess.run(["subst", drive, str(project_root)], check=True)
    try:
        yield Path(f"{drive}\\")
    finally:
        subprocess.run(["subst", drive, "/d"], check=False)


def parse_args(argv: list[str] | None = None) -> argparse.Namespace:
    parser = argparse.ArgumentParser(description="Build the PrinterTool executable.")
    parser.add_argument(
        "--show-maintenance-ui",
        action="store_true",
        help="Build a variant that shows the built-in maintenance trace window.",
    )
    parser.add_argument(
        "--both",
        dest="build_both",
        action="store_true",
        help="Build both the default and visible-maintenance variants.",
    )
    return parser.parse_args(argv)


def build_variants(*, show_maintenance_ui: bool, build_both: bool) -> list[BuildVariant]:
    if build_both:
        return [
            BuildVariant(app_name=APP_NAME, show_maintenance_ui=False),
            BuildVariant(
                app_name=MAINTENANCE_VISIBLE_APP_NAME,
                show_maintenance_ui=True,
            ),
        ]
    return [
        BuildVariant(
            app_name=MAINTENANCE_VISIBLE_APP_NAME if show_maintenance_ui else APP_NAME,
            show_maintenance_ui=show_maintenance_ui,
        )
    ]


def main(argv: list[str] | None = None) -> int:
    args = parse_args(argv)
    forwarded_args = list(argv if argv is not None else sys.argv[1:])
    project_root = Path(__file__).resolve().parent
    current_python = Path(sys.executable).resolve()
    real_project_root = project_root.resolve()

    if (
        os.getenv(ASCII_ALIAS_ENV) != "1"
        and path_needs_ascii_alias(current_python)
        and current_python.is_relative_to(real_project_root)
    ):
        with temporary_ascii_alias(real_project_root) as alias_root:
            alias_python = map_path_to_alias_root(current_python, real_project_root, alias_root)
            alias_script = alias_root / "build_exe.py"
            env = os.environ.copy()
            env[ASCII_ALIAS_ENV] = "1"
            print(f"检测到非 ASCII Python 路径，使用临时盘符 {alias_root.drive} 重新执行打包...")
            completed = subprocess.run(
                [str(alias_python), str(alias_script), *forwarded_args],
                cwd=alias_root,
                env=env,
                check=False,
            )
            return completed.returncode

    for variant in build_variants(
        show_maintenance_ui=args.show_maintenance_ui,
        build_both=args.build_both,
    ):
        command = build_pyinstaller_command(
            project_root,
            show_maintenance_ui=variant.show_maintenance_ui,
        )
        print(f"开始打包 {variant.app_name}.exe ...")
        completed = subprocess.run(command, cwd=project_root, check=False)
        if completed.returncode != 0:
            return completed.returncode
        print(f"打包完成: {project_root / 'dist' / f'{variant.app_name}.exe'}")

    return 0


if __name__ == "__main__":
    raise SystemExit(main())