mac_memory_optimizer.sh

#!/bin/bash
#==============================================================================
#  Apple Silicon Mac 内存优化脚本 v3.0
#  适用：macOS 14+ (M1/M2/M3/M4/M5 全系列)
#  功能：智能清理非必要进程 + 释放内存 + 缓存管理
#  用法：chmod +x mac_memory_optimizer.sh && ./mac_memory_optimizer.sh
#  恢复：./mac_memory_optimizer.sh --restore-apps
#  更新：2026-07-08 — 重写，修复原版多个安全问题
#==============================================================================
set -euo pipefail

RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'
BLUE='\033[0;34m'; CYAN='\033[0;36m'; MAGENTA='\033[0;35m'; NC='\033[0m'

SCRIPT_VERSION="3.0"
DRY_RUN=false
AGGRESSIVE=false
SKIP_CACHE=false
RESTORE_APPS=false
VERIFY_ONLY=false

# 解析参数
for arg in "$@"; do
    case $arg in
        --dry-run)      DRY_RUN=true ;;
        --aggressive)   AGGRESSIVE=true ;;
        --skip-cache)   SKIP_CACHE=true ;;
        --restore-apps) RESTORE_APPS=true ;;
        --verify)       VERIFY_ONLY=true ;;
        --help|-h)
            echo "用法: $0 [选项]"
            echo ""
            echo "模式："
            echo "  (无参数)      标准优化：关闭非白名单应用 + 清理缓存 + 释放内存"
            echo "  --dry-run     仅预览将要关闭的应用，不实际执行"
            echo "  --aggressive  激进模式：同时关闭后台非必要进程（慎用）"
            echo "  --skip-cache  跳过缓存清理，仅关闭应用和释放内存"
            echo "  --restore-apps 恢复上次被关闭的应用列表"
            echo "  --verify      仅检查当前内存状态，不做任何修改"
            echo "  --help        显示此帮助"
            echo ""
            echo "版本：v${SCRIPT_VERSION} — 适配 Apple Silicon 全系列"
            exit 0 ;;
        *)
            echo -e "${RED}未知参数: $arg${NC}"; exit 1 ;;
    esac
done

#==============================================================================
# 系统信息
#==============================================================================
OS_VER=$(sw_vers -productVersion 2>/dev/null || echo "0")
CHIP=$(sysctl -n machdep.cpu.brand_string 2>/dev/null | grep -q "Apple" && echo "Apple Silicon" || echo "Intel")
MEM_TOTAL_KB=$(sysctl -n hw.memsize 2>/dev/null | awk '{print $1/1024/1024}')  # MB
MEM_TOTAL_GB=$(echo "scale=1; $MEM_TOTAL_KB/1024" | bc 2>/dev/null || echo "?")
BACKUP_DIR="${HOME}/.mac_memory_optimizer_backup"
RESTORE_FILE="${BACKUP_DIR}/last_closed_apps.txt"

echo -e "${BLUE}========================================"
echo "  Apple Silicon Mac 内存优化脚本 v${SCRIPT_VERSION}"
echo -e "========================================${NC}"
echo -e "  系统: macOS ${OS_VER} | 芯片: ${CHIP} | 内存: ${MEM_TOTAL_GB} GB"
echo ""

#==============================================================================
# 白名单配置
#==============================================================================

# 用户保护应用（不会被关闭）
USER_WHITELIST=(
    "WeChat"                    # 微信
    "TencentLemonLite"          # 腾讯柠檬清理
    "Tencent Lemon Lite"        # 腾讯柠檬清理（别名）
    "Terminal"                  # 终端
    "iTerm2"                    # iTerm2
    "Alacritty"                 # Alacritty
    "kitty"                     # Kitty
    "Warp"                      # Warp Terminal
    "Visual Studio Code"        # VSCode
    "Code"                      # VSCode (Insiders)
    "Xcode"                     # Xcode
    "Activity Monitor"          # 活动监视器
    "System Settings"           # 系统设置
    "System Preferences"        # 系统偏好设置（旧版）
    "Finder"                    # 访达
    "Safari"                    # Safari
    "Google Chrome"             # Chrome
    "Firefox"                   # Firefox
)

