shell-kit/setup-mysql-cli.sh

#!/bin/bash
# ============================================
# MySQL CLI 工具安装脚本（官方 MySQL 客户端）
# 从 Oracle MySQL 官方仓库安装真正的 MySQL 客户端
# 替代 Debian Trixie 默认的 mariadb-client
# ============================================

set -e

MYSQL_VERSION="8.4.4"
BIN_INSTALL_DIR="/usr/local/bin"

# 颜色定义
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
NC='\033[0m'

info()  { echo -e "${GREEN}[info]${NC} $1"; }
warn()  { echo -e "${YELLOW}[warn]${NC} $1"; }
error() { echo -e "${RED}[error]${NC} $1"; }

# ------------------------------------------
# 检查是否已安装正确的 MySQL 客户端
# ------------------------------------------
check_mysql_client() {
  if command -v mysql >/dev/null 2>&1; then
    local ver
    ver=$(mysql --version 2>&1)

    if echo "$ver" | grep -qi "mariadb"; then
      warn "当前是 MariaDB 客户端: $ver"
      warn "（MariaDB 用 --ssl=0，MySQL 用 --ssl-mode=DISABLED）"
      warn "将替换为官方 MySQL 客户端..."
      return 1
    fi

    if mysql --help 2>&1 | grep -q "ssl-mode"; then
      info "✅ 官方 MySQL 客户端已安装: $ver"
      return 0
    else
      warn "MySQL 客户端版本过旧，不支持 --ssl-mode"
      return 1
    fi
  fi
  return 1
}

