install.sh

#!/bin/bash

# 检查是否有jq命令
if ! command -v jq &> /dev/null
then
  echo "❌ 请先安装jq命令！参考：https://command-not-found.com/jq"
  exit 1
fi

# 必须有unzip或者7z
if ! command -v unzip &> /dev/null && ! command -v 7z &> /dev/null
then
  echo "❌ 未找到unzip或7z命令，请先安装unzip或7z命令。"
  exit 1
fi

# 检查是否有docker命令
if ! command -v docker &> /dev/null
then
  echo "❌ 未找到docker命令，请先安装docker。"
  exit 1
fi

# 检查是否有docker-compose命令
if ! command -v docker-compose &> /dev/null
then
  echo "❌ 未找到docker-compose命令，请先安装docker-compose。"
  exit 1
fi

OS=$(uname)
FILE_INITIALIZED=".mdcx_initialized"

replace_in_file() {
  if [ "$OS" = 'Darwin' ]; then
    # for MacOS
    sed -i '' -r -e "$1" "$2"
  else
    # for Linux and Windows
    sed -i'' -r -e "$1" "$2"
  fi
}

# 发生错误时的退出处理
on_error() {
  local projectDir=$1
  
  echo ""
  # 询问是否删除目录
  read -p "❓ 是否删除项目目录 ${projectDir}？（y/n，默认为n）：" DELETE_DIR
  DELETE_DIR=${DELETE_DIR:-n}
  echo ""
  if [ "$DELETE_DIR" = "y" ]; then
    rm -rf "$projectDir"
    echo "🗑 已删除目录：${projectDir}"
  fi

  exit 1
}

# 询问用户选择的模版
echo "📖 下面请你回答几个问题，以完成MDCx Docker版的安装。"
echo ""
echo "❓ 请选择容器部署模版（输入数字进行选择）："
echo " 1) mdcx-builtin-gui-base      轻量版，内置编译版应用，通过网页使用"
echo " 2) mdcx-builtin-webtop-base   重量版，内置编译版应用，通过网页和远程桌面使用"
echo " 3) mdcx-src-gui-base          轻量版，自部署源码，通过网页使用"
echo " 4) mdcx-src-webtop-base       重量版，自部署源码，通过网页和远程桌面使用"

read -p "📌 请输入数字（1-4）: " TEMPLATE_NUM

case $TEMPLATE_NUM in
  1)
    TEMPLATE_NAME="mdcx-builtin-gui-base"
    ;;
  2)
    TEMPLATE_NAME="mdcx-builtin-webtop-base"
    ;;
  3)
    TEMPLATE_NAME="mdcx-src-gui-base"
    ;;
  4)
    TEMPLATE_NAME="mdcx-src-webtop-base"
    ;;
  *)
    echo "无效的输入！请输入数字（1-4）."
    exit 1
    ;;
esac

echo "📝 您选择的模版为：$TEMPLATE_NUM) $TEMPLATE_NAME"
echo ""

if [[ "$TEMPLATE_NAME" == *"gui-base"* ]]; then
  BASE=gui
else
  BASE=webtop
fi

if [[ "$TEMPLATE_NAME" == *"mdcx-src"* ]]; then
  TYPE=src
else
  TYPE=builtin
fi


#拼接模版文件下载链接
DOWNLOAD_URL="https://github.com/northsea4/mdcx-docker/releases/download/latest/template-$TEMPLATE_NAME.zip"

echo "🔗 模版文件下载链接：$DOWNLOAD_URL"
echo ""

echo "⏳ 正在下载模版文件，请稍候..."

# 下载zip文件并保存为随机文件名
# RANDOM_NAME=$(cat /dev/urandom | LC_ALL=C tr -dc 'a-zA-Z0-9-_' | fold -w 29 | sed 1q)
# fold命令在某些系统上不支持，使用head命令代替
RANDOM_NAME=$(cat /dev/urandom | LC_ALL=C tr -dc 'a-zA-Z0-9-_' | head -c 29)
if [ $? -ne 0 ]; then
  echo "❌ 生成随机文件名失败！"
  exit 1
fi

ZIP_FILE="${RANDOM_NAME}.zip"
curl "$DOWNLOAD_URL" -L --connect-timeout 30 --max-time 300 -o "$ZIP_FILE"

if [ $? -ne 0 ]; then
  echo "❌ 模版文件下载失败！"

  on_error "${DIR_FULL_PATH}"
fi