# 系统核心进程（绝对不能 kill）
SYSTEM_CORE=(
    "kernel_task" "launchd" "WindowServer" "Dock" "Finder"
    "loginwindow" "SystemUIServer" "ControlCenter" "NotificationCenter"
    "coreaudiod" "bluetoothd" "bluetoothaudiod" "mDNSResponder"
    "networkd" "configd" "powerd" "thermald" "thermalmonitord"
    "securityd" "trustd" "authd" "TCC" "keychain"
    "mds" "mds_stores" "Spotlight" "fseventsd"
    "cfprefsd" "distnoted" "notifyd" "syslogd" "logd"
    "opendirectoryd" "diskarbitrationd" "UserEventAgent"
    "hidd" "usbmuxd" "AppleMobileFileIntegrity"
    "sessionlogoutd" "xpcproxy" "backgroundtaskmanagementd"
    "sharingd" "rapportd" "swcd" "wirelessproxd"
)

# 激进模式额外关闭（非必要后台）
AGGRESSIVE_TARGETS=(
    "soagent" "callservicesd" "imagent" "imagents"
    "identityservicesd" "bird" "cloudd" "nsurlsessiond"
    "accountsd" "amsaccountsd" "akd" "AOSHeartbeat"
    "com.apple.geod" "com.apple.parsec-fbf" "parsecd"
    "suggestd" "knowledge-agent" "knowledge-construction"
    "assistantd" "Siri" "siriknowledged"
    "photoanalysisd" "mediaanalysisd" "photolibraryd"
    "gamed" "tipsd" "helpd" "storeuid"
    "com.apple.WebKit.Networking" "com.apple.WebKit.WebContent"
    "AMPDevicesAgent" "AMPLibraryAgent" "AMPArtworkAgent"
    "commerce" "appstoreagent" "softwareupdated"
    "analyticsd" "metrickitd" "powerlogHelperd"
    "ReportCrash" "spindump" "CrashReporterSupportHelper"
    "screentime" "UsageTrackingAgent" "parentalcontrols"
    "com.apple.MobileSoftwareUpdate" "com.apple.OTATaskingAgent"
    "com.apple.backgroundassets.user"
)

# 公用函数：检查是否在白名单
is_whitelisted() {
    local name="$1"; shift
    local -a list=("$@")
    for item in "${list[@]}"; do
        [[ "$name" == "$item" ]] && return 0
    done
    return 1
}

