From beaa0e86b43604cb85da5a1527b3fc11803a20e0 Mon Sep 17 00:00:00 2001 From: MacRimi <123239993+MacRimi@users.noreply.github.com> Date: Thu, 3 Apr 2025 14:39:33 +0200 Subject: [PATCH] Create id.sh --- scripts/test/id.sh | 290 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 290 insertions(+) create mode 100644 scripts/test/id.sh diff --git a/scripts/test/id.sh b/scripts/test/id.sh new file mode 100644 index 0000000..4b9a626 --- /dev/null +++ b/scripts/test/id.sh @@ -0,0 +1,290 @@ +#!/bin/bash + +# ========================================================== +# ProxMenu - A menu-driven script for Proxmox VE management +# ========================================================== +# Author : MacRimi +# Copyright : (c) 2024 MacRimi +# License : MIT (https://raw.githubusercontent.com/MacRimi/ProxMenux/main/LICENSE) +# Version : 1.0 +# Last Updated: 28/01/2025 +# ========================================================== +# Description: +# This script allows users to assign physical disks to existing +# Proxmox virtual machines (VMs) through an interactive menu. +# - Detects the system disk and excludes it from selection. +# - Lists all available VMs for the user to choose from. +# - Identifies and displays unassigned physical disks. +# - Allows the user to select multiple disks and attach them to a VM. +# - Supports interface types: SATA, SCSI, VirtIO, and IDE. +# - Ensures that disks are not already assigned to active VMs. +# - Warns about disk sharing between multiple VMs to avoid data corruption. +# - Configures the selected disks for the VM and verifies the assignment. +# +# The goal of this script is to simplify the process of assigning +# physical disks to Proxmox VMs, reducing manual configurations +# and preventing potential errors. +# ========================================================== + + +# Configuration ============================================ +REPO_URL="https://raw.githubusercontent.com/MacRimi/ProxMenux/main" +BASE_DIR="/usr/local/share/proxmenux" +UTILS_FILE="$BASE_DIR/utils.sh" +VENV_PATH="/opt/googletrans-env" + +if [[ -f "$UTILS_FILE" ]]; then + source "$UTILS_FILE" +fi +load_language +initialize_cache +# ========================================================== + + + + +# Function to get detailed disk information +get_disk_info() { + local disk=$1 + MODEL=$(lsblk -dn -o MODEL "$disk" | xargs) + SIZE=$(lsblk -dn -o SIZE "$disk" | xargs) + echo "$MODEL" "$SIZE" +} + + + + +# Display list of available VMs +VM_LIST=$(qm list | awk 'NR>1 {print $1, $2}') +if [ -z "$VM_LIST" ]; then + whiptail --title "$(translate "Error")" --msgbox "$(translate "No VMs available in the system.")" 8 40 + exit 1 +fi + +# Select VM +VMID=$(whiptail --title "$(translate "Select VM")" --menu "$(translate "Select the VM to which you want to add disks:")" 15 60 8 $VM_LIST 3>&1 1>&2 2>&3) + +if [ -z "$VMID" ]; then + whiptail --title "$(translate "Error")" --msgbox "$(translate "No VM was selected.")" 8 40 + exit 1 +fi + +VMID=$(echo "$VMID" | tr -d '"') + + + +clear +msg_ok "$(translate "VM selected successfully.")" + + +VM_STATUS=$(qm status "$VMID" | awk '{print $2}') +if [ "$VM_STATUS" == "running" ]; then + whiptail --title "$(translate "Warning")" --msgbox "$(translate "The VM is powered on. Turn it off before adding disks.")" 12 60 + exit 1 +fi + + +########################################## + +msg_info "$(translate "Detecting available disks...")" + +# Discos usados por LVM +USED_DISKS=$(lsblk -n -o PKNAME,TYPE | grep 'lvm' | awk '{print "/dev/" $1}') + +# Particiones montadas +MOUNTED_DISKS=$(lsblk -ln -o NAME,MOUNTPOINT | awk '$2!="" {print "/dev/" $1}') + +# Detectar discos usados por ZFS (desde wwn/ata/by-id) +ZFS_DISKS="" +ZFS_RAW=$(zpool list -v -H 2>/dev/null | awk '{print $1}' | grep -v '^NAME$' | grep -v '^-' | grep -v '^mirror') + +for entry in $ZFS_RAW; do + # Intentar resolver a /dev/NAME si es wwn o ata + path="" + if [[ "$entry" == wwn-* || "$entry" == ata-* ]]; then + if [ -e "/dev/disk/by-id/$entry" ]; then + path=$(readlink -f "/dev/disk/by-id/$entry") + fi + elif [[ "$entry" == /dev/* ]]; then + path="$entry" + fi + + # Obtener el disco base físico + if [ -n "$path" ]; then + base_disk=$(lsblk -no PKNAME "$path" 2>/dev/null) + if [ -n "$base_disk" ]; then + ZFS_DISKS+="/dev/$base_disk"$'\n' + fi + fi +done + +ZFS_DISKS=$(echo "$ZFS_DISKS" | sort -u) + +# Función para descartar discos en uso +is_disk_in_use() { + local disk="$1" + + # Verificar si alguna partición está montada o es zfs/raid + while read -r part fstype; do + case "$fstype" in + zfs_member|linux_raid_member) + return 0 ;; + esac + + if echo "$MOUNTED_DISKS" | grep -q "/dev/$part"; then + return 0 + fi + done < <(lsblk -ln -o NAME,FSTYPE "$disk" | tail -n +2) + + # Verificar si está en LVM o ZFS + if echo "$USED_DISKS" | grep -q "$disk" || echo "$ZFS_DISKS" | grep -q "$disk"; then + return 0 + fi + + return 1 +} + + +# Detectar discos libres +FREE_DISKS=() +while read -r DISK; do + [[ "$DISK" =~ /dev/zd ]] && continue + + if ! is_disk_in_use "$DISK" && ! qm config "$VMID" | grep -q "$DISK"; then + INFO=($(get_disk_info "$DISK")) + MODEL="${INFO[@]::${#INFO[@]}-1}" + SIZE="${INFO[-1]}" + DESCRIPTION=$(printf "%-40s %10s" "$MODEL" "$SIZE") + FREE_DISKS+=("$DISK" "$DESCRIPTION" "OFF") + fi +done < <(lsblk -dn -e 7,11 -o PATH) + +msg_ok "$(translate "Available disks detected.")" + +if [ "${#FREE_DISKS[@]}" -eq 0 ]; then + whiptail --title "$(translate "Error")" --msgbox "$(translate "No disks available for this VM.")" 8 40 + clear + exit 1 +fi + + + +###################################################### + + + + +MAX_WIDTH=$(printf "%s\n" "${FREE_DISKS[@]}" | awk '{print length}' | sort -nr | head -n1) +TOTAL_WIDTH=$((MAX_WIDTH + 20)) + +if [ $TOTAL_WIDTH -lt 70 ]; then + TOTAL_WIDTH=70 +fi + + +SELECTED=$(whiptail --title "$(translate "Select Disks")" --checklist \ + "$(translate "Select the disks you want to add:")" 20 $TOTAL_WIDTH 10 "${FREE_DISKS[@]}" 3>&1 1>&2 2>&3) + +if [ -z "$SELECTED" ]; then + whiptail --title "$(translate "Error")" --msgbox "$(translate "No disks were selected.")" 10 $TOTAL_WIDTH + clear + exit 1 +fi + +msg_ok "$(translate "Disks selected successfully.")" + +# Select interface type once for all disks +INTERFACE=$(whiptail --title "$(translate "Interface Type")" --menu "$(translate "Select the interface type for all disks:")" 15 40 4 \ + "sata" "$(translate "Add as SATA")" \ + "scsi" "$(translate "Add as SCSI")" \ + "virtio" "$(translate "Add as VirtIO")" \ + "ide" "$(translate "Add as IDE")" 3>&1 1>&2 2>&3) + +if [ -z "$INTERFACE" ]; then + whiptail --title "$(translate "Error")" --msgbox "$(translate "No interface type was selected for the disks.")" 8 40 + clear + exit 1 +fi + +msg_ok "$(translate "Interface type selected: $INTERFACE")" + +DISKS_ADDED=0 +ERROR_MESSAGES="" +SUCCESS_MESSAGES="" + +msg_info "$(translate "Processing selected disks...")" + +for DISK in $SELECTED; do + DISK=$(echo "$DISK" | tr -d '"') + DISK_INFO=$(get_disk_info "$DISK") + REAL_DISK=$(readlink -f "$DISK") + + # Verificar si el disco ya está asignado a otra VM + ASSIGNED_TO="" + while read -r VM_ID VM_NAME; do + [[ "$VM_ID" =~ ^[0-9]+$ ]] || continue + + while read -r line; do + # Limpiar parámetros adicionales después de la coma + raw_path=$(echo "$line" | awk -F ': ' '{print $2}' | awk '{print $1}' | cut -d',' -f1) + [ -z "$raw_path" ] && continue + + path_real=$(readlink -f "$raw_path" 2>/dev/null) + if [ "$path_real" = "$REAL_DISK" ]; then + ASSIGNED_TO+="$VM_ID $VM_NAME\n" + break + fi + done < <(qm config "$VM_ID" | grep -E '^(scsi|sata|virtio|ide)[0-9]+:') + done < <(qm list | awk 'NR>1 {print $1, $2}') + + CONTINUE=true + if [ -n "$ASSIGNED_TO" ]; then + RUNNING_VMS="" + while read -r VM_ID VM_NAME; do + if [[ "$VM_ID" =~ ^[0-9]+$ ]] && [ "$(qm status "$VM_ID" | awk '{print $2}')" == "running" ]; then + RUNNING_VMS+="$VM_ID $VM_NAME\n" + fi + done < <(echo -e "$ASSIGNED_TO") + + if [ -n "$RUNNING_VMS" ]; then + ERROR_MESSAGES+="$(translate "The disk") $DISK_INFO $(translate "is in use by the following running VM(s):")\\n$RUNNING_VMS\\n\\n" + CONTINUE=false + fi + fi + + if $CONTINUE; then + INDEX=0 + while qm config "$VMID" | grep -q "${INTERFACE}${INDEX}"; do + ((INDEX++)) + done + + RESULT=$(qm set "$VMID" -${INTERFACE}${INDEX} "$DISK" 2>&1) + + if [ $? -eq 0 ]; then + MESSAGE="$(translate "The disk") $DISK_INFO $(translate "has been successfully added to VM") $VMID." + if [ -n "$ASSIGNED_TO" ]; then + MESSAGE+="\n$(translate "WARNING: This disk is also assigned to the following VM(s):")\n$ASSIGNED_TO" + MESSAGE+="$(translate "Make sure not to start VMs that share this disk at the same time to avoid data corruption.")\n" + fi + SUCCESS_MESSAGES+="$MESSAGE\\n\\n" + ((DISKS_ADDED++)) + else + ERROR_MESSAGES+="$(translate "Could not add disk") $DISK_INFO $(translate "to VM") $VMID.\\n$(translate "Error:") $RESULT\\n\\n" + fi + fi +done + +msg_ok "$(translate "Disk processing completed.")" + +if [ -n "$SUCCESS_MESSAGES" ]; then + MSG_LINES=$(echo "$SUCCESS_MESSAGES" | wc -l) + whiptail --title "$(translate "Successful Operations")" --msgbox "$SUCCESS_MESSAGES" 16 70 +fi + +if [ -n "$ERROR_MESSAGES" ]; then + whiptail --title "$(translate "Warnings and Errors")" --msgbox "$ERROR_MESSAGES" 16 70 +fi + + + +exit 0