require 'tork/server'

module Tork
class CLIApp < Server

  def loop
    tell @clients, 'Absorbing test execution overhead...', false
    @driver = popen('tork-driver')
    super
  ensure
    pclose @driver
  end

protected

  def join client
    super
    help client
  end

  BACKTRACE_CENSOR = /\n\s+(?:from\s)?#{
    Regexp.union(File.expand_path('../../..', __FILE__), TORK_DOLLAR_ZERO)
  }[^:]*:\d+:.+$/

  def recv client, message
    case client
    when @driver
      event, *details = message

      case event_sym = event.to_sym
      when :absorb
        tell @clients, 'Overhead absorbed. Ready for testing!', false

      when :reabsorb
        tell @clients, 'Reabsorbing changed overhead files...', false

      when :test, :pass, :fail
        test_file, line_numbers, log_file, worker_number, exit_status = details
        message = [event.upcase, [test_file, *line_numbers].join(':'),
                   exit_status].compact.join(' ')

        color = case event_sym
                when :pass then "\e[34m%s\e[0m" # blue
                when :fail then "\e[31m%s\e[0m" # red
                end
        message = color % message if color and STDOUT.tty?

        if event_sym == :fail
          # censor Tork internals from test failure backtraces
          log = File.read(log_file).gsub(BACKTRACE_CENSOR, '')
          message = [message, log, message]
        end

        tell @clients, message, false

      when :idle
        tested, passed, failed = details.map(&:length)
        tell @clients, "#{tested} tested, #{passed} passed, #{failed} failed"
      end
    else
      key = message.shift.lstrip[0,1].downcase
      cmd = Array(COMMANDS.fetch(key, [:help, client])) + message
      if respond_to? cmd.first, true
        __send__(*cmd)
      else
        tell @clients, "Sending #{cmd.inspect} command...", false
        send @driver, cmd
      end
    end
  end

private

  COMMANDS = {
    't' => :run_test_file,
    'a' => :run_all_test_files,
    's' => :stop_running_test_files,
    'k' => [:stop_running_test_files, :SIGKILL],
    'p' => :rerun_passed_test_files,
    'f' => :rerun_failed_test_files,
    'l' => :list_failed_test_files,
    'o' => :reabsorb_overhead,
    'q' => :quit,
  }

  def help client
    tell client, <<HELP, false
Type a then ENTER to run all test files.
Type t then SPACE then a filename then ENTER to run a specific test file.
Type s then ENTER to stop currently running test files.
Type k then ENTER to kill currently running test files.
Type p then ENTER to re-run currently passing test files.
Type f then ENTER to re-run currently failing test files.
Type l then ENTER to list currently failing test files.
Type o then ENTER to re-absorb test execution overhead.
Type h then ENTER to see this help message.
Type q then ENTER to quit.
HELP
  end

end
end