Pull the stable product branches when on a stable docs branch

This ensures that when on a docs stable-version branch,
the respective product branches are set up correctly.

Closes https://gitlab.com/gitlab-com/gitlab-docs/issues/324

1 files changed, 15 insertions, 1 deletions

diff --git a/lib/task_helpers.rb b/lib/task_helpers.rb
index 29e7e108..c9c78472 100644
--- a/lib/task_helpers.rb
+++ b/lib/task_helpers.rb
@@ -1,6 +1,7 @@
 require 'yaml'
 
 PRODUCTS = %W[ce ee omnibus runner debug].freeze
+VERSION_FORMAT = /^(\d{1,2})\.(\d{1,2})$/
 
 def config
   # Parse the config file and create a hash.
@@ -17,7 +18,20 @@ def products
 end
 
 def retrieve_branch(slug)
-  ENV.fetch("BRANCH_#{slug.upcase}", 'master')
+  # If we're on a stable branch, catch the version and
+  # assign the right BRANCH_* variable.
+  if version = ENV["CI_COMMIT_REF_NAME"].match(VERSION_FORMAT)
+    case slug
+    when 'ee'
+      "#{version[1]}-#{version[2]}-stable-ee"
+    when 'ce', 'omnibus', 'runner'
+      "#{version[1]}-#{version[2]}-stable"
+    else
+      'master'
+    end
+  else
+    ENV.fetch("BRANCH_#{slug.upcase}", 'master')
+  end
 end
 
 def git_workdir_dirty?