Skip to content

Instantly share code, notes, and snippets.

@hanxue
Forked from skanev/README.md
Created June 5, 2017 15:29
Show Gist options
  • Save hanxue/394017bba6555c6cdbb1cb42b50b938d to your computer and use it in GitHub Desktop.
Save hanxue/394017bba6555c6cdbb1cb42b50b938d to your computer and use it in GitHub Desktop.
Hacky git diff syntax highlighting for the full code

Hacky syntax highlighting in git diff

Normally git diff would color additions green and deletions red. This is cool, but it would be even cooler if it adds syntax highlighting to those lines. This is a git pager that does so.

It parses the diff output and picks up the SHAs of files with additions and deletions. It uses CodeRay to highlight each file and then it extracts the lines that are shown in the diff. It then uses term/ansicolor to make a gradient from the CodeRay color and the diff color (red for deletion, green for addition) and uses it to replace the original.

I tried using rugged instead of shelling out to git show – it was faster overall, but it did incur a noticeable start up time.

Check out the image below for a demo.

#!/usr/bin/env ruby
require 'coderay'
require 'term/ansicolor'
require 'optparse'
# --- Adjusting colors -------------------------------------------------------
RGB = Term::ANSIColor::RGBTriple
Attribute = Term::ANSIColor::Attribute
Color = Term::ANSIColor
GREEN = RGB.new(0, 256, 0)
RED = RGB.new(256, 0, 0)
GRAY = RGB.new(180, 180, 180)
ADJUSTMENTS = {}
GRADIENTS = {}
def adjust(text, target, gradient = 8)
open = adjust_seq("\e[37m", target, gradient)
adjusted = text.chomp.gsub(/\e\[[0-9;]+m/) { |seq| adjust_seq(seq, target, gradient) }
"#{open}#{adjusted}"
end
def adjust_seq(seq, target, level)
key = [seq, target]
ADJUSTMENTS[key] ||=
begin
escapes = []
seq[/\d+(;\d+)*/].split(';').map(&:to_i).each do |number|
case number
when 0 then escapes << ["\e[0m", gradient_to(255, target, level)]
when 1 then escapes << "\e[1m"
when 4 then escapes << "\e[4m"
when 30..37 then escapes << gradient_to(number - 30, target, level)
when 40..47 then escapes << gradient_to(number - 30, target, level, :background)
else escapes << seq
end
end
escapes.join('')
end
end
def gradient_to(num, target, level, background = false)
key = [num, target, background]
GRADIENTS[key] ||=
begin
triple = triple(num)
target = triple.gradient_to(target)[level]
html = target.html
method = background ? "on_#{html}" : html
Attribute[method].apply
end
end
def triple(num)
Attribute[num].to_rgb_triple
end
# --- Parsing git diffs ------------------------------------------------------
FORMATS = {
'Gemfile' => :ruby,
'rb' => :ruby,
'c' => :c,
'h' => :c,
'cpp' => :cpp,
'cxx' => :cpp,
'hpp' => :cpp,
'clj' => :clojure,
'css' => :css,
'erb' => :erb,
'go' => :go,
'java' => :java,
'js' => :javascript,
'json' => :json,
'php' => :php,
'lua' => :lua,
'py' => :python,
'sass' => :sass,
'scss' => :scss,
'sql' => :sql,
'xml' => :xml,
'yml' => :yaml,
'yaml' => :yaml,
}
CACHED_OBJECTS = {}
def show(sha, path)
return [] if sha =~ /^0+$/
cached = CACHED_OBJECTS.delete sha
return cached if cached
format = FORMATS[path[/(\w+)$/, 1]]
code = `git show #{sha} 2> /dev/null`
code = File.read(path) if code.empty?
code = CodeRay.scan(code, format).terminal if format
CACHED_OBJECTS[sha] = code.lines
end
def process(options)
new = nil
old = nil
remaining = nil
old_file = nil
new_file = nil
old_hash = nil
new_hash = nil
while gets
stripped = $_.gsub(/(\e\[[0-9;]*m)*/, '')
case stripped
when /^index (\w+)..(\w+)/
old_hash = $1
new_hash = $2
puts $_
# @@ -1,4 +1,4 @@
when /^@@ -(\d+)(?:,(\d+))? \+(\d+)(?:,(\d+))? @@/
old_start, old_length = $1.to_i - 1, ($2 || 1).to_i
new_start, new_length = $3.to_i - 1, ($4 || 1).to_i
remaining = old_length + new_length
puts $_
when /^\+\+\+ (.*)$/
new_file = $1.sub(/^[ab]\//, '')
old = show old_hash, old_file
new = show new_hash, new_file
puts $_
when /^--- (.*)$/
old_file = $1.sub(/^[ab]\//, '')
puts $_
when /^ /
if remaining.nil? || remaining <= 0
puts $_
next
elsif options[:highlight]
puts adjust(" #{new[new_start]}", GRAY, 10)
else
puts $_
end
remaining -= 2
old_start += 1
new_start += 1
when /^\+/
puts adjust("+#{new[new_start]}", GREEN)
new_start += 1
remaining -= 1
when /^-/
puts adjust("-#{old[old_start]}", RED)
old_start += 1
remaining -= 1
else
puts $_
end
end
end
options = {}
OptionParser.new do |opts|
opts.banner = "Usage: #{File.basename(__FILE__)} [options]"
opts.on("-h", "--[no-]highlight", "Highlight all the code") do |v|
options[:highlight] = v
end
end.parse!
begin
process(options)
rescue Errno::EPIPE
exit 0
end
[core]
pager = /path/to/diff-syntax-highlight.rb | less
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment