112 lines
2.2 KiB
Ruby
112 lines
2.2 KiB
Ruby
module Mutant
|
|
# Class to create diffs from source code
|
|
class Diff
|
|
include Adamantium::Flat, Concord.new(:old, :new)
|
|
|
|
ADDITION = '+'.freeze
|
|
DELETION = '-'.freeze
|
|
NEWLINE = "\n".freeze
|
|
|
|
# Unified source diff between old and new
|
|
#
|
|
# @return [String]
|
|
# if there is exactly one diff
|
|
#
|
|
# @return [nil]
|
|
# otherwise
|
|
def diff
|
|
return if diffs.empty?
|
|
|
|
minimized_hunk.diff(:unified) << NEWLINE
|
|
end
|
|
memoize :diff
|
|
|
|
# Colorized unified source diff between old and new
|
|
#
|
|
# @return [String]
|
|
# if there is a diff
|
|
#
|
|
# @return [nil]
|
|
# otherwise
|
|
def colorized_diff
|
|
return unless diff
|
|
diff.lines.map(&self.class.method(:colorize_line)).join
|
|
end
|
|
memoize :colorized_diff
|
|
|
|
# Build new object from source strings
|
|
#
|
|
# @param [String] old
|
|
# @param [String] new
|
|
#
|
|
# @return [Diff]
|
|
def self.build(old, new)
|
|
new(lines(old), lines(new))
|
|
end
|
|
|
|
# Break up source into lines
|
|
#
|
|
# @param [String] source
|
|
#
|
|
# @return [Array<String>]
|
|
def self.lines(source)
|
|
source.lines.map(&:chomp)
|
|
end
|
|
private_class_method :lines
|
|
|
|
private
|
|
|
|
# Diffs between old and new
|
|
#
|
|
# @return [Array<Array>]
|
|
def diffs
|
|
::Diff::LCS.diff(old, new)
|
|
end
|
|
|
|
# Raw diff-lcs hunks
|
|
#
|
|
# @return [Array<Diff::LCS::Hunk>]
|
|
def hunks
|
|
diffs.map do |diff|
|
|
::Diff::LCS::Hunk.new(old, new, diff, max_length, 0)
|
|
end
|
|
end
|
|
|
|
# Minimized hunk
|
|
#
|
|
# @return Diff::LCS::Hunk
|
|
def minimized_hunk
|
|
head, *tail = hunks
|
|
|
|
tail.reduce(head) do |left, right|
|
|
right.merge(left)
|
|
right
|
|
end
|
|
end
|
|
|
|
# Max length of source line in new and old
|
|
#
|
|
# @return [Fixnum]
|
|
def max_length
|
|
[old, new].map(&:length).max
|
|
end
|
|
|
|
# Colorized a unified diff line
|
|
#
|
|
# @param [String] line
|
|
#
|
|
# @return [String]
|
|
def self.colorize_line(line)
|
|
case line[0]
|
|
when ADDITION
|
|
Color::GREEN
|
|
when DELETION
|
|
Color::RED
|
|
else
|
|
Color::NONE
|
|
end.format(line)
|
|
end
|
|
private_class_method :colorize_line
|
|
|
|
end # Diff
|
|
end # Mutant
|