mirror of
https://github.com/volcengine/OpenViking.git
synced 2026-09-28 11:43:00 +08:00
wutongyuonce
d436baa23c
The official npm bin (ov.mjs) starts with #!/usr/bin/env node. The Python entry-point fallback skipped every shebang file on PATH, so the Node wrapper was never exec'd and users got 'ov binary not found'. Only compare against and skip the current Python entry point itself; other PATH executables (including the Node shebang wrapper) are exec'd normally, and exec failures are no longer swallowed.
117 lines
3.9 KiB
Python
117 lines
3.9 KiB
Python
"""ov 命令的极简 Python 包装器
|
||
|
||
设计原则：
|
||
1. 职责单一：仅负责查找二进制并 execv
|
||
2. 无网络依赖：不实现下载功能
|
||
3. 极简代码：尽可能减少启动开销
|
||
4. 快速失败：找不到立即提示用户
|
||
|
||
性能说明：
|
||
- Python 虚拟机启动 + 导入基础模块：约 30-50ms
|
||
- 一旦 execv 执行，后续为纯 Rust 二进制，零开销
|
||
|
||
Rust CLI 独立发布能力完全保留，用户可通过以下方式获取：
|
||
- 官方安装脚本（零开销）
|
||
- GitHub Releases 手动下载（零开销）
|
||
- cargo install（零开销）
|
||
- 包管理器（未来）
|
||
"""
|
||
|
||
import os
|
||
import subprocess
|
||
import sys
|
||
from pathlib import Path
|
||
from shutil import which
|
||
|
||
|
||
def _exec_binary(binary: str, argv: list[str]) -> None:
|
||
"""Execute a binary, replacing the current process on Unix.
|
||
|
||
On Windows, ``os.execv`` does not truly replace the process — CPython's
|
||
MSVC implementation spawns a child process instead. This breaks console
|
||
handle inheritance and prevents the Rust TUI from receiving keyboard
|
||
input (see #587). We use ``subprocess.call`` on Windows to work around
|
||
this.
|
||
"""
|
||
if sys.platform == "win32":
|
||
sys.exit(subprocess.call([binary] + argv))
|
||
else:
|
||
os.execv(binary, [binary] + argv)
|
||
|
||
|
||
def main():
|
||
"""
|
||
极简入口点：查找 ov 二进制并执行
|
||
|
||
按优先级查找：
|
||
0. Python-native 子命令（doctor）
|
||
1. ./target/release/ov（开发环境）
|
||
2. Wheel 自带：{package_dir}/openviking/bin/ov
|
||
3. PATH 查找：系统全局安装的 ov
|
||
"""
|
||
# 0. Python-native subcommands (no Rust binary needed)
|
||
if len(sys.argv) > 1 and sys.argv[1] == "doctor":
|
||
from openviking_cli.doctor import main as doctor_main
|
||
|
||
sys.exit(doctor_main())
|
||
# 1. 检查开发环境（仅在直接运行脚本时有效）
|
||
try:
|
||
# __file__ is openviking_cli/rust_cli.py, so parent is openviking_cli directory
|
||
dev_binary = Path(__file__).parent.parent / "target" / "release" / "ov"
|
||
if dev_binary.exists() and os.access(dev_binary, os.X_OK):
|
||
_exec_binary(str(dev_binary), sys.argv[1:])
|
||
except Exception:
|
||
pass
|
||
|
||
# 2. 检查 Wheel 自带（不导入 openviking，避免额外开销）
|
||
try:
|
||
# __file__ is openviking_cli/rust_cli.py, so parent is openviking_cli directory
|
||
package_dir = Path(__file__).parent.parent / "openviking"
|
||
package_bin = package_dir / "bin"
|
||
for binary_name in ["ov", "ov.exe"]:
|
||
binary = package_bin / binary_name
|
||
if binary.exists() and os.access(binary, os.X_OK):
|
||
_exec_binary(str(binary), sys.argv[1:])
|
||
except Exception:
|
||
pass
|
||
|
||
# 3. 检查 PATH，但跳过当前 Python 入口点
|
||
path_binary = which("ov")
|
||
if path_binary:
|
||
try:
|
||
candidate_path = Path(path_binary).resolve()
|
||
current_path = Path(sys.argv[0])
|
||
if not current_path.is_absolute():
|
||
current_path = Path(which(sys.argv[0]) or current_path)
|
||
current_path = current_path.resolve()
|
||
except OSError:
|
||
pass
|
||
else:
|
||
if candidate_path != current_path:
|
||
_exec_binary(path_binary, sys.argv[1:])
|
||
|
||
# 都找不到，提示用户
|
||
print(
|
||
"""错误: 未找到 ov 二进制文件。
|
||
|
||
请选择以下方式之一安装：
|
||
|
||
1. 使用预构建 wheel（推荐）：
|
||
pip install openviking --upgrade --force-reinstall
|
||
|
||
2. 使用 npm 安装原生 CLI 包（零 Python 开销）：
|
||
npm i -g @openviking/cli
|
||
|
||
3. 从 GitHub Releases 下载（零 Python 开销）：
|
||
https://github.com/volcengine/OpenViking/releases
|
||
|
||
4. 从源码构建（零 Python 开销）：
|
||
cargo install --git https://github.com/volcengine/OpenViking ov_cli""",
|
||
file=sys.stderr,
|
||
)
|
||
return 1
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|