From c29d8b19a6d15e4ed83d3084c5c6e54b0d6b6f6e Mon Sep 17 00:00:00 2001
From: Leandro Mendes <lmendes@redhat.com>
Date: Mon, 21 Aug 2023 15:24:10 +0200
Subject: [PATCH] feat(HACBS-2227): snapshot data model can be used in update
 expressions

This PR adds the update-paths script to enable updating
yq expression in fileUpdates paths strings with the result
of these expressions

Signed-off-by: Leandro Mendes <lmendes@redhat.com>
---
 utils/update-paths | 88 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100755 utils/update-paths

diff --git a/utils/update-paths b/utils/update-paths
new file mode 100755
index 00000000..96fcb205
--- /dev/null
+++ b/utils/update-paths
@@ -0,0 +1,88 @@
+#!/bin/bash
+#
+# script:      update-paths
+# 
+# description: This script searches for "${{ }}" enclosed yq expression in fileUpdates paths then
+#              executes and replaces it with its result
+#
+# parameters:  
+#
+# -p, --paths  Json String containing fileUpdates paths with optional yq expression(s) enclosed by "${{ }}"
+#              The `paths` json string should be properly quoted
+#              Example: --paths="[{\"path\":\"foo.yaml\",\"replacements\":[{\"key\":\"bar\",\"replacement\":\"\${{'.components[0].containerImage'}}\"}]}]"
+#
+# -f, --file:  Path to the mapping_snapshots.json file
+#
+print_help(){
+    echo -e "$0 [ -p, --paths ] PATHS [ -f, --file ] FILE\n
+            \t-p, --paths\tJson String containing fileUpdates paths with optional yq expression(s) enclosed by \"\${{ }}\"
+            \t\t\tThe \`paths\` json string should be properly escaped/quoted
+            \t\t\tExample: --paths=\"[{\\\"path\\\":\\\"foo.yaml\\\",\\\"replacements\\\":[{\\\"key\\\":\\\"bar\\\",\\\"replacement\\\":\\\"\\\${{'.components[0].containerImage'}}\\\"}]}]\"\n
+            \t-f, --file\tPath to mapping_snapshots.json file\n"
+}
+
+OPTIONS=$(getopt -l "paths:,file:" -o "p:f:" -a -- "$@")
+eval set -- "$OPTIONS"
+while true; do
+    case "$1" in
+        -p|--paths)
+            shift
+            PATHS=$1
+            ;;
+        -f|--file)
+            shift
+            FILE="$1"
+            ;;
+        --)
+            shift
+            break
+            ;;
+    esac
+    shift
+done
+
+if [ -z "${PATHS}" ] || [ -z "${FILE}" ]; then
+    print_help
+    exit
+fi
+
+newPaths=('{"paths": []}')
+
+IFS=$'\n' # This is required because the JSON string might contain spaces
+for path in $(jq -cr '.[]' <<< "${PATHS}"); do
+  newPath=$(jq '{"path": .path}' <<< "${path}")
+  newReplacements=('{"replacements": []}')
+
+  replacementsLength=$(jq '.replacements | length' <<< "${path}")
+  for(( i=0; i<replacementsLength; i++ )); do
+      replacement=$(jq -cr ".replacements[${i}]" <<< "${path}")
+      # Here we are searching for the enclosed by "${{ }}" yq exp to later execute it.
+      # The PADDING = 4 means the begining of the expression "${{" +1 (the "'" char)
+      # to positioning it at the start of the yq expression.
+      yqExp=$(awk '{
+          PADDING = 4
+          match($0, "\\$\\{{2}.*\\}{2}")
+
+          EXP = substr($0, RSTART + PADDING, RLENGTH - (PADDING*2)+1)
+      } { 
+          gsub(/\\/, "", EXP); print EXP
+      }' <<< "${replacement}")
+
+      if [ -n "${yqExp}" ]; then
+          # execute the yq expression and replace the "replacement string" with the result
+          yqResult=$(yq "${yqExp}" "${FILE}")
+          newReplacement=$(sed "s|\${{.*}}|"${yqResult}"|g" <<< "${replacement}")
+          # appends to the newReplacements array
+          newReplacements[0]=$(jq -cr ".replacements += [${newReplacement}]" <<< "${newReplacements[0]}")
+      else
+          newReplacements[0]=$(jq -cr ".replacements += [${replacement}]" <<< "${newReplacements[0]}")
+      fi
+
+  done
+
+  # merge the two hashes and append to the newPaths array
+  merged=$(jq -cr -s add  <<< "${newPath[0]} ${newReplacements[0]}")
+  newPaths[0]=$(jq -cr ".paths += [ ${merged} ]" <<< "${newPaths[0]}")
+done
+
+jq -cr '.paths' <<< "${newPaths[0]}"