iamt 1.1.27__py3-none-any.whl

iamt/modules/github/ghcli.py

@@ -0,0 +1,197 @@
+"""GitHub CLI 模块 - 在远程 Linux 主机上安装和配置 GitHub CLI
+
+该模块通过 SSH 连接执行远程命令，实现 GitHub CLI 的自动化部署。
+"""
+
+from fabric import Connection, task
+
+from .._utils.github import download_file, resolve_version
+
+
+# region 模块常量配置
+
+# GitHub CLI 下载 URL 模板
+GH_DOWNLOAD_URL_TEMPLATE = "https://github.com/cli/cli/releases/download/v{version}/gh_{version}_linux_{arch}.tar.gz"
+
+# GitHub CLI latest release 重定向 URL (用于获取最新版本号)
+GH_LATEST_RELEASE_URL = "https://github.com/cli/cli/releases/latest"
+
+# 架构映射 (uname -m -> GitHub CLI 架构名称)
+GH_ARCH_MAP = {
+    "x86_64": "amd64",
+    "aarch64": "arm64",
+    "armv7l": "armv6",
+}
+
+# 默认安装目录
+GH_INSTALL_DIR = "/usr/local/bin"
+
+# 环境变量 (使用英文输出避免中文乱码)
+ENV_LANG_C = {"LANG": "C", "LC_ALL": "C"}
+
+# endregion
+
+
+# region 辅助函数
+
+def _get_remote_arch(conn: Connection) -> str:
+    """获取远程主机的架构标识
+    
+    优先使用 hostvars 中已收集的架构信息，避免重复执行命令
+    
+    Args:
+        conn: Fabric 连接对象
+        
+    Returns:
+        GitHub CLI 架构名称 (如 amd64, arm64)
+    """
+    # 优先从 hostvars 获取架构信息
+    if hasattr(conn, "hostvars") and conn.hostvars:
+        arch = conn.hostvars.get("architecture", "")
+        if arch:
+            return GH_ARCH_MAP.get(arch, "amd64")
+    
+    # 回退方案：直接执行命令获取
+    result = conn.run("uname -m", hide=True)
+    arch = result.stdout.strip()
+    return GH_ARCH_MAP.get(arch, "amd64")
+
+
+def _download_gh_tarball(version: str, arch: str, dest_dir: str) -> str:
+    """下载 GitHub CLI 压缩包到本地
+    
+    Args:
+        version: 版本号
+        arch: 架构标识
+        dest_dir: 下载目标目录
+        
+    Returns:
+        下载后的压缩包路径
+    """
+    version_number = resolve_version(version, GH_LATEST_RELEASE_URL)
+    url = GH_DOWNLOAD_URL_TEMPLATE.format(version=version_number, arch=arch)
+    filename = f"gh_{version_number}_linux_{arch}.tar.gz"
+    return download_file(url, dest_dir, filename)
+
+# endregion
+
+
+# region Task 函数
+
+@task
+def install_gh(conn: Connection, version: str = "latest", install_dir: str = GH_INSTALL_DIR) -> None:
+    """下载并安装 GitHub CLI (本地下载模式)
+    
+    先将 CLI 安装包下载到本地，然后上传到服务器并安装。
+    适用于本地网络条件较好的场景。
+    
+    Args:
+        conn: Fabric 连接对象
+        version: 版本号，默认为 "latest"，也可以指定具体版本如 "2.40.0"
+        install_dir: 安装目录路径，默认为 /usr/local/bin
+    """
+    arch = _get_remote_arch(conn)
+    print(f"[INFO] 目标架构: {arch}")
+    print(f"[INFO] 安装版本: {version}")
+    
+    remote_tmp_dir = "/tmp/gh_install"
+    remote_tarball = f"{remote_tmp_dir}/gh.tar.gz"
+    
+    # region 下载并上传压缩包
+    print(f"\n[任务] 下载 GitHub CLI {version}")
+    tarball_path = _download_gh_tarball(version, arch, str(conn.artifacts_dir))
+    
+    print("\n[任务] 上传压缩包到远程主机")
+    conn.run(f"mkdir -p {remote_tmp_dir}", env=ENV_LANG_C)
+    conn.put(tarball_path, remote_tarball)
+    print(f"  上传完成: {remote_tarball}")
+    # endregion
+    
+    # region 解压并安装
+    print("\n[任务] 解压并安装 GitHub CLI")
+    conn.run(f"tar -xzf {remote_tarball} -C {remote_tmp_dir}", env=ENV_LANG_C)
+    
+    # 查找解压后的目录
+    result = conn.run(f"ls -d {remote_tmp_dir}/gh_*_linux_{arch}", hide=True)
+    extracted_dir = result.stdout.strip()
+    
+    # 安装二进制文件
+    conn.sudo(f"cp {extracted_dir}/bin/gh {install_dir}/gh", env=ENV_LANG_C)
+    conn.sudo(f"chmod +x {install_dir}/gh", env=ENV_LANG_C)
+    print(f"  gh 已安装到: {install_dir}/gh")
+    # endregion
+    
+    # region 清理临时文件
+    print("\n[任务] 清理临时文件")
+    conn.run(f"rm -rf {remote_tmp_dir}", env=ENV_LANG_C)
+    print("  清理完成")
+    # endregion
+    
+    # region 验证安装
+    print("\n[任务] 验证安装")
+    result = conn.run(f"{install_dir}/gh --version", hide=True)
+    print(f"  {result.stdout.strip()}")
+    # endregion
+    
+    print(f"\n[OK] GitHub CLI {version} 安装完成 (本地下载模式)")
+
+
+@task
+def install_gh_remote(conn: Connection, version: str = "latest", install_dir: str = GH_INSTALL_DIR) -> None:
+    """下载并安装 GitHub CLI (远程下载模式)
+    
+    直接在远程主机上下载安装包并安装。
+    适用于远程主机网络条件较好的场景。
+    
+    Args:
+        conn: Fabric 连接对象
+        version: 版本号，默认为 "latest"，也可以指定具体版本如 "2.40.0"
+        install_dir: 安装目录路径，默认为 /usr/local/bin
+    """
+    arch = _get_remote_arch(conn)
+    version_number = resolve_version(version, GH_LATEST_RELEASE_URL)
+    print(f"[INFO] 目标架构: {arch}")
+    print(f"[INFO] 安装版本: {version_number}")
+    
+    download_url = GH_DOWNLOAD_URL_TEMPLATE.format(version=version_number, arch=arch)
+    print(f"[INFO] 下载地址: {download_url}")
+    
+    remote_tmp_dir = "/tmp/gh_install"
+    remote_tarball = f"{remote_tmp_dir}/gh.tar.gz"
+    
+    # region 在远程主机上下载压缩包
+    print(f"\n[任务] 在远程主机上下载 GitHub CLI {version}")
+    conn.run(f"mkdir -p {remote_tmp_dir}", env=ENV_LANG_C)
+    conn.run(f"curl -fsSL -o {remote_tarball} '{download_url}'", env=ENV_LANG_C)
+    print(f"  下载完成: {remote_tarball}")
+    # endregion
+    
+    # region 解压并安装
+    print("\n[任务] 解压并安装 GitHub CLI")
+    conn.run(f"tar -xzf {remote_tarball} -C {remote_tmp_dir}", env=ENV_LANG_C)
+    
+    # 查找解压后的目录
+    result = conn.run(f"ls -d {remote_tmp_dir}/gh_*_linux_{arch}", hide=True)
+    extracted_dir = result.stdout.strip()
+    
+    # 安装二进制文件
+    conn.sudo(f"cp {extracted_dir}/bin/gh {install_dir}/gh", env=ENV_LANG_C)
+    conn.sudo(f"chmod +x {install_dir}/gh", env=ENV_LANG_C)
+    print(f"  gh 已安装到: {install_dir}/gh")
+    # endregion
+    
+    # region 清理临时文件
+    print("\n[任务] 清理临时文件")
+    conn.run(f"rm -rf {remote_tmp_dir}", env=ENV_LANG_C)
+    print("  清理完成")
+    # endregion
+    
+    # region 验证安装
+    print("\n[任务] 验证安装")
+    result = conn.run(f"{install_dir}/gh --version", hide=True)
+    print(f"  {result.stdout.strip()}")
+    # endregion
+    
+    print(f"\n[OK] GitHub CLI {version} 安装完成 (远程下载模式)")
+
+# endregion

