module VagrantPlugins
  module Ansible
    module Config
      class Base < Vagrant.plugin("2", :config)

        attr_accessor :extra_vars
        attr_accessor :galaxy_role_file
        attr_accessor :galaxy_roles_path
        attr_accessor :galaxy_command
        attr_accessor :groups
        attr_accessor :inventory_path
        attr_accessor :limit
        attr_accessor :playbook
        attr_accessor :raw_arguments
        attr_accessor :skip_tags
        attr_accessor :start_at_task
        attr_accessor :sudo
        attr_accessor :sudo_user
        attr_accessor :tags
        attr_accessor :vault_password_file
        attr_accessor :verbose

        def initialize
          @extra_vars          = UNSET_VALUE
          @galaxy_role_file    = UNSET_VALUE
          @galaxy_roles_path   = UNSET_VALUE
          @galaxy_command      = UNSET_VALUE
          @groups              = UNSET_VALUE
          @inventory_path      = UNSET_VALUE
          @limit               = UNSET_VALUE
          @playbook            = UNSET_VALUE
          @raw_arguments       = UNSET_VALUE
          @skip_tags           = UNSET_VALUE
          @start_at_task       = UNSET_VALUE
          @sudo                = UNSET_VALUE
          @sudo_user           = UNSET_VALUE
          @tags                = UNSET_VALUE
          @vault_password_file = UNSET_VALUE
          @verbose             = UNSET_VALUE
        end

        def finalize!
          @extra_vars          = nil   if @extra_vars           == UNSET_VALUE
          @galaxy_role_file    = nil   if @galaxy_role_file     == UNSET_VALUE
          @galaxy_roles_path   = nil   if @galaxy_roles_path    == UNSET_VALUE
          @galaxy_command      = "ansible-galaxy install --role-file=%{ROLE_FILE} --roles-path=%{ROLES_PATH} --force" if @galaxy_command == UNSET_VALUE
          @groups              = {}    if @groups               == UNSET_VALUE
          @inventory_path      = nil   if @inventory_path       == UNSET_VALUE
          @limit               = nil   if @limit                == UNSET_VALUE
          @playbook            = nil   if @playbook             == UNSET_VALUE
          @raw_arguments       = nil   if @raw_arguments        == UNSET_VALUE
          @skip_tags           = nil   if @skip_tags            == UNSET_VALUE
          @start_at_task       = nil   if @start_at_task        == UNSET_VALUE
          @sudo                = false if @sudo                 != true
          @sudo_user           = nil   if @sudo_user            == UNSET_VALUE
          @tags                = nil   if @tags                 == UNSET_VALUE
          @vault_password_file = nil   if @vault_password_file  == UNSET_VALUE
          @verbose             = false if @verbose              == UNSET_VALUE
        end

        # Just like the normal configuration "validate" method except that
        # it returns an array of errors that should be merged into some
        # other error accumulator.
        def validate(machine)
          @errors = _detected_errors

          # Validate that a playbook path was provided
          if !playbook
            @errors << I18n.t("vagrant.provisioners.ansible.errors.no_playbook")
          end

          if playbook
            check_path_is_a_file(machine, playbook, "vagrant.provisioners.ansible.errors.playbook_path_invalid")
          end

          if inventory_path
            check_path_exists(machine, inventory_path, "vagrant.provisioners.ansible.errors.inventory_path_invalid")
          end

          if galaxy_role_file
            check_path_is_a_file(machine, galaxy_role_file, "vagrant.provisioners.ansible.errors.galaxy_role_file_invalid")
          end

          if vault_password_file
            check_path_is_a_file(machine, vault_password_file, "vagrant.provisioners.ansible.errors.vault_password_file_invalid")
          end

          # Validate that extra_vars is either a hash, or a path to an existing file
          if extra_vars
            extra_vars_is_valid = extra_vars.kind_of?(Hash) || extra_vars.kind_of?(String)
            if extra_vars.kind_of?(String)
              # Accept the usage of '@' prefix in Vagrantfile (e.g. '@vars.yml'
              # and 'vars.yml' are both supported)
              match_data = /^@?(.+)$/.match(extra_vars)
              extra_vars_path = match_data[1].to_s
              extra_vars_is_valid = check_path_is_a_file(machine, extra_vars_path)
              if extra_vars_is_valid
                @extra_vars = '@' + extra_vars_path
              end
            end

            if !extra_vars_is_valid
              @errors << I18n.t(
                "vagrant.provisioners.ansible.errors.extra_vars_invalid",
                type:  extra_vars.class.to_s,
                value: extra_vars.to_s)
            end
          end

        end
      end
    end
  end
end