feat: Included layers for aws_lambda_layers

Author: Marco Franceschi

src/services/lambda/data.ts

@@ -13,7 +13,8 @@ import Lambda, {
   ListEventSourceMappingsResponse,
   EventSourceMappingConfiguration,
   ListFunctionEventInvokeConfigsResponse,
-  FunctionEventInvokeConfig
+  FunctionEventInvokeConfig,
+  Layer
 } from 'aws-sdk/clients/lambda'
 import { AWSError } from 'aws-sdk/lib/error'
 import { Config } from 'aws-sdk/lib/config'
@@ -41,6 +42,7 @@ export interface RawAwsLambdaFunction extends FunctionConfiguration {
   }
   EventSourceMappings?: EventSourceMappingConfiguration[]
   EventInvokeConfigs?: FunctionEventInvokeConfig[]
+  Layers?: Layer[]
 }
 
 const listFunctionsForRegion = async ({
@@ -251,8 +253,9 @@ export default async ({
     logger.debug(lt.fetchedLambdas(lambdaData.length))
 
     // get all tags and policy for each Lambda
-    lambdaData.map(({ FunctionArn: arn, region, FunctionName: name }, idx) => {
+    lambdaData.map(({ FunctionArn: arn, region, FunctionName: name, Layers }, idx) => {
       const lambda = new Lambda({ ...config, region, endpoint })
+
       const additionalMetadataPromise = new Promise<void>(async resolveData => {
         const envTags: TagMap = await getResourceTags(lambda, arn)
         lambdaData[idx].Tags = envTags
@@ -262,6 +265,7 @@ export default async ({
         lambdaData[idx].EventSourceMappings = eventSourceMappings
         const eventInvokeConfigs = await getEventInvokeConfigs(lambda, name)
         lambdaData[idx].EventInvokeConfigs = eventInvokeConfigs
+        lambdaData[idx].Layers = Layers
         resolveData()
       })
       tagsPromises.push(additionalMetadataPromise)

src/services/lambda/format.ts

@@ -1,10 +1,10 @@
 import { generateUniqueId } from '@cloudgraph/sdk'
 import isEmpty from 'lodash/isEmpty'
 import t from '../../properties/translations'
-import { AwsLambda, AwsLambdaEventInvokeConfig, AwsLambdaEventSourceMappings } from '../../types/generated'
+import { AwsLambda, AwsLambdaEventInvokeConfig, AwsLambdaEventSourceMappings, AwsLambdaLayerVersion } from '../../types/generated'
 import { formatTagsFromMap, formatIamJsonPolicy } from '../../utils/format'
 import { RawAwsLambdaFunction } from './data'
-import { EventSourceMappingConfiguration, FunctionEventInvokeConfig } from 'aws-sdk/clients/lambda'
+import { EventSourceMappingConfiguration, FunctionEventInvokeConfig, Layer } from 'aws-sdk/clients/lambda'
 
 /**
  * Lambda
@@ -36,7 +36,8 @@ export default ({
     VpcConfig: vpcConfig,
     PolicyData: { Policy: policy = '', RevisionId: policyRevisionId = '' },
     EventSourceMappings: eventSourceMappings = [],
-    EventInvokeConfigs: eventInvokeConfigs = []
+    EventInvokeConfigs: eventInvokeConfigs = [],
+    Layers: layers = []
   } = rawData
   const environmentVariables = []
   const secretNames = [t.pass, t.secret, t.private, t.cert]
@@ -154,6 +155,29 @@ export default ({
     )
   }
 
+  const formatLayers = (
+    layers?: Layer[]
+  ): AwsLambdaLayerVersion[] => {
+    return (
+      layers?.map(l => {
+        const arnParts = l.Arn?.split(':')
+        // get layer name from arn:aws:lambda:_REGION_:_ACCOUNT_ID_:layer:_LAYER_NAME_:_LAYER_VERSION_
+        const layerName = arnParts[arnParts.length - 2]
+        return ({
+          id: generateUniqueId({
+            arn,
+            ...l,
+          }),
+          arn: l.Arn,
+          name: layerName,
+          codeSize: l.CodeSize,
+          signingProfileVersionArn: l.SigningProfileVersionArn,
+          signingJobArn: l.SigningJobArn,
+        })
+      }) || []
+    )
+  }
+
   const functionName = arn.split(':').pop()
   const functionPolicy = formatIamJsonPolicy(policy)
   const policyStatementIds = functionPolicy?.statement?.map(s => s.sid) ?? []
@@ -183,6 +207,7 @@ export default ({
     policyStatementIds,
     tags: formatTagsFromMap(Tags),
     eventSourceMappings: formatEventSourceMappings(eventSourceMappings),
-    eventInvokeConfigs: formatEventInvokeConfigs(eventInvokeConfigs)
+    eventInvokeConfigs: formatEventInvokeConfigs(eventInvokeConfigs),
+    layers: formatLayers(layers)
   }
 }

src/services/lambda/schema.graphql

@@ -75,6 +75,19 @@ type awsLambdaEventSourceMappings
   selfManagedKafkaEventSourceConfig: awsLambdaEventSourceConfig
 }
 
+type awsLambdaLayerVersion
+  @generate(
+    query: { get: false, query: true, aggregate: false }
+    mutation: { add: false, delete: false }
+    subscription: false
+  ) {
+  id: String! @id
+  arn: String @search(by: [hash, regexp])
+  name: String @search(by: [hash, regexp])
+  codeSize: Float
+  signingProfileVersionArn: String @search(by: [hash, regexp])
+  signingJobArn: String @search(by: [hash, regexp])
+}
 
 type awsLambda implements awsBaseService @key(fields: "arn") {
   description: String @search(by: [hash, regexp, fulltext])
@@ -97,6 +110,7 @@ type awsLambda implements awsBaseService @key(fields: "arn") {
   policyStatementIds: [String] @search(by: [hash, regexp])
   eventSourceMappings: [awsLambdaEventSourceMappings]
   eventInvokeConfigs: [awsLambdaEventInvokeConfig]
+  layers: [awsLambdaLayerVersion]
   tags: [awsRawTag]
   kms: [awsKms] @hasInverse(field: lambda)
   securityGroups: [awsSecurityGroup] @hasInverse(field: lambda)

src/types/generated.ts

@@ -3418,6 +3418,7 @@ export type AwsLambda = AwsBaseService & {
   kms?: Maybe<Array<Maybe<AwsKms>>>;
   kmsKeyArn?: Maybe<Scalars['String']>;
   lastModified?: Maybe<Scalars['String']>;
+  layers?: Maybe<Array<Maybe<AwsLambdaLayerVersion>>>;
   memorySize?: Maybe<Scalars['Int']>;
   name?: Maybe<Scalars['String']>;
   policy?: Maybe<AwsIamJsonPolicy>;
@@ -3491,6 +3492,15 @@ export type AwsLambdaEventSourceMappings = {
   uuid?: Maybe<Scalars['String']>;
 };
 
+export type AwsLambdaLayerVersion = {
+  arn?: Maybe<Scalars['String']>;
+  codeSize?: Maybe<Scalars['Float']>;
+  id: Scalars['String'];
+  name?: Maybe<Scalars['String']>;
+  signingJobArn?: Maybe<Scalars['String']>;
+  signingProfileVersionArn?: Maybe<Scalars['String']>;
+};
+
 export type AwsLambdaSourceAccessConfiguration = {
   id: Scalars['String'];
   type?: Maybe<Scalars['String']>;