refactor aws adjustEndpointAndAaaaIfNeeded

Signed-off-by: u-kai <76635578+u-kai@users.noreply.github.com>

Author: u-kai

provider/aws/aws.go

@@ -835,8 +835,83 @@ func (p *AWSProvider) AdjustEndpoints(endpoints []*endpoint.Endpoint) ([]*endpoi
 }
 
 func (p *AWSProvider) adjustEndpointAndNewAaaaIfNeeded(ep *endpoint.Endpoint) *endpoint.Endpoint {
-	// TODO: implement new logic here
-	return p.oldAdjustEndpointAndNewAaaaIfNeeded(ep)
+	setAliasConf := func(ep *endpoint.Endpoint) {
+		if ep.RecordTTL.IsConfigured() {
+			log.Debugf("Modifying endpoint: %v, setting ttl=%v", ep, defaultTTL)
+			ep.RecordTTL = defaultTTL
+		}
+		if prop, ok := ep.GetProviderSpecificProperty(providerSpecificEvaluateTargetHealth); ok {
+			if prop != "true" && prop != "false" {
+				ep.SetProviderSpecificProperty(providerSpecificEvaluateTargetHealth, "false")
+			}
+		} else {
+			ep.SetProviderSpecificProperty(providerSpecificEvaluateTargetHealth, strconv.FormatBool(p.evaluateTargetHealth))
+		}
+	}
+	cnameAliasCase := func(ep *endpoint.Endpoint) *endpoint.Endpoint {
+		setAliasConf(ep)
+		result := &endpoint.Endpoint{
+			DNSName:          ep.DNSName,
+			Targets:          ep.Targets,
+			RecordType:       endpoint.RecordTypeAAAA,
+			RecordTTL:        ep.RecordTTL,
+			Labels:           ep.Labels,
+			ProviderSpecific: ep.ProviderSpecific,
+			SetIdentifier:    ep.SetIdentifier,
+		}
+		ep.RecordType = endpoint.RecordTypeA
+		return result
+	}
+
+	var additionalAAAA *endpoint.Endpoint
+	switch ep.RecordType {
+	case endpoint.RecordTypeA, endpoint.RecordTypeAAAA:
+		aliasString, ok := ep.GetProviderSpecificProperty(providerSpecificAlias)
+		switch aliasString {
+		case "true":
+			setAliasConf(ep)
+		case "":
+			if ok {
+				ep.DeleteProviderSpecificProperty(providerSpecificAlias)
+			}
+			ep.DeleteProviderSpecificProperty(providerSpecificEvaluateTargetHealth)
+		default:
+			ep.DeleteProviderSpecificProperty(providerSpecificAlias)
+			ep.DeleteProviderSpecificProperty(providerSpecificEvaluateTargetHealth)
+		}
+	case endpoint.RecordTypeCNAME:
+		aliasString, _ := ep.GetProviderSpecificProperty(providerSpecificAlias)
+		switch aliasString {
+		case "true":
+			additionalAAAA = cnameAliasCase(ep)
+		case "":
+			alias := useAlias(ep, p.preferCNAME)
+			log.Debugf("Modifying endpoint: %v, setting %s=%v", ep, providerSpecificAlias, alias)
+			ep.SetProviderSpecificProperty(providerSpecificAlias, strconv.FormatBool(alias))
+			if alias {
+				additionalAAAA = cnameAliasCase(ep)
+			} else {
+				ep.DeleteProviderSpecificProperty(providerSpecificEvaluateTargetHealth)
+			}
+		default:
+			ep.SetProviderSpecificProperty(providerSpecificAlias, "false")
+			ep.DeleteProviderSpecificProperty(providerSpecificEvaluateTargetHealth)
+		}
+	default:
+		// TODO fix
+		aliasString, _ := ep.GetProviderSpecificProperty(providerSpecificAlias)
+		switch aliasString {
+		case "true":
+			ep.DeleteProviderSpecificProperty(providerSpecificAlias)
+			setAliasConf(ep)
+		default:
+			ep.DeleteProviderSpecificProperty(providerSpecificAlias)
+			ep.DeleteProviderSpecificProperty(providerSpecificEvaluateTargetHealth)
+		}
+	}
+
+	adjustGeoProximityLocationEndpoint(ep)
+	return additionalAAAA
 }
 
 func (p *AWSProvider) oldAdjustEndpointAndNewAaaaIfNeeded(ep *endpoint.Endpoint) *endpoint.Endpoint {