mirror of
https://github.com/oneclickvirt/pve.git
synced 2024-09-20 15:05:56 +08:00
247 lines
9 KiB
Bash
247 lines
9 KiB
Bash
#!/bin/bash
|
||
# from
|
||
# https://github.com/spiritLHLS/pve
|
||
# 2023.07.31
|
||
|
||
# cd /root
|
||
|
||
_red() { echo -e "\033[31m\033[01m$@\033[0m"; }
|
||
_green() { echo -e "\033[32m\033[01m$@\033[0m"; }
|
||
_yellow() { echo -e "\033[33m\033[01m$@\033[0m"; }
|
||
_blue() { echo -e "\033[36m\033[01m$@\033[0m"; }
|
||
reading(){ read -rp "$(_green "$1")" "$2"; }
|
||
utf8_locale=$(locale -a 2>/dev/null | grep -i -m 1 -E "UTF-8|utf8")
|
||
if [[ -z "$utf8_locale" ]]; then
|
||
echo "No UTF-8 locale found"
|
||
else
|
||
export LC_ALL="$utf8_locale"
|
||
export LANG="$utf8_locale"
|
||
export LANGUAGE="$utf8_locale"
|
||
echo "Locale set to $utf8_locale"
|
||
fi
|
||
|
||
get_system_arch() {
|
||
local sysarch="$(uname -m)"
|
||
if [ "${sysarch}" = "unknown" ] || [ "${sysarch}" = "" ]; then
|
||
local sysarch="$(arch)"
|
||
fi
|
||
# 根据架构信息设置系统位数并下载文件,其余 * 包括了 x86_64
|
||
case "${sysarch}" in
|
||
"i386" | "i686" | "x86_64")
|
||
system_arch="x86"
|
||
;;
|
||
"armv7l" | "armv8" | "armv8l" | "aarch64")
|
||
system_arch="arch"
|
||
;;
|
||
*)
|
||
system_arch=""
|
||
;;
|
||
esac
|
||
}
|
||
|
||
check_cdn() {
|
||
local o_url=$1
|
||
for cdn_url in "${cdn_urls[@]}"; do
|
||
if curl -sL -k "$cdn_url$o_url" --max-time 6 | grep -q "success" > /dev/null 2>&1; then
|
||
export cdn_success_url="$cdn_url"
|
||
return
|
||
fi
|
||
sleep 0.5
|
||
done
|
||
export cdn_success_url=""
|
||
}
|
||
|
||
check_cdn_file() {
|
||
check_cdn "https://raw.githubusercontent.com/spiritLHLS/ecs/main/back/test"
|
||
if [ -n "$cdn_success_url" ]; then
|
||
_yellow "CDN available, using CDN"
|
||
else
|
||
_yellow "No CDN available, no use CDN"
|
||
fi
|
||
}
|
||
|
||
cdn_urls=("https://cdn.spiritlhl.workers.dev/" "https://cdn3.spiritlhl.net/" "https://cdn1.spiritlhl.net/" "https://ghproxy.com/" "https://cdn2.spiritlhl.net/")
|
||
check_cdn_file
|
||
|
||
pre_check(){
|
||
home_dir=$(eval echo "~$(whoami)")
|
||
if [ "$home_dir" != "/root" ]; then
|
||
_red "The script will exit if the current path is not /root."
|
||
_red "当前路径不是/root，脚本将退出。"
|
||
exit 1
|
||
fi
|
||
if ! command -v dos2unix > /dev/null 2>&1; then
|
||
apt-get install dos2unix -y
|
||
fi
|
||
if [ ! -f "buildvm.sh" ]; then
|
||
curl -L ${cdn_success_url}https://raw.githubusercontent.com/spiritLHLS/pve/main/scripts/buildvm.sh -o buildvm.sh && chmod +x buildvm.sh
|
||
dos2unix buildvm.sh
|
||
fi
|
||
}
|
||
|
||
# files=$(find . -maxdepth 1 -name "vm*" | sort)
|
||
# if [ -n "$files" ]; then
|
||
# for file in $files
|
||
# do
|
||
# cat "$file" >> vmlog
|
||
# done
|
||
# fi
|
||
|
||
check_info(){
|
||
log_file="vmlog"
|
||
if [ ! -f "vmlog" ]; then
|
||
_yellow "vmlog file does not exist in the current directory"
|
||
_yellow "当前目录下不存在vmlog文件"
|
||
vm_num=202
|
||
web2_port=40003
|
||
port_end=50025
|
||
else
|
||
while read line; do
|
||
last_line="$line"
|
||
done < "$log_file"
|
||
last_line_array=($last_line)
|
||
vm_num="${last_line_array[0]}"
|
||
user="${last_line_array[1]}"
|
||
password="${last_line_array[2]}"
|
||
ssh_port="${last_line_array[6]}"
|
||
web1_port="${last_line_array[7]}"
|
||
web2_port="${last_line_array[8]}"
|
||
port_start="${last_line_array[9]}"
|
||
port_end="${last_line_array[10]}"
|
||
system="${last_line_array[11]}"
|
||
storage="${last_line_array[12]}"
|
||
_green "Current information corresponding to the last NAT server:"
|
||
_green "当前最后一个NAT服务器对应的信息："
|
||
echo "NAT服务器(NAT Server): $vm_num"
|
||
# echo "用户名: $user"
|
||
# echo "密码: $password"
|
||
echo "外网SSH端口(Extranet SSH port): $ssh_port"
|
||
echo "外网80端口(Extranet port 80): $web1_port"
|
||
echo "外网443端口(Extranet port 443): $web2_port"
|
||
echo "外网其他端口范围(Other port ranges): $port_start-$port_end"
|
||
echo "系统(System)：$system"
|
||
echo "存储盘(Storage Disk)：$storage"
|
||
fi
|
||
}
|
||
|
||
build_new_vms(){
|
||
while true; do
|
||
_green "How many more NAT servers need to be generated? (Enter how many new NAT servers to add):"
|
||
reading "还需要生成几个NAT服务器？(输入新增几个NAT服务器)：" new_nums
|
||
if [[ "$new_nums" =~ ^[1-9][0-9]*$ ]]; then
|
||
break
|
||
else
|
||
_yellow "Invalid input, please enter a positive integer."
|
||
_yellow "输入无效，请输入一个正整数。"
|
||
fi
|
||
done
|
||
while true; do
|
||
_green "How many CPUs are assigned to each virtual machine? (Enter 1 if 1 core is assigned to each virtual machine):"
|
||
reading "每个虚拟机分配几个CPU？(若每个虚拟机分配1核，则输入1)：" cpu_nums
|
||
if [[ "$cpu_nums" =~ ^[1-9][0-9]*$ ]]; then
|
||
break
|
||
else
|
||
_yellow "Invalid input, please enter a positive integer."
|
||
_yellow "输入无效，请输入一个正整数。"
|
||
fi
|
||
done
|
||
while true; do
|
||
_green "How much memory is allocated per virtual machine? (If 512 MB of memory is allocated per virtual machine, enter 512):"
|
||
reading "每个虚拟机分配多少内存？(若每个虚拟机分配512MB内存，则输入512)：" memory_nums
|
||
if [[ "$memory_nums" =~ ^[1-9][0-9]*$ ]]; then
|
||
break
|
||
else
|
||
_yellow "Invalid input, please enter a positive integer."
|
||
_yellow "输入无效，请输入一个正整数。"
|
||
fi
|
||
done
|
||
while true; do
|
||
_green "On which storage drive are the virtual machines opened? (Leave blank or enter 'local' if the virtual machine is to be opened on the system disk):"
|
||
reading "虚拟机们开设在哪个存储盘上？(若虚拟机要开设在系统盘上，则留空或输入local)：" storage
|
||
if [ -z "$storage" ]; then
|
||
storage="local"
|
||
fi
|
||
break
|
||
done
|
||
while true; do
|
||
_green "How many hard disks are allocated per virtual machine? (If 5G hard drives are allocated per virtual machine, enter 5):"
|
||
reading "每个虚拟机分配多少硬盘？(若每个虚拟机分配5G硬盘，则输入5)：" disk_nums
|
||
if [[ "$disk_nums" =~ ^[1-9][0-9]*$ ]]; then
|
||
break
|
||
else
|
||
_yellow "Invalid input, please enter a positive integer."
|
||
_yellow "输入无效，请输入一个正整数。"
|
||
fi
|
||
done
|
||
if [ "$system_arch" = "x86" ]; then
|
||
while true; do
|
||
sys_status="false"
|
||
_green "What system does each virtual machine use? (Leave blank or enter debian11 if all use debian11):"
|
||
reading "每个虚拟机都使用什么系统？(若都使用debian11，则留空或输入debian11)：" system
|
||
if [ -z "$system" ]; then
|
||
system="debian11"
|
||
fi
|
||
systems=("debian10" "debian11" "debian9" "ubuntu18" "ubuntu20" "ubuntu22" "archlinux" "centos9-stream" "centos8-stream" "almalinux8" "almalinux9" "fedora33" "fedora34" "opensuse-leap-15")
|
||
for sys in ${systems[@]}; do
|
||
if [[ "$system" == "$sys" ]]; then
|
||
sys_status="true"
|
||
break
|
||
fi
|
||
done
|
||
if [ "$sys_status" = "true" ]; then
|
||
break
|
||
else
|
||
_yellow "This system is not supported, please check https://github.com/spiritLHLS/Images for the names of supported systems"
|
||
_yellow "不支持该系统，请查看 https://github.com/spiritLHLS/Images 支持的系统名字"
|
||
fi
|
||
done
|
||
else
|
||
while true; do
|
||
sys_status="false"
|
||
_green "What system does each virtual machine use? (Leave blank or enter debian11 if all use debian11):"
|
||
reading "每个虚拟机都使用什么系统？(若都使用ubuntu22，则留空或输入ubuntu22)：" system
|
||
if [ -z "$system" ]; then
|
||
system="ubuntu22"
|
||
fi
|
||
systems=("ubuntu14" "ubuntu16" "ubuntu18" "ubuntu20" "ubuntu22")
|
||
for sys in ${systems[@]}; do
|
||
if [[ "$system" == "$sys" ]]; then
|
||
sys_status="true"
|
||
break
|
||
fi
|
||
done
|
||
if [ "$sys_status" = "true" ]; then
|
||
break
|
||
else
|
||
_yellow "Unable to install corresponding system, please check http://cloud-images.ubuntu.com for supported system images "
|
||
_yellow "无法安装对应系统，请查看 http://cloud-images.ubuntu.com 支持的系统镜像 "
|
||
fi
|
||
done
|
||
fi
|
||
for ((i=1; i<=$new_nums; i++)); do
|
||
vm_num=$(($vm_num + 1))
|
||
user=$(cat /dev/urandom | tr -dc 'a-zA-Z' | fold -w 4 | head -n 1)
|
||
ori=$(date | md5sum)
|
||
password=${ori: 2: 9}
|
||
ssh_port=$(($web2_port + 1))
|
||
web1_port=$(($web2_port + 2))
|
||
web2_port=$(($web1_port + 1))
|
||
port_start=$(($port_end + 1))
|
||
port_end=$(($port_start + 25))
|
||
./buildvm.sh $vm_num $user $password $cpu_nums $memory_nums $disk_nums $ssh_port $web1_port $web2_port $port_start $port_end $system $storage
|
||
cat "vm$vm_num" >> vmlog
|
||
rm -rf "vm$vm_num"
|
||
sleep 60
|
||
done
|
||
}
|
||
|
||
pre_check
|
||
get_system_arch
|
||
if [ -z "${system_arch}" ] || [ ! -v system_arch ]; then
|
||
_red "This script can only run on machines under x86_64 or arm architecture."
|
||
exit 1
|
||
fi
|
||
check_info
|
||
build_new_vms
|
||
check_info
|