iamt/modules/nvm/nvm.py

@@ -0,0 +1,183 @@
+"""NVM 模块 - 在远程 Linux 主机上安装和配置 Node Version Manager
+
+该模块通过 SSH 连接执行远程命令，实现 NVM 的自动化部署。
+https://github.com/nvm-sh/nvm
+"""
+
+from fabric import Connection, task
+
+from iamt.modules._utils.github import download_file, resolve_version
+
+
+# region 模块常量配置
+
+# NVM 下载 URL 模板
+NVM_DOWNLOAD_URL_TEMPLATE = "https://github.com/nvm-sh/nvm/archive/refs/tags/v{version}.tar.gz"
+
+# NVM latest release 重定向 URL (用于获取最新版本号)
+NVM_LATEST_RELEASE_URL = "https://github.com/nvm-sh/nvm/releases/latest"
+
+# 默认安装目录 (相对于用户 home)
+NVM_INSTALL_DIR = ".nvm"
+
+# 环境变量 (使用英文输出避免中文乱码)
+ENV_LANG_C = {"LANG": "C", "LC_ALL": "C"}
+
+# endregion
+
+
+# region 辅助函数
+
+def _download_nvm_tarball(version: str, dest_dir: str) -> tuple[str, str]:
+    """下载 NVM 压缩包到本地
+    
+    Args:
+        version: 版本号
+        dest_dir: 下载目标目录
+        
+    Returns:
+        (下载后的压缩包路径, 实际版本号)
+    """
+    version_number = resolve_version(version, NVM_LATEST_RELEASE_URL)
+    url = NVM_DOWNLOAD_URL_TEMPLATE.format(version=version_number)
+    filename = f"nvm-{version_number}.tar.gz"
+    dest_path = download_file(url, dest_dir, filename)
+    return dest_path, version_number
+
+
+def _get_shell_rc_file(conn: Connection) -> str:
+    """获取用户的 shell 配置文件路径
+    
+    Args:
+        conn: Fabric 连接对象
+        
+    Returns:
+        shell 配置文件路径 (如 ~/.bashrc, ~/.zshrc)
+    """
+    result = conn.run("echo $SHELL", hide=True)
+    shell = result.stdout.strip()
+    
+    if "zsh" in shell:
+        return "~/.zshrc"
+    return "~/.bashrc"
+
+# endregion
+
+
+# region Task 函数
+
+@task
+def install_nvm(conn: Connection, version: str = "latest") -> None:
+    """下载并安装 NVM (本地下载模式)
+    
+    先将 NVM 安装包下载到本地，然后上传到服务器并安装。
+    适用于本地网络条件较好的场景。
+    
+    Args:
+        conn: Fabric 连接对象
+        version: 版本号，默认为 "latest"，也可以指定具体版本如 "0.40.1"
+    """
+    # region 获取用户 home 目录
+    user_home = conn.hostvars.get("user_home", "~")
+    nvm_dir = f"{user_home}/{NVM_INSTALL_DIR}"
+    # endregion
+    
+    print(f"[INFO] 安装版本: {version}")
+    print(f"[INFO] 安装目录: {nvm_dir}")
+    
+    remote_tmp_dir = "/tmp/nvm_install"
+    remote_tarball = f"{remote_tmp_dir}/nvm.tar.gz"
+    
+    # region 下载并上传压缩包
+    print(f"\n[任务] 下载 NVM {version}")
+    tarball_path, version_number = _download_nvm_tarball(version, conn.artifacts_dir)
+    
+    print("\n[任务] 上传压缩包到远程主机")
+    conn.run(f"mkdir -p {remote_tmp_dir}", env=ENV_LANG_C)
+    conn.put(tarball_path, remote_tarball)
+    print(f"  上传完成: {remote_tarball}")
+    # endregion
+    
+    # region 解压并安装
+    print("\n[任务] 解压并安装 NVM")
+    conn.run(f"tar -xzf {remote_tarball} -C {remote_tmp_dir}", env=ENV_LANG_C)
+    
+    # 创建 NVM 目录并移动文件
+    conn.run(f"rm -rf {nvm_dir}", env=ENV_LANG_C)
+    conn.run(f"mv {remote_tmp_dir}/nvm-{version_number} {nvm_dir}", env=ENV_LANG_C)
+    print(f"  NVM 已安装到: {nvm_dir}")
+    # endregion
+    
+    # region 配置 shell 环境
+    print("\n[任务] 配置 shell 环境")
+    rc_file = _get_shell_rc_file(conn)
+    
+    nvm_config = f'''
+# NVM Configuration
+export NVM_DIR="{nvm_dir}"
+[ -s "$NVM_DIR/nvm.sh" ] && \\. "$NVM_DIR/nvm.sh"
+[ -s "$NVM_DIR/bash_completion" ] && \\. "$NVM_DIR/bash_completion"
+'''
+    
+    # 检查是否已配置
+    result = conn.run(f"grep -q 'NVM_DIR' {rc_file} 2>/dev/null", warn=True, hide=True)
+    if result.ok:
+        print(f"  NVM 配置已存在于 {rc_file}，跳过")
+    else:
+        conn.run(f"cat >> {rc_file} << 'EOF'{nvm_config}EOF", env=ENV_LANG_C)
+        print(f"  NVM 配置已添加到 {rc_file}")
+    # endregion
+    
+    # region 清理临时文件
+    print("\n[任务] 清理临时文件")
+    conn.run(f"rm -rf {remote_tmp_dir}", env=ENV_LANG_C)
+    print("  清理完成")
+    # endregion
+    
+    # region 验证安装
+    print("\n[任务] 验证安装")
+    result = conn.run(f'bash -c "source {nvm_dir}/nvm.sh && nvm --version"', hide=True)
+    print(f"  NVM 版本: {result.stdout.strip()}")
+    # endregion
+    
+    print(f"\n[OK] NVM {version_number} 安装完成")
+    print(f"[提示] 请执行 'source {rc_file}' 或重新登录以使 nvm 命令生效")
+
+
+@task
+def set_nodejs_mirror(conn: Connection, mirror: str = "https://mirrors.cernet.edu.cn/nodejs-release/") -> None:
+    """配置 NVM 的 Node.js 下载镜像源
+    
+    通过设置远程服务器rc文件当中的 NVM_NODEJS_ORG_MIRROR 环境变量来加速 Node.js 下载。
+    
+    Args:
+        conn: Fabric 连接对象
+        mirror: 镜像源 URL，默认为 CERNET 镜像
+    """
+    rc_file = _get_shell_rc_file(conn)
+    env_line = f'export NVM_NODEJS_ORG_MIRROR="{mirror}"'
+    
+    print(f"[INFO] 配置 Node.js 镜像源: {mirror}")
+    print(f"[INFO] 配置文件: {rc_file}")
+    
+    # region 检查并更新配置
+    # 检查是否已存在 NVM_NODEJS_ORG_MIRROR 配置
+    result = conn.run(f"grep -q 'NVM_NODEJS_ORG_MIRROR' {rc_file} 2>/dev/null", warn=True, hide=True)
+    
+    if result.ok:
+        # 已存在，使用 sed 替换
+        conn.run(
+            f"sed -i 's|^export NVM_NODEJS_ORG_MIRROR=.*|{env_line}|' {rc_file}",
+            env=ENV_LANG_C
+        )
+        print("  已更新 NVM_NODEJS_ORG_MIRROR 配置")
+    else:
+        # 不存在，追加到文件末尾
+        conn.run(f'echo \'{env_line}\' >> {rc_file}', env=ENV_LANG_C)
+        print("  已添加 NVM_NODEJS_ORG_MIRROR 配置")
+    # endregion
+    
+    print("\n[OK] Node.js 镜像源配置完成")
+    print(f"[提示] 请执行 'source {rc_file}' 或重新登录以使配置生效")
+
+# endregion

