d3ro-voice/apps/desktop/scripts/build-sidecar.py

"""
scripts/build-sidecar.py
faster-whisper sidecar를 PyInstaller로 빌드한다. (크로스 플랫폼)

사용법:
    pip install pyinstaller
    pip install -r sidecar/requirements.txt
    python scripts/build-sidecar.py

출력:
    sidecar-dist/sidecar/sidecar(.exe)  (onedir 모드)

플랫폼:
    Windows → --noconsole (콘솔 창 숨김)
    macOS   → --windowed (.app 번들 형식은 여기서는 미사용, onedir만)
    Linux   → 콘솔 옵션 없음
"""

import platform
import subprocess
import sys
import shutil
from pathlib import Path

PROJECT_ROOT = Path(__file__).parent.parent
SIDECAR_DIR = PROJECT_ROOT / "sidecar"
MAIN_PY = SIDECAR_DIR / "main.py"
OUTPUT_DIR = PROJECT_ROOT / "sidecar-dist"

IS_WINDOWS = platform.system() == "Windows"
IS_MACOS = platform.system() == "Darwin"
EXE_SUFFIX = ".exe" if IS_WINDOWS else ""


def check_prerequisites():
    """필수 도구가 설치되어 있는지 확인한다."""
    try:
        import PyInstaller  # noqa: F401
    except ImportError:
        print("PyInstaller가 설치되어 있지 않습니다.")
        print("실행: pip install pyinstaller")
        sys.exit(1)

    try:
        import faster_whisper  # noqa: F401
    except ImportError:
        print("faster-whisper가 설치되어 있지 않습니다.")
        print("실행: pip install -r sidecar/requirements.txt")
        sys.exit(1)

    if not MAIN_PY.exists():
        print(f"sidecar 소스를 찾을 수 없습니다: {MAIN_PY}")
        sys.exit(1)


def build():
    """PyInstaller로 sidecar를 빌드한다."""
    print("=" * 60)
    print(f"D3RO-VOICE Sidecar 빌드 시작 ({platform.system()} {platform.machine()})")
    print("=" * 60)

    # 기존 빌드 정리
    if OUTPUT_DIR.exists():
        shutil.rmtree(OUTPUT_DIR)

    cmd = [
        sys.executable, "-m", "PyInstaller",
        "--name", "sidecar",
        "--distpath", str(OUTPUT_DIR),
        "--workpath", str(PROJECT_ROOT / "build" / "sidecar-build"),
        "--specpath", str(PROJECT_ROOT / "build"),
        "--noconfirm",
        "--clean",
        # hidden imports
        "--hidden-import", "faster_whisper",
        "--hidden-import", "ctranslate2",
        "--hidden-import", "huggingface_hub",
        "--hidden-import", "tokenizers",
        "--hidden-import", "uvicorn.logging",
        "--hidden-import", "uvicorn.protocols.http",
        "--hidden-import", "uvicorn.protocols.http.auto",
        "--hidden-import", "uvicorn.protocols.http.h11_impl",
        "--hidden-import", "uvicorn.protocols.websockets",
        "--hidden-import", "uvicorn.protocols.websockets.auto",
        "--hidden-import", "uvicorn.lifespan",
        "--hidden-import", "uvicorn.lifespan.on",
        "--hidden-import", "uvicorn.lifespan.off",
    ]

    # 플랫폼별 콘솔 옵션
    if IS_WINDOWS:
        cmd.append("--noconsole")
    elif IS_MACOS:
        # macOS에서 --windowed는 .app 번들 생성을 의미.
        # onedir 모드로도 .app이 만들어질 수 있어서 명시적으로는 쓰지 않음.
        # electron-builder가 extraResources로 Binary만 포함하므로 기본 콘솔 형태 유지.
        pass
    # Linux: 옵션 없음

    cmd.append(str(MAIN_PY))

    print(f"\n실행 명령:\n  {' '.join(cmd)}\n")

    result = subprocess.run(cmd, cwd=str(PROJECT_ROOT))

    if result.returncode != 0:
        print(f"\nPyInstaller 빌드 실패 (exit code: {result.returncode})")
        sys.exit(1)

    # 빌드 결과 확인
    exe_path = OUTPUT_DIR / "sidecar" / f"sidecar{EXE_SUFFIX}"
    if exe_path.exists():
        size_mb = exe_path.stat().st_size / (1024 * 1024)
        print(f"\n빌드 성공!")
        print(f"  경로: {exe_path}")
        print(f"  크기: {size_mb:.1f} MB")

        total_size = sum(
            f.stat().st_size for f in (OUTPUT_DIR / "sidecar").rglob("*") if f.is_file()
        )
        total_mb = total_size / (1024 * 1024)
        print(f"  전체 디렉토리: {total_mb:.1f} MB")
    else:
        print(f"\n빌드 출력을 찾을 수 없습니다: {exe_path}")
        # onedir 모드에서는 디렉토리 내부에 바이너리가 있음
        pattern = "*.exe" if IS_WINDOWS else "sidecar"
        for item in OUTPUT_DIR.rglob(pattern):
            print(f"  발견: {item}")

    print("\n빌드 완료!")
    print("electron-builder에서 이 경로를 extraResources로 지정하세요:")
    print("  from: sidecar-dist/sidecar/")
    print("  to: sidecar/")


if __name__ == "__main__":
    check_prerequisites()
    build()