require File.join(File.dirname(__FILE__), '..', 'shindo')

@interrupt = proc do
  unless Thread.main[:exit]
    Formatador.display_line('Gracefully Exiting... (ctrl-c to force)')
    Thread.main[:exit] = true
  else
    Formatador.display_line('Exiting...')
    Thread.exit
  end
end
Kernel.trap('INT', @interrupt)

# if lib dir is available add it to load path
if File.directory?('lib')
  lib_dir = File.expand_path('lib')
  unless $LOAD_PATH.include?(lib_dir)
    $LOAD_PATH.unshift(lib_dir)
  end
end

helpers = Dir.glob(File.join('tests', '**', '*helper.rb')).sort_by {|helper| helper.count(File::SEPARATOR)}
tags, tests = [], nil
for argument in ARGV
  if argument.match(/^[\+\-]/)
    tags << argument
  else
    path = File.expand_path(argument)
    if File.directory?(path)
      tests ||= []
      tests |= Dir.glob(File.join(path, '**', '*tests.rb'))
    elsif File.exist?(path)
      tests ||= []
      tests << path
    else
      Formatador.display_line("[red][bold]#{argument}[/] [red]does not exist, please fix this path and try again.[/]")
      Kernel.exit(1)
    end
  end
end

# ARGV was empty or only contained tags
unless tests
  tests = Dir.glob(File.join('tests', '**', '*tests.rb'))
end

@started_at = Time.now
def run_in_thread(helpers, tests, thread_locals)
  shindo = Thread.new {
    for key, value in thread_locals
      Thread.current[key] = value
    end
    for file in helpers
      unless Thread.main[:exit]
        load(file)
      end
    end
    for file in tests
      Thread.current[:file] = file
      unless Thread.main[:exit]
        load(file)
      end
    end
  }
  shindo.join
  if shindo[:reload]
    run_in_thread(helpers, tests, thread_locals)
  else
    @totals = shindo[:totals]
  end
end
run_in_thread(helpers, tests, @thread_locals.merge({:tags => tags}))

@totals   ||= { :failed => 0, :errored => 0, :pending => 0, :succeeded => 0 }
@success  = @totals[:failed] + @totals[:errored] == 0
status = []
status << "[red]#{@totals[:failed]} failed[/]" if @totals[:failed] > 0
status << "[red]#{@totals[:errored]} errored[/]" if @totals[:errored] > 0
status << "[yellow]#{@totals[:pending]} pending[/]" if @totals[:pending] > 0
status << "[green]#{@totals[:succeeded]} succeeded[/]"
status = [status[0...-1].join(', '), 'and', status[-1]] if status.length >= 2
status << "in [bold]#{Time.now - @started_at}[/] seconds"
Formatador.display_lines(['', status.join(' '), ''])

if @success
  Kernel.exit(0)
else
  Kernel.exit(1)
end
