99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
|
"""Rust 工具链检测模块。"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import subprocess
|
|||
|
|
from pathlib import Path
|
|||
|
|
from typing import Any
|
|||
|
|
|
|||
|
|
from aide.env.modules.base import BaseModule, CheckResult, ModuleInfo
|
|||
|
|
|
|||
|
|
|
|||
|
|
class RustModule(BaseModule):
|
|||
|
|
"""Rust 工具链检测模块（类型A：无需配置）。"""
|
|||
|
|
|
|||
|
|
@property
|
|||
|
|
def info(self) -> ModuleInfo:
|
|||
|
|
return ModuleInfo(
|
|||
|
|
name="rust",
|
|||
|
|
description="Rust 工具链",
|
|||
|
|
capabilities=["check"],
|
|||
|
|
requires_config=False,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
def check(self, config: dict[str, Any], root: Path) -> CheckResult:
|
|||
|
|
"""检测 Rust 工具链（rustc 和 cargo）。"""
|
|||
|
|
rustc_version = self._get_version("rustc")
|
|||
|
|
cargo_version = self._get_version("cargo")
|
|||
|
|
|
|||
|
|
if not rustc_version:
|
|||
|
|
return CheckResult(
|
|||
|
|
success=False,
|
|||
|
|
message="rustc 未安装",
|
|||
|
|
can_ensure=False,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if not cargo_version:
|
|||
|
|
return CheckResult(
|
|||
|
|
success=False,
|
|||
|
|
message="cargo 未安装",
|
|||
|
|
can_ensure=False,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 检查最低版本要求（如果配置了）
|
|||
|
|
min_version = config.get("min_version")
|
|||
|
|
if min_version:
|
|||
|
|
if not self._version_satisfies(rustc_version, min_version):
|
|||
|
|
return CheckResult(
|
|||
|
|
success=False,
|
|||
|
|
version=rustc_version,
|
|||
|
|
message=f"版本不足，要求>={min_version}，当前 {rustc_version}",
|
|||
|
|
can_ensure=False,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
return CheckResult(
|
|||
|
|
success=True,
|
|||
|
|
version=rustc_version,
|
|||
|
|
message=f"cargo {cargo_version}",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
def _get_version(self, cmd: str) -> str | None:
|
|||
|
|
"""获取命令版本。"""
|
|||
|
|
try:
|
|||
|
|
result = subprocess.run(
|
|||
|
|
[cmd, "--version"],
|
|||
|
|
capture_output=True,
|
|||
|
|
text=True,
|
|||
|
|
timeout=10,
|
|||
|
|
)
|
|||
|
|
if result.returncode == 0:
|
|||
|
|
# rustc 1.75.0 (xxx) -> 1.75.0
|
|||
|
|
# cargo 1.75.0 (xxx) -> 1.75.0
|
|||
|
|
output = result.stdout.strip()
|
|||
|
|
parts = output.split()
|
|||
|
|
if len(parts) >= 2:
|
|||
|
|
return parts[1]
|
|||
|
|
return None
|
|||
|
|
except (FileNotFoundError, subprocess.TimeoutExpired):
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def _version_satisfies(self, current: str, minimum: str) -> bool:
|
|||
|
|
"""检查版本是否满足最低要求。"""
|
|||
|
|
current_parts = self._parse_version(current)
|
|||
|
|
min_parts = self._parse_version(minimum)
|
|||
|
|
return current_parts >= min_parts
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _parse_version(version: str) -> tuple[int, ...]:
|
|||
|
|
"""解析版本号字符串。"""
|
|||
|
|
parts = []
|
|||
|
|
for part in version.split("."):
|
|||
|
|
try:
|
|||
|
|
parts.append(int(part))
|
|||
|
|
except ValueError:
|
|||
|
|
break
|
|||
|
|
return tuple(parts)
|
|||
|
|
|
|||
|
|
|
|||
|
|
module = RustModule()
|