X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/135ec35a91bcca15637b6436dd7e546468c49690..b87983c2baf55d437c09e668188454d01b5f20b0:/app/models/note.rb?ds=sidebyside

diff --git a/app/models/note.rb b/app/models/note.rb
index fa9d0b1ae..6d8ca078f 100644
--- a/app/models/note.rb
+++ b/app/models/note.rb
@@ -18,10 +18,13 @@
 #  notes_updated_at_idx   (updated_at)
 #
 
-class Note < ActiveRecord::Base
+class Note < ApplicationRecord
   include GeoRecord
 
   has_many :comments, -> { left_joins(:author).where(:visible => true, :users => { :status => [nil, "active", "confirmed"] }).order(:created_at) }, :class_name => "NoteComment", :foreign_key => :note_id
+  has_many :all_comments, -> { left_joins(:author).order(:created_at) }, :class_name => "NoteComment", :foreign_key => :note_id, :inverse_of => :note
+  has_many :subscriptions, :class_name => "NoteSubscription"
+  has_many :subscribers, :through => :subscriptions, :source => :user
 
   validates :id, :uniqueness => true, :presence => { :on => :update },
                  :numericality => { :on => :update, :only_integer => true }
@@ -36,6 +39,8 @@ class Note < ActiveRecord::Base
 
   after_initialize :set_defaults
 
+  DEFAULT_FRESHLY_CLOSED_LIMIT = 7.days
+
   # Sanity check the latitude and longitude and add an error if it's broken
   def validate_position
     errors.add(:base, "Note is not in the world") unless in_world?
@@ -44,7 +49,7 @@ class Note < ActiveRecord::Base
   # Close a note
   def close
     self.status = "closed"
-    self.closed_at = Time.now.getutc
+    self.closed_at = Time.now.utc
     save
   end
 
@@ -65,6 +70,18 @@ class Note < ActiveRecord::Base
     !closed_at.nil?
   end
 
+  def freshly_closed?
+    return false unless closed?
+
+    Time.now.utc < freshly_closed_until
+  end
+
+  def freshly_closed_until
+    return nil unless closed?
+
+    closed_at + DEFAULT_FRESHLY_CLOSED_LIMIT
+  end
+
   # Return the author object, derived from the first comment
   def author
     comments.first.author