diff options
author | Douwe Maan <douwe@selenight.nl> | 2016-05-13 14:53:31 -0500 |
---|---|---|
committer | Douwe Maan <douwe@selenight.nl> | 2016-05-13 17:31:58 -0500 |
commit | 7848d54f5b0024f58d8ce2b0259347816be5bdbc (patch) | |
tree | f3b8edbd0be946edec4b2c4e2bd29070e3f68947 /app/models | |
parent | 9a3ed265e9c82ee55bc6a17737ef6c5c2a1ad7e3 (diff) | |
download | gitlab-ce-7848d54f5b0024f58d8ce2b0259347816be5bdbc.tar.gz |
Clean up LegacyDiffNote somewhat
Diffstat (limited to 'app/models')
-rw-r--r-- | app/models/legacy_diff_note.rb | 171 | ||||
-rw-r--r-- | app/models/note.rb | 4 |
2 files changed, 77 insertions, 98 deletions
diff --git a/app/models/legacy_diff_note.rb b/app/models/legacy_diff_note.rb index b5de85df99f..bbefc911b29 100644 --- a/app/models/legacy_diff_note.rb +++ b/app/models/legacy_diff_note.rb @@ -23,42 +23,65 @@ class LegacyDiffNote < Note @discussion_id ||= self.class.build_discussion_id(noteable_type, noteable_id || commit_id, line_code, active?) end - def find_diff - return nil unless noteable - return @diff if defined?(@diff) - - # Don't use ||= because nil is a valid value for @diff - @diff = noteable.diffs(Commit.max_diff_options).find do |d| - Digest::SHA1.hexdigest(d.new_path) == diff_file_index if d.new_path - end + def diff_file_hash + line_code.split('_')[0] if line_code end - def set_diff - # First lets find notes with same diff - # before iterating over all mr diffs - diff = diff_for_line_code unless for_merge_request? - diff ||= find_diff + def diff_old_line + line_code.split('_')[1].to_i if line_code + end - self.st_diff = diff.to_hash if diff + def diff_new_line + line_code.split('_')[2].to_i if line_code end def diff @diff ||= Gitlab::Git::Diff.new(st_diff) if st_diff.respond_to?(:map) end - def diff_for_line_code - attributes = { - noteable_type: noteable_type, - line_code: line_code - } + def diff_file_path + diff.new_path.presence || diff.old_path + end - if for_commit? - attributes[:commit_id] = commit_id - else - attributes[:noteable_id] = noteable_id + def diff_lines + @diff_lines ||= Gitlab::Diff::Parser.new.parse(diff.diff.each_line) + end + + def diff_line + @diff_line ||= diff_lines.find { |line| generate_line_code(line) == self.line_code } + end + + def diff_line_text + diff_line.try(:text) + end + + def diff_line_type + diff_line.try(:type) + end + + def highlighted_diff_lines + Gitlab::Diff::Highlight.new(diff_lines).highlight + end + + def truncated_diff_lines + max_number_of_lines = 16 + prev_match_line = nil + prev_lines = [] + + highlighted_diff_lines.each do |line| + if line.type == "match" + prev_lines.clear + prev_match_line = line + else + prev_lines << line + + break if generate_line_code(line) == self.line_code + + prev_lines.shift if prev_lines.length >= max_number_of_lines + end end - self.class.where(attributes).last.try(:diff) + prev_lines end # Check if this note is part of an "active" discussion @@ -69,17 +92,17 @@ class LegacyDiffNote < Note # If the note's current diff cannot be matched in the MergeRequest's current # diff, it's considered inactive. def active? + return @active if defined?(@active) return true if for_commit? return true unless self.diff return false unless noteable - return @active if defined?(@active) noteable_diff = find_noteable_diff if noteable_diff parsed_lines = Gitlab::Diff::Parser.new.parse(noteable_diff.diff.each_line) - @active = parsed_lines.any? { |line_obj| line_obj.text == diff_line } + @active = parsed_lines.any? { |line_obj| line_obj.text == diff_line_text } else @active = false end @@ -87,93 +110,45 @@ class LegacyDiffNote < Note @active end - def diff_file_index - line_code.split('_')[0] if line_code - end - - def diff_file_name - diff.new_path if diff - end - - def file_path - if diff.new_path.present? - diff.new_path - elsif diff.old_path.present? - diff.old_path - end - end - - def diff_old_line - line_code.split('_')[1].to_i if line_code - end - - def diff_new_line - line_code.split('_')[2].to_i if line_code - end - - def generate_line_code(line) - Gitlab::Diff::LineCode.generate(file_path, line.new_pos, line.old_pos) - end + private - def diff_line - return @diff_line if @diff_line + def find_diff + return nil unless noteable + return @diff if defined?(@diff) - if diff - diff_lines.each do |line| - if generate_line_code(line) == self.line_code - @diff_line = line.text - end - end + @diff = noteable.diffs(Commit.max_diff_options).find do |d| + d.new_path && Digest::SHA1.hexdigest(d.new_path) == diff_file_hash end - - @diff_line end - def diff_line_type - return @diff_line_type if @diff_line_type - - if diff - diff_lines.each do |line| - if generate_line_code(line) == self.line_code - @diff_line_type = line.type - end - end - end + def set_diff + # First lets find notes with same diff + # before iterating over all mr diffs + diff = diff_for_line_code unless for_merge_request? + diff ||= find_diff - @diff_line_type + self.st_diff = diff.to_hash if diff end - def truncated_diff_lines - max_number_of_lines = 16 - prev_match_line = nil - prev_lines = [] - - highlighted_diff_lines.each do |line| - if line.type == "match" - prev_lines.clear - prev_match_line = line - else - prev_lines << line - - break if generate_line_code(line) == self.line_code + def diff_for_line_code + attributes = { + noteable_type: noteable_type, + line_code: line_code + } - prev_lines.shift if prev_lines.length >= max_number_of_lines - end + if for_commit? + attributes[:commit_id] = commit_id + else + attributes[:noteable_id] = noteable_id end - prev_lines - end - - def diff_lines - @diff_lines ||= Gitlab::Diff::Parser.new.parse(diff.diff.each_line) + self.class.where(attributes).last.try(:diff) end - def highlighted_diff_lines - Gitlab::Diff::Highlight.new(diff_lines).highlight + def generate_line_code(line) + Gitlab::Diff::LineCode.generate(diff_file_path, line.new_pos, line.old_pos) end - private - # Find the diff on noteable that matches our own def find_noteable_diff diffs = noteable.diffs(Commit.max_diff_options) diff --git a/app/models/note.rb b/app/models/note.rb index 3bc55870704..7e5bdc09a84 100644 --- a/app/models/note.rb +++ b/app/models/note.rb @@ -112,6 +112,10 @@ class Note < ActiveRecord::Base false end + def active? + true + end + def discussion_id @discussion_id ||= if for_merge_request? |