#==============================================================================
# 验证模式
#==============================================================================
if [ "$VERIFY_ONLY" = true ]; then
    echo -e "${MAGENTA}[验证模式] 当前系统内存状态${NC}\n"

    # 内存压力
    PRESSURE=$(sysctl -n kern.memorystatus_vm_pressure_level 2>/dev/null || echo "?")
    case "$PRESSURE" in
        1) PRESS_STR="🟢 正常" ;;
        2) PRESS_STR="🟡 警告" ;;
        4) PRESS_STR="🔴 严重" ;;
        *) PRESS_STR="⚪ 未知" ;;
    esac

    # 内存统计
    VM_STAT=$(vm_stat)
    PAGE_SIZE=16384  # Apple Silicon
    FREE=$(echo "$VM_STAT" | grep "Pages free" | awk '{print $3}' | tr -d '.')
    ACTIVE=$(echo "$VM_STAT" | grep "Pages active" | awk '{print $3}' | tr -d '.')
    INACTIVE=$(echo "$VM_STAT" | grep "Pages inactive" | awk '{print $3}' | tr -d '.')
    WIRED=$(echo "$VM_STAT" | grep "Pages wired down" | awk '{print $4}' | tr -d '.')
    COMPRESSED=$(echo "$VM_STAT" | grep "Pages occupied by compressor" | awk '{print $5}' | tr -d '.')
    SPECULATIVE=$(echo "$VM_STAT" | grep "Pages speculative" | awk '{print $3}' | tr -d '.')

    FREE_MB=$(( FREE * PAGE_SIZE / 1048576 ))
    ACTIVE_MB=$(( ACTIVE * PAGE_SIZE / 1048576 ))
    INACTIVE_MB=$(( INACTIVE * PAGE_SIZE / 1048576 ))
    WIRED_MB=$(( WIRED * PAGE_SIZE / 1048576 ))
    COMPRESSED_MB=$(( COMPRESSED * PAGE_SIZE / 1048576 ))
    USED_MB=$(( ACTIVE_MB + WIRED_MB + COMPRESSED_MB ))
    AVAIL_MB=$(( FREE_MB + INACTIVE_MB + SPECULATIVE * PAGE_SIZE / 1048576 ))

    echo -e "  内存压力: ${PRESS_STR}"
    echo -e "  总内存:   ${MEM_TOTAL_KB} MB"
    echo -e "  已使用:   ${USED_MB} MB (活跃+固定+压缩)"
    echo -e "  可用:     ${AVAIL_MB} MB"
    echo -e "  压缩:     ${COMPRESSED_MB} MB"
    echo ""

    # 进程统计
    echo -e "${CYAN}当前运行应用:${NC}"
    IFS=',' read -ra APP_LIST <<< "$(osascript -e 'tell application "System Events" to get name of every application process where visible is true' 2>/dev/null || echo '')"
    for a in "${APP_LIST[@]}"; do
        a=$(echo "$a" | xargs)
        [ -z "$a" ] && continue
        is_whitelisted "$a" "${USER_WHITELIST[@]}" && echo -e "  ${GREEN}✓${NC} $a" || echo -e "  ${YELLOW}○${NC} $a"
    done
    echo ""
    echo -e "${BLUE}验证完毕${NC}"
    exit 0
fi

#==============================================================================
# 恢复模式
#==============================================================================
if [ "$RESTORE_APPS" = true ]; then
    echo -e "${YELLOW}[恢复模式] 重新打开上次关闭的应用...${NC}\n"
    if [ -f "$RESTORE_FILE" ]; then
        while IFS= read -r app_path; do
            [ -z "$app_path" ] && continue
            echo -e "  ${GREEN}→ 打开:${NC} $(basename "$app_path")"
            [ "$DRY_RUN" = false ] && open "$app_path" 2>/dev/null || true
        done < "$RESTORE_FILE"
        echo -e "\n${GREEN}✅ 恢复完毕${NC}"
        [ "$DRY_RUN" = false ] && rm -f "$RESTORE_FILE"
    else
        echo -e "  ${YELLOW}未找到上次关闭的应用记录${NC}"
    fi
    exit 0
fi

#==============================================================================
# 主逻辑
#==============================================================================

echo -e "${YELLOW}模式: $( [ "$DRY_RUN" = true ] && echo '预览' || echo '执行' ) | $( [ "$AGGRESSIVE" = true ] && echo '激进' || echo '标准' ) | $( [ "$SKIP_CACHE" = true ] && echo '跳过缓存' || echo '含缓存清理' )${NC}"
echo ""

# --- 记录优化前内存 ---
BEFORE_FREE=$(vm_stat | grep "Pages free" | awk '{print $3}' | tr -d '.')
BEFORE_INACTIVE=$(vm_stat | grep "Pages inactive" | awk '{print $3}' | tr -d '.')
BEFORE_AVAIL=$(( (BEFORE_FREE + BEFORE_INACTIVE) * 16384 / 1048576 ))

# --- Step 1: 关闭非白名单应用 ---
echo -e "${CYAN}========================================="
echo " [1] 检查并关闭非白名单应用"
echo -e "=========================================${NC}"

CLOSED_APPS=()
KEPT_APPS=()
mkdir -p "$BACKUP_DIR"

