57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
package middlewares
|
|
|
|
import (
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/snyk/driftctl/enumeration/resource"
|
|
"github.com/snyk/driftctl/enumeration/resource/aws"
|
|
)
|
|
|
|
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.ResourceType() != aws.AwsInstanceResourceType {
|
|
continue
|
|
}
|
|
|
|
if a.hasEIP(remoteResource, resourcesFromState) {
|
|
logrus.WithFields(logrus.Fields{
|
|
"instance": remoteResource.ResourceId(),
|
|
}).Debug("Ignore instance public ip and dns as it has an eip attached")
|
|
a.ignorePublicIpAndDns(remoteResource, remoteResources, resourcesFromState)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a AwsInstanceEIP) hasEIP(instance *resource.Resource, resources *[]*resource.Resource) bool {
|
|
for _, res := range *resources {
|
|
if res.ResourceType() == aws.AwsEipResourceType {
|
|
if (*res.Attrs)["instance"] == instance.ResourceId() {
|
|
return true
|
|
}
|
|
}
|
|
if res.ResourceType() == aws.AwsEipAssociationResourceType {
|
|
if (*res.Attrs)["instance_id"] == instance.ResourceId() {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (a AwsInstanceEIP) ignorePublicIpAndDns(instance *resource.Resource, resourcesSet ...*[]*resource.Resource) {
|
|
for _, resources := range resourcesSet {
|
|
for _, res := range *resources {
|
|
if res.ResourceType() == instance.ResourceType() &&
|
|
res.ResourceId() == instance.ResourceId() {
|
|
res.Attrs.SafeDelete([]string{"public_dns"})
|
|
res.Attrs.SafeDelete([]string{"public_ip"})
|
|
}
|
|
}
|
|
}
|
|
}
|