require_relative 'docker_run'
require_relative '../lib/inspec'
#
# BUGON: These requires are to get around concurrency issues with
# autoloading in Ruby
#
require 'train'
require 'train/plugins'
require 'train/plugins/transport'
require 'train/transports/docker'

tests = ARGV
if tests.empty?
  puts 'Nothing to do.'
  exit 0
end

class DockerTester
  def initialize(tests)
    @tests = tests
    @docker = DockerRunner.new
  end

  def run
    puts ['Running tests:', @tests].flatten.join("\n- ")
    puts ''

    conf = RSpec.configuration
    reporter = conf.reporter
    results = nil

    # start reporting loop
    reporter.report(0) do |report|
      results = @docker.run_all do |name, container|
        status = test_container(container, report)
        status.all? ? nil : "Failed to run tests on #{name}"
      end
    end

    # check if we were successful
    failures = results.compact
    failures.each { |f| puts "\033[31;1m#{f}\033[0m\n\n" }
    failures.empty? or fail 'Test failures'
  end

  def test_container(container, report)
    puts "--> run test on docker #{container.id}"
    opts = { 'target' => "docker://#{container.id}" }
    runner = Inspec::Runner.new(opts)
    @tests.each { |test| runner.add_target(test) }
    runner.tests.map { |g| g.run(report) }
  end
end

DockerTester.new(tests).run