# frozen_string_literal: true

require 'rake_factory'
require 'ruby_terraform'
require 'ostruct'
require 'colored2'

module RakeTerraform
  module Tasks
    class Provision < RakeFactory::Task
      default_name :provision
      default_prerequisites(RakeFactory::DynamicValue.new do |t|
        [t.ensure_task_name]
      end)
      default_description(RakeFactory::DynamicValue.new do |t|
        "Provision #{t.configuration_name} using terraform"
      end)

      parameter :configuration_name, required: true
      parameter :source_directory, required: true
      parameter :work_directory, required: true

      parameter :environment, default: {}

      parameter :backend_config

      parameter :vars, default: {}
      parameter :var_file
      parameter :state_file

      parameter :debug, default: false
      parameter :input, default: false
      parameter :no_color, default: false
      parameter :no_backup, default: false

      parameter :backup_file

      parameter :ensure_task_name, default: :'terraform:ensure'

      action do |task|
        Colored2.disable! if task.no_color

        module_directory =
          File.join(FileUtils.pwd, task.source_directory)
        configuration_directory =
          File.join(task.work_directory, task.source_directory)

        Kernel.puts("Provisioning #{task.configuration_name}".cyan)

        prepare_configuration_directory(configuration_directory)
        init_configuration(configuration_directory, module_directory, task)
        apply_configuration(configuration_directory, task)
      end

      def prepare_configuration_directory(configuration_directory)
        FileUtils.rm_rf(configuration_directory)
        FileUtils.mkdir_p(configuration_directory)
      end

      def init_configuration(configuration_directory, module_directory, task)
        RubyTerraform.init(
          {
            chdir: configuration_directory,
            from_module: module_directory,
            backend_config: task.backend_config,
            no_color: task.no_color,
            input: task.input
          },
          { environment: task.environment }
        )
      end

      # rubocop:disable Metrics/MethodLength
      def apply_configuration(configuration_directory, task)
        RubyTerraform.apply(
          {
            chdir: configuration_directory,
            auto_approve: true,
            input: task.input,
            no_color: task.no_color,
            no_backup: task.no_backup,
            backup: task.backup_file,
            state: task.state_file,
            vars: task.vars,
            var_file: task.var_file
          },
          { environment: task.environment }
        )
      end
      # rubocop:enable Metrics/MethodLength
    end
  end
end