require 'docker'
require 'yaml'
require_relative '../lib/vulcano'

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

class DockerTester
  def initialize(tests)
    @tests = tests
    @images = docker_images_by_tag
    @conf = tests_conf
  end

  def run
    # test all images
    @conf['images'].each{|n|
      test_image(n)
    }.all? or raise "Test failures"
  end

  def docker_images_by_tag
    # get all docker image tags
    images = {}
    Docker::Image.all.map do |img|
      Array(img.info['RepoTags']).each do |tag|
        images[tag] = img
      end
    end
    images
  end

  def tests_conf
    # get the test configuration
    conf_path = File::join(File::dirname(__FILE__), '..', '.tests.yaml')
    raise "Can't find tests config in #{conf_path}" unless File::file?(conf_path)
    conf = YAML.load(File::read(conf_path))
  end

  def test_container(container_id)
    opts = { target: "docker://#{container_id}" }
    runner = Vulcano::Runner.new(nil, opts)
    runner.add_tests(@tests)
    runner.run
  end

  def test_image(name)
    dname = "docker-#{name}:latest"
    image = @images[dname]
    raise "Can't find docker image #{dname}" if image.nil?

    container = Docker::Container.create(
      'Cmd' => [ '/bin/bash' ],
      'Image' => image.id,
      'OpenStdin' => true,
    )
    container.start

    res = test_container(container.id)

    container.kill
    container.delete(force: true)
    res
  end
end

DockerTester.new(tests).run