class Snippet < ActiveRecord::Base include Gitlab::VisibilityLevel include CacheMarkdownField include Noteable include Participable include Referable include Sortable include Awardable include Mentionable include Spammable include Editable cache_markdown_field :title, pipeline: :single_line cache_markdown_field :description cache_markdown_field :content # 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 # 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 default_value_for(:visibility_level) { current_application_settings.default_snippet_visibility } belongs_to :author, class_name: 'User' belongs_to :project has_many :notes, as: :noteable, dependent: :destroy delegate :name, :email, to: :author, prefix: true, allow_nil: true validates :author, presence: true validates :title, presence: true, length: { maximum: 255 } validates :file_name, length: { maximum: 255 }, format: { with: Gitlab::Regex.file_name_regex, message: Gitlab::Regex.file_name_regex_message } validates :content, presence: true validates :visibility_level, inclusion: { in: Gitlab::VisibilityLevel.values } # Scopes 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]) } scope :fresh, -> { order("created_at DESC") } participant :author participant :notes_with_associations attr_spammable :title, spam_title: true attr_spammable :content, spam_description: true def self.reference_prefix '$' end # Pattern used to extract `$123` snippet references from text # # This pattern supports cross-project references. def self.reference_pattern @reference_pattern ||= %r{ (#{Project.reference_pattern})? #{Regexp.escape(reference_prefix)}(?\d+) }x end def self.link_reference_pattern @link_reference_pattern ||= super("snippets", /(?\d+)/) end def to_reference(from_project = nil, full: false) reference = "#{self.class.reference_prefix}#{id}" if project.present? "#{project.to_reference(from_project, full: full)}#{reference}" else reference end end def self.content_types [ ".rb", ".py", ".pl", ".scala", ".c", ".cpp", ".java", ".haml", ".html", ".sass", ".scss", ".xml", ".php", ".erb", ".js", ".sh", ".coffee", ".yml", ".md" ] end def blob @blob ||= Blob.decorate(SnippetBlob.new(self), nil) end def hook_attrs attributes end def file_name super.to_s end def sanitized_file_name file_name.gsub(/[^a-zA-Z0-9_\-\.]+/, '') end def visibility_level_field :visibility_level end def notes_with_associations notes.includes(:author) end def check_for_spam? visibility_level_changed?(to: Snippet::PUBLIC) || (public? && (title_changed? || content_changed?)) end def spammable_entity_type 'snippet' end class << self # 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. def search(query) t = arel_table pattern = "%#{query}%" where(t[:title].matches(pattern).or(t[:file_name].matches(pattern))) end # 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. def search_code(query) table = Snippet.arel_table pattern = "%#{query}%" where(table[:content].matches(pattern)) end end end