# ------------------------------------------
# 清理磁盘空间
# ------------------------------------------
cleanup_space() {
  local needed_kb=${1:-51200}  # 默认需要 50MB

  local avail_kb
  avail_kb=$(df /var/cache/apt/archives/ 2>/dev/null | awk 'NR==2 {print $4}')  # 单位 KB

  if [ -z "$avail_kb" ] || [ "$avail_kb" -lt "$needed_kb" ]; then
    warn "磁盘空间不足，尝试清理..."
    apt-get clean 2>/dev/null || true
    rm -rf /tmp/* /var/tmp/* 2>/dev/null || true

    avail_kb=$(df /var/cache/apt/archives/ 2>/dev/null | awk 'NR==2 {print $4}')
    if [ -n "$avail_kb" ] && [ "$avail_kb" -lt "$needed_kb" ]; then
      warn "清理后空间仍不足 ($((avail_kb / 1024)) MB 可用)"
      return 1
    fi
  fi
  return 0
}

# ------------------------------------------
# 方式一：从 MySQL 官方 APT 仓库安装
# ------------------------------------------
install_from_apt() {
  info "方式一：从 MySQL 官方 APT 仓库安装..."

  # 检查磁盘空间
  if ! cleanup_space; then
    warn "磁盘空间不足，跳过 APT 安装方式"
    return 1
  fi

  # 确保 wget 可用
  if ! command -v wget >/dev/null 2>&1; then
    apt-get update -qq && apt-get install -y -qq wget 2>/dev/null || {
      warn "wget 不可用，跳过 APT 安装方式"
      return 1
    }
  fi

  local mysql_repo_deb="/tmp/mysql-apt-config.deb"

  # 清理旧配置
  rm -f /etc/apt/sources.list.d/mysql.list /etc/apt/sources.list.d/mysql-*.list

  # 下载 MySQL APT 配置包
  info "下载 MySQL APT 配置包..."
  wget -q "https://repo.mysql.com/mysql-apt-config_0.8.33-1_all.deb" -O "$mysql_repo_deb" || {
    warn "下载 MySQL APT 配置包失败"
    return 1
  }

  # 非交互式安装配置包
  DEBIAN_FRONTEND=noninteractive dpkg -i "$mysql_repo_deb" 2>/dev/null || true

  # 修正 sources.list 中的 codename
  # MySQL APT 仓库不支持 Debian Trixie，强制使用 bookworm
  if [ -f /etc/apt/sources.list.d/mysql.list ]; then
    sed -i "s/$(lsb_release -sc 2>/dev/null || echo 'unknown')/bookworm/g" \
      /etc/apt/sources.list.d/mysql.list 2>/dev/null || true
  fi

  # 更新 APT 缓存
  info "更新 APT 缓存..."
  apt-get update -qq 2>/dev/null || {
    warn "APT 更新失败"
    return 1
  }

  # 安装 MySQL 客户端（优先 mysql-client，备用 mysql-community-client）
  info "安装 mysql-client..."
  apt-get install -y -qq mysql-client 2>/dev/null || {
    warn "mysql-client 安装失败，尝试 mysql-community-client..."
    apt-get install -y -qq mysql-community-client 2>/dev/null || {
      warn "APT 安装 MySQL 客户端失败"
      return 1
    }
  }

  return 0
}

# ------------------------------------------
# 方式二：直接下载 MySQL 客户端二进制
# ------------------------------------------
install_from_binary() {
  info "方式二：直接下载 MySQL 客户端二进制..."

  # 检查磁盘空间（二进制包大约 300MB 解压，但只需要 2 个文件 ~5MB）
  local needed_kb=102400  # 100MB for download + extract
  local avail_kb
  avail_kb=$(df /tmp 2>/dev/null | awk 'NR==2 {print $4}')
  if [ -n "$avail_kb" ] && [ "$avail_kb" -lt "$needed_kb" ]; then
    error "磁盘空间不足 ($((avail_kb / 1024)) MB 可用，需要 ~100 MB)"
    return 1
  fi

  # 确保 wget 可用
  if ! command -v wget >/dev/null 2>&1; then
    error "wget 不可用，无法下载"
    return 1
  fi

  local tmp_dir
  tmp_dir=$(mktemp -d)
  local tarball="${tmp_dir}/mysql-${MYSQL_VERSION}-linux-glibc2.28-x86_64.tar.xz"
  local base_url="https://dev.mysql.com/get/Downloads/MySQL-8.4"

  info "下载 MySQL ${MYSQL_VERSION} 客户端 (约 300MB)..."
  wget -q --show-progress "${base_url}/$(basename "$tarball")" -O "$tarball" || {
    error "下载失败"
    rm -rf "$tmp_dir"
    return 1
  }

  # 确保 xz 解压工具可用（.tar.xz 格式需要）
  if ! command -v xz >/dev/null 2>&1; then
    warn "xz 未安装，尝试安装 xz-utils..."
    apt-get install -y -qq xz-utils 2>/dev/null || {
      warn "apt 安装 xz-utils 失败，改用 Python lzma 模块解压..."
      # Python 内置了 lzma 模块，无需额外安装
    }
  fi

  info "解压（仅提取 mysql 和 mysqladmin）..."
  local mysql_bin mysqladmin_bin
  local extract_dir="${tmp_dir}/extract"
  mkdir -p "$extract_dir"

  if command -v xz >/dev/null 2>&1; then
    # 有 xz：直接用 tar 解压
    mysql_bin=$(tar -tJf "$tarball" 2>/dev/null | grep -E 'bin/mysql$' | head -1)
    mysqladmin_bin=$(tar -tJf "$tarball" 2>/dev/null | grep -E 'bin/mysqladmin$' | head -1)
    if [ -n "$mysql_bin" ] && [ -n "$mysqladmin_bin" ]; then
      tar -xJf "$tarball" -C "$extract_dir" "$mysql_bin" "$mysqladmin_bin"
    else
      tar -xJf "$tarball" -C "$extract_dir"
    fi
  else
    # 没有 xz：用 Python lzma 模块解压
    info "使用 Python lzma 模块解压..."
    python3 -c "
import lzma, tarfile, sys, os

tarball = '$tarball'
extract_to = '$extract_dir'

with lzma.open(tarball) as f:
    with tarfile.open(fileobj=f, mode='r|') as tar:
        for member in tar:
            # 只提取 bin/mysql 和 bin/mysqladmin
            if member.name.endswith('/bin/mysql') or member.name.endswith('/bin/mysqladmin'):
                tar.extract(member, path=extract_to)
" 2>/dev/null || {
      # 如果 Python 也失败，解压全部
      info "Python 选择性提取失败，解压全部文件..."
      python3 -c "
import lzma, tarfile

tarball = '$tarball'
extract_to = '$extract_dir'

with lzma.open(tarball) as f:
    with tarfile.open(fileobj=f, mode='r|') as tar:
        tar.extractall(path=extract_to)
" 2>/dev/null || {
        error "Python 解压也失败"
        rm -rf "$tmp_dir"
        return 1
      }
    }
  fi

  # 从解压目录复制二进制文件
  local mysql_src mysqladmin_src
  mysql_src=$(find "$extract_dir" -name 'mysql' -type f -path '*/bin/mysql' | head -1)
  mysqladmin_src=$(find "$extract_dir" -name 'mysqladmin' -type f -path '*/bin/mysqladmin' | head -1)

  if [ -n "$mysql_src" ] && [ -n "$mysqladmin_src" ]; then
    cp "$mysql_src" "${BIN_INSTALL_DIR}/mysql"
    cp "$mysqladmin_src" "${BIN_INSTALL_DIR}/mysqladmin"
  else
    # 最后尝试：直接找任何 mysql 二进制
    mysql_src=$(find "$extract_dir" -name 'mysql' -type f | head -1)
    mysqladmin_src=$(find "$extract_dir" -name 'mysqladmin' -type f | head -1)
    if [ -z "$mysql_src" ]; then
      error "解压后找不到 mysql 二进制文件"
      rm -rf "$tmp_dir"
      return 1
    fi
    cp "$mysql_src" "${BIN_INSTALL_DIR}/mysql"
    [ -n "$mysqladmin_src" ] && cp "$mysqladmin_src" "${BIN_INSTALL_DIR}/mysqladmin"
  fi

  chmod +x "${BIN_INSTALL_DIR}/mysql" "${BIN_INSTALL_DIR}/mysqladmin"
  rm -rf "$tmp_dir"

  info "二进制安装完成"
  return 0
}

