Add methods to resolve diff notes and discussions

1 files changed, 18 insertions, 0 deletions

diff --git a/app/models/diff_note.rb b/app/models/diff_note.rb
index d76ac51af0f..3edbdf9f966 100644
--- a/app/models/diff_note.rb
+++ b/app/models/diff_note.rb
@@ -85,6 +85,24 @@ class DiffNote < Note
     self.resolved_at.present?
   end
 
+  def resolve!(current_user)
+    return unless resolvable?
+    return if resolved?
+
+    self.resolved_at = Time.now
+    self.resolved_by = current_user
+    save!
+  end
+
+  def unresolve!
+    return unless resolvable?
+    return unless resolved?
+
+    self.resolved_at = nil
+    self.resolved_by = nil
+    save!
+  end
+
   private
 
   def supported?