1 files changed, 13 insertions, 0 deletions
diff --git a/scripts/lib/gitlab.rb b/scripts/lib/gitlab.rb
new file mode 100644
index 00000000000..556e2037edf
--- /dev/null
+++ b/scripts/lib/gitlab.rb
@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module_function
+
+  def ee?
+    File.exist?(File.expand_path('../../ee/app/models/license.rb', __dir__)) && !%w[true 1].include?(ENV['FOSS_ONLY'].to_s)
+  end
+
+  def jh?
+    ee? && Dir.exist?(File.expand_path('../../jh', __dir__)) && !%w[true 1].include?(ENV['EE_ONLY'].to_s)
+  end
+end