X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/8181cb4bf77548088f959562f24be17f8ec053a3..e89a78fff858d5db07f6959a39dd9655dc46008e:/app/models/diary_comment.rb

diff --git a/app/models/diary_comment.rb b/app/models/diary_comment.rb
index aaf48be4f..a88c69f12 100644
--- a/app/models/diary_comment.rb
+++ b/app/models/diary_comment.rb
@@ -1,7 +1,53 @@
-class DiaryComment < ActiveRecord::Base
-  belongs_to :user
+# == Schema Information
+#
+# Table name: diary_comments
+#
+#  id             :bigint(8)        not null, primary key
+#  diary_entry_id :bigint(8)        not null
+#  user_id        :bigint(8)        not null
+#  body           :text             not null
+#  created_at     :datetime         not null
+#  updated_at     :datetime         not null
+#  visible        :boolean          default(TRUE), not null
+#  body_format    :enum             default("markdown"), not null
+#
+# Indexes
+#
+#  diary_comment_user_id_created_at_index  (user_id,created_at)
+#  diary_comments_entry_id_idx             (diary_entry_id,id) UNIQUE
+#
+# Foreign Keys
+#
+#  diary_comments_diary_entry_id_fkey  (diary_entry_id => diary_entries.id)
+#  diary_comments_user_id_fkey         (user_id => users.id)
+#
+
+class DiaryComment < ApplicationRecord
+  belongs_to :user, :counter_cache => true
   belongs_to :diary_entry
 
-  validates_presence_of :body
-  validates_associated :diary_entry
+  scope :visible, -> { where(:visible => true) }
+
+  validates :body, :presence => true, :characters => true
+  validates :diary_entry, :user, :associated => true
+
+  after_save :spam_check
+
+  def body
+    RichText.new(self[:body_format], self[:body])
+  end
+
+  def notification_token(subscriber)
+    sha256 = Digest::SHA256.new
+    sha256 << Rails.application.key_generator.generate_key("openstreetmap/diary_comment")
+    sha256 << id.to_s
+    sha256 << subscriber.to_s
+    Base64.urlsafe_encode64(sha256.digest)[0, 8]
+  end
+
+  private
+
+  def spam_check
+    user.spam_check
+  end
 end