X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/67dd9e4c9d487bdb5f38a09dc0c99def4989326e..7b057545c0b2030aad9981bd93699f9e33ad7d5f:/lib/redactable.rb

diff --git a/lib/redactable.rb b/lib/redactable.rb
index b994e8563..d827cfd74 100644
--- a/lib/redactable.rb
+++ b/lib/redactable.rb
@@ -1,15 +1,22 @@
-require 'osm'
+require "osm"
 
 module Redactable
+  def self.included(base)
+    # this is used to extend activerecord bases, as these aren't
+    # in scope for the module itself.
+    base.scope :unredacted, -> { base.where(:redaction_id => nil) }
+  end
+
   def redacted?
-    not self.redaction.nil?
+    !redaction.nil?
   end
 
   def redact!(redaction)
     # check that this version isn't the current version
-    raise OSM::APICannotRedactError.new if self.is_latest_version?
+    raise OSM::APICannotRedactError if is_latest_version?
 
     # make the change
     self.redaction = redaction
+    save!
   end
 end