config.rb in vidar-1.4.0

- old
+ new

@@ -1,12 +1,13 @@
 module Vidar
   class Config
     DEFAULT_MANIFEST_FILE = "vidar.yml".freeze
+    DEFAULT_BRANCHES = %w[main master].freeze
 
     DEFAULT_OPTIONS = {
       compose_file:       -> { "docker-compose.ci.yml" },
-      default_branch:     -> { "master" },
+      default_branch:     -> { (DEFAULT_BRANCHES & branches).first || DEFAULT_BRANCHES.first },
       current_branch:     -> { `git rev-parse --abbrev-ref HEAD`.strip.tr("/", "-") },
       revision:           -> { `git rev-parse HEAD`.strip },
       revision_name:      -> { `git show --pretty=format:"%s (%h)" -s HEAD`.strip },
       kubectl_context:    -> { `kubectl config current-context`.strip },
       shell_command:      -> { "/bin/sh" },
@@ -69,9 +70,13 @@
 
         deployment.transform_keys!(&:to_sym)
         deployment.transform_values! { |value| Vidar::Interpolation.call(value, self) }
 
         DeployConfig.new(deployment)
+      end
+
+      def branches
+        `git for-each-ref --format='%(refname:short)' refs/heads/*`.split("\n")
       end
 
       def default_branch?
         get!(:current_branch) == get!(:default_branch)
       end