Extending libvirt image handler to create pool and volume

Added code to check if Libvirt pool and volume exists if not creating them through handler

Signed-off-by: Saripalli Lavanya <[email protected]>

Author: Saripalli-lavanya

config/peerpods/podvm/libvirt-podvm-image-handler.sh

@@ -39,6 +39,7 @@ function verify_vars() {
 
 # Function to create the libvirt image based on the type.
 function create_libvirt_image() {
+    check_pool_and_volume_existence
     # Based on the value of `IMAGE_TYPE` the image is either build from scratch or using the prebuilt artifact.
     if [[ "${IMAGE_TYPE}" == "operator-built" ]]; then
         create_libvirt_image_from_scratch
@@ -48,6 +49,46 @@ function create_libvirt_image() {
 
 }
 
+# Function to create Libvirt pool and configure libvirt volume to upload qcow2 image
+function create_pool_and_volume() {
+ 
+    virsh -d 0 -c "${LIBVIRT_URI}" pool-define-as "${LIBVIRT_POOL}" --type dir --target "${LIBVIRT_DIR_NAME}" \
+        || error_exit "Failed to define libvirt pool '${LIBVIRT_POOL}'."
+
+    virsh -d 0 -c "${LIBVIRT_URI}" pool-start "${LIBVIRT_POOL}" \
+        || error_exit "Failed to start libvirt pool '${LIBVIRT_POOL}'."
+
+    # Create the libvirt volume
+    virsh -d 0 -c "${LIBVIRT_URI}" vol-create-as --pool "${LIBVIRT_POOL}" \
+        --name "${LIBVIRT_VOL_NAME}" \
+        --capacity 20G \
+        --allocation 2G \
+        --prealloc-metadata \
+        --format qcow2 \
+        || error_exit "Failed to create libvirt volume '${LIBVIRT_VOL_NAME}' in pool '${LIBVIRT_POOL}'."
+
+    echo "Created libvirt pool and volume successfully."
+
+}
+
+# Function that check if defined volume and pool are already present
+function check_pool_and_volume_existence() {
+
+    echo "Checking existence of libvirt pool '${LIBVIRT_POOL}' and volume '${LIBVIRT_VOL_NAME}'..."
+    
+        virsh -d 0 -c "${LIBVIRT_URI}" pool-info "${LIBVIRT_POOL}" >/dev/null 2>&1
+        POOL_EXISTS=$?
+        virsh -d 0 -c "${LIBVIRT_URI}" vol-info --pool "${LIBVIRT_POOL}" "${LIBVIRT_VOL_NAME}" >/dev/null 2>&1
+        VOL_EXISTS=$?
+
+        if [ "$POOL_EXISTS" -eq 0 ] && [ "$VOL_EXISTS" -eq 0 ]; then
+            echo "Disclaimer: A Libvirt pool named '${LIBVIRT_POOL}' with volume '${LIBVIRT_VOL_NAME}' already exists on the KVM host. Image will be uploaded to same volume."
+        else
+            echo "Libvirt pool '${LIBVIRT_POOL}' or volume '${LIBVIRT_VOL_NAME}' does not exist. Proceeding to create..."
+            create_pool_and_volume
+        fi
+}
+
 # Function to create the libvirt image from a prebuilt artifact.
 function create_libvirt_image_from_prebuilt_artifact() {
     echo "Pulling the podvm image from the provided path"
@@ -204,14 +245,31 @@ function delete_libvirt_image() {
     # LIBVIRT_POOL shouldn't be empty
     [[ -z "${LIBVIRT_POOL}" ]] && error_exit "LIBVIRT_POOL is empty"
 
-    # Delete the libvirt pool
-    echo "Deleting libvirt pool."
-    virsh -d 0 -c "${LIBVIRT_URI}" pool-destroy "${LIBVIRT_POOL}" ||
+    # Delete the libvirt volume
+    virsh -d 0 -c "${LIBVIRT_URI}" vol-delete --pool "${LIBVIRT_POOL}" "${LIBVIRT_VOL_NAME}" || 
+    error_exit "Failed to delete volume '${LIBVIRT_VOL_NAME}' from pool '${LIBVIRT_POOL}'"
+
+    echo "Volume '${LIBVIRT_VOL_NAME}' deleted successfully."
+
+    # Check remaining volumes in the pool
+    VOLUMES=$(virsh -d 0 -c "${LIBVIRT_URI}" vol-list "${LIBVIRT_POOL}" | awk 'NR>2 {print $1}' | grep -v "^$LIBVIRT_VOL_NAME\$")
+
+    #Deleting pool if there are other volumes in the pool
+    if [ -z "$VOLUMES" ]; then
+        echo "No other volumes found in the pool. Deleting the pool"
+
+        virsh -d 0 -c "${LIBVIRT_URI}" pool-destroy "${LIBVIRT_POOL}" ||
         error_exit "Failed to destroy the libvirt pool"
 
-    virsh -d 0 -c "${LIBVIRT_URI}" pool-undefine "${LIBVIRT_POOL}" ||
+        echo "Pool '${LIBVIRT_POOL}' destroyed successfully."
+
+        virsh -d 0 -c "${LIBVIRT_URI}" pool-undefine "${LIBVIRT_POOL}" ||
         error_exit "Failed to undefine the libvirt pool"
 
+        echo "Pool '${LIBVIRT_POOL}' undefined successfully."
+    else
+        echo "Other volumes are present in the pool. Pool will not be deleted."
+    fi    
     echo "Deleted libvirt image successfully"
 
     # Remove the libvirt_volume_name from peer-pods-cm configmap