Allow FoundBlob to access language from gitattributes

Extract language_from_git_attributes as a concern so it can
ben included in two blob classes.

1 files changed, 5 insertions, 6 deletions

diff --git a/lib/gitlab/search_results.rb b/lib/gitlab/search_results.rb
index 6c86ad11385..b46c18c4364 100644
--- a/lib/gitlab/search_results.rb
+++ b/lib/gitlab/search_results.rb
@@ -5,8 +5,9 @@ module Gitlab
     class FoundBlob
       include EncodingHelper
       include Presentable
+      include BlobLanguageFromGitAttributes
 
-      attr_reader :id, :filename, :basename, :ref, :startline, :data, :project_id
+      attr_reader :id, :filename, :basename, :ref, :startline, :data, :project
 
       def initialize(opts = {})
         @id = opts.fetch(:id, nil)
@@ -16,17 +17,15 @@ module Gitlab
         @startline = opts.fetch(:startline, nil)
         @data = encode_utf8(opts.fetch(:data, nil))
         @per_page = opts.fetch(:per_page, 20)
-        @project_id = opts.fetch(:project_id, nil)
+        @project = opts.fetch(:project, nil)
       end
 
       def path
         filename
       end
 
-      # Since search results often contain many items,
-      # not triggering lookup can avoid n+1 queries.
-      def language_from_gitattributes
-        nil
+      def project_id
+        @project&.id
       end
 
       def present