feat(script): allow script stages to customize branch

spinnaker · Apr 4, 2017 · f842e16 · f842e16
1 parent 9e9c0c0
commit f842e16
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/orca-igor/src/main/groovy/com/netflix/spinnaker/orca/igor/tasks/StartScriptTask.groovy b/orca-igor/src/main/groovy/com/netflix/spinnaker/orca/igor/tasks/StartScriptTask.groovy
@@ -52,6 +52,7 @@ class StartScriptTask implements Task {
     String cluster = stage.context.cluster
     String cmc = stage.context.cmc
     String repoUrl = stage.context.repoUrl
+    String repoBranch = stage.context.repoBranch
 
     if (stage.execution instanceof Pipeline && stage.execution.trigger.parameters?.strategy == true) {
       Map trigger = ((Pipeline) stage.execution).trigger
@@ -75,6 +76,10 @@ class StartScriptTask implements Task {
       parameters.REPO_URL = repoUrl
     }
 
+    if (repoBranch) {
+      parameters.REPO_BRANCH = repoBranch
+    }
+
     String queuedBuild = buildService.build(master, job, parameters)
     new TaskResult(ExecutionStatus.SUCCEEDED, [master: master, job: job, queuedBuild: queuedBuild])
   }