snippet.rb 4,2 КБ
Newer Older
gitlabhq's avatar
gitlabhq включено в состав коммита
1
class Snippet < ActiveRecord::Base
Valery Sizov's avatar
Valery Sizov включено в состав коммита
2
  include Gitlab::VisibilityLevel
Nick Thomas's avatar
Nick Thomas включено в состав коммита
3
  include CacheMarkdownField
Douwe Maan's avatar
Douwe Maan включено в состав коммита
4
  include Noteable
Douwe Maan's avatar
Douwe Maan включено в состав коммита
5
  include Participable
Robert Speicher's avatar
Robert Speicher включено в состав коммита
6
7
  include Referable
  include Sortable
Z.J. van de Weg's avatar
Z.J. van de Weg включено в состав коммита
8
  include Awardable
Felipe Artur's avatar
Felipe Artur включено в состав коммита
9
  include Mentionable
Sean McGivern's avatar
Sean McGivern включено в состав коммита
10
  include Spammable
gitlabhq's avatar
gitlabhq включено в состав коммита
11

Nick Thomas's avatar
Nick Thomas включено в состав коммита
12
13
14
  cache_markdown_field :title, pipeline: :single_line
  cache_markdown_field :content

blackst0ne's avatar
blackst0ne включено в состав коммита
15
16
17
18
19
  # Aliases to make application_helper#edited_time_ago_with_tooltip helper work properly with snippets.
  # See https://gitlab.com/gitlab-org/gitlab-ce/merge_requests/10392/diffs#note_28719102
  alias_attribute :last_edited_at, :updated_at
  alias_attribute :last_edited_by, :updated_by

Nick Thomas's avatar
Nick Thomas включено в состав коммита
20
21
22
23
24
25
  # If file_name changes, it invalidates content
  alias_method :default_content_html_invalidator, :content_html_invalidated?
  def content_html_invalidated?
    default_content_html_invalidator || file_name_changed?
  end

Sean McGivern's avatar
Sean McGivern включено в состав коммита
26
  default_value_for(:visibility_level) { current_application_settings.default_snippet_visibility }
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
27

Robert Speicher's avatar
Robert Speicher включено в состав коммита
28
29
  belongs_to :author, class_name: 'User'
  belongs_to :project
Andrew8xx8's avatar
Andrew8xx8 включено в состав коммита
30

Robert Speicher's avatar
Robert Speicher включено в состав коммита
31
  has_many :notes, as: :noteable, dependent: :destroy
gitlabhq's avatar
gitlabhq включено в состав коммита
32

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
33
  delegate :name, :email, to: :author, prefix: true, allow_nil: true
gitlabhq's avatar
gitlabhq включено в состав коммита
34

Andrey Kumanyaev's avatar
Andrey Kumanyaev включено в состав коммита
35
  validates :author, presence: true
Rémy Coutable's avatar
Rémy Coutable включено в состав коммита
36
  validates :title, presence: true, length: { maximum: 255 }
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
37
  validates :file_name,
Rémy Coutable's avatar
Rémy Coutable включено в состав коммита
38
    length: { maximum: 255 },
Douwe Maan's avatar
Douwe Maan включено в состав коммита
39
40
    format: { with: Gitlab::Regex.file_name_regex,
              message: Gitlab::Regex.file_name_regex_message }
http://jneen.net/'s avatar
http://jneen.net/ включено в состав коммита
41

Valeriy Sizov's avatar
Valeriy Sizov включено в состав коммита
42
  validates :content, presence: true
Valery Sizov's avatar
Valery Sizov включено в состав коммита
43
  validates :visibility_level, inclusion: { in: Gitlab::VisibilityLevel.values }
gitlabhq's avatar
gitlabhq включено в состав коммита
44

Andrey Kumanyaev's avatar
Andrey Kumanyaev включено в состав коммита
45
  # Scopes
Valery Sizov's avatar
Valery Sizov включено в состав коммита
46
47
48
49
  scope :are_internal,  -> { where(visibility_level: Snippet::INTERNAL) }
  scope :are_private, -> { where(visibility_level: Snippet::PRIVATE) }
  scope :are_public, -> { where(visibility_level: Snippet::PUBLIC) }
  scope :public_and_internal, -> { where(visibility_level: [Snippet::PUBLIC, Snippet::INTERNAL]) }
Andrew8xx8's avatar
Andrew8xx8 включено в состав коммита
50
  scope :fresh,   -> { order("created_at DESC") }
Nihad Abbasov's avatar
Nihad Abbasov включено в состав коммита
51

Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
52
53
  participant :author
  participant :notes_with_associations
Douwe Maan's avatar
Douwe Maan включено в состав коммита
54

Sean McGivern's avatar
Sean McGivern включено в состав коммита
55
56
57
  attr_spammable :title, spam_title: true
  attr_spammable :content, spam_description: true

Robert Speicher's avatar
Robert Speicher включено в состав коммита
58
59
60
61
  def self.reference_prefix
    '$'
  end

