71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
package middlewares
|
|
|
|
import (
|
|
"github.com/snyk/driftctl/pkg/resource"
|
|
"github.com/snyk/driftctl/pkg/resource/aws"
|
|
)
|
|
|
|
/**
|
|
Fetching eip association from remote return every association but some of them are embedded in eip.
|
|
This middleware will check for every eip_association that here is no corresponding association_id inside eip.
|
|
*/
|
|
|
|
type EipAssociationExpander struct {
|
|
resourceFactory resource.ResourceFactory
|
|
}
|
|
|
|
func NewEipAssociationExpander(resourceFactory resource.ResourceFactory) EipAssociationExpander {
|
|
return EipAssociationExpander{resourceFactory}
|
|
}
|
|
|
|
func (m EipAssociationExpander) Execute(_, resourcesFromState *[]*resource.Resource) error {
|
|
var newResources []*resource.Resource
|
|
for _, res := range *resourcesFromState {
|
|
newResources = append(newResources, res)
|
|
|
|
if res.ResourceType() != aws.AwsEipResourceType {
|
|
continue
|
|
}
|
|
if m.haveMatchingEipAssociation(res, resourcesFromState) {
|
|
continue
|
|
}
|
|
// This EIP have no association, check if we need to create one
|
|
assocID := res.Attributes().GetString("association_id")
|
|
if assocID == nil || *assocID == "" {
|
|
continue
|
|
}
|
|
|
|
attributes := *res.Attributes()
|
|
newAssoc := m.resourceFactory.CreateAbstractResource(
|
|
aws.AwsEipAssociationResourceType,
|
|
*assocID,
|
|
map[string]interface{}{
|
|
"allocation_id": res.ResourceId(),
|
|
"id": *assocID,
|
|
"instance_id": attributes["instance"],
|
|
"network_interface_id": attributes["network_interface"],
|
|
"private_ip_address": attributes["private_ip"],
|
|
"public_ip": attributes["public_ip"],
|
|
},
|
|
)
|
|
|
|
newResources = append(newResources, newAssoc)
|
|
}
|
|
*resourcesFromState = newResources
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m EipAssociationExpander) haveMatchingEipAssociation(cur *resource.Resource, stateRes *[]*resource.Resource) bool {
|
|
for _, res := range *stateRes {
|
|
if res.ResourceType() != aws.AwsEipAssociationResourceType {
|
|
continue
|
|
}
|
|
assocId := cur.Attributes().GetString("association_id")
|
|
if assocId != nil && res.ResourceId() == *assocId {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|