lambda.go

package recipes

import (
	"reflect"
	"strings"

	"github.com/aws/aws-sdk-go-v2/service/lambda"
	"github.com/aws/aws-sdk-go-v2/service/lambda/types"
	"github.com/cloudquery/plugin-sdk/codegen"
	"github.com/cloudquery/plugin-sdk/schema"

	"github.com/OpsHelmInc/cloudquery/resources/services/lambda/models"
)

func LambdaResources() []*Resource {
	resources := []*Resource{
		{
			SubService:          "functions",
			Struct:              &lambda.GetFunctionOutput{},
			SkipFields:          []string{},
			PreResourceResolver: "getFunction",
			ExtraColumns: append(
				defaultRegionalColumns,
				[]codegen.ColumnDefinition{
					{
						Name:     "arn",
						Type:     schema.TypeString,
						Resolver: `schema.PathResolver("Configuration.FunctionArn")`,
					},
					{
						Name: "policy_revision_id",
						Type: schema.TypeString,
					},
					{
						Name: "policy_document",
						Type: schema.TypeJSON,
					},
					{
						Name: "code_signing_config",
						Type: schema.TypeJSON,
					},
					{
						Name:     "code_repository_type",
						Type:     schema.TypeString,
						Resolver: `schema.PathResolver("Code.RepositoryType")`,
					},
					{
						Name:     ohResourceTypeColumn,
						Type:     schema.TypeString,
						Resolver: `client.StaticValueResolver("AWS::Lambda::Function")`,
					},
				}...),
			PostResourceResolver: `resolvePolicyCodeSigningConfig`,
			Relations: []string{
				"FunctionEventInvokeConfigs()",
				"FunctionAliases()",
				"FunctionVersions()",
				"FunctionConcurrencyConfigs()",
				"FunctionEventSourceMappings()",
			},
		},
		{
			SubService:  "function_event_invoke_configs",
			Struct:      &types.FunctionEventInvokeConfig{},
			Description: "https://docs.aws.amazon.com/lambda/latest/dg/API_FunctionEventInvokeConfig.html",
			SkipFields:  []string{"FunctionArn"},
			ExtraColumns: append(
				defaultRegionalColumns,
				[]codegen.ColumnDefinition{
					{
						Name:     "function_arn",
						Type:     schema.TypeString,
						Resolver: `schema.ParentColumnResolver("arn")`,
					},
				}...),
		},
		{
			SubService:          "function_aliases",
			Struct:              &models.AliasWrapper{},
			SkipFields:          []string{"AliasArn"},
			PreResourceResolver: "getFunctionAliasURLConfig",
			ExtraColumns: append(
				defaultRegionalColumns,
				[]codegen.ColumnDefinition{
					{
						Name:     "function_arn",
						Type:     schema.TypeString,
						Resolver: `schema.ParentColumnResolver("arn")`,
					},
					{
						Name:     "arn",
						Type:     schema.TypeString,
						Resolver: `schema.PathResolver("AliasArn")`,
						Options:  schema.ColumnCreationOptions{PrimaryKey: true},
					},
				}...),
		},
		{
			SubService:  "function_versions",
			Struct:      &types.FunctionConfiguration{},
			Description: "https://docs.aws.amazon.com/lambda/latest/dg/API_FunctionConfiguration.html",
			SkipFields:  []string{"FunctionArn"},
			ExtraColumns: append(
				defaultRegionalColumns,
				[]codegen.ColumnDefinition{
					{
						Name:     "function_arn",
						Type:     schema.TypeString,
						Resolver: `schema.ParentColumnResolver("arn")`,
					},
				}...),
		},
		{
			SubService:  "function_concurrency_configs",
			Struct:      &types.ProvisionedConcurrencyConfigListItem{},
			Description: "https://docs.aws.amazon.com/lambda/latest/dg/API_ProvisionedConcurrencyConfigListItem.html",
			SkipFields:  []string{"FunctionArn"},
			ExtraColumns: append(
				defaultRegionalColumns,
				[]codegen.ColumnDefinition{
					{
						Name:     "function_arn",
						Type:     schema.TypeString,
						Resolver: `schema.ParentColumnResolver("arn")`,
					},
				}...),
		},
		{
			SubService:  "function_event_source_mappings",
			Struct:      &types.EventSourceMappingConfiguration{},
			Description: "https://docs.aws.amazon.com/lambda/latest/dg/API_EventSourceMappingConfiguration.html",
			SkipFields:  []string{"FunctionArn"},
			ExtraColumns: append(
				defaultRegionalColumns,
				[]codegen.ColumnDefinition{
					{
						Name:     "function_arn",
						Type:     schema.TypeString,
						Resolver: `schema.ParentColumnResolver("arn")`,
					},
				}...),
		},
		{
			SubService:  "layers",
			Struct:      &types.LayersListItem{},
			Description: "https://docs.aws.amazon.com/lambda/latest/dg/API_LayersListItem.html",
			SkipFields:  []string{},
			ExtraColumns: append(
				defaultRegionalColumns,
				[]codegen.ColumnDefinition{
					{
						Name:     "arn",
						Type:     schema.TypeString,
						Resolver: `schema.PathResolver("LayerArn")`,
						Options:  schema.ColumnCreationOptions{PrimaryKey: true},
					},
					{
						Name:     ohResourceTypeColumn,
						Type:     schema.TypeString,
						Resolver: `client.StaticValueResolver("AWS::Lambda::LayerVersion")`,
					},
				}...),
			Relations: []string{
				"LayerVersions()",
			},
		},
		{
			SubService:  "layer_versions",
			Struct:      &types.LayerVersionsListItem{},
			Description: "https://docs.aws.amazon.com/lambda/latest/dg/API_LayerVersionsListItem.html",
			SkipFields:  []string{"LayerVersionArn"},
			ExtraColumns: append(
				defaultRegionalColumns,
				[]codegen.ColumnDefinition{
					{
						Name:     "arn",
						Type:     schema.TypeString,
						Resolver: `schema.PathResolver("LayerVersionArn")`,
					},
					{
						Name:     "layer_arn",
						Type:     schema.TypeString,
						Resolver: `schema.ParentColumnResolver("arn")`,
					},
				}...),
			Relations: []string{
				"LayerVersionPolicies()",
			},
		},
		{
			SubService: "layer_version_policies",
			Struct:     &lambda.GetLayerVersionPolicyOutput{},
			SkipFields: []string{},
			ExtraColumns: append(
				defaultRegionalColumns,
				[]codegen.ColumnDefinition{
					{
						Name:     "layer_version_arn",
						Type:     schema.TypeString,
						Resolver: `schema.ParentColumnResolver("arn")`,
					},
					{
						Name:     "layer_version",
						Type:     schema.TypeInt,
						Resolver: `schema.ParentColumnResolver("version")`,
					},
				}...),
		},
		{
			SubService: "runtimes",
			Struct:     &models.RuntimeWrapper{},
			SkipFields: []string{"Name"},
			ExtraColumns: append(
				defaultRegionalColumns,
				[]codegen.ColumnDefinition{
					{
						Name:    "name",
						Type:    schema.TypeString,
						Options: schema.ColumnCreationOptions{PrimaryKey: true},
					},
				}...),
		},
	}

	// set default values
	for _, r := range resources {
		r.Service = "lambda"
		r.Multiplex = `client.ServiceAccountRegionMultiplexer("lambda")`
		structName := reflect.ValueOf(r.Struct).Elem().Type().Name()
		if strings.Contains(structName, "Wrapper") {
			r.UnwrapEmbeddedStructs = true
		}
	}
	return resources
}