debian-mirror-gitlab/app/models/todo.rb

112 lines
2.5 KiB
Ruby
Raw Normal View History

2016-04-02 18:10:28 +05:30
class Todo < ActiveRecord::Base
2016-09-13 17:45:13 +05:30
include Sortable
2016-08-24 12:49:21 +05:30
ASSIGNED = 1
MENTIONED = 2
BUILD_FAILED = 3
MARKED = 4
APPROVAL_REQUIRED = 5 # This is an EE-only feature
ACTION_NAMES = {
ASSIGNED => :assigned,
MENTIONED => :mentioned,
BUILD_FAILED => :build_failed,
MARKED => :marked,
APPROVAL_REQUIRED => :approval_required
}
2016-04-02 18:10:28 +05:30
belongs_to :author, class_name: "User"
belongs_to :note
belongs_to :project
belongs_to :target, polymorphic: true, touch: true
belongs_to :user
delegate :name, :email, to: :author, prefix: true, allow_nil: true
2016-06-02 11:05:42 +05:30
validates :action, :project, :target_type, :user, presence: true
validates :target_id, presence: true, unless: :for_commit?
validates :commit_id, presence: true, if: :for_commit?
2016-04-02 18:10:28 +05:30
default_scope { reorder(id: :desc) }
scope :pending, -> { with_state(:pending) }
scope :done, -> { with_state(:done) }
state_machine :state, initial: :pending do
event :done do
2016-06-02 11:05:42 +05:30
transition [:pending] => :done
2016-04-02 18:10:28 +05:30
end
state :pending
state :done
end
2016-08-24 12:49:21 +05:30
after_save :keep_around_commit
2016-09-13 17:45:13 +05:30
class << self
def sort(method)
method == "priority" ? order_by_labels_priority : order_by(method)
end
# Order by priority depending on which issue/merge request the Todo belongs to
# Todos with highest priority first then oldest todos
# Need to order by created_at last because of differences on Mysql and Postgres when joining by type "Merge_request/Issue"
def order_by_labels_priority
2016-11-03 12:29:30 +05:30
params = {
target_type_column: "todos.target_type",
target_column: "todos.target_id",
project_column: "todos.project_id"
}
highest_priority = highest_label_priority(params).to_sql
2016-09-13 17:45:13 +05:30
select("#{table_name}.*, (#{highest_priority}) AS highest_priority").
order(Gitlab::Database.nulls_last_order('highest_priority', 'ASC')).
order('todos.created_at')
end
end
2016-06-02 11:05:42 +05:30
def build_failed?
action == BUILD_FAILED
end
2016-08-24 12:49:21 +05:30
def action_name
ACTION_NAMES[action]
end
2016-04-02 18:10:28 +05:30
def body
if note.present?
note.note
else
target.title
end
end
2016-06-02 11:05:42 +05:30
def for_commit?
target_type == "Commit"
end
# override to return commits, which are not active record
def target
if for_commit?
project.commit(commit_id) rescue nil
else
super
end
end
def target_reference
if for_commit?
target.short_id
else
target.to_reference
end
end
2016-08-24 12:49:21 +05:30
private
def keep_around_commit
project.repository.keep_around(self.commit_id)
end
2016-04-02 18:10:28 +05:30
end