blob.rb 5,1 КБ
Newer Older
Robert Speicher's avatar
Robert Speicher включено в состав коммита
1
2
# Blob is a Rails-specific wrapper around Gitlab::Git::Blob objects
class Blob < SimpleDelegator
Jacob Vosmaer's avatar
Jacob Vosmaer включено в состав коммита
3
4
5
  CACHE_TIME = 60 # Cache raw blobs referred to by a (mutable) ref for 1 minute
  CACHE_TIME_IMMUTABLE = 3600 # Cache blobs referred to by an immutable reference for 1 hour

Douwe Maan's avatar
Douwe Maan включено в состав коммита
6
7
  MAXIMUM_TEXT_HIGHLIGHT_SIZE = 1.megabyte

Douwe Maan's avatar
Douwe Maan включено в состав коммита
8
9
10
11
12
13
14
15
16
17
18
19
20
21
  # Finding a viewer for a blob happens based only on extension and whether the
  # blob is binary or text, which means 1 blob should only be matched by 1 viewer,
  # and the order of these viewers doesn't really matter.
  #
  # However, when the blob is an LFS pointer, we cannot know for sure whether the
  # file being pointed to is binary or text. In this case, we match only on
  # extension, preferring binary viewers over text ones if both exist, since the
  # large files referred to in "Large File Storage" are much more likely to be
  # binary than text.
  #
  # `.stl` files, for example, exist in both binary and text forms, and are
  # handled by different viewers (`BinarySTL` and `TextSTL`) depending on blob
  # type. LFS pointers to `.stl` files are assumed to always be the binary kind,
  # and use the `BinarySTL` viewer.
Douwe Maan's avatar
Douwe Maan включено в состав коммита
22
  RICH_VIEWERS = [
Douwe Maan's avatar
Douwe Maan включено в состав коммита
23
24
25
26
    BlobViewer::Markup,
    BlobViewer::Notebook,
    BlobViewer::SVG,

Douwe Maan's avatar
Douwe Maan включено в состав коммита
27
28
    BlobViewer::Image,
    BlobViewer::Sketch,
Luke "Jared" Bennett's avatar
Luke "Jared" Bennett включено в состав коммита
29
    BlobViewer::Balsamiq,
Douwe Maan's avatar
Douwe Maan включено в состав коммита
30

Douwe Maan's avatar
Douwe Maan включено в состав коммита
31
    BlobViewer::Video,
Douwe Maan's avatar
Douwe Maan включено в состав коммита
32

Douwe Maan's avatar
Douwe Maan включено в состав коммита
33
34
    BlobViewer::PDF,

Douwe Maan's avatar
Douwe Maan включено в состав коммита
35
    BlobViewer::BinarySTL,
Rémy Coutable's avatar
Rémy Coutable включено в состав коммита
36
    BlobViewer::TextSTL
Douwe Maan's avatar
Douwe Maan включено в состав коммита
37
  ].sort_by { |v| v.binary? ? 0 : 1 }.freeze
Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
38

Douwe Maan's avatar
Douwe Maan включено в состав коммита
39
40
41
  AUXILIARY_VIEWERS = [
    BlobViewer::GitlabCiYml,
    BlobViewer::RouteMap,
Douwe Maan's avatar
Douwe Maan включено в состав коммита
42

Douwe Maan's avatar
Douwe Maan включено в состав коммита
43
    BlobViewer::Readme,
Douwe Maan's avatar
Douwe Maan включено в состав коммита
44
    BlobViewer::License,
Douwe Maan's avatar
Douwe Maan включено в состав коммита
45
46
    BlobViewer::Contributing,
    BlobViewer::Changelog
Douwe Maan's avatar
Douwe Maan включено в состав коммита
47
  ].freeze
Douwe Maan's avatar
Douwe Maan включено в состав коммита
48

Douwe Maan's avatar
Douwe Maan включено в состав коммита
49
50
  attr_reader :project

Robert Speicher's avatar
Robert Speicher включено в состав коммита
51
52
53
54
55
56
57
58
59
60
  # Wrap a Gitlab::Git::Blob object, or return nil when given nil
  #
  # This method prevents the decorated object from evaluating to "truthy" when
  # given a nil value. For example:
  #
  #     blob = Blob.new(nil)
  #     puts "truthy" if blob # => "truthy"
  #
  #     blob = Blob.decorate(nil)
  #     puts "truthy" if blob # No output
Douwe Maan's avatar
Douwe Maan включено в состав коммита
61
  def self.decorate(blob, project = nil)
Robert Speicher's avatar
Robert Speicher включено в состав коммита
62
63
    return if blob.nil?

Douwe Maan's avatar
Douwe Maan включено в состав коммита
64
65
66
    new(blob, project)
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
67
  def initialize(blob, project = nil)
Douwe Maan's avatar
Douwe Maan включено в состав коммита
68
69
70
    @project = project

    super(blob)
Robert Speicher's avatar
Robert Speicher включено в состав коммита
71
72
  end

Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
73
74
75
76
77
78
79
80
81
82
83
84
  # Returns the data of the blob.
  #
  # If the blob is a text based blob the content is converted to UTF-8 and any
  # invalid byte sequences are replaced.
  def data
    if binary?
      super
    else
      @data ||= super.encode(Encoding::UTF_8, invalid: :replace, undef: :replace)
    end
  end

Zeger-Jan van de Weg's avatar
Zeger-Jan van de Weg включено в состав коммита
85
  def no_highlighting?
Douwe Maan's avatar
Douwe Maan включено в состав коммита
86
87
88
89
90
    raw_size && raw_size > MAXIMUM_TEXT_HIGHLIGHT_SIZE
  end

  def empty?
    raw_size == 0