# ------------------------------------------
# 方式三：提示手动安装
# ------------------------------------------
manual_install_hint() {
  echo ""
  warn "============================================"
  warn "  自动安装均失败，请手动下载安装："
  warn "============================================"
  echo ""
  echo "  1. 进入容器:   docker compose exec dev bash"
  echo ""
  echo "  2. 下载二进制:"
  echo "     wget https://dev.mysql.com/get/Downloads/MySQL-8.4/mysql-${MYSQL_VERSION}-linux-glibc2.28-x86_64.tar.xz"
  echo ""
  echo "  3. 解压并复制:"
  echo "     tar -xf mysql-${MYSQL_VERSION}-linux-glibc2.28-x86_64.tar.xz \\"
  echo "       --strip-components 1 -C /usr/local/bin \\"
  echo "       mysql-${MYSQL_VERSION}-linux-glibc2.28-x86_64/bin/mysql \\"
  echo "       mysql-${MYSQL_VERSION}-linux-glibc2.28-x86_64/bin/mysqladmin"
  echo ""
  echo "  4. 验证:"
  echo "     mysql --version"
  echo "     # 应输出: mysql  Ver 8.4.4 for Linux on x86_64 ..."
  echo ""
  echo "  5. 连接测试:"
  echo "     mysql -h mysql -P 3306 -u root -proot --ssl-mode=DISABLED"
  echo ""
}

# ==========================================
# 主流程
# ==========================================
echo ""
echo "======================================"
echo "  MySQL CLI 安装 (官方客户端)"
echo "======================================"
echo ""

# 步骤 1：检查是否已安装正确的版本
if check_mysql_client; then
  exit 0
fi

# 步骤 2：清理旧版 MariaDB 客户端（确保后续 mysql 命令走新版本）
if dpkg -l mariadb-client 2>/dev/null | grep -q "^ii"; then
  info "移除旧版 MariaDB 客户端..."
  apt-get remove -y -qq mariadb-client mariadb-client-core 2>/dev/null || true
fi

# 步骤 3：优先从官方 APT 仓库安装
if install_from_apt; then
  info "✅ 方式一成功：从 APT 仓库安装"
else
  warn "方式一失败，尝试方式二（二进制下载）..."
  if install_from_binary; then
    info "✅ 方式二成功：从二进制下载安装"
  else
    error "❌ 所有安装方案均失败"
    manual_install_hint
    exit 1
  fi
fi

# 步骤 4：最终验证
echo ""
if command -v mysql >/dev/null 2>&1; then
  mysql_ver=$(mysql --version)
  info "✅ MySQL 客户端已就绪"
  info "📊 版本: $mysql_ver"
  echo ""
  echo "📋 使用说明:"
  echo "   连接命令: mysql -h <主机> -P <端口> -u <用户> -p<密码> --ssl-mode=DISABLED"
  echo "   Docker 内连接: mysql -h mysql -P 3306 -u root -proot --ssl-mode=DISABLED"
  echo ""
else
  error "❌ MySQL 客户端安装失败（未知错误）"
  exit 1
fi