require 'beaker-hostgenerator/data'
require 'beaker-hostgenerator/hypervisor'
require 'deep_merge/rails_compat'

module BeakerHostGenerator
  module Hypervisor
    class Docker < BeakerHostGenerator::Hypervisor::Interface
      include BeakerHostGenerator::Data

      def generate_node(node_info, base_config, bhg_version)
        base_config['docker_cmd'] = ['/sbin/init']

        ostype = node_info['ostype']

        if (commands = image_commands(ostype))
          base_config['docker_image_commands'] = commands
        end

        base_config['image'] = image(ostype)
        base_config['image'].prepend('amd64/') if node_info['bits'] == '64' && !base_config['image'].start_with?('quay.io/')

        base_generate_node(node_info, base_config, bhg_version, :docker)
      end

      private

      def image(ostype)
        image = ostype.sub(/(\d)/, ':\1')

        case ostype
        when /^centos/
          version = ostype.delete_prefix('centos')
          tag = (version.to_i >= 8) ? "stream#{version}" : "centos#{version}"
          image = "quay.io/centos/centos:#{tag}"
        when /^oracle/
          image.sub!(/\w+/, 'oraclelinux')
        when /^opensuse/
          image.sub!(/(\w+)/, '\1/leap')
        when /^ubuntu/
          image.sub!(/(\d{2})/, '\1.')
        when /^rocky/
          image.sub!(/(\w+)/, 'rockylinux')
        when /^alma/
          image.sub!(/(\w+)/, 'almalinux')
        end

        image
      end

      def image_commands(ostype)
        case ostype
        when /^(almalinux|centos|rocky|oracle)/
          [
            'cp /bin/true /sbin/agetty',
            el_package_install_command(ostype.delete_prefix(Regexp.last_match(1)).to_i),
          ]
        when /^debian/
          [
            'cp /bin/true /sbin/agetty',
            'rm -f /usr/sbin/policy-rc.d',
            'apt-get update && apt-get install -y cron locales-all net-tools wget gnupg iproute2',
          ]
        when /^opensuse/
          [
            'cp /bin/true /sbin/agetty',
            'zypper install -y cron iproute2 tar wget which',
          ]
        when /^ubuntu/
          [
            'cp /bin/true /sbin/agetty',
            'apt-get install -y net-tools wget locales iproute2 gnupg',
            'locale-gen en_US.UTF-8',
            'echo LANG=en_US.UTF-8 > /etc/default/locale',
          ]
        end
      end

      def el_package_install_command(version)
        if version >= 8
          'dnf install -y cronie crontabs initscripts iproute openssl wget which glibc-langpack-en hostname'
        else
          'yum install -y crontabs initscripts iproute openssl wget which sysvinit-tools tar ss'
        end
      end
    end
  end
end