Zeger-Jan van de Weg's avatar
Zeger-Jan van de Weg включено в состав коммита
91
92
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
93
  def too_large?
Stan Hu's avatar
Stan Hu включено в состав коммита
94
    size && truncated?
Zeger-Jan van de Weg's avatar
Zeger-Jan van de Weg включено в состав коммита
95
96
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
97
98
99
100
101
102
103
104
105
106
107
108
109
110
  def external_storage_error?
    if external_storage == :lfs
      !project&.lfs_enabled?
    else
      false
    end
  end

  def stored_externally?
    return @stored_externally if defined?(@stored_externally)

    @stored_externally = external_storage && !external_storage_error?
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
111
112
113
  # Returns the size of the file that this blob represents. If this blob is an
  # LFS pointer, this is the size of the file stored in LFS. Otherwise, this is
  # the size of the blob itself.
Douwe Maan's avatar
Douwe Maan включено в состав коммита
114
  def raw_size
Douwe Maan's avatar
Douwe Maan включено в состав коммита
115
116
    if stored_externally?
      external_size
Douwe Maan's avatar
Douwe Maan включено в состав коммита
117
118
119
    else
      size
    end
Douwe Maan's avatar
Douwe Maan включено в состав коммита
120
121
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
122
123
124
125
  # Returns whether the file that this blob represents is binary. If this blob is
  # an LFS pointer, we assume the file stored in LFS is binary, unless a
  # text-based rich blob viewer matched on the file's extension. Otherwise, this
  # depends on the type of the blob itself.
Douwe Maan's avatar
Douwe Maan включено в состав коммита
126
  def raw_binary?
Douwe Maan's avatar
Douwe Maan включено в состав коммита
127
    if stored_externally?
Douwe Maan's avatar
Douwe Maan включено в состав коммита
128
129
      if rich_viewer
        rich_viewer.binary?
Douwe Maan's avatar
Douwe Maan включено в состав коммита
130
131
132
133
      elsif Linguist::Language.find_by_filename(name).any?
        false
      elsif _mime_type
        _mime_type.binary?
Douwe Maan's avatar
Douwe Maan включено в состав коммита
134
135
136
      else
        true
      end
Douwe Maan's avatar
Douwe Maan включено в состав коммита
137
138
139
    else
      binary?
    end
Robert Speicher's avatar
Robert Speicher включено в состав коммита
140
141
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
142
143
  def extension
    @extension ||= extname.downcase.delete('.')
Sam Rose's avatar
Sam Rose включено в состав коммита
144
145
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
146
147
  def video?
    UploaderHelper::VIDEO_EXT.include?(extension)
Phil Hughes's avatar
Phil Hughes включено в состав коммита
148
149
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
150
  def readable_text?
Douwe Maan's avatar
Douwe Maan включено в состав коммита
151
    text? && !stored_externally? && !too_large?
Phil Hughes's avatar
Phil Hughes включено в состав коммита
152
153
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
154
155
156
157
158
159
160
  def simple_viewer
    @simple_viewer ||= simple_viewer_class.new(self)
  end

  def rich_viewer
    return @rich_viewer if defined?(@rich_viewer)

Douwe Maan's avatar
Douwe Maan включено в состав коммита
161
    @rich_viewer = rich_viewer_class&.new(self)
Douwe Maan's avatar
Douwe Maan включено в состав коммита
162
163
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
164
165
166
167
168
169
  def auxiliary_viewer
    return @auxiliary_viewer if defined?(@auxiliary_viewer)

    @auxiliary_viewer = auxiliary_viewer_class&.new(self)
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
170
  def rendered_as_text?(ignore_errors: true)
Douwe Maan's avatar
Douwe Maan включено в состав коммита
171
    simple_viewer.text? && (ignore_errors || simple_viewer.render_error.nil?)
Douwe Maan's avatar
Douwe Maan включено в состав коммита
172
173
174
  end

  def show_viewer_switcher?
Douwe Maan's avatar
Douwe Maan включено в состав коммита
175
176
177
178
179
180
    rendered_as_text? && rich_viewer
  end

  def override_max_size!
    simple_viewer&.override_max_size = true
    rich_viewer&.override_max_size = true
Douwe Maan's avatar
Douwe Maan включено в состав коммита
181
182
183
184
  end

  private

Douwe Maan's avatar
Douwe Maan включено в состав коммита
185
186
187
188
189
190
191
192
193
194
195
  def simple_viewer_class
    if empty?
      BlobViewer::Empty
    elsif raw_binary?
      BlobViewer::Download
    else # text
      BlobViewer::Text
    end
  end

  def rich_viewer_class
Douwe Maan's avatar
Douwe Maan включено в состав коммита
196
197
198
199
200
201
202
203
    viewer_class_from(RICH_VIEWERS)
  end

  def auxiliary_viewer_class
    viewer_class_from(AUXILIARY_VIEWERS)
  end

  def viewer_class_from(classes)
Douwe Maan's avatar
Douwe Maan включено в состав коммита
204
    return if empty? || external_storage_error?
Douwe Maan's avatar
Douwe Maan включено в состав коммита
205

Douwe Maan's avatar
Douwe Maan включено в состав коммита
206
    verify_binary = !stored_externally?
Douwe Maan's avatar
Douwe Maan включено в состав коммита
207

Douwe Maan's avatar
Douwe Maan включено в состав коммита
208
    classes.find { |viewer_class| viewer_class.can_render?(self, verify_binary: verify_binary) }
Douwe Maan's avatar
Douwe Maan включено в состав коммита
209
  end
Robert Speicher's avatar
Robert Speicher включено в состав коммита
210
end