misc/build.func

@@ -3086,29 +3086,21 @@ check_container_resources() {
 # ------------------------------------------------------------------------------
 # check_container_storage()
 #
-# - Checks root (/) partition usage
+# - Checks /boot partition usage
 # - Warns if usage >80% and asks user confirmation before proceeding
 # ------------------------------------------------------------------------------
 check_container_storage() {
-  usage=$(df / -P | awk 'NR==2 {print $5}' | tr -d '%')
-
-  if [ -z "$usage" ] || [ "$usage" -lt 0 ]; then
-    echo -e "${CROSS}${HOLD}${RD}Error: Failed to check disk usage.${CL}"
-    exit 1
-  fi
-
-  if [ "$usage" -gt 80 ]; then
+  total_size=$(df /boot --output=size | tail -n 1)
+  local used_size=$(df /boot --output=used | tail -n 1)
+  usage=$((100 * used_size / total_size))
+  if ((usage > 80)); then
     echo -e "${INFO}${HOLD} ${YWB}Warning: Storage is dangerously low (${usage}%).${CL}"
-    printf "Continue anyway? <y/N>  "
+    echo -ne "Continue anyway? <y/N>  "
     read -r prompt
-
-    case "$prompt" in
-    [yY][eE][sS] | [yY]) ;;
-    *)
+    if [[ ! ${prompt,,} =~ ^(y|yes)$ ]]; then
       echo -e "${CROSS}${HOLD}${YWB}Exiting based on user input.${CL}"
       exit 1
-      ;;
-    esac
+    fi
   fi
 }