module Beaker
  module Answers
    # This class provides answer file information for PE version 3.x
    #
    # @api private
    module Version30
      # Return answer data for a host
      #
      # @param [Beaker::Host] host Host to return data for
      # @param [String] master_certname Hostname of the puppet master.
      # @param [Beaker::Host] master Host object representing the master
      # @param [Beaker::Host] dashboard Host object representing the dashboard
      # @param [Hash] options options for answer files
      # @option options [Symbol] :type Should be one of :upgrade or :install.
      # @return [Hash] A hash (keyed from hosts) containing hashes of answer file
      #   data.
      def self.host_answers(host, master_certname, master, database, dashboard, options)
        # Windows hosts don't have normal answers...
        return nil if host['platform'] =~ /windows/

        # Everything's an agent
        agent_a = {
          :q_puppetagent_install => 'y',
          :q_puppet_cloud_install => 'y',
          :q_verify_packages => ENV['q_verify_packages'] || 'y',
          :q_puppet_symlinks_install => 'y',
          :q_puppetagent_certname => host,
          :q_puppetagent_server => master_certname,

          # Disable database, console, and master by default
          # This will be overridden by other blocks being merged in.
          :q_puppetmaster_install => 'n',
          :q_all_in_one_install => 'n',
          :q_puppet_enterpriseconsole_install => 'n',
          :q_puppetdb_install => 'n',
          :q_database_install => 'n',
        }

        # These base answers are needed by all
        common_a = {
          :q_install => 'y',
          :q_vendor_packages_install => 'y',
        }

        # master/database answers
        master_database_a = {
          :q_puppetmaster_certname => master_certname
        }

        # Master/dashboard answers
        master_console_a = {
          :q_puppetdb_hostname => database,
          :q_puppetdb_port => 8081
        }

        # Master only answers
        master_a = {
          :q_puppetmaster_install => 'y',
          :q_puppetmaster_dnsaltnames => master_certname+",puppet",
          :q_puppetmaster_enterpriseconsole_hostname => dashboard,
          :q_puppetmaster_enterpriseconsole_port => 443,
        }

        if master['ip']
          master_a[:q_puppetmaster_dnsaltnames]+= "," + master['ip']
        end

        # Common answers for console and database
        dashboard_password = "'#{ENV['q_puppet_enterpriseconsole_auth_password'] || '~!@#$%^*-/ aZ'}'"
        puppetdb_password = "'#{ENV['q_puppetdb_password'] || '~!@#$%^*-/ aZ'}'"

        console_database_a = {
          :q_puppetdb_database_name => 'pe-puppetdb',
          :q_puppetdb_database_user => 'mYpdBu3r',
          :q_puppetdb_database_password => puppetdb_password,
          :q_puppet_enterpriseconsole_auth_database_name => 'console_auth',
          :q_puppet_enterpriseconsole_auth_database_user => 'mYu7hu3r',
          :q_puppet_enterpriseconsole_auth_database_password => dashboard_password,
          :q_puppet_enterpriseconsole_database_name => 'console',
          :q_puppet_enterpriseconsole_database_user => 'mYc0nS03u3r',
          :q_puppet_enterpriseconsole_database_password => dashboard_password,

          :q_database_host => database,
          :q_database_port => 5432
        }

        # Console only answers
        dashboard_user = "'#{ENV['q_puppet_enterpriseconsole_auth_user_email'] || 'admin@example.com'}'"

        smtp_host = "'#{ENV['q_puppet_enterpriseconsole_smtp_host'] || dashboard}'"
        smtp_port = "'#{ENV['q_puppet_enterpriseconsole_smtp_port'] || 25}'"
        smtp_username = ENV['q_puppet_enterpriseconsole_smtp_username']
        smtp_password = ENV['q_puppet_enterpriseconsole_smtp_password']
        smtp_use_tls = "'#{ENV['q_puppet_enterpriseconsole_smtp_use_tls'] || 'n'}'"

        console_a = {
          :q_puppet_enterpriseconsole_install => 'y',
          :q_puppet_enterpriseconsole_inventory_hostname => host,
          :q_puppet_enterpriseconsole_inventory_certname => host,
          :q_puppet_enterpriseconsole_inventory_dnsaltnames => dashboard,
          :q_puppet_enterpriseconsole_inventory_port => 8140,
          :q_puppet_enterpriseconsole_master_hostname => master,

          :q_puppet_enterpriseconsole_auth_user_email => dashboard_user,
          :q_puppet_enterpriseconsole_auth_password => dashboard_password,

          :q_puppet_enterpriseconsole_httpd_port => 443,

          :q_puppet_enterpriseconsole_smtp_host => smtp_host,
          :q_puppet_enterpriseconsole_smtp_use_tls => smtp_use_tls,
          :q_puppet_enterpriseconsole_smtp_port => smtp_port,
        }

        if smtp_password and smtp_username
          console_a.merge!({
                             :q_puppet_enterpriseconsole_smtp_password => "'#{smtp_password}'",
                             :q_puppet_enterpriseconsole_smtp_username => "'#{smtp_username}'",
                             :q_puppet_enterpriseconsole_smtp_user_auth => 'y'
                           })
        end

        # Database only answers
        database_a = {
          :q_puppetdb_install => 'y',
          :q_database_install => 'y',
          :q_database_root_password => "'=ZYdjiP3jCwV5eo9s1MBd'",
          :q_database_root_user => 'pe-postgres',
        }

        # Special answers for special hosts
        aix_a = {
          :q_run_updtvpkg => 'y',
        }

        answers = common_a.dup

        unless options[:type] == :upgrade
          answers.merge! agent_a
        end

        if host == master
          answers.merge! master_console_a
          unless options[:type] == :upgrade
            answers.merge! master_a
            answers.merge! master_database_a
          end
        end

        if host == dashboard
          answers.merge! master_console_a
          answers.merge! console_database_a
          answers[:q_pe_database] = 'y'
          unless options[:type] == :upgrade
            answers.merge! console_a
          else
            answers[:q_database_export_dir] = '/tmp'
          end
        end

        if host == database
          if database != master
            if options[:type] == :upgrade
              # This is kinda annoying - if we're upgrading to 3.0 and are
              # puppetdb, we're actually doing a clean install. We thus
              # need the core agent answers.
              answers.merge! agent_a
            end
            answers.merge! master_database_a
          end
          answers.merge! database_a
          answers.merge! console_database_a
        end

        if host == master and host == database and host == dashboard
          answers[:q_all_in_one_install] = 'y'
        end

        if host['platform'].include? 'aix'
          answers.merge! aix_a
        end

        return answers
      end

      # Return answer data for all hosts.
      #
      # @param [Array<Beaker::Host>] hosts An array of host objects.
      # @param [String] master_certname Hostname of the puppet master.
      # @param [Hash] options options for answer files
      # @option options [Symbol] :type Should be one of :upgrade or :install.
      # @return [Hash] A hash (keyed from hosts) containing hashes of answer file
      #   data.
      def self.answers(hosts, master_certname, options)
        the_answers = {}
        database = only_host_with_role(hosts, 'database')
        dashboard = only_host_with_role(hosts, 'dashboard')
        master = only_host_with_role(hosts, 'master')
        hosts.each do |h|
          if options[:type] == :upgrade and h[:pe_ver] =~ /\A3.0/
            # 3.0.x to 3.0.x should require no answers
            the_answers[h.name] = {
              :q_install => 'y',
              :q_install_vendor_packages => 'y',
            }
          else
            the_answers[h.name] = host_answers(h, master_certname, master, database, dashboard, options)
          end
        end
        return the_answers
      end
    end
  end
end