# 创建以文件名为名称的目录并解压zip文件
mkdir "$RANDOM_NAME"
# 如果有7z命令，则使用7z解压
if command -v 7z &> /dev/null; then
  7z x "$ZIP_FILE" -o"$RANDOM_NAME"
else
  unzip "$ZIP_FILE" -d "$RANDOM_NAME"
fi

if [ $? -ne 0 ]; then
  echo "❌ 模版文件解压失败！"

  on_error "${DIR_FULL_PATH}"
fi

echo "🎉 模版文件下载完成！"
echo ""


# 询问用户目录名称，默认为 `mdcx-docker`
echo "选择一个目录作为本docker项目的根目录(存放应用或容器的相关数据)，可以是目录路径或目录名称。"
read -p "❓ 请输入目录名称（默认为 mdcx-docker）：" DIR_NAME
DIR_NAME=${DIR_NAME:-mdcx-docker}

# 检查目录是否已存在
while [ -d "$DIR_NAME" ]; do
  read -p "❌ 目录已存在，请输入其他目录名称：" DIR_NAME
done

# 移动mdcx-docker模版目录并重命名为用户输入的目录名称
mv "$RANDOM_NAME/mdcx-docker" "$DIR_NAME"
# 删除临时目录和zip文件
rm -rf "$RANDOM_NAME"
rm "$ZIP_FILE"

# 进入用户输入的目录名称
cd "$DIR_NAME"
DIR_FULL_PATH=$(pwd)
echo "📁 已创建并进入目录：$(pwd)"

source .env

USER_ID=$(id -u)
GROUP_ID=$(id -g)
# 不同模版使用不同的环境变量名称
if [[ "$BASE" == "gui" ]]; then
  USER_ID_KEY="USER_ID"
  GROUP_ID_KEY="GROUP_ID"
else
  USER_ID_KEY="PUID"
  GROUP_ID_KEY="PGID"
fi

echo ""
echo "❓ 请输入${USER_ID_KEY}（容器使用的UID），默认为$(id -u)"
read -p "${USER_ID_KEY}: " USER_ID
USER_ID=${USER_ID:-$(id -u)}

echo ""
echo "❓ 请输入${GROUP_ID_KEY}（容器使用的GID），默认为$(id -g)"
read -p "${GROUP_ID_KEY}: " GROUP_ID
GROUP_ID=${GROUP_ID:-$(id -g)}


# 不同的模版使用不同的端口环境变量名称
if [[ "$BASE" == "gui" ]]; then
  echo ""
  echo "❓ 请输入WEB访问端口号， 默认为5800"
  read -p "WEB_PORT: " WEB_PORT
  WEB_PORT=${WEB_PORT:-5800}

  echo ""
  echo "❓ 请输入VNC端口号， 默认为5900"
  read -p "VNC_PORT: " VNC_PORT
  VNC_PORT=${VNC_PORT:-5900}
else
  echo ""
  echo "❓ 请输入WEB访问端口号， 默认为3000"
  read -p "WEB_PORT: " WEB_PORT
  WEB_PORT=${WEB_PORT:-3000}
  # echo "📝 你输入的WEB访问端口号为：$WEB_PORT"

  echo ""
  echo "❓ 请输入RDP访问端口号， 默认为3389"
  read -p "RDP_LISTEN_PORT: " RDP_LISTEN_PORT
  RDP_LISTEN_PORT=${RDP_LISTEN_PORT:-3389}
fi


echo ""
while true; do
    read -p "❓ 请输入需要映射的影片目录，格式为/path/to/movies:/movies，留空则跳过： " MOVIE_DIR
    if [ -z "$MOVIE_DIR" ]; then
        break
    elif ! echo "$MOVIE_DIR" | grep -qE '^[^:]+:[^:]+$'; then
        echo "❌ 错误：输入格式不正确，请按格式输入"
        continue
    fi
    VOLUMES="$VOLUMES\n      - $MOVIE_DIR"
done


# 展示用户所输入的信息，并询问确认信息是否正确
echo ""
echo "📝 您输入的信息如下："
echo "🔘 $USER_ID_KEY: $USER_ID"
echo "🔘 $GROUP_ID_KEY: $GROUP_ID"

# 根据不同的模版，展示不同的端口信息
if [[ "$BASE" == "gui" ]]; then
  echo "🔘 WEB_PORT: $WEB_PORT"
  echo "🔘 VNC_PORT: $VNC_PORT"
else
  echo "🔘 WEB_PORT: $WEB_PORT"
  echo "🔘 RDP_LISTEN_PORT: $RDP_LISTEN_PORT"
