feat: AppRunner add support for Connections (#3602)

bbernays · web-flow · commit b6c17a299bc5 · 2022-11-04T12:58:06.000-05:00
#### Summary
diff --git a/plugins/source/aws/codegen/recipes/apprunner.go b/plugins/source/aws/codegen/recipes/apprunner.go
@@ -9,6 +9,22 @@ import (
 func ApprunnerResources() []*Resource {
 	resources := []*Resource{
 		{
+			SubService:  "connections",
+			Struct:      &types.ConnectionSummary{},
+			Description: "https://docs.aws.amazon.com/apprunner/latest/api/API_Connection.html",
+			SkipFields:  []string{"ConnectionArn"},
+			Multiplex:   `client.ServiceAccountRegionMultiplexer("apprunner")`,
+			ExtraColumns: append(
+				defaultRegionalColumns,
+				[]codegen.ColumnDefinition{
+					{
+						Name:     "arn",
+						Type:     schema.TypeString,
+						Resolver: `schema.PathResolver("ConnectionArn")`,
+						Options:  schema.ColumnCreationOptions{PrimaryKey: true},
+					},
+				}...),
+		}, {
 			SubService:          "services",
 			Struct:              &types.Service{},
 			Description:         "https://docs.aws.amazon.com/apprunner/latest/api/API_Service.html",
diff --git a/plugins/source/aws/docs/tables/README.md b/plugins/source/aws/docs/tables/README.md
@@ -34,6 +34,7 @@
 | ↳ [aws_apigatewayv2_domain_name_rest_api_mappings](aws_apigatewayv2_domain_name_rest_api_mappings.md) |
 | [aws_apigatewayv2_vpc_links](aws_apigatewayv2_vpc_links.md) |
 | [aws_applicationautoscaling_policies](aws_applicationautoscaling_policies.md) |
+| [aws_apprunner_connections](aws_apprunner_connections.md) |
 | [aws_apprunner_services](aws_apprunner_services.md) |
 | ↳ [aws_apprunner_operations](aws_apprunner_operations.md) |
 | ↳ [aws_apprunner_custom_domains](aws_apprunner_custom_domains.md) |
diff --git a/plugins/source/aws/docs/tables/aws_apprunner_connections.md b/plugins/source/aws/docs/tables/aws_apprunner_connections.md
@@ -0,0 +1,21 @@
+# Table: aws_apprunner_connections
+
+https://docs.aws.amazon.com/apprunner/latest/api/API_Connection.html
+
+The primary key for this table is **arn**.
+
+
+## Columns
+| Name          | Type          |
+| ------------- | ------------- |
+|_cq_source_name|String|
+|_cq_sync_time|Timestamp|
+|_cq_id|UUID|
+|_cq_parent_id|UUID|
+|account_id|String|
+|region|String|
+|arn (PK)|String|
+|connection_name|String|
+|created_at|Timestamp|
+|provider_type|String|
+|status|String|
diff --git a/plugins/source/aws/resources/plugin/tables.go b/plugins/source/aws/resources/plugin/tables.go
diff --git a/plugins/source/aws/resources/services/apprunner/connections.go b/plugins/source/aws/resources/services/apprunner/connections.go
diff --git a/plugins/source/aws/resources/services/apprunner/connections_fetch.go b/plugins/source/aws/resources/services/apprunner/connections_fetch.go
@@ -0,0 +1,23 @@
+package apprunner
+
+import (
+	"context"
+
+	"github.com/aws/aws-sdk-go-v2/service/apprunner"
+	"github.com/cloudquery/cloudquery/plugins/source/aws/client"
+	"github.com/cloudquery/plugin-sdk/schema"
+)
+
+func fetchApprunnerConnections(ctx context.Context, meta schema.ClientMeta, parent *schema.Resource, res chan<- interface{}) error {
+	var config apprunner.ListConnectionsInput
+	svc := meta.(*client.Client).Services().Apprunner
+	paginator := apprunner.NewListConnectionsPaginator(svc, &config)
+	for paginator.HasMorePages() {
+		output, err := paginator.NextPage(ctx)
+		if err != nil {
+			return err
+		}
+		res <- output.ConnectionSummaryList
+	}
+	return nil
+}
diff --git a/plugins/source/aws/resources/services/apprunner/connections_mock_test.go b/plugins/source/aws/resources/services/apprunner/connections_mock_test.go
@@ -0,0 +1,34 @@
+package apprunner
+
+import (
+	"testing"
+
+	"github.com/aws/aws-sdk-go-v2/service/apprunner"
+	"github.com/aws/aws-sdk-go-v2/service/apprunner/types"
+	"github.com/cloudquery/cloudquery/plugins/source/aws/client"
+	"github.com/cloudquery/cloudquery/plugins/source/aws/client/mocks"
+	"github.com/cloudquery/plugin-sdk/faker"
+	"github.com/golang/mock/gomock"
+)
+
+func buildConnections(t *testing.T, ctrl *gomock.Controller) client.Services {
+	m := mocks.NewMockApprunnerClient(ctrl)
+	s := types.ConnectionSummary{}
+	err := faker.FakeObject(&s)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	m.EXPECT().ListConnections(gomock.Any(), gomock.Any(), gomock.Any()).Return(
+		&apprunner.ListConnectionsOutput{
+			ConnectionSummaryList: []types.ConnectionSummary{s},
+		}, nil)
+
+	return client.Services{
+		Apprunner: m,
+	}
+}
+
+func TestConnections(t *testing.T) {
+	client.AwsMockTestHelper(t, Connections(), buildConnections, client.TestOptions{})
+}
