require 'rspactor'

module RSpactor
  class Runner
    def self.start(options = {})
      run_in = options.delete(:run_in) || Dir.pwd
      new(run_in, options).start
    end
    
    attr_reader :dir, :options, :inspector, :interactor, :rspec
    
    def initialize(dir, options = {})
      @dir = dir
      @options = options
      @rspec = File.exists?(File.join(dir, 'spec'))
      read_git_head
    end
    
    def start
      load_dotfile
      puts "** RSpactor is now watching at '#{dir}'"
      start_interactor
      start_listener
    end
    
    def start_interactor
      @interactor = Interactor.new
      aborted = @interactor.wait_for_enter_key("** Hit <enter> to skip initial spec run", 3)
      @interactor.start_termination_handler
      run_all_tests unless aborted
    end
    
    def start_listener
      @inspector_class = rspec? ? RSpecInspector : TestInspector
      @inspector = @inspector_class.new(dir)

      Listener.new(@inspector_class::EXTENSIONS) do |files|
        test_changed_files(files) unless git_head_changed?
      end.run(dir)
    end
    
    def load_dotfile
      dotfile = File.join(ENV['HOME'], '.rspactor')
      if File.exists?(dotfile)
        begin
          Kernel.load dotfile
        rescue => e
          $stderr.puts "Error while loading #{dotfile}: #{e}"
        end
      end
    end

    def run_all_tests
      if rspec?
        run_spec_command(File.join(dir, 'spec'))
      else
        patterns = ["test/unit/**/*_test.rb", "test/lib/**/*_test.rb", 
          "test/functional/**/*_test.rb", "test/integration/**/*_test.rb"]
        
        patterns.each do |pattern|
          run_test_command(Dir[pattern])
        end
      end
    end

    def run_spec_command(paths)
      paths = Array(paths)
      if paths.empty?
        @last_run_failed = nil
      else
        cmd = [ruby_opts, spec_runner, paths, spec_opts].flatten.join(' ')
        @last_run_failed = run_command(cmd)
      end
    end
    
    def run_test_command(paths)
      paths = Array(paths)
      if paths.empty?
        @last_run_failed = nil
      else
        cmd = [ruby_opts, test_runner, paths].flatten.join(' ')
        @last_run_failed = run_command(cmd)
      end
    end
    
    def last_run_failed?
      @last_run_failed == false
    end
    
    protected

    def run_command(cmd)
      system(cmd)
      $?.success?
    end
    
    def test_changed_files(files)
      files_to_spec = files.inject([]) do |all, file|
        all.concat inspector.determine_spec_files(file)
      end
      unless files_to_spec.empty?
        puts files_to_spec.join("\n")
        
        previous_run_failed = last_run_failed?
        run_spec_command(files_to_spec)
        
        if options[:retry_failed] and previous_run_failed and not last_run_failed?
          run_all_tests
        end
      end
    end
    
    def rspec?
      rspec ? true : false
    end
    
    private

    def spec_opts
      if File.exist?('spec/spec.opts')
        opts = File.read('spec/spec.opts').gsub("\n", ' ')
      else
        opts = "--color"
      end

      opts << ' ' << formatter_opts
      # only add the "progress" formatter unless no other (besides growl) is specified
      opts << ' -f progress' unless opts.scan(/\s(?:-f|--format)\b/).length > 1

      opts
    end

    def formatter_opts
      "-r #{File.dirname(__FILE__)}/../rspec_growler.rb -f RSpecGrowler:STDOUT"
    end

    def spec_runner
      if File.exist?("script/spec")
        "script/spec"
      else
        "spec"
      end
    end
    
    def test_runner
      "ruby -e 'ARGV.each{|f| load f}'"
    end

    def ruby_opts
      other = ENV['RUBYOPT'] ? " #{ENV['RUBYOPT']}" : ''
      other << ' -rcoral' if options[:coral]
      if rspec?
        %(RUBYOPT='-Ilib:spec#{other}')
      else
        %(RUBYOPT='-Ilib:test#{other}')
      end
    end
    
    def git_head_changed?
      old_git_head = @git_head
      read_git_head
      @git_head and old_git_head and @git_head != old_git_head
    end
    
    def read_git_head
      git_head_file = File.join(dir, '.git', 'HEAD')
      @git_head = File.exists?(git_head_file) && File.read(git_head_file)
    end
    
  end
end

# backward compatibility
Runner = RSpactor::Runner