require 'pty'
require 'open3'
require 'fileutils'
require 'terminal-table'

module Scan
  class Runner
    def run
      handle_results(test_app)
    end

    def test_app
      open_simulator_for_device(Scan.device)

      command = TestCommandGenerator.generate
      prefix_hash = [
        {
          prefix: "Running Tests: ",
          block: proc do |value|
            value.include?("Touching")
          end
        }
      ]
      exit_status = 0
      FastlaneCore::CommandExecutor.execute(command: command,
                                          print_all: true,
                                      print_command: true,
                                             prefix: prefix_hash,
                                            loading: "Loading...",
                                              error: proc do |error_output|
                                                begin
                                                  exit_status = $?.exitstatus
                                                  ErrorHandler.handle_build_error(error_output)
                                                rescue => ex
                                                  SlackPoster.new.run({
                                                    build_errors: 1
                                                  })
                                                  raise ex
                                                end
                                              end)
      exit_status
    end

    def handle_results(tests_exit_status)
      # First, generate a JUnit report to get the number of tests
      require 'tempfile'
      output_file = Tempfile.new("junit_report")

      report_collector = ReportCollector.new(Scan.config[:open_report],
                                             Scan.config[:output_types],
                                             Scan.config[:output_directory])

      cmd = report_collector.generate_commands(TestCommandGenerator.xcodebuild_log_path,
                                               types: 'junit',
                                               output_file_name: output_file.path).values.last
      system(cmd)

      result = TestResultParser.new.parse_result(output_file.read)
      SlackPoster.new.run(result)

      if result[:failures] > 0
        failures_str = result[:failures].to_s.red
      else
        failures_str = result[:failures].to_s.green
      end

      puts Terminal::Table.new({
        title: "Test Results",
        rows: [
          ["Number of tests", result[:tests]],
          ["Number of failures", failures_str]
        ]
      })
      puts ""

      report_collector.parse_raw_file(TestCommandGenerator.xcodebuild_log_path)

      UI.user_error!("Test execution failed. Exit status: #{tests_exit_status}") unless tests_exit_status == 0
      UI.user_error!("Tests failed") unless result[:failures] == 0
    end

    def open_simulator_for_device(device)
      return unless ENV['FASTLANE_EXPLICIT_OPEN_SIMULATOR']

      UI.message("Killing all running simulators")
      `killall Simulator &> /dev/null`

      UI.message("Explicitly opening simulator for device: #{device.name}")
      `open -a Simulator --args -CurrentDeviceUDID #{device.udid}`
    end
  end
end