From c9fd4380807ec23f0f7f91ea0f97c149ed502cd8 Mon Sep 17 00:00:00 2001
From: Balasankar C <balasankar@gitlab.com>
Date: Fri, 1 Dec 2017 10:57:12 +0530
Subject: [PATCH] Set EE variable

---
 scripts/trigger-build-omnibus | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/scripts/trigger-build-omnibus b/scripts/trigger-build-omnibus
index 0c662ac19d2..3c5c22c9372 100755
--- a/scripts/trigger-build-omnibus
+++ b/scripts/trigger-build-omnibus
@@ -30,12 +30,16 @@ module Omnibus
 
     private
 
+    def ee?
+      File.exist?('CHANGELOG-EE.md')
+    end
+
     def env_params
       {
         "ref" => ENV["OMNIBUS_BRANCH"] || "master",
         "variables[GITLAB_VERSION]" => ENV["CI_COMMIT_SHA"],
         "variables[ALTERNATIVE_SOURCES]" => true,
-        "variables[ee]" => ENV["EE_PACKAGE"] || "false"
+        "variables[ee]" => ee? ? 'true' : 'false'
       }
     end