require 'rspactor'

module RSpactor
  class Runner
    def self.start(options = {})
      new(Dir.pwd, options).start
    end
    
    attr_reader :dir, :options, :inspector, :interactor
    
    def initialize(dir, options = {})
      @dir = dir
      @options = options
      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_specs unless aborted
    end
    
    def start_listener
      @inspector = Inspector.new(dir)

      Listener.new(:extensions => Inspector::DEFAULT_EXTENSIONS) do |files|
        spec_changed_files(files) unless git_head_changed?
      end.start(dir).enter_run_loop
    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_specs
      run_spec_command('spec')
    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 last_run_failed?
      @last_run_failed == false
    end
    
    protected

    def run_command(cmd)
      system(cmd)
      $?.success?
    end
    
    def spec_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_specs
        end
      end
    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 ruby_opts
      other = ENV['RUBYOPT'] ? " #{ENV['RUBYOPT']}" : ''
      other << ' -rcoral' if options[:coral]
      %(RUBYOPT='-Ilib:spec#{other}')
    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