Add git-blame-color script

git-blame-color

@@ -0,0 +1,154 @@
+#!/usr/bin/env ruby
+
+# Colorize string
+class String
+	def colorize(color_code)
+		"\e[#{color_code}m#{self}\e[0m"
+	end
+
+	def truncate(max)
+		length > max ? "#{self[0...max]}" : self
+	end
+end
+
+class Colors
+	@@user_color = 34
+	@@colors = [31, 32, 33, 35, 36]
+	@@index = 0
+
+	def self.user
+		@@user_color
+	end
+
+	def self.next
+		color = @@colors[@@index]
+
+		# Should take care of case when more users than colors
+		if @@index < @@colors.count
+			@@index += 1
+		else
+			@@index = 0
+		end
+		color
+	end
+end
+
+class Authors
+	@@authors = {}
+
+	def self.[](key)
+		@@authors[key]
+	end
+
+	def self.[]=(key, value)
+		@@authors[key] = value
+	end
+
+	def self.include?(key)
+		@@authors.include?(key)
+	end
+
+	def self.all
+		@@authors.collect{ |k,v| v.to_s }.join(' ')
+	end
+end
+
+class Author
+	def initialize(name)
+		@lines = 0
+		@name = name
+		if name == $current_user
+			@color = Colors.user
+		else
+			@color = Colors.next
+		end
+		@initials = @name.split.collect { |word| word[0] }.join
+	end
+
+	def initials
+		@initials.colorize(@color)
+	end
+
+	def add_line
+		@lines += 1
+	end
+
+	def getColor
+		return @color
+	end
+
+	def to_s
+		"#{@name.colorize(@color)}(#{@lines})"
+	end
+end
+
+def blame(filename)
+	output = "#{filename}: "
+
+	content = `git blame --line-porcelain #{filename} 2>&1`
+
+	if content =~ /^fatal:/
+		output << content.match(/^fatal: (.*)/)[1] + "\n\n"
+		return output
+	else
+		output << "\n"
+	end
+
+	content.split(/(?<=^[a-f0-9]{40})/).each_with_index do |line, line_number|
+		if line_number == 0
+			@hash = line.split("\n").last.truncate(12)
+		end
+
+		next if line_number == 0
+
+		# Get code line
+		if line =~ /^(previous|boundary)/
+			code = line.split("\n")[12]
+		else
+			code = line.split("\n")[11]
+		end
+		code.gsub!("\t", "  ")
+
+		# Get author
+		if line =~ /^author /
+			author = line.match(/^author (.*)$/)[1]
+			if !Authors.include?(author)
+				Authors[author] = Author.new(author)
+			end
+			Authors[author].add_line
+		end
+
+		output << "%s   %s\t%-4s %s\n" % [@hash.colorize(Authors[author].getColor()), Authors[author].initials, line_number, code]
+
+		@hash = line.split("\n").last.truncate(12)
+	end
+
+	output << "\n"
+	return output
+end
+
+
+###   MAIN   ###
+
+# Make sure a file was specified
+filenames = ARGV
+if filenames.empty?
+	puts "Specify a file to blame"
+	exit 1
+end
+
+# Get the current user's name
+$current_user = `git config --get user.name`.chomp
+
+output = filenames.collect { |filename| blame(filename) }.join
+output.chomp!
+
+# Print all output
+print <<EOS
+
+#{Authors.all}
+
+#{output}
+#{Authors.all}
+
+EOS