SHELL赋能：JumpServer自动化部署的灵动乐章

#!/usr/bin/env bash
#
# Jumpserver 安装脚本优化版
#

# 版本号
VERSION=v3.6.2
# 下载地址
DOWNLOAD_URL=https://github.com
# 默认安装目录，可通过环境变量 INSTALL_DIR 自定义
INSTALL_DIR=${INSTALL_DIR:-/opt}
# 下载重试次数
DOWNLOAD_RETRIES=3

# 日志打印函数
log() {
  echo -e "[`date '+%Y-%m-%d %H:%M:%S'`] $1"
}

# 安装必要软件包的函数
function install_soft() {
    package="$1"
    log "尝试安装依赖：$package"
    if command -v dnf > /dev/null; then
      dnf -q -y install "$package"
    elif command -v yum > /dev/null; then
      yum -q -y install "$package"
    elif command -v apt > /dev/null; then
      apt-get -qqy install "$package"
    elif command -v zypper > /dev/null; then
      zypper -q -n install "$package"
    elif command -v apk > /dev/null; then
      apk add -q "$package"
      # 安装gettext-dev和python3（如果未安装gettext）
      command -v gettext >/dev/null || apk add -q gettext-dev python3
    else
      echo -e "[\033[31m ERROR \033[0m] 未找到适用的包管理器，请先手动安装 $package"
      exit 1
    fi
}

# 准备安装：检查并安装所需工具
function prepare_install() {
  local deps=(curl wget tar iptables)
  for dep in "${deps[@]}"; do
    if ! command -v $dep &>/dev/null; then
      install_soft $dep
    else
      log "$dep 已安装"
    fi
  done
}

# 下载并解压安装包
function get_installer() {
  log "准备将安装脚本下载到 ${INSTALL_DIR}/jumpserver-installer-${VERSION}"
  cd "${INSTALL_DIR}" || { log "无法进入 ${INSTALL_DIR} 目录，退出。"; exit 1; }

  if [ ! -d "${INSTALL_DIR}/jumpserver-installer-${VERSION}" ]; then
    local tarball="jumpserver-installer-${VERSION}.tar.gz"
    local download_url="${DOWNLOAD_URL}/jumpserver/installer/releases/download/${VERSION}/${tarball}"
    
    # 下载重试逻辑
    local attempt=1
    while [ $attempt -le $DOWNLOAD_RETRIES ]; do
      log "下载尝试 $attempt/$DOWNLOAD_RETRIES：$download_url"
      if wget --tries=1 --timeout=60 -qO "$tarball" "$download_url"; then
        log "下载成功"
        break
      else
        log "下载失败，正在重试..."
        ((attempt++))
      fi
    done

    if [ $attempt -gt $DOWNLOAD_RETRIES ]; then
      rm -f "$tarball"
      echo -e "[\033[31m ERROR \033[0m] 下载 jumpserver-installer-${VERSION} 失败，退出。"
      exit 1
    fi

    # 解压文件
    if tar -xf "$tarball" -C "${INSTALL_DIR}"; then
      log "解压成功"
    else
      rm -rf "${INSTALL_DIR}/jumpserver-installer-${VERSION}"
      echo -e "[\033[31m ERROR \033[0m] 解压 jumpserver-installer-${VERSION} 失败，退出。"
      exit 1
    fi
    # 删除压缩包以节省空间
    rm -f "$tarball"
  else
    log "安装目录已存在，跳过下载和解压步骤"
  fi
}

# 配置安装程序并启动 Jumpserver
function config_installer() {
  local installer_path="${INSTALL_DIR}/jumpserver-installer-${VERSION}"
  cd "$installer_path" || { log "无法进入安装目录 $installer_path，退出。"; exit 1; }

  # 修改配置文件中的 VERSION 变量
  if sed -i "s/VERSION=.*/VERSION=${VERSION}/g" static.env; then
    log "配置文件 static.env 更新成功"
  else
    echo -e "[\033[31m ERROR \033[0m] 更新 static.env 失败，退出。"
    exit 1
  fi

  # 安装 Jumpserver
  if ./jmsctl.sh install; then
    log "Jumpserver 安装成功"
  else
    echo -e "[\033[31m ERROR \033[0m] Jumpserver 安装失败，退出。"
    exit 1
  fi

  # 启动 Jumpserver
  if ./jmsctl.sh start; then
    log "Jumpserver 启动成功"
  else
    echo -e "[\033[31m ERROR \033[0m] Jumpserver 启动失败，请检查日志。"
    exit 1
  fi
}

# 主函数，组织安装流程
function main(){
  # 检查操作系统是否为 macOS（Darwin）
  if [ "$(uname)" == "Darwin" ]; then
    echo -e "[\033[31m ERROR \033[0m] macOS 系统不支持该安装脚本，退出。"
    exit 1
  fi

  log "开始准备安装环境..."
  prepare_install

  log "开始下载并解压安装包..."
  get_installer

  log "开始配置并启动 Jumpserver..."
  config_installer

  log "安装和启动过程完成！"
}

# 执行主函数
main