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