Poc runner checkout merge refpoc-runner-checkout-merge-ref

1 files changed, 9 insertions, 2 deletions

diff --git a/lib/api/entities.rb b/lib/api/entities.rb
index 22403664c21..2472eb72b85 100644
--- a/lib/api/entities.rb
+++ b/lib/api/entities.rb
@@ -1312,12 +1312,19 @@ module API
       class GitInfo < Grape::Entity
         expose :repo_url, :ref, :sha, :before_sha
         expose :ref_type do |model|
-          if model.tag
+          if model.pipeline.branch?
+            'branch'
+          elsif model.pipeline.merge_request?
+            'merge-requests'
+          elsif model.pipeline.tag?
             'tag'
           else
-            'branch'
+            raise ArgumentError, 'Invalid ref type!'
           end
         end
+        expose :merge_request_iid, if: ->(job, _) { job.pipeline.merge_request? } do
+          job.pipeline.merge_request.iid
+        end
       end
 
       class RunnerInfo < Grape::Entity