fi


if [ -z "$VOLUMES" ]; then
  echo "🔘 映射目录：没有指定"
else
  echo "🔘 映射目录："
  echo -e "${VOLUMES[*]}\n"
fi


echo ""
read -p "❓ 确认信息是否填写正确（yes/y确认，no/n退出）：" CONFIRMED

if [[ "$CONFIRMED" =~ ^[nN](o)?$ ]]; then
  echo "❗ 操作已取消"
  
  on_error "${DIR_FULL_PATH}"
fi


echo "⏳ 替换环境变量..."
# 根据不同的模版，替换不同的环境变量名称
if [[ "$BASE" == "gui" ]]; then
  replace_in_file "s/USER_ID=[0-9]+/USER_ID=$USER_ID/g" .env
  replace_in_file "s/GROUP_ID=[0-9]+/GROUP_ID=$GROUP_ID/g" .env
else
  replace_in_file "s/PUID=[0-9]+/PUID=$USER_ID/g" .env
  replace_in_file "s/PGID=[0-9]+/PGID=$GROUP_ID/g" .env
fi


# 根据不同的模版，替换不同的端口信息
if [[ "$BASE" == "gui" ]]; then
  replace_in_file "s/WEB_PORT=[0-9]+/WEB_PORT=$WEB_PORT/g" .env
  replace_in_file "s/VNC_PORT=[0-9]+/VNC_PORT=$VNC_PORT/g" .env
else
  replace_in_file "s/WEB_PORT=[0-9]+/WEB_PORT=$WEB_PORT/g" .env
  replace_in_file "s/RDP_LISTEN_PORT=[0-9]+/RDP_LISTEN_PORT=$RDP_LISTEN_PORT/g" .env
fi

echo "✅ 替换环境变量完成"

echo "⏳ 替换挂载卷..."
# $VOLUMES不为空时才进行替换
if [[ -n "$VOLUMES" ]]; then
  replace_in_file "s|# VOLUMES_REPLACEMENT|$VOLUMES|" docker-compose.yml
  echo "✅ 替换挂载卷完成"
else
  echo "❗ 你没有指定映射影片目录，你可以之后在docker-compose.yml中手动添加。"
fi

generate_app_version() {
  local published_at="$1"

  # 去除非数字字符
  published_at=$(echo "$published_at" | tr -dc '0-9')

  # 取前8位数字作为年月日，前缀为d
  echo "d${published_at:0:8}"
}

find_release_by_tag_name() {
  local repo=$1
  local target_tag_name=$2
  
  local url="https://api.github.com/repos/${repo}/releases"

  # echo "URL: $url"

  local target_release=""

  let found=false
  local page=1
  while true; do
    local response=$(curl -s "${url}?per_page=100&page=${page}")
    if [[ -z "$response" ]]; then
      break
    fi

    local releases=$(printf '%s' $response | jq -c '.[]')
    for release in $releases; do
      tag_name=$(printf '%s' $release | jq -r '.tag_name')
      if [[ "$tag_name" == "$target_tag_name" ]]; then
        found=true
        echo $release
        break
      fi
    done

    if [[ $found ]]; then
      break
    fi

    page=$((page + 1))
  done
}

# 获取指定仓库和tag_name的release，并解析得到release信息
# 返回json对象:
# {
#   "tag_name": "v1.0.0",
#   "published_at": "2022-01-01T00:00:00Z",
#   "release_version": "120220101",
#   "tar_url": "https://api.github.com/repos/sqzw-x/mdcx/tarball/daily_release",
#   "zip_url": "https://api.github.com/repos/sqzw-x/mdcx/zipball/daily_release"
# }
get_release_info() {
  local repo="$1"
  local tag_name="$2"

  # echo "⏳ 正在获取仓库 ${repo} 中 tag_name=${tag_name} 的release..."
  local release=$(find_release_by_tag_name "$repo" "$tag_name")

  if [[ -z "$release" ]]; then
    echo "❌ 找不到 tag_name=${tag_name} 的release！"
    return 1
  fi

  tag_name=$(printf '%s' $release | jq -r '.tag_name')
  if [[ -z "$tag_name" ]]; then
    echo "❌ 找不到 tag_name！"
    return 1
  fi

  published_at=$(printf '%s' $release | jq -r '.published_at')
  if [[ -z "$published_at" ]]; then
    echo "❌ 找不到 published_at！"
    return 1
  fi

  release_version=$(generate_app_version "$published_at")

  tar_url=$(printf '%s' $release | jq -r '.tarball_url')
  if [[ -z "$tar_url" ]]; then
    echo "❌ 从请求结果获取源码压缩包文件下载链接失败！"
    return 1
  fi

  zip_url=$(printf '%s' $release | jq -r '.zipball_url')
  if [[ -z "$zip_url" ]]; then
    echo "❌ 从请求结果获取源码压缩包文件下载链接失败！"
    return 1
  fi

  # 构建一个json对象
  local data="{
    \"tag_name\": \"${tag_name}\",
    \"published_at\": \"${published_at}\",
    \"release_version\": \"${release_version}\",
    \"tar_url\": \"${tar_url}\",
    \"zip_url\": \"${zip_url}\"
  }"
  echo $data
  return 0
}

