X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/acbc99ac16103e1be5c1ef1841794f259e3b6a03..f77a3c75bee528caaa6fc8e237ee6eaef704efe0:/app/models/map_bug_comment.rb

diff --git a/app/models/map_bug_comment.rb b/app/models/map_bug_comment.rb
index 166e3fb42..9a130dd34 100644
--- a/app/models/map_bug_comment.rb
+++ b/app/models/map_bug_comment.rb
@@ -1,15 +1,19 @@
 class MapBugComment < ActiveRecord::Base
-
   set_table_name 'map_bug_comment'
 
-  belongs_to :map_bug, :foreign_key => 'bug_id'
-  belongs_to :user, :foreign_key => 'commenter_id'
-
-   validates_inclusion_of :event, :in => [ "opened", "closed", "reopened", "commented", "hidden" ]
+  belongs_to :map_bug, :foreign_key => :bug_id
+  belongs_to :author, :class_name => "User", :foreign_key => :author_id
 
+  validates_inclusion_of :event, :in => [ "opened", "closed", "reopened", "commented", "hidden" ]
   validates_presence_of :id, :on => :update
   validates_uniqueness_of :id
   validates_presence_of :visible
-  validates_presence_of :date_created
 
+  def author_name
+    if self.author_id.nil?
+      self.read_attribute(:author_name)
+    else
+      self.author.display_name
+    end
+  end
 end