#!/usr/bin/env ruby

require 'optparse'
require 'fileutils'
require 'date'
require 'rubygems'

require 'haml'
require 'sass'
require 'compass'
require 'gnuplot'

$LOAD_PATH.unshift(File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib')))

require 'gitstats'

$options = {
  :out => 'stats',
  :template => nil,
  :verbose => false,
  :debug => false,
  :quiet => false,
  :cache => false,
  :statcache => nil,
  :commitcache => nil,
  :commitcache_dir => nil,
  :future => true,
  :maxage => 0,
  :withmail => false,
}

parser = OptionParser.new do |opts|
  opts.banner = 'Usage: gitstats.rb [options] <[name1:]gitdir1[:ref1]> [<[name2:]gitdir2[:ref2]> ...]'

  opts.on('-o', '--out=arg', 'output directory') do |arg|
    $options[:out] = arg
  end

  opts.on('-t', '--template=arg', 'template directory') do |arg|
    $options[:template] = arg
  end

  opts.on('-c', '--[no-]cache', 'use the statcache file') do |arg|
    $options[:cache] = arg
  end

  opts.on('-C', '--[no-]commitcache', 'use the commit cache') do |arg|
    $options[:commitcache] = arg
  end

  opts.on('-s', '--statcache=arg', 'statcache file to use') do |arg|
    $options[:statcache] = arg
  end

  opts.on('--commitcache=arg', 'commit cache directory to use') do |arg|
    $options[:commitcache_dir] = arg
  end

  opts.on('--[no-]future', 'count future commits') do |arg|
    $options[:future] = arg
  end

  opts.on('-m', '--max-age=arg', Integer, 'set max age of commit in days') do |arg|
    $options[:maxage] = arg
  end

  opts.on('--[no-]mail', 'include mail in author names') do |arg|
    $options[:withmail] = arg
  end

  opts.on('-v', '--[no-]verbose', 'verbose mode') do |arg|
    $options[:verbose] = arg
  end

  opts.on('-q', '--[no-]quiet', 'quiet mode') do |arg|
    $options[:quiet] = arg
  end

  opts.on('-d', '--[no-]debug', 'print debug messages') do |arg|
    $options[:debug] = arg
  end

  opts.on_tail('-h', '--help', 'this help') do
    puts opts
    exit 0
  end
end

parser.parse!

if $options[:quiet] && $options[:verbose]
  STDERR.puts 'cannot specify --quiet and --verbose at the same time!'
  exit 1
end

$options[:statcache] = File.join($options[:out], '.statcache') if $options[:statcache].nil?
$options[:commitcache_dir] = $options[:out] if $options[:commitcache_dir].nil?
$options[:template] = File.expand_path(File.join(File.dirname(__FILE__), '..', 'template')) if $options[:template].nil?

FileUtils.mkdir_p($options[:out])

stat = nil
cache_loaded = false
if $options[:cache]
  begin
    puts 'trying to load cache ...' unless $options[:quiet]
    stat = Marshal::load(IO::readlines($options[:statcache]).join(''))
    stat.clear_repos
    cache_loaded = true
  rescue
  end
end

if stat.nil?
  if ARGV.empty?
    puts parser
    exit 1
  end

  stat = StatGen.new
end

if cache_loaded
  if stat.include_mail != $options[:withmail]
    puts 'cannot change --[no-]mail option when using statcache'
    exit 1
  end

  if stat.future != $options[:future]
    puts 'cannot change --[no-]future option when using statcache'
    exit 1
  end

  if stat.maxage != $options[:maxage] * 24 * 60 * 60
    puts 'cannot change --max-age option when using statcache'
    exit 1
  end
end

Author::include_mail = $options[:withmail]
stat.include_mail = $options[:withmail]
stat.verbose = $options[:verbose]
stat.debug = $options[:debug]
stat.quiet = $options[:quiet]
stat.future = $options[:future]
stat.maxage = $options[:maxage] * 24 * 60 * 60
stat.commitcache = $options[:commitcache] ? $options[:commitcache_dir] : nil

ARGV.each do |path|
  name, path, ref = path.split(':')
  path ||= name
  ref ||= 'HEAD'
  stat << [name, path, ref]
end

if cache_loaded
  unless stat.check_repostate
    puts 'cannot use cache when working on different repositories!'
    exit 1
  end
end

puts 'fetching statistics ...' unless $options[:quiet]
begin
  stat.calc
ensure
  if $options[:cache]
    puts 'writing cache ...' unless $options[:quiet]
    cache = Marshal::dump(stat)
    File.new($options[:statcache], 'w').write(cache)
  end
end

puts 'rendering ...' unless $options[:quiet]
renderer = Renderer.new($options[:template], $options[:out], $options[:verbose])
renderer.render(stat)