# 获取所有运行的应用（含后台）
RAW_APPS=$(osascript -e 'tell application "System Events" to get name of every application process' 2>/dev/null || echo '')
IFS=',' read -ra ALL_APPS <<< "$RAW_APPS"

for app in "${ALL_APPS[@]}"; do
    app=$(echo "$app" | xargs)
    [ -z "$app" ] && continue

    if is_whitelisted "$app" "${USER_WHITELIST[@]}"; then
        echo -e "  ${GREEN}[保留]${NC} $app"
        KEPT_APPS+=("$app")
        continue
    fi

    # 获取应用路径用于恢复
    APP_PATH=$(mdfind "kMDItemKind == 'Application' && kMDItemDisplayName == '$app'" 2>/dev/null | head -1 || echo "")

    echo -e "  ${RED}[关闭]${NC} $app"
    if [ "$DRY_RUN" = false ]; then
        # 使用 SIGTERM 先友好退出，失败再用 SIGKILL
        killall "$app" 2>/dev/null || killall -9 "$app" 2>/dev/null || true
        sleep 0.1
        [ -n "$APP_PATH" ] && echo "$APP_PATH" >> "$RESTORE_FILE"
    fi
    CLOSED_APPS+=("$app")
done

echo -e "  已关闭: ${#CLOSED_APPS[@]} | 已保留: ${#KEPT_APPS[@]}"

# --- Step 2: 激进模式额外清理 ---
if [ "$AGGRESSIVE" = true ]; then
    echo ""
    echo -e "${CYAN}========================================="
    echo " [2] 激进模式：清理后台非必要进程"
    echo -e "=========================================${NC}"

    AGGRESSIVE_CLOSED=0
    AGGRESSIVE_KEPT=0

    while IFS= read -r line; do
        [[ "$line" != *"$(whoami)"* ]] && continue
        PID=$(echo "$line" | awk '{print $2}')
        PROC_NAME=$(echo "$line" | awk '{for(i=11;i<=NF;i++) printf "%s ", $i; print ""}' | xargs basename 2>/dev/null)
        [ -z "$PROC_NAME" ] || [ -z "$PID" ] || [ "$PID" = "$$" ] && continue

        if is_whitelisted "$PROC_NAME" "${SYSTEM_CORE[@]}" "${USER_WHITELIST[@]}"; then
            ((AGGRESSIVE_KEPT++)) || true
            continue
        fi

        if is_whitelisted "$PROC_NAME" "${AGGRESSIVE_TARGETS[@]}"; then
            echo -e "  ${RED}[关闭]${NC} $PROC_NAME (PID:$PID)"
            [ "$DRY_RUN" = false ] && kill -15 "$PID" 2>/dev/null || kill -9 "$PID" 2>/dev/null || true
            ((AGGRESSIVE_CLOSED++)) || true
        fi
    done < <(ps aux)

    echo -e "  激进关闭: ${AGGRESSIVE_CLOSED} | 保留: ${AGGRESSIVE_KEPT}"
fi

