X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/65e1dbb4a69cf5251c638e50e480aee3b77bca50..fc0096289dce3c89c1a9d3f7af4e2620f347df11:/app/models/issue.rb

diff --git a/app/models/issue.rb b/app/models/issue.rb
index 40e0bb82a..567ee819e 100644
--- a/app/models/issue.rb
+++ b/app/models/issue.rb
@@ -1,3 +1,32 @@
+# == Schema Information
+#
+# Table name: issues
+#
+#  id               :integer          not null, primary key
+#  reportable_type  :string           not null
+#  reportable_id    :integer          not null
+#  reported_user_id :integer          not null
+#  status           :integer
+#  issue_type       :string
+#  resolved_at      :datetime
+#  resolved_by      :integer
+#  updated_by       :integer
+#  created_at       :datetime         not null
+#  updated_at       :datetime         not null
+#  reports_count    :integer          default(0)
+#
+# Indexes
+#
+#  index_issues_on_reportable_id_and_reportable_type  (reportable_id,reportable_type)
+#  index_issues_on_reported_user_id                   (reported_user_id)
+#  index_issues_on_updated_by                         (updated_by)
+#
+# Foreign Keys
+#
+#  issues_reported_user_id_fkey  (reported_user_id => users.id) ON DELETE => cascade
+#  issues_updated_by_fkey        (updated_by => users.id) ON DELETE => cascade
+#
+
 class Issue < ActiveRecord::Base
   belongs_to :reportable, :polymorphic => true
   belongs_to :reported_user, :class_name => "User", :foreign_key => :reported_user_id
@@ -9,6 +38,8 @@ class Issue < ActiveRecord::Base
   validates :reportable_id, :uniqueness => { :scope => [:reportable_type] }
   validates :reported_user_id, :presence => true
 
+  before_validation :set_reported_user
+
   # Check if more statuses are needed
   enum :status => %w[open ignored resolved]
   enum :type => %w[administrator moderator]
@@ -45,4 +76,17 @@ class Issue < ActiveRecord::Base
       transitions :from => :ignored, :to => :open
     end
   end
+
+  private
+
+  def set_reported_user
+    self.reported_user = case reportable.class.name
+                         when "User"
+                           reportable
+                         when "Note"
+                           reportable.author
+                         else
+                           reportable.user
+                         end
+  end
 end