#!/bin/bash
#from https://github.com/oneclickvirt/pve


## China_IP
if [[ -z "${CN}" ]]; then
  if [[ $(curl -m 10 -s https://ipapi.co/json | grep 'China') != "" ]]; then
      _yellow "根据ipapi.co提供的信息，当前IP可能在中国"
      read -e -r -p "是否选用中国镜像完成安装? [Y/n] " input
      case $input in
          [yY][eE][sS] | [yY])
              echo "使用中国镜像"
              CN=true
          ;;
          [nN][oO] | [nN])
              echo "不使用中国镜像"
          ;;
          *)
              echo "使用中国镜像"
              CN=true
          ;;
      esac
  fi
fi

echo "请选择要加载到的模板目录："
echo "1. Proxmox VE 的模板目录（/var/lib/vz/template/iso/）"
echo "2. LXC 的模板目录（/var/lib/vz/template/cache/）"
echo "3. 全都要"
read -p "请输入选项编号（1或2）: " choice

# 将镜像文件移动到Proxmox VE的模板目录中
case "$choice" in
  1)
    if [[ -n "${CN}" ]]; then
      wget -P /root/ https://ghproxy.com/https://github.com/spiritLHLS/realServer-Template/releases/download/1.1/ubuntu20.qcow2
      wget -P /root/ https://ghproxy.com/https://github.com/spiritLHLS/realServer-Template/releases/download/1.0/debian11.qcow2
    else
      wget -P /root/ https://github.com/spiritLHLS/realServer-Template/releases/download/1.1/ubuntu20.qcow2
      wget -P /root/ https://github.com/spiritLHLS/realServer-Template/releases/download/1.0/debian11.qcow2
    fi
    qemu-img convert -f qcow2 -O raw /root/ubuntu20.qcow2 /root/ubuntu20.raw
    mkisofs -o /root/ubuntu20.iso /root/ubuntu20.raw
    mv /root/ubuntu20.iso /var/lib/vz/template/iso/
    qemu-img convert -f qcow2 -O raw /root/debian11.qcow2 /root/debian11.raw
    mkisofs -o /root/debian11.iso /root/debian11.raw
    mv /root/debian11.iso /var/lib/vz/template/iso/
    rm -rf /root/ubuntu20.qcow2 /root/debian11.qcow2 /root/debian11.raw /root/ubuntu20.raw
    echo "已将镜像文件加载到 Proxmox VE 的模板目录"
    ;;
  2)
    if [[ -n "${CN}" ]]; then
      wget -P /root/ https://mirrors.tuna.tsinghua.edu.cn/proxmox/images/system/debian-11-standard_11.3-0_amd64.tar.gz
      wget -P /root/ https://mirrors.tuna.tsinghua.edu.cn/proxmox/images/system/ubuntu-20.10-standard_20.10-1_amd64.tar.gz
      wget -P /root/ https://mirrors.tuna.tsinghua.edu.cn/proxmox/images/system/debian-10-standard_10.7-1_amd64.tar.gz
    else
      wget -P /root/ http://download.proxmox.com/images/system/ubuntu-20.10-standard_20.10-1_amd64.tar.gz
      wget -P /root/ http://download.proxmox.com/images/system/debian-11-standard_11.3-0_amd64.tar.gz
      wget -P /root/ http://download.proxmox.com/images/system/debian-10-standard_10.7-1_amd64.tar.gz
    fi
    mv /root/ubuntu-20.10-standard_20.10-1_amd64.tar.gz /var/lib/vz/template/cache/
    mv /root/debian-11-standard_11.3-0_amd64.tar.gz /var/lib/vz/template/cache/
    mv /root/debian-10-standard_10.7-1_amd64.tar.gz /var/lib/vz/template/cache/
    echo "已将镜像文件移动到 LXC 的模板目录"
    ;;
  3)
    
    if [[ -n "${CN}" ]]; then
      wget -P /root/ https://ghproxy.com/https://github.com/spiritLHLS/realServer-Template/releases/download/1.1/ubuntu20.qcow2
      wget -P /root/ https://ghproxy.com/https://github.com/spiritLHLS/realServer-Template/releases/download/1.0/debian11.qcow2
      wget -P /root/ https://mirrors.tuna.tsinghua.edu.cn/proxmox/images/system/debian-11-standard_11.3-0_amd64.tar.gz
      wget -P /root/ https://mirrors.tuna.tsinghua.edu.cn/proxmox/images/system/ubuntu-20.10-standard_20.10-1_amd64.tar.gz
    else
      wget -P /root/ https://github.com/spiritLHLS/realServer-Template/releases/download/1.1/ubuntu20.qcow2
      wget -P /root/ https://github.com/spiritLHLS/realServer-Template/releases/download/1.0/debian11.qcow2
      wget -P /root/ http://download.proxmox.com/images/system/ubuntu-20.10-standard_20.10-1_amd64.tar.gz
      wget -P /root/ http://download.proxmox.com/images/system/debian-11-standard_11.3-0_amd64.tar.gz
    fi
    qemu-img convert -f qcow2 -O raw /root/ubuntu20.qcow2 /root/ubuntu20.raw
    mkisofs -o /root/ubuntu20.iso /root/ubuntu20.raw
    mv /root/ubuntu20.iso /var/lib/vz/template/iso/
    qemu-img convert -f qcow2 -O raw /root/debian11.qcow2 /root/debian11.raw
    mkisofs -o /root/debian11.iso /root/debian11.raw
    mv /root/debian11.iso /var/lib/vz/template/iso/
    rm -rf /root/ubuntu20.qcow2 /root/debian11.qcow2 /root/debian11.raw /root/ubuntu20.raw
    mv /root/ubuntu-20.10-standard_20.10-1_amd64.tar.gz /var/lib/vz/template/cache/
    mv /root/debian-11-standard_11.3-0_amd64.tar.gz /var/lib/vz/template/cache/
    echo "已全部加载"
    ;;
  *)
    echo "无效的选项，程序退出"
    ;;
esac