iamt/modules/nvm/roles/role.py

@@ -0,0 +1,24 @@
+from rich.console import Console
+
+from ..nvm import Connection, install_nvm, set_nodejs_mirror
+from ..._utils.steps import Step, run_steps
+
+console = Console()
+
+
+def setup_nvm(c: Connection) -> None:
+    """在远程服务器上安装配置 NVM
+    
+    执行以下步骤:
+    1. 下载并安装 NVM
+    2. 配置 Node.js 镜像源
+    
+    Args:
+        c: Fabric 连接对象
+    """
+    steps: list[Step] = [
+        ("安装 NVM", install_nvm),
+        ("配置 Node.js 镜像源", set_nodejs_mirror),
+    ]
+    run_steps(c, steps)
+    console.print("[bold green]✓ NVM 安装配置完成[/bold green]")

iamt/modules/runlinetask/runlinetask.py

@@ -0,0 +1,22 @@
+from fabric import Connection, task
+from invoke.exceptions import UnexpectedExit
+
+
+@task
+def run_line(conn: Connection, command: str, sudo: bool = False) -> None:
+    """在目标服务器上执行一行命令
+
+    Args:
+        conn: Fabric Connection 对象
+        command: 要执行的命令
+        sudo: 是否使用 sudo 执行，默认 False
+    """
+    # 使用 bash -ilc 以 interactive login shell 方式执行，确保加载 ~/.bashrc（nvm 等工具需要）
+    wrapped_command = f'bash -ilc {command!r}'
+    try:
+        if sudo:
+            conn.sudo(wrapped_command)
+        else:
+            conn.run(wrapped_command)
+    except UnexpectedExit as e:
+        print(f"\n命令执行失败 (退出码: {e.result.exited})")

