require 'itamae-mitsurin'
require 'itamae-mitsurin/mitsurin'

module ItamaeMitsurin
  module Resource
    class AwsRoute53RrsetAlias < Base

        define_attribute :region, type: String, required: true
        define_attribute :action, default: :create
        define_attribute :name, type: String, default_name: true
        define_attribute :type, type: String, required: true
        define_attribute :hosted_zone_id, type: String, required: true
        define_attribute :weight, type: Integer
        define_attribute :failover, type: String
        define_attribute :alias_target, type: [TrueClass, FalseClass], default: false, required: true
        define_attribute :alias_hosted_zone_id, type: String, required: true
        define_attribute :dns_name, type: String, required: true
        define_attribute :evaluate_target_health, type: [TrueClass, FalseClass], default: false
        define_attribute :health_check_id, type: String
        define_attribute :traffic_policy_instance_id, type: String

        def pre_action
          @route53 = ::Aws::Route53::Client.new(region: attributes.region)

          @record = @route53.list_resource_record_sets({
              hosted_zone_id: attributes.hosted_zone_id,
              start_record_name: attributes.name,
              start_record_type: attributes.type,
              max_items: 1,
          })

          if attributes.failover == "PRIMARY"
            set_identifier = "PRIMARY-" + attributes.name.split(".")[0]
          elsif attributes.failover == "SECONDARY"
            set_identifier = "SECONDARY-" + attributes.name.split(".")[0]
          else
            set_identifier = nil
          end

          @rrset_hash = {
            hosted_zone_id: attributes.hosted_zone_id,
            change_batch: {
              comment: nil,
              changes: [
                {
                  action: attributes.action.to_s.upcase,
                  resource_record_set: {
                    name: attributes.name,
                    type: attributes.type,
                    set_identifier: set_identifier,
                    weight: attributes.weight,
                    failover: attributes.failover,
                    alias_target: {
                      hosted_zone_id: attributes.alias_hosted_zone_id,
                      dns_name: attributes.dns_name,
                      evaluate_target_health: attributes.evaluate_target_health,
                    },
                    health_check_id: attributes.health_check_id,
                    traffic_policy_instance_id: attributes.traffic_policy_instance_id,
                    },
                  },
                ],
              },
            }

        end

        def action_create(options)
          unless @record[0][0][0] == attributes.name
            resp = @route53.change_resource_record_sets(@rrset_hash)
            ItamaeMitsurin.logger.debug "#{resp}"
            ItamaeMitsurin.logger.info "created record #{attributes.name}"
            updated!
          end
        end

        def action_upsert(options)
          resp = @route53.change_resource_record_sets(@rrset_hash)
            ItamaeMitsurin.logger.debug "#{resp}"
            ItamaeMitsurin.logger.info "upserted record #{attributes.name}"
          updated!
        end

        def action_delete(options)
          if @record[0][0][0] == attributes.name
            resp = @route53.change_resource_record_sets(@rrset_hash)
            ItamaeMitsurin.logger.debug "#{resp}"
            ItamaeMitsurin.logger.info "deleted record #{attributes.name}"
            updated!
          end
        end

    end
  end
end