cloudquery · kodiakhq · Mar 2, 2023 · Mar 2, 2023 · Mar 2, 2023 · Mar 2, 2023
diff --git a/plugins/source/aws/docs/tables/README.md b/plugins/source/aws/docs/tables/README.md
diff --git a/plugins/source/aws/docs/tables/aws_ec2_launch_template_versions.md b/plugins/source/aws/docs/tables/aws_ec2_launch_template_versions.md
@@ -0,0 +1,29 @@
+# Table: aws_ec2_launch_template_versions
+
+https://docs.aws.amazon.com/AWSEC2/latest/APIReference/API_LaunchTemplateVersion.html
+
+The composite primary key for this table is (**arn**, **version_number**).
+
+## Relations
+
+This table depends on [aws_ec2_launch_templates](aws_ec2_launch_templates.md).
+
+## 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|
+|version_number (PK)|Int|
+|create_time|Timestamp|
+|created_by|String|
+|default_version|Bool|
+|launch_template_data|JSON|
+|launch_template_id|String|
+|launch_template_name|String|
+|version_description|String|
diff --git a/plugins/source/aws/docs/tables/aws_ec2_launch_templates.md b/plugins/source/aws/docs/tables/aws_ec2_launch_templates.md
@@ -0,0 +1,29 @@
+# Table: aws_ec2_launch_templates
+
+https://docs.aws.amazon.com/AWSEC2/latest/APIReference/API_LaunchTemplate.html
+
+The primary key for this table is **arn**.
+
+## Relations
+
+The following tables depend on aws_ec2_launch_templates:
+  - [aws_ec2_launch_template_versions](aws_ec2_launch_template_versions.md)
+
+## 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|
+|tags|JSON|
+|create_time|Timestamp|
+|created_by|String|
+|default_version_number|Int|
+|latest_version_number|Int|
+|launch_template_id|String|
+|launch_template_name|String|
@@ -195,6 +195,7 @@ func tables() []*schema.Table {
 		ec2.InstanceTypes(),
 		ec2.InternetGateways(),
 		ec2.KeyPairs(),
+		ec2.LaunchTemplates(),
 		ec2.ManagedPrefixLists(),
 		ec2.NatGateways(),
 		ec2.NetworkAcls(),

@@ -0,0 +1,38 @@
+package ec2
+
+import (
+	"github.com/aws/aws-sdk-go-v2/service/ec2/types"
+	"github.com/cloudquery/cloudquery/plugins/source/aws/client"
+	"github.com/cloudquery/plugin-sdk/schema"
+	"github.com/cloudquery/plugin-sdk/transformers"
+)
+
+func LaunchTemplateVersions() *schema.Table {
+	return &schema.Table{
+		Name:        "aws_ec2_launch_template_versions",
+		Description: `https://docs.aws.amazon.com/AWSEC2/latest/APIReference/API_LaunchTemplateVersion.html`,
+		Resolver:    fetchEc2LaunchTemplateVersions,
+		Transform:   transformers.TransformWithStruct(&types.LaunchTemplateVersion{}),
+		Multiplex:   client.ServiceAccountRegionMultiplexer("ec2"),
+		Columns: []schema.Column{
+			client.DefaultAccountIDColumn(false),
+			client.DefaultRegionColumn(false),
+			{
+				Name:     "arn",
+				Type:     schema.TypeString,
+				Resolver: schema.ParentColumnResolver("arn"),
+				CreationOptions: schema.ColumnCreationOptions{
+					PrimaryKey: true,
+				},
+			},
+			{
+				Name:     "version_number",
+				Type:     schema.TypeInt,
+				Resolver: schema.PathResolver("VersionNumber"),
+				CreationOptions: schema.ColumnCreationOptions{
+					PrimaryKey: true,
+				},
+			},
+		},
+	}
+}
@@ -0,0 +1,31 @@
+package ec2
+
+import (
+	"context"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"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/schema"
+)
+
+func fetchEc2LaunchTemplateVersions(ctx context.Context, meta schema.ClientMeta, parent *schema.Resource, res chan<- any) error {
+	config := ec2.DescribeLaunchTemplateVersionsInput{
+		LaunchTemplateId: parent.Item.(types.LaunchTemplate).LaunchTemplateId,
+	}
+	c := meta.(*client.Client)
+	svc := c.Services().Ec2
+	for {
+		output, err := svc.DescribeLaunchTemplateVersions(ctx, &config)
+		if err != nil {
+			return err
+		}
+		res <- output.LaunchTemplateVersions
+		if aws.ToString(output.NextToken) == "" {
+			break
+		}
+		config.NextToken = output.NextToken
+	}
+	return nil
+}
@@ -0,0 +1,39 @@
+package ec2
+
+import (
+	"github.com/aws/aws-sdk-go-v2/service/ec2/types"
+	"github.com/cloudquery/cloudquery/plugins/source/aws/client"
+	"github.com/cloudquery/plugin-sdk/schema"
+	"github.com/cloudquery/plugin-sdk/transformers"
+)
+
+func LaunchTemplates() *schema.Table {
+	return &schema.Table{
+		Name:        "aws_ec2_launch_templates",
+		Description: `https://docs.aws.amazon.com/AWSEC2/latest/APIReference/API_LaunchTemplate.html`,
+		Resolver:    fetchEc2LaunchTemplates,
+		Transform:   transformers.TransformWithStruct(&types.LaunchTemplate{}),
+		Multiplex:   client.ServiceAccountRegionMultiplexer("ec2"),
+		Columns: []schema.Column{
+			client.DefaultAccountIDColumn(false),
+			client.DefaultRegionColumn(false),
+			{
+				Name:     "arn",
+				Type:     schema.TypeString,
+				Resolver: resolveEc2LaunchTemplateArn,
+				CreationOptions: schema.ColumnCreationOptions{
+					PrimaryKey: true,
+				},
+			},
+			{
+				Name:     "tags",
+				Type:     schema.TypeJSON,
+				Resolver: client.ResolveTags,
+			},
+		},
+
+		Relations: []*schema.Table{
+			LaunchTemplateVersions(),
+		},
+	}
+}
@@ -0,0 +1,43 @@
+package ec2
+
+import (
+	"context"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/aws/arn"
+	"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/schema"
+)
+
+func fetchEc2LaunchTemplates(ctx context.Context, meta schema.ClientMeta, parent *schema.Resource, res chan<- any) error {
+	var config ec2.DescribeLaunchTemplatesInput
+	c := meta.(*client.Client)
+	svc := c.Services().Ec2
+	for {
+		output, err := svc.DescribeLaunchTemplates(ctx, &config)
+		if err != nil {
+			return err
+		}
+		res <- output.LaunchTemplates
+		if aws.ToString(output.NextToken) == "" {
+			break
+		}
+		config.NextToken = output.NextToken
+	}
+	return nil
+}
+
+func resolveEc2LaunchTemplateArn(_ context.Context, meta schema.ClientMeta, resource *schema.Resource, c schema.Column) error {
+	cl := meta.(*client.Client)
+	item := resource.Item.(types.LaunchTemplate)
+	a := arn.ARN{
+		Partition: cl.Partition,
+		Service:   "ec2",
+		Region:    cl.Region,
+		AccountID: cl.AccountID,
+		Resource:  "launch-template/" + aws.ToString(item.LaunchTemplateId),
+	}
+	return resource.Set(c.Name, a.String())
+}
@@ -0,0 +1,44 @@
+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/faker"
+	"github.com/golang/mock/gomock"
+)
+
+func buildEc2LaunchTemplates(t *testing.T, ctrl *gomock.Controller) client.Services {
+	m := mocks.NewMockEc2Client(ctrl)
+	lt := types.LaunchTemplate{}
+	ltv := types.LaunchTemplateVersion{}
+	err := faker.FakeObject(&lt)
+	if err != nil {
+		t.Fatal(err)
+	}
+	err = faker.FakeObject(&ltv)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	m.EXPECT().DescribeLaunchTemplates(gomock.Any(), gomock.Any(), gomock.Any()).Return(
+		&ec2.DescribeLaunchTemplatesOutput{
+			LaunchTemplates: []types.LaunchTemplate{lt},
+		}, nil)
+
+	m.EXPECT().DescribeLaunchTemplateVersions(gomock.Any(), gomock.Any(), gomock.Any()).Return(
+		&ec2.DescribeLaunchTemplateVersionsOutput{
+			LaunchTemplateVersions: []types.LaunchTemplateVersion{ltv},
+		}, nil)
+
+	return client.Services{
+		Ec2: m,
+	}
+}
+
+func TestEc2LaunchTemplates(t *testing.T) {
+	client.AwsMockTestHelper(t, LaunchTemplates(), buildEc2LaunchTemplates, client.TestOptions{})
+}