package middlewares

import (
	"github.com/cloudskiff/driftctl/pkg/resource"
	"github.com/cloudskiff/driftctl/pkg/resource/aws"
	"github.com/sirupsen/logrus"
)

type AwsInstanceEIP struct{}

func (a AwsInstanceEIP) Execute(remoteResources, resourcesFromState *[]resource.Resource) error {
	for _, remoteResource := range *remoteResources {
		// Ignore all resources other than aws_instance
		if remoteResource.TerraformType() != aws.AwsInstanceResourceType {
			continue
		}

		instance, _ := remoteResource.(*resource.AbstractResource)

		if a.hasEIP(instance, resourcesFromState) {
			logrus.WithFields(logrus.Fields{
				"instance": instance.TerraformId(),
			}).Debug("Ignore instance public ip and dns as it has an eip attached")
			a.ignorePublicIpAndDns(instance, remoteResources, resourcesFromState)
		}
	}

	return nil
}

func (a AwsInstanceEIP) hasEIP(instance *resource.AbstractResource, resources *[]resource.Resource) bool {
	for _, res := range *resources {
		if res.TerraformType() == aws.AwsEipResourceType {
			eip, _ := res.(*resource.AbstractResource)
			if (*eip.Attrs)["instance"] == instance.TerraformId() {
				return true
			}
		}
		if res.TerraformType() == aws.AwsEipAssociationResourceType {
			eip, _ := res.(*resource.AbstractResource)
			if (*eip.Attrs)["instance_id"] == instance.TerraformId() {
				return true
			}
		}
	}

	return false
}

func (a AwsInstanceEIP) ignorePublicIpAndDns(instance *resource.AbstractResource, resourcesSet ...*[]resource.Resource) {
	for _, resources := range resourcesSet {
		for _, res := range *resources {
			if res.TerraformType() == instance.TerraformType() &&
				res.TerraformId() == instance.TerraformId() {
				instance, _ := res.(*resource.AbstractResource)
				instance.Attrs.SafeDelete([]string{"public_dns"})
				instance.Attrs.SafeDelete([]string{"public_ip"})
			}
		}
	}
}