180 lines
4 KiB
Ruby
180 lines
4 KiB
Ruby
module TimeCost
|
|
class CLI
|
|
def initialize
|
|
# FIXME: accept multiple authors
|
|
@config = {
|
|
:author_filter_enable => false,
|
|
:author_filter => ".*?",
|
|
|
|
:date_filter_enable => false,
|
|
:date_filter => ".*?",
|
|
|
|
:input_dump => [],
|
|
:output_dump => nil,
|
|
|
|
:range_granularity => 0.5, # in decimal hours
|
|
|
|
:verbose => false
|
|
}
|
|
@rangelist = nil
|
|
@authorlist = nil
|
|
end
|
|
|
|
def parse_cmdline args
|
|
opts = OptionParser.new do |opts|
|
|
opts.banner = "Usage: #{File.basename $0} [options]"
|
|
|
|
opts.on_tail("-v","--verbose", "Run verbosely") do |v|
|
|
@config[:verbose] = true
|
|
end
|
|
|
|
opts.on_tail("-h","--help", "Show this help") do
|
|
puts opts
|
|
exit 0
|
|
end
|
|
|
|
opts.on("-i","--input FILE", "Set input dump file") do |file|
|
|
@config[:input_dump] << file
|
|
end
|
|
|
|
opts.on("-o","--output FILE", "Set output dump file") do |file|
|
|
@config[:output_dump] = file
|
|
end
|
|
|
|
opts.on("-d","--date DATE", "Keep only commits since DATE") do |date|
|
|
puts "set date filter to #{date}"
|
|
@config[:date_filter] = DateTime.parse(date);
|
|
@config[:date_filter_enable] = true
|
|
end
|
|
|
|
opts.on("-t","--time TIME", "Keep only commits on last TIME datys") do |time|
|
|
puts "set time filter to latest #{time} days"
|
|
@config[:date_filter] = DateTime.now - time.to_f;
|
|
puts "set date filter to date = #{@config[:date_filter]}"
|
|
@config[:date_filter_enable] = true
|
|
end
|
|
|
|
opts.on("-a","--author AUTHOR", "Keep only commits by AUTHOR") do |author|
|
|
puts "set author filter to #{author}"
|
|
@config[:author_filter] = author
|
|
@config[:author_filter_enable] = true
|
|
end
|
|
|
|
# overlap :
|
|
#
|
|
opts.on("-s","--scotch GRANULARITY", "Use GRANULARITY (decimal hours) to merge ranges") do |granularity|
|
|
puts "set scotch to #{granularity}"
|
|
@config[:range_granularity] = granularity.to_f
|
|
end
|
|
end
|
|
opts.parse! args
|
|
|
|
end
|
|
|
|
|
|
def analyze_git
|
|
# git log
|
|
# foreach, create time range (before) + logs
|
|
process = IO.popen ["git", "log",
|
|
"--date=iso",
|
|
"--no-patch",
|
|
"--","."]
|
|
|
|
|
|
@rangelist = RangeList.new
|
|
commit = nil
|
|
loop do
|
|
line = process.gets
|
|
break if line.nil?
|
|
# utf-8 fix ?
|
|
# line.encode!( line.encoding, "binary", :invalid => :replace, :undef => :replace)
|
|
line.strip!
|
|
|
|
case line
|
|
when /^commit (.*)$/ then
|
|
id = $1
|
|
# merge ranges & push
|
|
unless commit.nil? then
|
|
range = Range.new @config, commit
|
|
@rangelist.add range
|
|
end
|
|
commit = Commit.new id
|
|
# puts "commit #{id}"
|
|
|
|
when /^Author:\s*(.*?)\s*$/ then
|
|
unless commit.nil? then
|
|
commit.author = $1
|
|
|
|
if @config[:author_filter_enable] and
|
|
(not commit.author =~ /#{@config[:author_filter]}/) then
|
|
commit = nil
|
|
# reject
|
|
end
|
|
end
|
|
|
|
when /^Date:\s*(.*?)\s*$/ then
|
|
unless commit.nil? then
|
|
commit.date = $1
|
|
|
|
if @config[:date_filter_enable] and
|
|
(DateTime.parse(commit.date) < @config[:date_filter]) then
|
|
commit = nil
|
|
# reject
|
|
end
|
|
end
|
|
|
|
when /^\s*$/ then
|
|
# skip
|
|
|
|
else
|
|
# add as note
|
|
unless commit.nil? then
|
|
commit.note = if commit.note.nil? then line
|
|
else commit.note + "\n" + line
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
def analyze_dumps
|
|
#read ranges
|
|
@rangelist = RangeList.new
|
|
|
|
@config[:input_dump].each do |filename|
|
|
rangelist = YAML::load(File.open(filename,"r"))
|
|
rangelist.each do |range|
|
|
@rangelist.add range
|
|
end
|
|
end
|
|
end
|
|
|
|
def analyze
|
|
if @config[:input_dump].empty? then
|
|
analyze_git
|
|
else
|
|
analyze_dumps
|
|
end
|
|
end
|
|
|
|
def export
|
|
return if @config[:output_dump].nil?
|
|
puts "Exporting to %s" % @config[:output_dump]
|
|
File.open(@config[:output_dump], "w") do |file|
|
|
file.puts YAML::dump(@rangelist)
|
|
end
|
|
end
|
|
|
|
def report
|
|
return if not @config[:output_dump].nil?
|
|
|
|
@rangelist.each do |r|
|
|
puts r.to_s + "\n"
|
|
end
|
|
puts "TOTAL: %.2f hours" % @rangelist.sum
|
|
end
|
|
end
|
|
end
|
|
|