1 files changed, 11 insertions, 0 deletions
diff --git a/app/models/note.rb b/app/models/note.rb
index b0875a0761b..8714db2e10e 100644
--- a/app/models/note.rb
+++ b/app/models/note.rb
@@ -99,10 +99,21 @@ class Note < ActiveRecord::Base
     diff.new_path if diff
   end
 
+  def diff_old_line
+    line_code.split('_')[1].to_i
+  end
+
   def diff_new_line
     line_code.split('_')[2].to_i
   end
 
+  def diff_line
+    if diff
+      @diff_line ||= diff.diff.lines.select { |line| line =~ /\A\+/ }[diff_new_line] ||
+        diff.diff.lines.select { |line| line =~ /\A\-/ }[diff_old_line]
+    end
+  end
+
   def discussion_id
     @discussion_id ||= [:discussion, noteable_type.try(:underscore), noteable_id || commit_id, line_code].join("-").to_sym
   end