git-timecost/lib/timecost/cli.rb

module TimeCost
	class CLI
		def initialize
			# FIXME: accept multiple authors
			@config = {
				:author_filter_enable => false,
				:author_filter => ".*?",

				:date_filter_enable => false,
				:date_filter => [],

				:branches_filter_enable => true,

				:input_dump => [],
				:output_dump => nil,

				:range_granularity => 0.5, # in decimal hours

				:verbose => false
			}
			@rangelist = {}
			@authorlist = nil
		end

		def parse_cmdline args
			options = 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("--before DATE", "Keep only commits before DATE") do |date|
					puts "set date filter to <= #{date}"
					@config[:date_filter] << lambda { |other|
						return (other <= DateTime.parse(date))
					}
					@config[:date_filter_enable] = true
				end

				opts.on("--after DATE", "Keep only commits after DATE") do |date|
					puts "set date filter to >= #{date}"
					@config[:date_filter] << lambda { |other|
						return (other >= DateTime.parse(date))
					}
					@config[:date_filter_enable] = true
				end

				opts.on("-t","--time TIME", "Keep only commits on last TIME days") 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

				opts.on_tail("--all", "Collect from all branches and refs") do
					@config[:branches_filter_enable] = false
				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
			options.parse! args

		end


		def analyze_git
			# git log
			# foreach, create time range (before) + logs

			cmd = [
				"git", "log",
		   		"--date=iso", 
		   		"--no-patch"
			]
			if not @config[:branches_filter_enable] then  
				cmd << "--all"
			end
			cmd.concat ["--", "."]
			process = IO.popen cmd

			@rangelist = {}
			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 commit, granularity: @config[:range_granularity]

						if not @rangelist.include? commit.author then
							@rangelist[commit.author] = RangeList.new
						end
						@rangelist[commit.author].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

						# reject if a some filter does not validate date
						filter_keep = true
						filters = @config[:date_filter]
						filters.each do |f|
							filter_keep &= f.call(DateTime.parse(commit.date))
						end

						if not filter_keep then
							commit = nil
						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 |author,rangelist|
				rangelist.each do |range|
					puts range.to_s(!@config[:author_filter_enable]) + "\n"
				end
			end
			total = 0
			@rangelist.each do |author,rangelist|
				puts "SUB-TOTAL for %s: %.2f hours\n" % [author, rangelist.sum]
				total += rangelist.sum
			end
			puts "TOTAL: %.2f hours" % total
		end
	end
end