free_mutant/lib/mutant/diff.rb

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_hunks.map do |hunk|
        hunk.diff(:unified) << NEWLINE
      end.join
    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 hunks
    #
    # @return [Array<Diff::LCS::Hunk>]
    def minimized_hunks
      head, *tail = hunks

      tail.each_with_object([head]) do |right, aggregate|
        left = aggregate.last
        if right.overlaps?(left)
          right.merge(left)
          aggregate.pop
        end
        aggregate << 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