#!/bin/sh

if [[ $EUID -ne 0 ]]; then
    clear
    echo "Error: This script must be run as root!" 1>&2
    exit 1
fi

function CopyRight() {
  clear
  echo "########################################################"
  echo "#                                                      #"
  echo "#  New Reinstall Script V4.0                           #"
  echo "#                                                      #"
  echo "#  Author: YUCHEN 815494.COM                           #"
  echo "#  Last Modified: 2022-01-11                           #"
  echo "#                                                      #"
  echo "#                                                      #"
  echo "#                                                      #"
  echo "########################################################"
  echo -e "\n"
}

function isValidIp() {
  local ip=$1
  local ret=1
  if [[ $ip =~ ^[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}$ ]]; then
    ip=(${ip//\./ })
    [[ ${ip[0]} -le 255 && ${ip[1]} -le 255 && ${ip[2]} -le 255 && ${ip[3]} -le 255 ]]
    ret=$?
  fi
  return $ret
}

function ipCheck() {
  isLegal=0
  for add in $MAINIP $GATEWAYIP $NETMASK; do
    isValidIp $add
    if [ $? -eq 1 ]; then
      isLegal=1
    fi
  done
  return $isLegal
}

function GetIp() {
  MAINIP=$(ip route get 1 | awk -F 'src ' '{print $2}' | awk '{print $1}')
  GATEWAYIP=$(ip route | grep default | awk '{print $3}')
  SUBNET=$(ip -o -f inet addr show | awk '/scope global/{sub(/[^.]+\//,"0/",$4);print $4}' | head -1 | awk -F '/' '{print $2}')
  value=$(( 0xffffffff ^ ((1 << (32 - $SUBNET)) - 1) ))
  NETMASK="$(( (value >> 24) & 0xff )).$(( (value >> 16) & 0xff )).$(( (value >> 8) & 0xff )).$(( value & 0xff ))"
}

function UpdateIp() {
  read -r -p "Your IP: " MAINIP
  read -r -p "Your Gateway: " GATEWAYIP
  read -r -p "Your Netmask: " NETMASK
}

function SetNetwork() {
  isAuto='0'
  if [[ -f '/etc/network/interfaces' ]];then
    [[ ! -z "$(sed -n '/iface.*inet static/p' /etc/network/interfaces)" ]] && isAuto='1'
    [[ -d /etc/network/interfaces.d ]] && {
      cfgNum="$(find /etc/network/interfaces.d -name '*.cfg' |wc -l)" || cfgNum='0'
      [[ "$cfgNum" -ne '0' ]] && {
        for netConfig in `ls -1 /etc/network/interfaces.d/*.cfg`
        do 
          [[ ! -z "$(cat $netConfig | sed -n '/iface.*inet static/p')" ]] && isAuto='1'
        done
      }
    }
  fi
  
  if [[ -d '/etc/sysconfig/network-scripts' ]];then
    cfgNum="$(find /etc/network/interfaces.d -name '*.cfg' |wc -l)" || cfgNum='0'
    [[ "$cfgNum" -ne '0' ]] && {
      for netConfig in `ls -1 /etc/sysconfig/network-scripts/ifcfg-* | grep -v 'lo$' | grep -v ':[0-9]\{1,\}'`
      do 
        [[ ! -z "$(cat $netConfig | sed -n '/BOOTPROTO.*[sS][tT][aA][tT][iI][cC]/p')" ]] && isAuto='1'
      done
    }
  fi
}

function NetMode() {
  CopyRight

  if [ "$isAuto" == '0' ]; then
    read -r -p "Using DHCP to configure network automatically? [Y/n]:" input
    case $input in
      [yY][eE][sS]|[yY]) NETSTR='' ;;
      [nN][oO]|[nN]) isAuto='1' ;;
      *) clear; echo "Canceled by user!"; exit 1;;
    esac
  fi

  if [ "$isAuto" == '1' ]; then
    GetIp
    ipCheck
    if [ $? -ne 0 ]; then
      echo -e "Error occurred when detecting ip. Please input manually.\n"
      UpdateIp
    else
      CopyRight
      echo "IP: $MAINIP"
      echo "Gateway: $GATEWAYIP"
      echo "Netmask: $NETMASK"
      echo -e "\n"
      read -r -p "Confirm? [Y/n]:" input
      case $input in
        [yY][eE][sS]|[yY]) ;;
        [nN][oO]|[nN])
          echo -e "\n"
          UpdateIp
          ipCheck
          [[ $? -ne 0 ]] && {
            clear
            echo -e "Input error!\n"
            exit 1
          }
        ;;
        *) clear; echo "Canceled by user!"; exit 1;;
      esac
    fi
    NETSTR="--ip-addr ${MAINIP} --ip-gate ${GATEWAYIP} --ip-mask ${NETMASK}"
  fi
}

