From e1fb8b35e7a298233fc635c5ec4be30debc1045c Mon Sep 17 00:00:00 2001
From: Achilleas Pipinellis <axil@gitlab.com>
Date: Wed, 15 Nov 2017 13:05:53 +0100
Subject: [PATCH] Change logic of repo name and slugs in docs review script

We need to make room for Runner and Omnibus docs.
This change makes sure the correct values are defined
based on CI_PROJECT_NAME.
---
 scripts/trigger-build-docs | 35 ++++++++++++++++++++++++-----------
 1 file changed, 24 insertions(+), 11 deletions(-)

diff --git a/scripts/trigger-build-docs b/scripts/trigger-build-docs
index d3a9f5ff4ea..89ad6a99467 100755
--- a/scripts/trigger-build-docs
+++ b/scripts/trigger-build-docs
@@ -30,13 +30,6 @@ def docs_branch
   "preview-#{ENV["CI_COMMIT_REF_SLUG"]}"[0...max]
 end
 
-#
-# Dummy way to find out in which repo we are, CE or EE
-#
-def ee?
-  File.exist?('CHANGELOG-EE.md')
-end
-
 #
 # Create a remote branch in gitlab-docs
 #
@@ -55,15 +48,35 @@ def remove_remote_branch
   puts "Remote branch '#{docs_branch}' deleted"
 end
 
+#
+# Define suffix in review app URL based on project
+#
+def slug
+  case ENV["CI_PROJECT_NAME"]
+  when 'gitlab-ce'
+    'ce'
+  when 'gitlab-ee'
+    'ee'
+  when 'gitlab-runner'
+    'runner'
+  when 'omnibus-gitlab'
+    'omnibus'
+  end
+end
+
+#
+# Overriding vars in https://gitlab.com/gitlab-com/gitlab-docs/blob/master/.gitlab-ci.yml
+#
+def param_name
+  "BRANCH_#{slug.upcase}"
+end
+
 #
 # Trigger a pipeline in gitlab-docs
 #
 def trigger_pipeline
-  # Overriding vars in https://gitlab.com/gitlab-com/gitlab-docs/blob/master/.gitlab-ci.yml
-  param_name = ee? ? 'BRANCH_EE' : 'BRANCH_CE'
-
   # The review app URL
-  app_url = "http://#{docs_branch}.#{ENV["DOCS_REVIEW_APPS_DOMAIN"]}/#{ee? ? 'ee' : 'ce'}"
+  app_url = "http://#{docs_branch}.#{ENV["DOCS_REVIEW_APPS_DOMAIN"]}/#{slug}"
 
   # Create the pipeline
   puts "=> Triggering a pipeline..."