iamt/modules/ubuntud/antigravity.py

@@ -0,0 +1,191 @@
+from fabric import Connection, task
+
+from .._utils.arch import normalize_arch
+from .._utils.github import download_file
+
+
+# region 模块常量配置
+
+# Antigravity APT 仓库配置
+ANTIGRAVITY_REPO_BASE = "https://us-central1-apt.pkg.dev/projects/antigravity-auto-updater-dev"
+ANTIGRAVITY_PACKAGES_URL = f"{ANTIGRAVITY_REPO_BASE}/dists/antigravity-debian/main/binary-{{arch}}/Packages"
+
+# 环境变量 (使用英文输出避免中文乱码)
+ENV_LANG_C = {"LANG": "C", "LC_ALL": "C"}
+
+# endregion
+
+
+# region 辅助函数
+
+def _get_remote_arch(conn: Connection) -> str:
+    """获取远程主机的 dpkg 架构标识
+    
+    优先使用 hostvars 中已收集的架构信息
+    
+    Args:
+        conn: Fabric 连接对象
+        
+    Returns:
+        dpkg 架构名称 (如 amd64, arm64)
+    """
+    if hasattr(conn, "hostvars") and conn.hostvars:
+        arch = conn.hostvars.get("architecture", "")
+        if arch:
+            return normalize_arch(arch)
+    
+    result = conn.run("uname -m", hide=True)
+    return normalize_arch(result.stdout.strip())
+
+
+def _download_antigravity_deb(arch: str, dest_dir: str) -> str:
+    """从 APT 仓库下载 Antigravity .deb 包到本地
+    
+    直接从仓库的 Packages 索引获取包信息，无需远程主机配置 APT 源。
+    
+    Args:
+        arch: dpkg 架构标识 (如 amd64, arm64)
+        dest_dir: 下载目标目录
+        
+    Returns:
+        下载后的 .deb 包路径
+    """
+    import urllib.request
+    
+    # region 获取 Packages 索引并解析
+    packages_url = ANTIGRAVITY_PACKAGES_URL.format(arch=arch)
+    print(f"  正在获取包索引: {packages_url}")
+    
+    with urllib.request.urlopen(packages_url) as response:
+        packages_content = response.read().decode("utf-8")
+    
+    filename = None
+    version = None
+    for line in packages_content.split("\n"):
+        if line.startswith("Filename:"):
+            filename = line.split(":", 1)[1].strip()
+        elif line.startswith("Version:"):
+            version = line.split(":", 1)[1].strip()
+    
+    if not filename:
+        raise RuntimeError("无法从 Packages 索引中解析 .deb 文件路径")
+    # endregion
+    
+    # region 下载 .deb 文件
+    deb_url = f"{ANTIGRAVITY_REPO_BASE}/{filename}"
+    local_filename = f"antigravity_{version}_{arch}.deb" if version else f"antigravity_{arch}.deb"
+    return download_file(deb_url, dest_dir, local_filename)
+    # endregion
+
+# endregion
+
+
+# region Task 函数
+
+@task
+def install_antigravity(conn: Connection) -> None:
+    """在 Ubuntu 上安装 antigravity"""
+    
+    # region 检查系统类型
+    dist_info: dict = conn.hostvars["distribution"]
+    dist_id: str = dist_info.get("id", "").lower()
+    
+    if dist_id not in ("ubuntu", "debian"):
+        print(f"不支持的发行版: {dist_id}，仅支持 Ubuntu 和 Debian")
+        return
+    # endregion
+    
+    # region 配置 APT 仓库
+    keyring_dir = "/etc/apt/keyrings"
+    keyring_file = f"{keyring_dir}/antigravity-repo-key.gpg"
+    sources_list = "/etc/apt/sources.list.d/antigravity.list"
+    repo_url = "https://us-central1-apt.pkg.dev/projects/antigravity-auto-updater-dev/"
+    key_url = "https://us-central1-apt.pkg.dev/doc/repo-signing-key.gpg"
+    
+    # 创建 keyrings 目录
+    conn.sudo(f"mkdir -p {keyring_dir}")
+    
+    # 下载并转换 GPG 密钥
+    print("正在下载 GPG 密钥...")
+    conn.sudo(f"curl -fsSL {key_url} -o /tmp/repo-signing-key.gpg", hide=True)
+    conn.sudo(f"gpg --dearmor --yes -o {keyring_file} /tmp/repo-signing-key.gpg", hide=True)
+    conn.sudo("rm -f /tmp/repo-signing-key.gpg", hide=True)
+    
+    # 添加 APT 源
+    print("正在配置 APT 源...")
+    sources_content = f"deb [signed-by={keyring_file}] {repo_url} antigravity-debian main"
+    conn.run(f"echo '{sources_content}' > /tmp/antigravity.list")
+    conn.sudo(f"mv /tmp/antigravity.list {sources_list}")
+    # endregion
+    
+    # region 安装 antigravity
+    print("正在更新软件包索引...")
+    conn.sudo("apt-get update", hide=True)
+    
+    print("正在安装 antigravity...")
+    result = conn.sudo("apt-get install -y antigravity", hide=True, warn=True)
+    
+    if result.ok:
+        print("antigravity 安装完成")
+    else:
+        print("antigravity 安装失败，请检查错误信息")
+    # endregion
+
+
+@task
+def install_antigravity_local(conn: Connection) -> None:
+    """在 Ubuntu 上安装 antigravity (本地下载模式)
+    
+    先将安装包下载到本地 artifacts 目录，然后上传到服务器并安装。
+    适用于本地网络条件较好或远程主机无法访问外网的场景。
+    """
+    # region 检查系统类型
+    dist_info: dict = conn.hostvars["distribution"]
+    dist_id: str = dist_info.get("id", "").lower()
+    
+    if dist_id not in ("ubuntu", "debian"):
+        print(f"不支持的发行版: {dist_id}，仅支持 Ubuntu 和 Debian")
+        return
+    # endregion
+    
+    arch = _get_remote_arch(conn)
+    print(f"[INFO] 目标架构: {arch}")
+    
+    remote_tmp_dir = "/tmp/antigravity_install"
+    remote_deb = f"{remote_tmp_dir}/antigravity.deb"
+    
+    # region 下载并上传安装包
+    print("\n[任务] 下载 antigravity 安装包")
+    deb_path = _download_antigravity_deb(arch, conn.artifacts_dir)
+    
+    print("\n[任务] 上传安装包到远程主机")
+    conn.run(f"mkdir -p {remote_tmp_dir}", env=ENV_LANG_C)
+    conn.put(deb_path, remote_deb)
+    print(f"  上传完成: {remote_deb}")
+    # endregion
+    
+    # region 安装 antigravity
+    print("\n[任务] 安装 antigravity")
+    conn.sudo("apt-get update", env=ENV_LANG_C, hide=True)
+    result = conn.sudo(
+        f"DEBIAN_FRONTEND=noninteractive apt-get install -y {remote_deb}",
+        env=ENV_LANG_C,
+        warn=True
+    )
+    
+    if result.ok:
+        print("  antigravity 已安装")
+    else:
+        print("  antigravity 安装失败，请检查错误信息")
+        return
+    # endregion
+    
+    # region 清理临时文件
+    print("\n[任务] 清理临时文件")
+    conn.run(f"rm -rf {remote_tmp_dir}", env=ENV_LANG_C)
+    print("  清理完成")
+    # endregion
+    
+    print("\n[OK] antigravity 安装完成 (本地下载模式)")
+
+# endregion