Validate the assigned role, and remove the :type activrecord enum
[rails.git] / app / models / issue.rb
1 # == Schema Information
2 #
3 # Table name: issues
4 #
5 #  id               :integer          not null, primary key
6 #  reportable_type  :string           not null
7 #  reportable_id    :integer          not null
8 #  reported_user_id :integer          not null
9 #  status           :integer
10 #  assigned_role    :enum             not null
11 #  resolved_at      :datetime
12 #  resolved_by      :integer
13 #  updated_by       :integer
14 #  reports_count    :integer          default(0)
15 #  created_at       :datetime         not null
16 #  updated_at       :datetime         not null
17 #
18 # Indexes
19 #
20 #  index_issues_on_reportable_id_and_reportable_type  (reportable_id,reportable_type)
21 #  index_issues_on_reported_user_id                   (reported_user_id)
22 #  index_issues_on_updated_by                         (updated_by)
23 #
24 # Foreign Keys
25 #
26 #  issues_reported_user_id_fkey  (reported_user_id => users.id) ON DELETE => cascade
27 #  issues_updated_by_fkey        (updated_by => users.id) ON DELETE => cascade
28 #
29
30 class Issue < ActiveRecord::Base
31   belongs_to :reportable, :polymorphic => true
32   belongs_to :reported_user, :class_name => "User", :foreign_key => :reported_user_id
33   belongs_to :user_updated, :class_name => "User", :foreign_key => :updated_by
34
35   has_many :reports, :dependent => :destroy
36   has_many :comments, :class_name => "IssueComment", :dependent => :destroy
37
38   validates :reportable_id, :uniqueness => { :scope => [:reportable_type] }
39   validates :reported_user_id, :presence => true
40   validates :assigned_role, :presence => true, :inclusion => %w[administrator moderator]
41
42   before_validation :set_default_assigned_role
43   before_validation :set_reported_user
44
45   # Check if more statuses are needed
46   enum :status => %w[open ignored resolved]
47
48   scope :with_status, ->(issue_status) { where(:status => statuses[issue_status]) }
49
50   def read_reports
51     resolved_at.present? ? reports.where("updated_at < ?", resolved_at) : nil
52   end
53
54   def unread_reports
55     resolved_at.present? ? reports.where("updated_at >= ?", resolved_at) : reports
56   end
57
58   include AASM
59   aasm :column => :status, :no_direct_assignment => true do
60     state :open, :initial => true
61     state :ignored
62     state :resolved
63
64     event :ignore do
65       transitions :from => :open, :to => :ignored
66     end
67
68     event :resolve do
69       transitions :from => :open, :to => :resolved
70       after do
71         self.resolved_at = Time.now.getutc
72       end
73     end
74
75     event :reopen do
76       transitions :from => :resolved, :to => :open
77       transitions :from => :ignored, :to => :open
78     end
79   end
80
81   private
82
83   def set_reported_user
84     self.reported_user = case reportable.class.name
85                          when "User"
86                            reportable
87                          when "Note"
88                            reportable.author
89                          else
90                            reportable.user
91                          end
92   end
93
94   def set_default_assigned_role
95     role = %w[Changeset Note].include?(reportable.class.name) ? "moderator" : "administrator"
96     self.assigned_role = role if assigned_role.blank?
97   end
98 end