app/controllers/concerns/snippets/blobs_actions.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

# frozen_string_literal: true

module Snippets::BlobsActions
  extend ActiveSupport::Concern

  include Gitlab::Utils::StrongMemoize
  include ExtractsRef
  include Snippets::SendBlob

  included do
    before_action :authorize_read_snippet!, only: [:raw]
    before_action :ensure_repository
    before_action :ensure_blob
  end

  def raw
    send_snippet_blob(snippet, blob)
  end

  private

  def repository_container
    snippet
  end

  # rubocop:disable Gitlab/ModuleWithInstanceVariables
  def blob
    strong_memoize(:blob) do
      assign_ref_vars

      next unless @commit

      @repo.blob_at(@commit.id, @path)
    end
  end
  # rubocop:enable Gitlab/ModuleWithInstanceVariables

  def ensure_blob
    render_404 unless blob
  end

  def ensure_repository
    unless snippet.repo_exists?
      Gitlab::AppLogger.error(message: "Snippet raw blob attempt with no repo", snippet: snippet.id)

      respond_422
    end
  end

  def snippet_id
    params[:snippet_id]
  end
end

Snippets::BlobsActions.prepend_mod