label.rb 4.47 KB
Newer Older
1
class Label < ActiveRecord::Base
2
  include CacheMarkdownField
3
  include Referable
4 5
  include Subscribable

6 7
  # Represents a "No Label" state used for filtering Issues and Merge
  # Requests that have no label assigned.
8 9
  LabelStruct = Struct.new(:title, :name)
  None = LabelStruct.new('No Label', 'No Label')
10
  Any = LabelStruct.new('Any Label', '')
11

12 13
  cache_markdown_field :description, pipeline: :single_line

14
  DEFAULT_COLOR = '#428BCA'
15

Douwe Maan's avatar
Douwe Maan committed
16 17
  default_value_for :color, DEFAULT_COLOR

18
  has_many :lists, dependent: :destroy
19
  has_many :priorities, class_name: 'LabelPriority'
20
  has_many :label_links, dependent: :destroy
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
21
  has_many :issues, through: :label_links, source: :target, source_type: 'Issue'
22
  has_many :merge_requests, through: :label_links, source: :target, source_type: 'MergeRequest'
23

24
  validates :color, color: true, allow_blank: false
25

26
  # Don't allow ',' for label titles
27
  validates :title, presence: true, format: { with: /\A[^,]+\z/ }
28
  validates :title, uniqueness: { scope: [:group_id, :project_id] }
29

30
  default_scope { order(title: :asc) }
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
31

32 33
  scope :templates, -> { where(template: true) }
  scope :with_title, ->(title) { where(title: title) }
34

35 36 37 38
  def self.prioritized(project)
    joins(:priorities)
      .where(label_priorities: { project_id: project })
      .reorder('label_priorities.priority ASC, labels.title ASC')
Alfredo Sumaran's avatar
Alfredo Sumaran committed
39
  end
40

41 42
  def self.unprioritized(project)
    where.not(id: prioritized(project).select(:id))
Thijs Wouters's avatar
Thijs Wouters committed
43 44
  end

45
  alias_attribute :name, :title
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
46

47 48 49 50
  def self.reference_prefix
    '~'
  end

51
  ##
52
  # Pattern used to extract label references from text
53 54 55
  #
  # This pattern supports cross-project references.
  #
56
  def self.reference_pattern
57 58 59
    # NOTE: The id pattern only matches when all characters on the expression
    # are digits, so it will match ~2 but not ~2fa because that's probably a
    # label name and we want it to be matched as such.
60
    @reference_pattern ||= %r{
61 62
      (#{Project.reference_pattern})?
      #{Regexp.escape(reference_prefix)}
63
      (?:
64
        (?<label_id>\d+(?!\S\w)\b) | # Integer-based label ID, or
65
        (?<label_name>
66
          [A-Za-z0-9_\-\?\.&]+ | # String-based single-word label title, or
67
          ".+?"                  # String-based multi-word label surrounded in quotes
68 69 70 71 72
        )
      )
    }x
  end

73 74 75 76
  def self.link_reference_pattern
    nil
  end

77 78
  def open_issues_count(user = nil, project = nil)
    issues_count(user, project_id: project.try(:id) || project_id, state: 'opened')
Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
79
  end
Valery Sizov's avatar
Valery Sizov committed
80

81 82
  def closed_issues_count(user = nil, project = nil)
    issues_count(user, project_id: project.try(:id) || project_id, state: 'closed')
83 84
  end

85 86
  def open_merge_requests_count(user = nil, project = nil)
    merge_requests_count(user, project_id: project.try(:id) || project_id, state: 'opened')
87 88
  end

Valery Sizov's avatar
Valery Sizov committed
89 90 91
  def template?
    template
  end
92

93
  def text_color
94
    LabelsHelper.text_color_for_bg(self.color)
95 96
  end

97
  def title=(value)
98
    write_attribute(:title, sanitize_title(value)) if value.present?
99 100
  end

101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124
  ##
  # Returns the String necessary to reference this Label in Markdown
  #
  # format - Symbol format to use (default: :id, optional: :name)
  #
  # Examples:
  #
  #   Label.first.to_reference                     # => "~1"
  #   Label.first.to_reference(format: :name)      # => "~\"bug\""
  #   Label.first.to_reference(project1, project2) # => "gitlab-org/gitlab-ce~1"
  #
  # Returns a String
  #
  def to_reference(source_project = nil, target_project = nil, format: :id)
    format_reference = label_format_reference(format)
    reference = "#{self.class.reference_prefix}#{format_reference}"

    if cross_project_reference?(source_project, target_project)
      source_project.to_reference + reference
    else
      reference
    end
  end

125 126
  private

127 128 129 130
  def cross_project_reference?(source_project, target_project)
    source_project && target_project && source_project != target_project
  end

131
  def issues_count(user, params = {})
132
    IssuesFinder.new(user, params.reverse_merge(label_name: title, scope: 'all'))
133 134 135 136 137
                .execute
                .count
  end

  def merge_requests_count(user, params = {})
138
    MergeRequestsFinder.new(user, params.reverse_merge(label_name: title, scope: 'all'))
139 140 141 142
                       .execute
                       .count
  end

143 144 145 146
  def label_format_reference(format = :id)
    raise StandardError, 'Unknown format' unless [:id, :name].include?(format)

    if format == :name && !name.include?('"')
147
      %("#{name}")
148
    else
149
      id
150 151
    end
  end
Thijs Wouters's avatar
Thijs Wouters committed
152

153
  def sanitize_title(value)
154
    CGI.unescapeHTML(Sanitize.clean(value.to_s))
155
  end
156
end