优化 pve.sh 脚本，添加存储参数支持，改进存储获取逻辑

scripts/pve.sh

@@ -12,6 +12,7 @@ REPO="https://github.com/RROrg/rr"
 ONBOOT=1      # 开机启动，默认1
 EFI=1         # 启用 UEFI 引导，默认1
 BLTYPE="sata" # 引导盘类型， 支持 sata,usb,nvme 默认 sata
+STORAGE=""    # 存储，默认自动获取
 NINEPPATH=""  # 9p路径，默认空不添加
 TAG=""        # 镜像tag，默认自动获取
 IMG=""        # 本地镜像路径，默认空
@@ -22,6 +23,7 @@ usage() {
 	echo "  --onboot <0|1>           Enable VM on boot, default 1 (enable)"
 	echo "  --efi <0|1>              Enable UEFI boot, default 1 (enable)"
 	echo "  --bltype <sata|usb|nvme> Bootloader disk type, default sata"
+	echo "  --storage <name>         Storage name for images, default auto get"
 	echo "  --9ppath <path>          Set to /path/to/9p to mount 9p share"
 	echo "  --tag <tag>              Image tag, download latest release if not set"
 	echo "  --img <path>             Local image path, use local image if set"
@@ -51,6 +53,16 @@ while true; do
     echo "$BLTYPE" | grep -qvE '^(sata|usb|nvme)$' && BLTYPE="sata"
     shift 2
     ;;
+  --storage)
+    STORAGE="$2"
+		[ -n "${STORAGE}" ] && pvesm status -content images | grep -qw "^${STORAGE}" || STORAGE=""
+    shift 2
+    ;;
+  --9ppath)
+    NINEPPATH="$2"
+		[ -d "${NINEPPATH}" ] && NINEPPATH="$(realpath "${NINEPPATH}")" || NINEPPATH=""
+    shift 2
+    ;;
   --tag)
     TAG="$2"
 		[ "${TAG:0:1}" = "v" ] && TAG="${TAG:1}"
@@ -62,12 +74,6 @@ while true; do
 		[ -n "${IMG}" ] && IMG="$(realpath "${IMG}")"
     shift 2
     ;;
-  --9ppath)
-    NINEPPATH="$2"
-		[ ! -d "${NINEPPATH}" ] && NINEPPATH=""
-		[ -z "${NINEPPATH}" ] || NINEPPATH="$(realpath "${NINEPPATH}")"
-    shift 2
-    ;;
   --)
     shift
     break
@@ -106,7 +112,7 @@ else
   if ! command -v unzip >/dev/null 2>&1; then
     apt-get update >/dev/null 2>&1 && apt-get install -y unzip >/dev/null 2>&1
   fi
-  IMG_FILE=$(unzip -l "/tmp/rr-${TAG}.img.zip" | awk '{print $4}' | grep '\.img$' | head -n 1)
+  IMG_FILE=$(unzip -l "/tmp/rr-${TAG}.img.zip" | awk '{print $4}' | grep '\.img$' | head -1)
   if [ -z "${IMG_FILE}" ]; then
     echo "No img file found in rr-${TAG}.img.zip"
     exit 1
@@ -137,9 +143,17 @@ if [ $? -ne 0 ]; then
   exit 1
 fi
 
+# 获取 存储
+[ -z "${STORAGE}" ] && STORAGE=$(pvesm status -content images | awk 'NR>1 {print $1}' | grep local | tail -1)
+if [ -z "${STORAGE}" ]; then
+	echo "No storage for images"
+	qm destroy ${VMID} --purge
+	exit 1
+fi
+
 # 启用 UEFI 引导
 if [ "${EFI:-1}" -eq 1 ]; then
-  if ! qm set ${VMID} --bios ovmf --efidisk0 local-lvm:4,efitype=4m,pre-enrolled-keys=0; then
+  if ! qm set ${VMID} --bios ovmf --efidisk0 ${STORAGE}:4,efitype=4m,pre-enrolled-keys=0; then
     echo "Set UEFI failed"
     qm destroy ${VMID} --purge
     exit 1
@@ -147,12 +161,6 @@ if [ "${EFI:-1}" -eq 1 ]; then
 fi
 
 # 导入 RR 镜像
-STORAGE=$(pvesm status -content images | awk 'NR>1 {print $1}')
-if [ -z "${STORAGE}" ]; then
-	echo "No storage for images"
-	qm destroy ${VMID} --purge
-	exit 1
-fi
 BLDISK=$(qm importdisk ${VMID} "${IMG_PATH}" "${STORAGE}" | grep 'successfully imported disk' | sed -n "s/.*'\(.*\)'.*/\1/p")
 STATUS=$?
 if [ "${STATUS:-0}" -ne 0 ] || [ -z "${BLDISK}" ]; then
@@ -203,9 +211,9 @@ if [ $? -ne 0 ]; then
 fi
 
 # 添加 32G 数据盘
-qm set ${VMID} --sata1 local-lvm:32
+qm set ${VMID} --sata1 ${STORAGE}:32
 
-BRIDGE=$(awk -F: '/^iface vmbr/ {print $1}' /etc/network/interfaces | awk '{print $2}' | head -n 1)
+BRIDGE=$(awk -F: '/^iface vmbr/ {print $1}' /etc/network/interfaces | awk '{print $2}' | head -1)
 if [ -z "${BRIDGE}" ]; then
   echo "Get bridge failed"
   qm destroy ${VMID} --purge