From 4248917f947fd7430a5523ed0f384f23f4b25190 Mon Sep 17 00:00:00 2001
From: paulovmr <832830+paulovmr@users.noreply.github.com>
Date: Tue, 13 Aug 2024 13:46:33 -0300
Subject: [PATCH] RHOAIENG-11300: Improve the library upgrade script for
 alternative sources

---
 scripts/update_library_version.sh | 54 ++++++++++++++++++-------------
 1 file changed, 32 insertions(+), 22 deletions(-)

diff --git a/scripts/update_library_version.sh b/scripts/update_library_version.sh
index fc64875f2..2cbc47fbf 100755
--- a/scripts/update_library_version.sh
+++ b/scripts/update_library_version.sh
@@ -108,38 +108,48 @@ update_library_version_pipfile() {
   fi
 
   # Extract the current version and qualifier from the Pipfile
-  current_line=$(grep -E "^\"?$lib\"? = " "$file")
+  current_line=$(grep -E "^\"?$lib\"?[[:space:]]*=[[:space:]]*" "$file")
+
   if [[ $current_line =~ ^\"?$lib\"?[[:space:]]*=[[:space:]]*\"?([=><!~]*)?([0-9]+(\.[0-9]+)*)\"?$ ]]; then
     current_qualifier="${BASH_REMATCH[1]}"
     current_ver="${BASH_REMATCH[2]}"
+  elif [[ $current_line =~ version[[:space:]]*=[[:space:]]*\"?([=><!~]*)?([0-9]+(\.[0-9]+)*)\" ]]; then
+    current_qualifier="${BASH_REMATCH[1]}"
+    current_ver="${BASH_REMATCH[2]}"
+  else
+    echo "$lib not found in $file."
+    return
+  fi
 
-    # Compare the current and new versions
-    is_version_higher "$new_ver" "$current_ver"
-    comparison_result=$?
+  # Compare the current and new versions
+  is_version_higher "$new_ver" "$current_ver"
+  comparison_result=$?
 
-    if [ "$comparison_result" -eq 0 ]; then
-      # Keep the original qualifier and update to the new version if it is higher
+  if [ "$comparison_result" -eq 0 ]; then
+    # Keep the original qualifier and update to the new version if it is higher
+    if [[ $current_line =~ ^\"?$lib\"?[[:space:]]*=[[:space:]]*\"?([=><!~]*)?[0-9]+(\.[0-9]+)*\"?$ ]]; then
       new_line="${lib} = \"${current_qualifier}${new_ver}\""
       sed -i -E "s/^\"?$lib\"?[[:space:]]*=[[:space:]]*\"?[=><!~]*[0-9]+(\.[0-9]+)*\"?/${new_line}/" "$file"
-      echo "Updated $lib in $file to version ${current_qualifier}${new_ver}"
-
-      # Handle renaming and pipenv lock, if necessary
-      if [ "$lock_files" == "true" ]; then
-        if [ "$is_specific" = true ]; then
-          mv "$file" "${directory}/Pipfile"
-          mv "${directory}/$lockfile" "${directory}/Pipfile.lock"
-          (cd "$directory" && pipenv lock)
-          mv "${directory}/Pipfile" "$file"
-          mv "${directory}/Pipfile.lock" "${directory}/$lockfile"
-        else
-          (cd "$directory" && pipenv lock)
-        fi
+    elif [[ $current_line =~ version[[:space:]]*=[[:space:]]*\"?([=><!~]*)?([0-9]+(\.[0-9]+)*)\" ]]; then
+      new_line=$(echo "$current_line" | sed -E "s/(version[[:space:]]*=[[:space:]]*\")([=><!~]*[0-9]+(\.[0-9]+)*)\"/\1${current_qualifier}${new_ver}\"/")
+      sed -i "s/^$current_line/$new_line/" "$file"
+    fi
+    echo "Updated $lib in $file to version ${current_qualifier}${new_ver}"
+
+    # Handle renaming and pipenv lock, if necessary
+    if [ "$lock_files" == "true" ]; then
+      if [ "$is_specific" = true ]; then
+        mv "$file" "${directory}/Pipfile"
+        mv "${directory}/$lockfile" "${directory}/Pipfile.lock"
+        (cd "$directory" && pipenv lock)
+        mv "${directory}/Pipfile" "$file"
+        mv "${directory}/Pipfile.lock" "${directory}/$lockfile"
+      else
+        (cd "$directory" && pipenv lock)
       fi
-    else
-      echo "$lib in $file is already up-to-date or has a higher version ($current_ver)."
     fi
   else
-    echo "$lib not found in $file."
+    echo "$lib in $file is already up-to-date or has a higher version ($current_ver)."
   fi
 }