require 'yaml'
require_relative '../test_helper'

class TestCLI < Minitest::Test
  include Orats::Test

  attr_accessor :target_path, :extra_flags

  def startup
    @target_path = ''
    @extra_flags = ''
  end

  def teardown
    assert_nuked unless @target_path.nil?
  end

  def test_project
    assert_project ansible: :assert
  end

  def test_project_with_auth
    assert_project '--template auth', ansible: :assert
  end

  def test_project_without_ansible
    assert_project '--skip-ansible', ansible: :refute
  end

  def test_project_with_invalid_template
    @target_path = generate_app_name
    @extra_flags = "#{ORATS_NEW_FLAGS} --template foo"

    assert_orats 'project', 'not a valid template'
  end

  def test_inventory
    assert_inventory
  end

  def test_playbook
    assert_playbook
  end

  def test_playbook_update_roles
    target_path = generate_app_name

    assert_playbook target_path
    assert_playbook target_path
  end

  def test_diff
    assert_diff

    out, err = capture_orats('diff')

    assert_count out, 'gem', 1
    assert_count out, 'missing', 0
    assert_count out, 'outdated', 0
    assert_count out, 'extra', 0
  end

  def test_diff_with_differences
    file_paths = assert_diff

    hosts_path      = "#{file_paths[0]}/hosts"
    inventory_path  = "#{file_paths[0]}/group_vars/all.yml"
    galaxyfile_path = "#{file_paths[1]}/Galaxyfile"
    site_path       = "#{file_paths[1]}/site.yml"

    gsub_file(hosts_path, '[cache]', '[something]')
    gsub_file(inventory_path, 'postgres_user', 'hello_world')
    gsub_file(galaxyfile_path, /nickjj.ruby,v.*$/, "nickjj.ruby,v0.1.2\nfoo")
    gsub_file(site_path, 'nickjj.whenever', 'bar')

    out, err = capture_orats('diff')

    assert_count out, 'gem', 1
    assert_count out, 'missing', 3
    assert_count out, 'outdated', 1
    assert_count out, 'extra', 4
  end

  def test_templates
    assert_orats 'templates', '--template auth'
  end

  def test_version
    assert_orats 'version', 'Orats'
  end

  private

  def assert_orats(command, match_regex, ansible: nil)
    out, err = capture_orats(command)

    assert_match /#{match_regex}/, out, err unless match_regex.empty?

    assert_or_refute_ansible ansible if ansible
  end

  def assert_project(flags = '', ansible: nil)
    @target_path         = generate_app_name
    absolute_target_path = "#{TEST_PATH}/#{@target_path}"
    @extra_flags         = "#{ORATS_NEW_FLAGS} #{flags}"

    assert_orats 'project', 'Start your server', ansible: ansible

    if ansible == :assert
      assert_ansible_yaml "#{absolute_target_path}/inventory/group_vars/all.yml"
    end

    absolute_target_path << "/services/#{@target_path}"
    assert_project_tests_pass absolute_target_path
  end

  def assert_project_tests_pass(target_path)
    out, err = capture_subprocess_io do
      system "cd #{target_path} && bundle exec rake test"
    end

    log_rails_test_results out
    assert out.include?('0 failures') && out.include?('0 errors'), err
  end

  def assert_inventory
    @target_path = generate_app_name

    assert_orats 'inventory', 'success'
    assert_path "#{TEST_PATH}/#{@target_path}/inventory/hosts"
    assert_ansible_yaml "#{TEST_PATH}/#{@target_path}/inventory/group_vars/all.yml"
  end

  def assert_playbook(target_path = generate_app_name)
    @target_path = target_path

    assert_orats 'playbook', 'success'
    assert_path "#{TEST_PATH}/#{@target_path}/Galaxyfile"
    assert_ansible_yaml "#{TEST_PATH}/#{@target_path}/site.yml"
  end

  def assert_diff
    assert_inventory
    inventory_path = "#{TEST_PATH}/#{@target_path}/inventory"

    assert_playbook
    playbook_path = "#{TEST_PATH}/#{@target_path}"

    @target_path = ''
    @extra_flags = "-i #{inventory_path} -p #{playbook_path}"

    [inventory_path, playbook_path]
  end

  def assert_nuked(options = {})
    out, err = capture_subprocess_io do
      orats "nuke #{@target_path}", flags: options[:flags], answer: 'y'
    end

    assert_match /#{@target_path}/, out
    system "rm -rf #{TEST_PATH}"
  end

  def assert_count(input, regex, expected)
    assert input.scan(regex).size == expected,
           "Found #{input.scan(regex).size} matches for '#{regex}' but expected #{expected}"
  end

  def assert_in_file(file_path, match_regex)
    file_contents = `cat #{file_path}`
    assert_match /#{match_regex}/, file_contents
  end

  def assert_path(file_or_dir)
    assert File.exists?(file_or_dir), "Expected path '#{file_or_dir}' to exist"
  end

  def refute_path(file_or_dir)
    refute File.exists?(file_or_dir), "Expected path '#{file_or_dir}' to exist"
  end

  def assert_or_refute_ansible(assert_or_refute)
    absolute_target_path = "#{TEST_PATH}/#{@target_path}"

    assert_path "#{absolute_target_path}/services"
    send("#{assert_or_refute.to_s}_path",
         "#{absolute_target_path}/inventory")
    send("#{assert_or_refute.to_s}_path",
         "#{absolute_target_path}/secrets")
  end

  def assert_ansible_yaml(file_path)
    begin
      file                 = IO.read(file_path)
      invalid_ansible_yaml = file.match(/^([^#].\w): {/)

      if invalid_ansible_yaml
        assert false, "Invalid yaml syntax found near:\n #{invalid_ansible_yaml}\n\nYou need to include quotes around values that start with jinja template tags:\n Example, foo: '{{ bar }}'"
        return
      end

      assert YAML.load(file)
    rescue Psych::SyntaxError => ex
      assert false, ex.message
    end
  end

  def capture_orats(command)
    out, err = capture_subprocess_io do
      orats "#{command} #{@target_path}", flags: @extra_flags
    end

    [out, err]
  end

  def log_rails_test_results(out)
    out_lines = out.split("\n")

    out_lines.delete_if do |line|
      line.include?('Sidekiq') || line.start_with?('.') ||
          line.include?('Running') || line.include?('Run options') ||
          line.empty?
    end

    puts
    puts '-'*80
    puts 'Results of running `bundle exec rake test` on the generated test app:'
    puts '-'*80
    puts out_lines.join("\n\n").rstrip
    puts '-'*80
    puts
  end

  def gsub_file(file_path, replace, with)
    IO.write(file_path, File.open(file_path) do |f|
      f.read.gsub(replace, with)
    end
    )
  end
end