Hide survey banner on stable branches

We don't want the banner to be shown on stable branches.
This introduces a new function that checks if CI_COMMIT_REF_NAME
is defined, and if not, it uses Git to find the current branch name.

If show_banner is false and current_branch is not a stable branch,
hide the banner.

1 files changed, 12 insertions, 0 deletions

diff --git a/lib/helpers/generic.rb b/lib/helpers/generic.rb
index 6dfcaf96..264ba497 100644
--- a/lib/helpers/generic.rb
+++ b/lib/helpers/generic.rb
@@ -25,5 +25,17 @@ module Nanoc::Helpers
     def omnibus?
       ENV['NANOC_ENV'] == 'omnibus'
     end
+
+    #
+    # Find the current branch. If CI_COMMIT_BRANCH is not defined, that means
+    # we're working locally, and Git is used to find the branch.
+    #
+    def current_branch
+      if ENV['CI_COMMIT_REF_NAME'].nil?
+        `git branch --show-current`.tr("\n", '')
+      else
+        ENV['CI_COMMIT_REF_NAME']
+      end
+    end
   end
 end