Merge pull request #316 from github/solmaz/medium-restore

[PERF] Provide version and size input for restore

Author: solmazabbaspour

.github/workflows/restore.yml

@@ -19,13 +19,28 @@ on:
         required: false
         type: string
         default: 'master'
+      version:
+        description: 'Version of the dataset to restore (3.8/3.9)'
+        required: false
+        default: '3.8'
+        type: string
 
   workflow_call:
     inputs:
       hostname:
         description: 'Hostname of the server'
         required: true
         type: string
+      size:
+        description: 'Size of the dataset to restore (small/medium)'
+        required: false
+        default: 'small'
+        type: string
+      version:
+        description: 'Version of the dataset to restore (3.8/3.9)'
+        required: false
+        default: '3.8'
+        type: string
       ref:
         description: 'Branch ref to use'
         required: false
@@ -63,22 +78,41 @@ jobs:
     - name: Load docker container
       run: docker load -i backup-utils.tar
 
+    - name: Find backup file version
+      id: file
+      run: |
+        version="${{ inputs.version }}"
+        size="${{ inputs.size }}"
+        V3_8_COMPATIBLE="3.6 3.7 3.8 3.9 3.10"
+        echo "$V3_8_COMPATIBLE" | tr " " '\n' | grep -F -q -x "$version"
+        exit_code="$?"
+        if [ "$exit_code" -eq "0" ]; then
+            echo "Version $version is acceptable"
+            file_version=3.8
+            echo "version=3.8" >> "$GITHUB_OUTPUT"
+            echo "name=v$file_version-$size.tar.gz" >> "$GITHUB_OUTPUT"
+        else
+            echo "Version $version is not acceptable"
+            exit 1
+        fi
+
     - name: Download from blob storage
       run: |
         mkdir ghes-data
         az storage blob download \
         --account-name ghesresults \
         --container-name ghes-data \
-        --name v3.8-small-docs.tar.gz \
-        --file ghes-data/3.8-small-docs.tar.gz \
+        --name "${{ steps.file.outputs.name }}" \
+        --file "ghes-data/${{ steps.file.outputs.name }}" \
         --connection-string "${{ secrets.CONNECTIONSTRING }}"
 
     - name: Unzip backup and setup symlink
       run: |
           mkdir "$HOME/ghe-backup-data"
           dir_name=$(date +%s)
           mkdir "$HOME/ghe-backup-data/$dir_name"
-          tar -xvf "ghes-data/3.8-small-docs.tar.gz" -C "$HOME/ghe-backup-data/$dir_name"
+
+          tar -xvf "ghes-data/${{ steps.file.outputs.name }}" -C "$HOME/ghe-backup-data/$dir_name"
 
           ln -s "$dir_name" "$HOME/ghe-backup-data/current"
 
@@ -89,7 +123,7 @@ jobs:
       run: chmod 0600 "$HOME/backup"
 
     - name: change version
-      run: echo "3.8.0" > "$HOME/version"
+      run: echo "${{ inputs.version }}.0" > "$HOME/version"
 
     - name: Prepare for restore
       run: ssh -p122 -i "$HOME/backup" -o StrictHostKeyChecking=no admin@${{ inputs.hostname }} "ghe-maintenance -s"