Feature/block96 Undo the changes made in case of pipeline generation failure

scripts/pipelines/azure-devops/pipeline_generator.sh

@@ -43,6 +43,9 @@ red='\e[0;31m'
 # Common var
 commonTemplatesPath="scripts/pipelines/azure-devops/templates/common"
 
+# Undo-Level for the script. Used to clean up the resources in case of a failure
+undoStage=0
+
 function help {
     echo ""
     echo "Generates a pipeline on Azure DevOps based on the given definition."
@@ -152,11 +155,18 @@ function createNewBranch {
     echo -e "${green}Creating the new branch: ${sourceBranch}..."
     echo -ne ${white}
 
-    # Create the new branch.
     cd "${localDirectory}"
+
+    # store current branch into a variable (only used for rollback/undo) 
+    originalBranch=$(git branch --show-current)
+
     [ $? != "0" ] && echo -e "${red}The local directory: '${localDirectory}' cannot be found, please check the path." && exit 1
 
+    # Create the new branch.
     git checkout -b ${sourceBranch}
+
+    # set undo-stage (clear local branches)
+    undoStage=1
 }
 
 function copyYAMLFile {
@@ -201,14 +211,30 @@ function commitCommonFiles {
 
     git commit -m "Adding the source YAML"
     git push -u origin ${sourceBranch}
+
+    # clean up remote-branches
+    undoStage=2
 }
 
 function createPipeline {
     echo -e "${green}Generating the pipeline from the YAML template..."
     echo -ne ${white}
 
     # Create Azure Pipeline
-    az pipelines create --name $pipelineName --yml-path "${pipelinePath}/${yamlFile}" --skip-first-run true
+
+    pipelineResult=$(az pipelines create --name $pipelineName --yml-path "${pipelinePath}/${yamlFile}" --skip-first-run true) || {
+        # if the script fails, clean pollution
+        echo "There was an error creating the pipeline! Please check if you specified the name and got all settings right!"
+
+        undoPreviousSteps
+
+        exit 127
+    }
+
+    pipelineId=$(echo "$pipelineResult" | python -c "import sys, json; print(json.load(sys.stdin)['id'])")
+
+    # pipeline got created successfully     
+    undoStage=3
 }
 
 # Function that adds the variables to be used in the pipeline.
@@ -218,7 +244,15 @@ function addCommonPipelineVariables {
         echo "Skipping creation of the variable artifactPath as the flag has not been used."
     else
         # Add the extra artifact to store variable.
-        az pipelines variable create --name "artifactPath" --pipeline-name "$pipelineName" --value "${artifactPath}"
+        variableResult=$(az pipelines variable create --pipeline-name "$pipelineName" --value "${artifactPath}") || {
+            # if the variable-creation fails, clean up
+
+            echo "There was an error creating the pipeline-artifact-path variable!. Exiting"
+
+            undoPreviousSteps
+
+            exit 127
+        }
     fi
 }
 
@@ -272,6 +306,66 @@ function createPR {
     fi
 }
 
+function removeLocalBranches {
+    # visit the directory and switch to the branch which was present before
+    cd "${localDirectory}"
+
+    git checkout $originalBranch
+
+    git branch -D ${sourceBranch}
+}
+
+function removeRemoteBranches {
+    cd "${localDirectory}"
+
+    # update list of remotes
+    git fetch
+
+    # delete mapped remote-branch
+    git push origin --delete ${sourceBranch}
+}
+
+function removePipelineFiles {
+    # clear the generated pipeline-files
+    cd "${localDirectory}"
+
+    if [ -d "./pipelines" ]; then
+        rm -rf ./pipelines
+    fi
+}
+
+function removePipeline {
+    az pipelines delete --id ${pipelineId}
+}
+
+function abandonPullRequest {
+    az repos pr update --id id --status abandoned
+}
+
+function undoPreviousSteps {
+    # free all resources
+
+    if [ ${undoStage} -gt 2 ]; then
+        echo "Removing pipeline with id: ${pipelineId}"
+
+        removePipeline
+    fi
+
+    if [ ${undoStage} -gt 1 ]; then
+        echo "Removing all pipeline-files and remote branches!"
+
+        removePipelineFiles
+        removePullRequest
+        removeRemoteBranches
+    fi
+
+    if [ ${undoStage} -gt 0 ]; then 
+        echo "Removing all local branches!"
+
+        removeLocalBranches
+    fi
+}  
+
 if [[ "$help" == "true" ]]; then help; fi
 
 importConfigFile