moved lfs_blob_ids method into ExtractsPath modulelfs-badge

3 files changed, 6 insertions, 8 deletions

diff --git a/app/controllers/projects/tree_controller.rb b/app/controllers/projects/tree_controller.rb
index 2998ec21dca..f752a46f828 100644
--- a/app/controllers/projects/tree_controller.rb
+++ b/app/controllers/projects/tree_controller.rb
@@ -26,10 +26,8 @@ class Projects::TreeController < Projects::ApplicationController
 
     respond_to do |format|
       format.html do
-        blob_ids = tree.blobs.map(&:id)
-
+        lfs_blob_ids
         @last_commit = @repository.last_commit_for_path(@commit.id, @tree.path) || @commit
-        @lfs_blob_ids = Gitlab::Git::Blob.batch_lfs_pointers(@repo, blob_ids).map(&:id)
       end
 
       format.js do
diff --git a/app/controllers/projects_controller.rb b/app/controllers/projects_controller.rb
index bce3a8d42e4..b06981b4b1d 100644
--- a/app/controllers/projects_controller.rb
+++ b/app/controllers/projects_controller.rb
@@ -404,9 +404,4 @@ class ProjectsController < Projects::ApplicationController
     #
     redirect_to request.original_url.sub(/\.git\/?\Z/, '') if params[:format] == 'git'
   end
-
-  def lfs_blob_ids
-    blob_ids = tree.blobs.map(&:id)
-    @lfs_blob_ids = Gitlab::Git::Blob.batch_lfs_pointers(@repo, blob_ids).map(&:id)
-  end
 end
diff --git a/lib/extracts_path.rb b/lib/extracts_path.rb
index f576ef74603..27c712a84d4 100644
--- a/lib/extracts_path.rb
+++ b/lib/extracts_path.rb
@@ -135,6 +135,11 @@ module ExtractsPath
     @tree ||= @repo.tree(@commit.id, @path)
   end
 
+  def lfs_blob_ids
+    blob_ids = tree.blobs.map(&:id)
+    @lfs_blob_ids = Gitlab::Git::Blob.batch_lfs_pointers(@project.repository, blob_ids).map(&:id)
+  end
+
   private
 
   # overriden in subclasses, do not remove