Basic ref fetching for commitsapi-refs-for-commit

2 files changed, 32 insertions, 3 deletions

diff --git a/lib/api/commits.rb b/lib/api/commits.rb
index 2685dc27252..14d9a3ee985 100644
--- a/lib/api/commits.rb
+++ b/lib/api/commits.rb
@@ -155,6 +155,31 @@ module API
         end
       end
 
+      desc 'Get all reference a commit is pushed to' do
+        detail 'This feature was introduced in GitLab 10.2'
+        success Entities::Ref
+      end
+      params do
+        requires :sha, type: String, desc: 'A commit sha'
+        optional :type, type: String, values: %w(branches tags all), default: 'all', desc: 'Scope'
+      end
+      get ':id/repository/commits/:sha/refs', requirements: API::COMMIT_ENDPOINT_REQUIREMENTS do
+        commit = user_project.commit(params[:sha])
+        not_found!('Commit') unless commit
+
+        case params[:type]
+        when 'branches'
+          refs = user_project.repository.branch_names_contains(commit.id)
+        when 'tags'
+          refs = user_project.repository.tag_names_contains(commit.id)
+        else
+          refs = user_project.repository.branch_names_contains(commit.id)
+          refs << user_project.repository.branch_names_contains(commit.id)
+        end
+
+        present refs, with: Entities::BasicRef
+      end
+
       desc 'Post comment to commit' do
         success Entities::CommitNote
       end
diff --git a/lib/api/entities.rb b/lib/api/entities.rb
index 5f0bad14839..ad3ca18db3c 100644
--- a/lib/api/entities.rb
+++ b/lib/api/entities.rb
@@ -238,7 +238,11 @@ module API
       expose :last_pipeline, using: 'API::Entities::PipelineBasic'
     end
 
-    class Branch < Grape::Entity
+    class BasicRef < Grape::Entity
+      expose :name
+    end
+
+    class Branch < BasicRef
       expose :name
 
       expose :commit, using: Entities::Commit do |repo_branch, options|
@@ -774,8 +778,8 @@ module API
       expose :description
     end
 
-    class Tag < Grape::Entity
-      expose :name, :message
+    class Tag < BasicRef
+      expose :message
 
       expose :commit, using: Entities::Commit do |repo_tag, options|
         options[:project].repository.commit(repo_tag.dereferenced_target)