download_src() {
  local REPO="sqzw-x/mdcx"
  local TAG_NAME="daily_release"

  local info=$(get_release_info "$REPO" "$TAG_NAME")
  if [[ $? -ne 0 ]]; then
    echo "❌ 获取仓库 ${REPO} 中 tag_name=${TAG_NAME} 的release信息失败！"
    exit 1
  else
    echo "✅ 获取仓库 ${REPO} 中 tag_name=${TAG_NAME} 的release信息成功！"
  fi
  echo $info | jq

  # 发布时间
  local published_at=$(printf '%s' $info | jq -r ".published_at")
  echo "📅 发布时间: $published_at"

  # 版本号
  local release_version=$(printf '%s' $info | jq -r ".release_version")
  echo "🔢 版本号: $release_version"

  # 源码链接
  local file_url=$(printf '%s' $info | jq -r ".tar_url")
  echo "🔗 下载链接: $file_url"

  local file_path="$release_version.tar.gz"

  curl -o $file_path $file_url -L
  if [[ $? -ne 0 ]]; then
    echo "❌ 下载文件失败！"
    exit 1
  fi

  echo "✅ 下载成功"
  echo "⏳ 开始解压..."

  local appPath="./app"
  mkdir -p $appPath

  # 解压
  tar -zxvf $file_path -C $appPath --strip-components=1
  
  # 删除压缩包
  rm -f $file_path
  echo "✅ 源码已覆盖到 $appPath"

  echo "🔘 删除标记文件 $appPath/$FILE_INITIALIZED"
  rm -f "$appPath/$FILE_INITIALIZED"

  echo "✅ 源码已更新成功！"
}

# 如果是src版，则需要下载源码
if [[ "$TYPE" == "src" ]]; then
  echo ""
  echo "⏳ 下载源码..."
  download_src
fi

# 询问输入容器名称
echo ""
echo "❓ 请输入容器名称（默认：${MDCX_CONTAINER_NAME}）"
read -p "容器名称：" CONTAINER_NAME
CONTAINER_NAME=${CONTAINER_NAME:-$MDCX_CONTAINER_NAME}

echo "⏳ 替换容器名称..."
replace_in_file "s/MDCX_CONTAINER_NAME=.*/MDCX_CONTAINER_NAME=$CONTAINER_NAME/g" .env
echo "✅ 替换容器名称完成"

# 拉取镜像
echo ""
echo "⏳ 拉取镜像..."
docker-compose pull
if [ $? -eq 0 ]; then
  echo "✅ 拉取镜像完成"
else
  echo "❌ 拉取镜像失败，请检查错误日志。如果是网络问题，在解决后你可以使用以下命令重新拉取和运行: "
  echo "cd ${DIR_FULL_PATH}"
  echo "docker-compose pull"
  echo "docker-compose up -d"

  on_error "${DIR_FULL_PATH}"
fi

echo ""
read -p "❓ 是否运行容器？[y/n] " RUN_CONTAINER
if [[ "$RUN_CONTAINER" =~ ^[Yy](es)?$ ]]; then
  docker-compose up -d
  if [ $? -eq 0 ]; then
      echo "✅ 容器已经成功运行"
      echo ""
      echo "🔘 可以通过以下命令查看容器运行状态:"
      echo "🔘 docker ps -a | grep $CONTAINER_NAME"
  else
      echo "❌ 容器启动失败，请检查错误日志"

      on_error "${DIR_FULL_PATH}"
  fi
else
  docker-compose create

  echo "🔘 你可以之后通过以下命令启动容器:"
  echo "cd ${DIR_FULL_PATH} && docker-compose up -d"
fi

echo ""
echo "👋🏻 Enjoy!"