Orion
2470bdc278
新增 brew-upgrade-manager.sh 脚本，实现 Homebrew 自动化升级及 Cask 更新。引入 Python PTY 伪终端处理机制，修复了自动化过程中因终端尺寸导致的 Ruby 渲染崩溃问题，并对部分项目文件进行了归类整理。
208 lines
7.0 KiB
Bash
208 lines
7.0 KiB
Bash
#!/usr/bin/env bash
|
||
# Homebrew 智能升级脚本（强制 Cask 更新增强版 v5.2 - 修复 PTY 终端尺寸导致 Ruby 渲染崩溃问题）
|
||
# ❗❗️❗️️使用前请搜索 SUDO_PWD 并添加自己的 root 密码！！！
|
||
|
||
# ================== 脚本环境设置 ==================
|
||
|
||
# set -e：当命令返回非零退出状态（表示失败）时，脚本会立即退出。
|
||
set -e
|
||
# set -o pipefail：在管道命令中，如果任何一个子命令失败，整个管道即为失败。
|
||
set -o pipefail
|
||
|
||
# --- 颜色定义 (自动检测终端是否支持) ---
|
||
if [ -t 1 ]; then
|
||
GREEN='\033[1;32m'
|
||
YELLOW='\033[1;33m'
|
||
BLUE='\033[1;34m'
|
||
CYAN='\033[1;36m'
|
||
NC='\033[0m'
|
||
else
|
||
GREEN=''
|
||
YELLOW=''
|
||
BLUE=''
|
||
CYAN=''
|
||
NC=''
|
||
fi
|
||
|
||
# --- 终端宽度和打印函数 ---
|
||
DEFAULT_FALLBACK_WIDTH="130"
|
||
TERMINAL_WIDTH_OVERRIDE=""
|
||
|
||
# 解析命令行参数
|
||
while [[ $# -gt 0 ]]; do
|
||
case "$1" in
|
||
--width)
|
||
if [[ -n "$2" && "$2" =~ ^[0-9]+$ ]]; then
|
||
TERMINAL_WIDTH_OVERRIDE="$2"
|
||
shift 2
|
||
else
|
||
echo -e "${YELLOW}Error: '--width' parameter requires a valid numeric value.${NC}"
|
||
exit 1
|
||
fi
|
||
;;
|
||
--width=*)
|
||
TERMINAL_WIDTH_OVERRIDE="${1#*=}"
|
||
if ! [[ "$TERMINAL_WIDTH_OVERRIDE" =~ ^[0-9]+$ ]]; then
|
||
echo -e "${YELLOW}Error: '--width' parameter requires a valid numeric value.${NC}"
|
||
exit 1
|
||
fi
|
||
shift
|
||
;;
|
||
*)
|
||
shift
|
||
;;
|
||
esac
|
||
done
|
||
|
||
# 确定最终的 TERMINAL_WIDTH
|
||
if [[ -n "$TERMINAL_WIDTH_OVERRIDE" ]]; then
|
||
TERMINAL_WIDTH="$TERMINAL_WIDTH_OVERRIDE"
|
||
elif [[ -n "$HB_TERMINAL_WIDTH" && "$HB_TERMINAL_WIDTH" =~ ^[0-9]+$ ]]; then
|
||
TERMINAL_WIDTH="$HB_TERMINAL_WIDTH"
|
||
elif command -v stty &>/dev/null && stty size &>/dev/null; then
|
||
TERMINAL_WIDTH=$(stty size 2>/dev/null | awk '{print $2}')
|
||
if [[ -z "$TERMINAL_WIDTH" || "$TERMINAL_WIDTH" -le 0 ]]; then
|
||
TERMINAL_WIDTH=$(tput cols 2>/dev/null || echo "$DEFAULT_FALLBACK_WIDTH")
|
||
fi
|
||
else
|
||
TERMINAL_WIDTH=$(tput cols 2>/dev/null || echo "$DEFAULT_FALLBACK_WIDTH")
|
||
fi
|
||
|
||
separator() { printf '=%.0s' $(seq 1 "$TERMINAL_WIDTH"); printf "\n"; }
|
||
print_header() { echo -e "${BLUE}$1${NC}"; }
|
||
|
||
# ================== 流程开始 ==================
|
||
separator
|
||
|
||
print_header "Step 1: Updating Homebrew repositories (brew update -v)"
|
||
brew update -v
|
||
separator
|
||
printf "\n"
|
||
|
||
separator
|
||
print_header "Step 2: Performing health check (brew doctor)"
|
||
if ! brew doctor; then
|
||
echo -e "${YELLOW}Warning: 'brew doctor' detected issues. Manual review and resolution are recommended.${NC}"
|
||
else
|
||
echo "Homebrew environment is in good health."
|
||
fi
|
||
separator
|
||
printf "\n"
|
||
|
||
separator
|
||
print_header "Step 3: Verifying brew-cu extension for GUI Apps"
|
||
if ! brew tap | grep -q "buo/cask-upgrade"; then
|
||
echo -e "${YELLOW}Extension 'brew-cu' not found. Installing now...${NC}"
|
||
brew tap buo/cask-upgrade
|
||
else
|
||
echo -e "${GREEN}Extension 'brew-cu' is already active.${NC}"
|
||
fi
|
||
separator
|
||
printf "\n"
|
||
|
||
separator
|
||
print_header "Step 4: Executing comprehensive upgrades (Formulae & Casks)"
|
||
echo -e "${NC}"
|
||
|
||
# 使用前请先修改这里的密码！！
|
||
SUDO_PWD="caiao1226"
|
||
|
||
# 1. 升级命令行工具 (Formulae)
|
||
echo -e "\n${CYAN}>>> [1/2] Upgrading CLI Formulae (brew upgrade --formula)...${NC}"
|
||
brew upgrade --formula
|
||
|
||
# 2. 升级图形界面软件 (Casks)
|
||
echo -e "\n${CYAN}>>> [2/2] Upgrading GUI Casks (brew cu -yaq)...${NC}"
|
||
|
||
# 强制注入环境变量，确保终端输出依然保留 ANSI 颜色格式
|
||
export HOMEBREW_COLOR=1
|
||
|
||
# --- 核心修复：环境变量宽度兜底 ---
|
||
# 强制将 Bash 计算好的真实终端宽度传递给底层，防止 Ruby 绘表时发生 negative argument 崩溃
|
||
export COLUMNS="$TERMINAL_WIDTH"
|
||
|
||
# 使用 Python 原生 PTY 自建极简轮询引擎
|
||
python3 -c '
|
||
import pty, os, sys, select, fcntl, termios, struct
|
||
|
||
# 将传入的明文密码转换为底层的原始字节流，并在末尾追加换行符模拟按下回车键
|
||
pwd = sys.argv[1].encode() + b"\n"
|
||
cmd = sys.argv[2:]
|
||
|
||
# 在操作系统底层分叉出一个携带完整伪终端(PTY)特性的子进程
|
||
pid, fd = pty.fork()
|
||
|
||
if pid == 0:
|
||
# --- 逻辑分支：子进程空间 ---
|
||
# 使用 execvp 顶替当前进程，正式开始执行 brew cu 命令
|
||
os.execvp(cmd[0], cmd)
|
||
else:
|
||
# --- 逻辑分支：父进程空间 (监控端) ---
|
||
|
||
# --- 核心修复：物理终端尺寸投影 ---
|
||
# 尝试捕获外层真实物理终端的行列尺寸，并硬塞进伪终端的文件描述符中
|
||
try:
|
||
# 打包一个空的 4 短整型结构体准备接收数据
|
||
s = struct.pack("HHHH", 0, 0, 0, 0)
|
||
# 通过 ioctl 系统调用，向操作系统的标准输出请求 TIOCGWINSZ (获取窗口大小)
|
||
winsize = fcntl.ioctl(sys.stdout.fileno(), termios.TIOCGWINSZ, s)
|
||
# 将获取到的真实尺寸，立刻通过 TIOCSWINSZ (设置窗口大小) 写入刚刚生成的伪终端 fd 中
|
||
fcntl.ioctl(fd, termios.TIOCSWINSZ, winsize)
|
||
except Exception:
|
||
# 如果当前环境不是交互式终端(如 crontab 后台运行)，捕获异常静默跳过，
|
||
# 此时外层 bash 注入的 export COLUMNS="$TERMINAL_WIDTH" 将作为完美兜底
|
||
pass
|
||
|
||
exit_code = 0
|
||
while True:
|
||
try:
|
||
# 开启 IO 多路复用轮询，设置 0.1 秒的极短超时时间防止物理死锁
|
||
rfds, _, _ = select.select([fd], [], [], 0.1)
|
||
|
||
# 如果伪终端有数据吐出
|
||
if fd in rfds:
|
||
# 以 8KB 为块，读取原始字节流 (Raw Bytes)
|
||
data = os.read(fd, 8192)
|
||
# 读到空字节代表通道已被操作系统关闭
|
||
if not data:
|
||
break
|
||
|
||
# 在纯净字节流中精准狙击 sudo 发出的密码请求信号
|
||
if b"assword:" in data or b"Password:" in data:
|
||
# 发现目标，将密码字节流强行注入到伪终端标准输入中
|
||
os.write(fd, pwd)
|
||
|
||
# 将读取到的原汁原味的字节流直接刷入真实屏幕，确保 Emoji 不乱码
|
||
os.write(sys.stdout.fileno(), data)
|
||
|
||
except OSError:
|
||
# 捕获对侧进程已死导致的 EIO 错误，安全打破循环
|
||
break
|
||
|
||
# 主动进程心跳侦测：非阻塞探查子进程生死
|
||
try:
|
||
wpid, wstatus = os.waitpid(pid, os.WNOHANG)
|
||
if wpid == pid:
|
||
if os.WIFEXITED(wstatus):
|
||
exit_code = os.WEXITSTATUS(wstatus)
|
||
else:
|
||
exit_code = 1
|
||
break
|
||
except ChildProcessError:
|
||
break
|
||
|
||
# 将真实状态码原样返回给外层的 shell
|
||
sys.exit(exit_code)
|
||
' "$SUDO_PWD" brew cu -yaq
|
||
|
||
separator
|
||
printf "\n"
|
||
|
||
separator
|
||
print_header "Step 5: Cleaning up old files and caches (brew cleanup --prune=all)"
|
||
brew cleanup --prune=all
|
||
separator
|
||
printf "\n"
|
||
|
||
echo -e "${GREEN}All operations completed!${NC}"
|
||
printf "\n" |