function Start() {
  CopyRight
  
  isCN='0'
  geoip=$(wget --no-check-certificate -qO- https://api.ip.sb/geoip | grep "\"country_code\":\"CN\"")
  if [[ "$geoip" != "" ]];then
    isCN='1'
  fi

  if [ "$isAuto" == '0' ]; then
    echo "Using DHCP mode."
  else
    echo "IP: $MAINIP"
    echo "Gateway: $GATEWAYIP"
    echo "Netmask: $NETMASK"
  fi

  [[ "$isCN" == '1' ]] && echo "Using domestic mode."

  if [ -f "/tmp/Core_Install.sh" ]; then
    rm -f /tmp/Core_Install.sh
  fi
  wget --no-check-certificate -qO /tmp/Core_Install.sh 'https://www.815494.com/html/shell/dd/Core_Install.sh' && chmod a+x /tmp/Core_Install.sh
  
  CMIRROR=''
  CVMIRROR=''
  DMIRROR=''
  UMIRROR=''
  SYSMIRROR1='https://www.815494.com/html/disk/os/cxthhhhh/CentOS-7.img.gz'
  SYSMIRROR2='https://www.815494.com/html/disk/share/cxthhhhh/CentOS_7.X_x64_Legacy_NetInstallation_Final_v9.8.vhd.gz'
  SYSMIRROR3='https://www.815494.com/html/disk/os/cxthhhhh/CentOS_8.X_x64_Legacy_NetInstallation_Stable_v6.8.vhd.gz'
  SYSMIRROR12='https://www.815494.com/html/disk//os/laosiji/VirtAll-win7-sp1-ent-x64-cn.vhd.gz'
  SYSMIRROR13='https://www.815494.com/html/disk/share/laosiji/guajibao-win7-sp1-ent-x64-cn-efi.vhd.gz'
  SYSMIRROR14='https://www.815494.com/html/disk/os/laosiji/guajibao-win10-ent-ltsc-2021-x64-cn.vhd.gz'
  SYSMIRROR15='https://www.815494.com/html/disk/share/laosiji/guajibao-win10-ent-ltsc-2021-x64-cn-efi.vhd.gz'
  SYSMIRROR16='https://www.815494.com/html/disk/os/laosiji/WinSrv2003x86-Chinese-C10G.vhd.gz'
  SYSMIRROR17='https://www.815494.com/html/disk/os/laosiji/winsrv2008r2-data-sp1-x64-cn.vhd.gz'
  SYSMIRROR18='https://www.815494.com/html/disk/share/laosiji/winsrv2008r2-data-sp1-x64-cn-efi.vhd.gz'
  SYSMIRROR19='https://www.815494.com/html/disk/os/laosiji/guajibao-winsrv2012r2-data-x64-cn.vhd.gz'
  SYSMIRROR20='https://www.815494.com/html/disk/share/laosiji/guajibao-winsrv2012r2-data-x64-cn-efi.vhd.gz'
  SYSMIRROR21='https://www.815494.com/html/disk/os/laosiji/guajibao-winsrv2016-data-x64-cn.vhd.gz'
  SYSMIRROR22='https://www.815494.com/html/disk/share/laosiji/guajibao-winsrv2016-data-x64-cn-efi.vhd.gz'
  SYSMIRROR23='https://www.815494.com/html/disk/share/cxthhhhh/Disk_Windows_Server_2019_DataCenter_CN_v5.1.vhd.gz'
  SYSMIRROR24='https://www.815494.com/html/disk/os/laosiji/guajibao-winsrv2022-data-x64-cn.vhd.gz'
  SYSMIRROR25='https://www.815494.com/html/disk/share/laosiji/guajibao-winsrv2022-data-x64-cn-efi.vhd.gz'

  if [[ "$isCN" == '1' ]];then
    sed -i 's#https://www.815494.com/html/disk/os/reinstall/wget_udeb_amd64.tar.gz#http://raw.githubusercontent.com/fcurrk/reinstall/master/wget_udeb_amd64.tar.gz#' /tmp/Core_Install.sh
    CMIRROR="--mirror http://mirrors.aliyun.com/centos/"
    CVMIRROR="--mirror http://mirrors.tuna.tsinghua.edu.cn/centos-vault/"
    DMIRROR="--mirror http://mirrors.ustc.edu.cn/debian/"
    UMIRROR="--mirror http://mirrors.aliyun.com/ubuntu/"
  SYSMIRROR1='https://www.815494.com/html/disk/os/cxthhhhh/CentOS-7.img.gz'
  SYSMIRROR2='https://www.815494.com/html/disk/share/cxthhhhh/CentOS_7.X_x64_Legacy_NetInstallation_Final_v9.8.vhd.gz'
  SYSMIRROR3='https://www.815494.com/html/disk/os/cxthhhhh/CentOS_8.X_x64_Legacy_NetInstallation_Stable_v6.8.vhd.gz'
  SYSMIRROR12='https://www.815494.com/html/disk//os/laosiji/VirtAll-win7-sp1-ent-x64-cn.vhd.gz'
  SYSMIRROR13='https://www.815494.com/html/disk/share/laosiji/guajibao-win7-sp1-ent-x64-cn-efi.vhd.gz'
  SYSMIRROR14='https://www.815494.com/html/disk/os/laosiji/guajibao-win10-ent-ltsc-2021-x64-cn.vhd.gz'
  SYSMIRROR15='https://www.815494.com/html/disk/share/laosiji/guajibao-win10-ent-ltsc-2021-x64-cn-efi.vhd.gz'
  SYSMIRROR16='https://www.815494.com/html/disk/os/laosiji/WinSrv2003x86-Chinese-C10G.vhd.gz'
  SYSMIRROR17='https://www.815494.com/html/disk/os/laosiji/winsrv2008r2-data-sp1-x64-cn.vhd.gz'
  SYSMIRROR18='https://www.815494.com/html/disk/share/laosiji/winsrv2008r2-data-sp1-x64-cn-efi.vhd.gz'
  SYSMIRROR19='https://www.815494.com/html/disk/os/laosiji/guajibao-winsrv2012r2-data-x64-cn.vhd.gz'
  SYSMIRROR20='https://www.815494.com/html/disk/share/laosiji/guajibao-winsrv2012r2-data-x64-cn-efi.vhd.gz'
  SYSMIRROR21='https://www.815494.com/html/disk/os/laosiji/guajibao-winsrv2016-data-x64-cn.vhd.gz'
  SYSMIRROR22='https://www.815494.com/html/disk/share/laosiji/guajibao-winsrv2016-data-x64-cn-efi.vhd.gz'
  SYSMIRROR23='https://www.815494.com/html/disk/share/cxthhhhh/Disk_Windows_Server_2019_DataCenter_CN_v5.1.vhd.gz'
  SYSMIRROR24='https://www.815494.com/html/disk/os/laosiji/guajibao-winsrv2022-data-x64-cn.vhd.gz'
  SYSMIRROR25='https://www.815494.com/html/disk/share/laosiji/guajibao-winsrv2022-data-x64-cn-efi.vhd.gz'

  fi

  echo -e "\nPlease select an OS:"
  echo "   1) CentOS 7.7 (ext4)"
  echo "   2) CentOS 7 (cxthhhh)"
  echo "   3) CentOS 8 (cxthhhh)"
  echo "   4) CentOS 6"
  echo "   5) Debian 11"
  echo "   6) Debian 10"
  echo "   7) Debian 9"
  echo "   8) Debian 8"
  echo "   9) Ubuntu 20.04"
  echo "  10) Ubuntu 18.04"
  echo "  11) Ubuntu 16.04"
  echo "  12) Windows 7 x64 Lite [X64-Legacy-nat.ee]"
  echo "  13) Windows 7 x64 Lite [X64-UEFI-nat.ee]"
  echo "  14) Windows 10 LTSC Lite [X64-Legacy-nat.ee]"
  echo "  15) Windows 10 LTSC Lite [X64-UEFI-nat.ee"
  echo "  16) Windows Server 2003 Lite [X86-Legacy-nat.ee]"
  echo "  17) Windows Server 2008 Lite [X64-Legacy-nat.ee]"
  echo "  18) Windows Server 2008 Lite [X64-UEFI-nat.ee]"
  echo "  19) Windows Server 2012 Lite [X64-Legacy-nat.ee]"
  echo "  20) Windows Server 2012 Lite [X64-UEFI-nat.ee]"
  echo "  21) Windows Server 2016 Lite [X64-Legacy-nat.ee]"
  echo "  22) Windows Server 2016 Lite [X64-UEFI-nat.ee]"
  echo "  23) Windows Server 2019 [X64-Legacy-cxthhhhh]"
  echo "  24) Windows Server 2022 Lite [X64-Legacy-nat.ee]"
  echo "  25) Windows Server 2022 Lite [X64-UEFI-nat.ee]"
  echo "  99) Custom install"
  echo "   0) Exit"
  echo -ne "\nYour option: "
  read N
  case $N in
    1) echo -e "\nPassword: Pwd@CentOS\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR1 $DMIRROR ;;
    2) echo -e "\nPassword: cxthhhhh.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR2 $DMIRROR ;;
    3) echo -e "\nPassword: cxthhhhh.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR3 $DMIRROR ;;
    4) echo -e "\nPassword: 815494.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh -c 6.10 -v 64 -a -p 815494.com $NETSTR $CMIRROR ;;
    5) echo -e "\nPassword: 815494.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh -d 11 -v 64 -a -p 815494.com $NETSTR $DMIRROR ;;
    6) echo -e "\nPassword: 815494.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh -d 10 -v 64 -a -p 815494.com $NETSTR $DMIRROR ;;
    7) echo -e "\nPassword: 815494.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh -d 9 -v 64 -a -p 815494.com $NETSTR $DMIRROR ;;
    8) echo -e "\nPassword: 815494.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh -d 8 -v 64 -a -p 815494.com $NETSTR $DMIRROR ;;
    9) echo -e "\nPassword: 815494.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh -u 20.04 -v 64 -a -p 815494.com $NETSTR $UMIRROR ;;
    10) echo -e "\nPassword: 815494.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh -u 18.04 -v 64 -a -p 815494.com $NETSTR $UMIRROR ;;
    11) echo -e "\nPassword: 815494.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh -u 16.04 -v 64 -a -p 815494.com $NETSTR $UMIRROR ;;
    12) echo -e "\nPassword: cxthhhhh.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR12 $DMIRROR ;;
    13) echo -e "\nPassword: cxthhhhh.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR13 $DMIRROR ;;
    14) echo -e "\nPassword: cxthhhhh.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR14 $DMIRROR ;;
    15) echo -e "\nPassword: nat.ee\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR15 $DMIRROR ;;
    16) echo -e "\nPassword: cxthhhhh.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR16 $DMIRROR ;;
    17) echo -e "\nPassword: nat.ee\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR17 $DMIRROR ;;
    18) echo -e "\nPassword: cxthhhhh.com\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR18 $DMIRROR ;;
    19) echo -e "\nPassword: WinSrv2003x86-Chinese\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR19 $DMIRROR ;;
    20) echo -e "\nPassword: www.nat.ee\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR20 $DMIRROR ;;
    21) echo -e "\nPassword: Windows7x86-Chinese\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR21 $DMIRROR ;;
    22) echo -e "\nPassword: nat.ee\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR22 $DMIRROR ;;   
    23) echo -e "\nPassword: nat.ee\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR23 $DMIRROR ;;   
    24) echo -e "\nPassword: nat.ee\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR24 $DMIRROR ;;   
    25) echo -e "\nPassword: nat.ee\n"; read -s -n1 -p "Press any key to continue..." ; bash /tmp/Core_Install.sh $NETSTR -dd $SYSMIRROR25 $DMIRROR ;;  
    99)
      echo -e "\n"
      read -r -p "Custom image URL: " imgURL
      echo -e "\n"
      read -r -p "Are you sure start reinstall? [Y/n]: " input
      case $input in
        [yY][eE][sS]|[yY]) bash /tmp/Core_Install.sh $NETSTR -dd $imgURL $DMIRROR ;;
        *) clear; echo "Canceled by user!"; exit 1;;
      esac
      ;;
    0) exit 0;;
    *) echo "Wrong input!"; exit 1;;
  esac
}

SetNetwork
NetMode
Start