# --- Step 3: 清理缓存 ---
if [ "$SKIP_CACHE" = false ]; then
    echo ""
    echo -e "${CYAN}========================================="
    echo " [3] 选择性清理缓存"
    echo -e "=========================================${NC}"

    CACHE_CLEANED=0

    if [ "$DRY_RUN" = false ]; then
        # 只清理安全的缓存目录，不碰应用数据
        CACHE_DIRS=(
            "$HOME/Library/Caches/com.apple.Safari"
            "$HOME/Library/Caches/com.apple.helpd"
            "$HOME/Library/Caches/com.apple.appstore"
            "$HOME/Library/Caches/com.apple.parsecd"
            "$HOME/Library/Caches/com.apple.suggestd"
            "$HOME/Library/Caches/com.apple.geod"
            "$HOME/Library/Caches/com.apple.weatherd"
            "$HOME/Library/Caches/com.apple.stocks"
            "$HOME/Library/Caches/com.apple.gamed"
            "$HOME/Library/Caches/com.apple.assistant"
            "$HOME/Library/Caches/com.apple.siri"
            "$HOME/Library/Caches/CloudKit"
            "$HOME/Library/Caches/com.apple.nsservicescache"
            "$HOME/Library/Caches/com.apple.quicklook.ThumbnailsAgent"
            "$HOME/Library/Caches/com.apple.Safari.SafeBrowsing"
            "/tmp/com.apple.*"
        )

        for d in "${CACHE_DIRS[@]}"; do
            # 展开通配符
            for real_d in $d; do
                if [ -d "$real_d" ] || [ -f "$real_d" ]; then
                    SIZE=$(du -sh "$real_d" 2>/dev/null | awk '{print $1}' || echo "?")
                    rm -rf "$real_d" 2>/dev/null && {
                        echo -e "  ${GREEN}✓${NC} 清理: $(basename "$real_d") (${SIZE})"
                        ((CACHE_CLEANED++)) || true
                    }
                fi
            done
        done
    else
        echo -e "  ${YELLOW}(预览模式) 将清理以下缓存:${NC}"
        for d in "${CACHE_DIRS[@]}"; do
            for real_d in $d; do
                [ -d "$real_d" ] || [ -f "$real_d" ] && echo -e "  → $real_d"
            done
        done
    fi

    echo -e "  已清理缓存项: ${CACHE_CLEANED}"
fi

# --- Step 4: 释放内存 ---
echo ""
echo -e "${CYAN}========================================="
echo " [4] 释放系统内存"
echo -e "=========================================${NC}"

if [ "$DRY_RUN" = false ]; then
    sudo purge 2>/dev/null && echo -e "  ${GREEN}✓${NC} sudo purge 已执行" || \
        echo -e "  ${YELLOW}⚠${NC} sudo purge 失败（可能需要输入密码）"
else
    echo -e "  ${YELLOW}(预览模式) 将执行: sudo purge${NC}"
fi

# --- 汇总 ---
AFTER_FREE=$(vm_stat 2>/dev/null | grep "Pages free" | awk '{print $3}' | tr -d '.')
AFTER_INACTIVE=$(vm_stat 2>/dev/null | grep "Pages inactive" | awk '{print $3}' | tr -d '.')
AFTER_AVAIL=$(( (AFTER_FREE + AFTER_INACTIVE) * 16384 / 1048576 ))
FREED_MB=$(( AFTER_AVAIL - BEFORE_AVAIL ))
[ "$FREED_MB" -lt 0 ] && FREED_MB=0

echo ""
echo -e "${BLUE}========================================"
echo "  ✅ 优化完成"
echo -e "========================================${NC}"
echo ""
echo -e "  📊 汇总："
echo -e "     关闭应用: ${#CLOSED_APPS[@]} 个"
echo -e "     保留应用: ${#KEPT_APPS[@]} 个"
echo -e "     优化前可用: ${BEFORE_AVAIL} MB"
echo -e "     优化后可用: ${AFTER_AVAIL} MB"
echo -e "     释放内存:   ${FREED_MB} MB"
echo ""
echo -e "  💡 提示："
echo -e "     • 恢复被关闭的应用: $0 --restore-apps"
echo -e "     • 预览模式: $0 --dry-run"
echo -e "     • 激进模式: $0 --aggressive"
echo -e "     • 仅查看状态: $0 --verify"
echo ""
echo -e "  📋 当前运行应用："
CURRENT_APPS=$(osascript -e 'tell application "System Events" to get name of every application process where visible is true' 2>/dev/null || echo '')
if [ -n "$CURRENT_APPS" ]; then
    echo "$CURRENT_APPS" | tr ',' '\n' | while read a; do
        a=$(echo "$a" | xargs)
        [ -z "$a" ] && continue
        echo -e "    • $a"
    done
fi
echo ""