Robert Speicher's avatar
Robert Speicher включено в состав коммита
62
63
64
65
  # Pattern used to extract `$123` snippet references from text
  #
  # This pattern supports cross-project references.
  def self.reference_pattern
Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
66
    @reference_pattern ||= %r{
Douwe Maan's avatar
Douwe Maan включено в состав коммита
67
68
      (#{Project.reference_pattern})?
      #{Regexp.escape(reference_prefix)}(?<snippet>\d+)
Robert Speicher's avatar
Robert Speicher включено в состав коммита
69
70
71
    }x
  end

Douwe Maan's avatar
Douwe Maan включено в состав коммита
72
  def self.link_reference_pattern
Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
73
    @link_reference_pattern ||= super("snippets", /(?<snippet>\d+)/)
Douwe Maan's avatar
Douwe Maan включено в состав коммита
74
75
  end

James Lopez's avatar
James Lopez включено в состав коммита
76
  def to_reference(from_project = nil, full: false)
Robert Speicher's avatar
Robert Speicher включено в состав коммита
77
78
    reference = "#{self.class.reference_prefix}#{id}"

Oswaldo Ferreira's avatar
Oswaldo Ferreira включено в состав коммита
79
    if project.present?
James Lopez's avatar
James Lopez включено в состав коммита
80
      "#{project.to_reference(from_project, full: full)}#{reference}"
Oswaldo Ferreira's avatar
Oswaldo Ferreira включено в состав коммита
81
82
    else
      reference
Robert Speicher's avatar
Robert Speicher включено в состав коммита
83
84
85
    end
  end

gitlabhq's avatar
gitlabhq включено в состав коммита
86
  def self.content_types
Nihad Abbasov's avatar
Nihad Abbasov включено в состав коммита
87
    [
gitlabhq's avatar
gitlabhq включено в состав коммита
88
89
90
91
92
      ".rb", ".py", ".pl", ".scala", ".c", ".cpp", ".java",
      ".haml", ".html", ".sass", ".scss", ".xml", ".php", ".erb",
      ".js", ".sh", ".coffee", ".yml", ".md"
    ]
  end
gitlabhq's avatar
gitlabhq включено в состав коммита
93

Douwe Maan's avatar
Douwe Maan включено в состав коммита
94
95
  def blob
    @blob ||= Blob.decorate(SnippetBlob.new(self), nil)
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
96
97
  end

Stan Hu's avatar
Stan Hu включено в состав коммита
98
99
100
101
  def hook_attrs
    attributes
  end

Rémy Coutable's avatar
Rémy Coutable включено в состав коммита
102
103
104
105
  def file_name
    super.to_s
  end

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
106
107
108
109
  def sanitized_file_name
    file_name.gsub(/[^a-zA-Z0-9_\-\.]+/, '')
  end

Valery Sizov's avatar
Valery Sizov включено в состав коммита
110
  def visibility_level_field
Toon Claes's avatar
Toon Claes включено в состав коммита
111
    :visibility_level
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets включено в состав коммита
112
  end
Valery Sizov's avatar
Valery Sizov включено в состав коммита
113

Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
114
  def notes_with_associations
115
    notes.includes(:author)
Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
116
117
  end

Sean McGivern's avatar
Sean McGivern включено в состав коммита
118
  def check_for_spam?
Oswaldo Ferreira's avatar
Oswaldo Ferreira включено в состав коммита
119
120
    visibility_level_changed?(to: Snippet::PUBLIC) ||
      (public? && (title_changed? || content_changed?))
Sean McGivern's avatar
Sean McGivern включено в состав коммита
121
122
123
124
125
126
  end

  def spammable_entity_type
    'snippet'
  end

Charles Bushong's avatar
Charles Bushong включено в состав коммита
127
  class << self
Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
128
129
130
131
132
133
134
    # Searches for snippets with a matching title or file name.
    #
    # This method uses ILIKE on PostgreSQL and LIKE on MySQL.
    #
    # query - The search query as a String.
    #
    # Returns an ActiveRecord::Relation.
Charles Bushong's avatar
Charles Bushong включено в состав коммита
135
    def search(query)
Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
136
      t = arel_table
Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
137
138
139
      pattern = "%#{query}%"

      where(t[:title].matches(pattern).or(t[:file_name].matches(pattern)))
Charles Bushong's avatar
Charles Bushong включено в состав коммита
140
141
    end

Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
142
143
144
145
146
147
148
    # Searches for snippets with matching content.
    #
    # This method uses ILIKE on PostgreSQL and LIKE on MySQL.
    #
    # query - The search query as a String.
    #
    # Returns an ActiveRecord::Relation.
Charles Bushong's avatar
Charles Bushong включено в состав коммита
149
    def search_code(query)
Yorick Peterse's avatar
Yorick Peterse включено в состав коммита
150
151
152
153
      table   = Snippet.arel_table
      pattern = "%#{query}%"

      where(table[:content].matches(pattern))
Charles Bushong's avatar
Charles Bushong включено в состав коммита
154
155
    end
  end
gitlabhq's avatar
gitlabhq включено в состав коммита
156
end