From 7eb2617e48449080f1f50d754b5268dd8a440ca6 Mon Sep 17 00:00:00 2001
From: Piotr Sarnacki <drogus@gmail.com>
Date: Tue, 24 Nov 2015 15:50:38 +0100
Subject: [PATCH] Add branch_is_default info to commits

---
 lib/travis/api/v2/http/build.rb     | 9 +++++++--
 lib/travis/api/v2/http/job.rb       | 9 +++++++--
 spec/unit/api/v2/http/build_spec.rb | 1 +
 spec/unit/api/v2/http/job_spec.rb   | 1 +
 4 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/lib/travis/api/v2/http/build.rb b/lib/travis/api/v2/http/build.rb
index 9d8203b5..394c4ac8 100644
--- a/lib/travis/api/v2/http/build.rb
+++ b/lib/travis/api/v2/http/build.rb
@@ -17,7 +17,7 @@ module Travis
           def data
             {
               'build'  => build_data(build),
-              'commit' => commit_data(build.commit),
+              'commit' => commit_data(build.commit, build.repository),
               'jobs'   => options[:include_jobs] ? build.matrix.map { |job| job_data(job) } : [],
               'annotations' => options[:include_jobs] ? Annotations.new(annotations(build), @options).data["annotations"] : [],
             }
@@ -44,11 +44,12 @@ module Travis
               }
             end
 
-            def commit_data(commit)
+            def commit_data(commit, repository)
               {
                 'id' => commit.id,
                 'sha' => commit.commit,
                 'branch' => commit.branch,
+                'branch_is_default' => branch_is_default(commit, repository),
                 'message' => commit.message,
                 'committed_at' => format_date(commit.committed_at),
                 'author_name' => commit.author_name,
@@ -78,6 +79,10 @@ module Travis
               }
             end
 
+            def branch_is_default(commit, repository)
+              repository.default_branch == commit.branch
+            end
+
             def annotations(build)
               build.matrix.map(&:annotations).flatten
             end
diff --git a/lib/travis/api/v2/http/job.rb b/lib/travis/api/v2/http/job.rb
index 4b712b40..92366474 100644
--- a/lib/travis/api/v2/http/job.rb
+++ b/lib/travis/api/v2/http/job.rb
@@ -15,7 +15,7 @@ module Travis
           def data
             {
               'job' => job_data(job),
-              'commit' => commit_data(job.commit),
+              'commit' => commit_data(job.commit, job.repository),
               'annotations' => Annotations.new(job.annotations, @options).data["annotations"],
             }
           end
@@ -42,11 +42,12 @@ module Travis
               }
             end
 
-            def commit_data(commit)
+            def commit_data(commit, repository)
               {
                 'id' => commit.id,
                 'sha' => commit.commit,
                 'branch' => commit.branch,
+                'branch_is_default' => branch_is_default(commit, repository),
                 'message' => commit.message,
                 'committed_at' => format_date(commit.committed_at),
                 'author_name' => commit.author_name,
@@ -56,6 +57,10 @@ module Travis
                 'compare_url' => commit.compare_url,
               }
             end
+
+            def branch_is_default(commit, repository)
+              repository.default_branch == commit.branch
+            end
         end
       end
     end
diff --git a/spec/unit/api/v2/http/build_spec.rb b/spec/unit/api/v2/http/build_spec.rb
index 11a6dbea..f276b2b8 100644
--- a/spec/unit/api/v2/http/build_spec.rb
+++ b/spec/unit/api/v2/http/build_spec.rb
@@ -29,6 +29,7 @@ describe Travis::Api::V2::Http::Build do
       'id' => 1,
       'sha' => '62aae5f70ceee39123ef',
       'branch' => 'master',
+      'branch_is_default' => true,
       'message' => 'the commit message',
       'compare_url' => 'https://github.com/svenfuchs/minimal/compare/master...develop',
       'committed_at' => json_format_time(Time.now.utc - 1.hour),
diff --git a/spec/unit/api/v2/http/job_spec.rb b/spec/unit/api/v2/http/job_spec.rb
index a9211b6f..22fda7b6 100644
--- a/spec/unit/api/v2/http/job_spec.rb
+++ b/spec/unit/api/v2/http/job_spec.rb
@@ -31,6 +31,7 @@ describe Travis::Api::V2::Http::Job do
       'sha' => '62aae5f70ceee39123ef',
       'message' => 'the commit message',
       'branch' => 'master',
+      'branch_is_default' => true,
       'message' => 'the commit message',
       'committed_at' => json_format_time(Time.now.utc - 1.hour),
       'committer_name' => 'Sven Fuchs',