feat: Add support for AWS EC2 VPC Endpoint Connections (#14044)

bbernays · web-flow · commit 0fa640e1ca45 · 2023-09-26T16:06:10.000Z
#### Summary closes: #14030
diff --git a/plugins/source/aws/docs/tables/README.md b/plugins/source/aws/docs/tables/README.md
@@ -244,6 +244,7 @@
   - [aws_ec2_transit_gateway_peering_attachments](../../../../../website/tables/aws/aws_ec2_transit_gateway_peering_attachments.md)
   - [aws_ec2_transit_gateway_route_tables](../../../../../website/tables/aws/aws_ec2_transit_gateway_route_tables.md)
   - [aws_ec2_transit_gateway_vpc_attachments](../../../../../website/tables/aws/aws_ec2_transit_gateway_vpc_attachments.md)
+- [aws_ec2_vpc_endpoint_connections](../../../../../website/tables/aws/aws_ec2_vpc_endpoint_connections.md)
 - [aws_ec2_vpc_endpoint_service_configurations](../../../../../website/tables/aws/aws_ec2_vpc_endpoint_service_configurations.md)
 - [aws_ec2_vpc_endpoint_services](../../../../../website/tables/aws/aws_ec2_vpc_endpoint_services.md)
   - [aws_ec2_vpc_endpoint_service_permissions](../../../../../website/tables/aws/aws_ec2_vpc_endpoint_service_permissions.md)
diff --git a/plugins/source/aws/resources/plugin/tables.go b/plugins/source/aws/resources/plugin/tables.go
@@ -282,6 +282,7 @@ func getTables() schema.Tables {
 		ec2.SpotInstanceRequests(),
 		ec2.Subnets(),
 		ec2.TransitGateways(),
+		ec2.VpcEndpointConnections(),
 		ec2.VpcEndpoints(),
 		ec2.VpcEndpointServiceConfigurations(),
 		ec2.VpcEndpointServices(),
diff --git a/plugins/source/aws/resources/services/ec2/vpc_endpoint_connections.go b/plugins/source/aws/resources/services/ec2/vpc_endpoint_connections.go
@@ -0,0 +1,51 @@
+package ec2
+
+import (
+	"context"
+
+	sdkTypes "github.com/cloudquery/plugin-sdk/v4/types"
+
+	"github.com/aws/aws-sdk-go-v2/service/ec2"
+	"github.com/aws/aws-sdk-go-v2/service/ec2/types"
+	"github.com/cloudquery/cloudquery/plugins/source/aws/client"
+	"github.com/cloudquery/plugin-sdk/v4/schema"
+	"github.com/cloudquery/plugin-sdk/v4/transformers"
+)
+
+func VpcEndpointConnections() *schema.Table {
+	tableName := "aws_ec2_vpc_endpoint_connections"
+	return &schema.Table{
+		Name:        tableName,
+		Description: `https://docs.aws.amazon.com/AWSEC2/latest/APIReference/API_VpcEndpointConnection.html`,
+		Resolver:    fetchEc2VpcEndpointConnections,
+		Multiplex:   client.ServiceAccountRegionMultiplexer(tableName, "ec2"),
+		Transform:   transformers.TransformWithStruct(&types.VpcEndpointConnection{}, transformers.WithPrimaryKeys("VpcEndpointConnectionId", "VpcEndpointOwner")),
+		Columns: []schema.Column{
+			client.DefaultAccountIDColumn(true),
+			client.DefaultRegionColumn(true),
+			{
+				Name:     "tags",
+				Type:     sdkTypes.ExtensionTypes.JSON,
+				Resolver: client.ResolveTags,
+			},
+		},
+	}
+}
+func fetchEc2VpcEndpointConnections(ctx context.Context, meta schema.ClientMeta, _ *schema.Resource, res chan<- any) error {
+	var config ec2.DescribeVpcEndpointConnectionsInput
+	cl := meta.(*client.Client)
+	svc := cl.Services(client.AWSServiceEc2).Ec2
+	// No paginator available
+	paginator := ec2.NewDescribeVpcEndpointConnectionsPaginator(svc, &config)
+
+	for paginator.HasMorePages() {
+		page, err := paginator.NextPage(ctx, func(options *ec2.Options) {
+			options.Region = cl.Region
+		})
+		if err != nil {
+			return err
+		}
+		res <- page.VpcEndpointConnections
+	}
+	return nil
+}
diff --git a/plugins/source/aws/resources/services/ec2/vpc_endpoint_connections_mock_test.go b/plugins/source/aws/resources/services/ec2/vpc_endpoint_connections_mock_test.go
@@ -0,0 +1,31 @@
+package ec2
+
+import (
+	"testing"
+
+	"github.com/aws/aws-sdk-go-v2/service/ec2"
+	"github.com/aws/aws-sdk-go-v2/service/ec2/types"
+	"github.com/cloudquery/cloudquery/plugins/source/aws/client"
+	"github.com/cloudquery/cloudquery/plugins/source/aws/client/mocks"
+	"github.com/cloudquery/plugin-sdk/v4/faker"
+	"github.com/golang/mock/gomock"
+	"github.com/stretchr/testify/require"
+)
+
+func buildEc2VpcEndpointConnections(t *testing.T, ctrl *gomock.Controller) client.Services {
+	m := mocks.NewMockEc2Client(ctrl)
+	sc := types.VpcEndpointConnection{}
+	require.NoError(t, faker.FakeObject(&sc))
+
+	m.EXPECT().DescribeVpcEndpointConnections(gomock.Any(), gomock.Any(), gomock.Any()).Return(
+		&ec2.DescribeVpcEndpointConnectionsOutput{
+			VpcEndpointConnections: []types.VpcEndpointConnection{sc},
+		}, nil)
+	return client.Services{
+		Ec2: m,
+	}
+}
+
+func TestEc2VpcEndpointConnections(t *testing.T) {
+	client.AwsMockTestHelper(t, VpcEndpointConnections(), buildEc2VpcEndpointConnections, client.TestOptions{})
+}
diff --git a/website/pages/docs/plugins/sources/aws/tables.md b/website/pages/docs/plugins/sources/aws/tables.md
diff --git a/website/tables/aws/aws_ec2_vpc_endpoint_connections.md b/website/tables/aws/aws_ec2_vpc_endpoint_connections.md
@@ -0,0 +1,27 @@
+# Table: aws_ec2_vpc_endpoint_connections
+
+This table shows data for Amazon Elastic Compute Cloud (EC2) VPC Endpoint Connections.
+
+https://docs.aws.amazon.com/AWSEC2/latest/APIReference/API_VpcEndpointConnection.html
+
+The composite primary key for this table is (**account_id**, **region**, **vpc_endpoint_connection_id**, **vpc_endpoint_owner**).
+
+## Columns
+
+| Name          | Type          |
+| ------------- | ------------- |
+|_cq_id|`uuid`|
+|_cq_parent_id|`uuid`|
+|account_id (PK)|`utf8`|
+|region (PK)|`utf8`|
+|tags|`json`|
+|creation_timestamp|`timestamp[us, tz=UTC]`|
+|dns_entries|`json`|
+|gateway_load_balancer_arns|`list<item: utf8, nullable>`|
+|ip_address_type|`utf8`|
+|network_load_balancer_arns|`list<item: utf8, nullable>`|
+|service_id|`utf8`|
+|vpc_endpoint_connection_id (PK)|`utf8`|
+|vpc_endpoint_id|`utf8`|
+|vpc_endpoint_owner (PK)|`utf8`|
+|vpc_endpoint_state|`utf8`|
