feat: Add TypeScriptCodeCollection for building multiple Lambda functions

examples/typescript/multi-lambda/app.ts

@@ -0,0 +1,41 @@
+import * as cdk from 'aws-cdk-lib';
+import * as lambda from 'aws-cdk-lib/aws-lambda';
+import { TypeScriptCodeCollection } from '@mrgrain/cdk-esbuild';
+
+export class MultiLambdaStack extends cdk.Stack {
+  constructor(scope: cdk.App, id: string, props?: cdk.StackProps) {
+    super(scope, id, props);
+
+    const codeCollection = new TypeScriptCodeCollection(this, 'MultiLambda', {
+      entryPoints: {
+        'api': './src/api.ts',
+        'auth': './src/auth.ts',
+        'notifications': './src/notifications.ts',
+      },
+      buildOptions: {
+        minify: true,
+      },
+    });
+
+    new lambda.Function(this, 'ApiFunction', {
+      runtime: lambda.Runtime.NODEJS_18_X,
+      handler: 'api.handler',
+      code: codeCollection.getCode('api'),
+    });
+
+    new lambda.Function(this, 'AuthFunction', {
+      runtime: lambda.Runtime.NODEJS_18_X,
+      handler: 'auth.handler',
+      code: codeCollection.getCode('auth'),
+    });
+
+    new lambda.Function(this, 'NotificationsFunction', {
+      runtime: lambda.Runtime.NODEJS_18_X,
+      handler: 'notifications.handler',
+      code: codeCollection.getCode('notifications'),
+    });
+  }
+}
+
+const app = new cdk.App();
+new MultiLambdaStack(app, 'MultiLambdaStack');

examples/typescript/multi-lambda/package.json

@@ -0,0 +1,18 @@
+{
+  "name": "multi-lambda-example",
+  "version": "1.0.0",
+  "description": "Example of building multiple Lambda functions with cdk-esbuild",
+  "main": "app.ts",
+  "scripts": {
+    "deploy": "cdk deploy",
+    "synth": "cdk synth"
+  },
+  "dependencies": {
+    "aws-cdk-lib": "^2.0.0",
+    "constructs": "^10.0.0",
+    "@mrgrain/cdk-esbuild": "^5.0.0"
+  },
+  "devDependencies": {
+    "typescript": "^5.0.0"
+  }
+}

examples/typescript/multi-lambda/src/api.ts

@@ -0,0 +1,6 @@
+export async function handler(event: any) {
+  return {
+    statusCode: 200,
+    body: JSON.stringify({ message: 'API Handler' }),
+  };
+}

examples/typescript/multi-lambda/src/auth.ts

@@ -0,0 +1,6 @@
+export async function handler(event: any) {
+  return {
+    statusCode: 200,
+    body: JSON.stringify({ message: 'Auth Handler' }),
+  };
+}

examples/typescript/multi-lambda/src/notifications.ts

@@ -0,0 +1,6 @@
+export async function handler(event: any) {
+  return {
+    statusCode: 200,
+    body: JSON.stringify({ message: 'Notifications Handler' }),
+  };
+}

src/index.ts

@@ -49,3 +49,8 @@ export {
   TypeScriptSource,
   TypeScriptSourceProps,
 } from './source';
+
+export {
+  TypeScriptCodeCollection,
+  TypeScriptCodeCollectionProps,
+} from './typescript-code-collection';

src/typescript-code-collection.ts

@@ -0,0 +1,95 @@
+import { Construct } from 'constructs';
+import { BundlerProps } from './bundler';
+import { TypeScriptCode, TypeScriptCodeProps } from './code';
+
+/**
+ * Properties for TypeScriptCodeCollection
+ *
+ * @stability stable
+ */
+export interface TypeScriptCodeCollectionProps extends BundlerProps {
+  /**
+   * Entry points to bundle as a collection.
+   *
+   * Key: logical function name (used to retrieve the code later).
+   * Value: path to the entry point file.
+   *
+   * @stability stable
+   */
+  readonly entryPoints: Record<string, string>;
+
+  /**
+   * A hash of this asset, which is available at construction time.
+   *
+   * As this is a plain string, it can be used in construct IDs in order to enforce creation of a new resource when the content hash has changed.
+   *
+   * Defaults to a hash of all files in the resulting bundle.
+   *
+   * @stability stable
+   */
+  readonly assetHash?: string;
+}
+
+/**
+ * Manages multiple Lambda function entry points that share the same build configuration.
+ *
+ * Creates a {@link TypeScriptCode} asset for each entry point, allowing related
+ * functions to be organized with common build options. This is primarily a
+ * convenience construct for managing multiple Lambda functions that share
+ * the same esbuild configuration.
+ *
+ * @stability stable
+ */
+export class TypeScriptCodeCollection extends Construct {
+  private readonly codes: { [name: string]: TypeScriptCode } = {};
+
+  constructor(scope: Construct, id: string, props: TypeScriptCodeCollectionProps) {
+    super(scope, id);
+
+    const { entryPoints, assetHash, ...bundlerProps } = props;
+
+    // Create individual TypeScriptCode instances for each entry point
+    Object.entries(entryPoints).forEach(([name, entryPoint]) => {
+      const codeProps: TypeScriptCodeProps = {
+        ...bundlerProps,
+        assetHash,
+      };
+
+      this.codes[name] = new TypeScriptCode(entryPoint, codeProps);
+    });
+  }
+
+  /**
+   * Get the bundled TypeScript code for a specific function.
+   *
+   * @param functionName The logical function name (key from entryPoints)
+   * @returns TypeScriptCode instance that can be used with Lambda.Function
+   *
+   * @stability stable
+   */
+  public getCode(functionName: string): TypeScriptCode {
+    const code = this.codes[functionName];
+    if (!code) {
+      throw new Error(`No entry point found for function: ${functionName}`);
+    }
+    return code;
+  }
+
+  /**
+   * The names of all functions in this collection.
+   *
+   * @stability stable
+   */
+  public get functionNames(): string[] {
+    return Object.keys(this.codes);
+  }
+
+  /**
+   * All bundled code instances.
+   *
+   * @stability stable
+   */
+  public get allCodes(): { [name: string]: TypeScriptCode } {
+    return { ...this.codes };
+  }
+}

test/integ/integ-multi-lambda.py

@@ -0,0 +1,74 @@
+import aws_cdk as cdk
+import aws_cdk.aws_lambda as lambda_
+import aws_cdk.integ_tests_alpha as integ
+from constructs import Construct
+from mrgrain.cdk_esbuild import (
+    BuildOptions,
+    TypeScriptCodeCollection,
+    EsbuildProvider,
+    EsbuildSource,
+)
+
+
+class MultiLambdaStack(cdk.Stack):
+    def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
+        super().__init__(scope, construct_id, **kwargs)
+
+        # Set a new default
+        EsbuildProvider.override_default_provider(
+            EsbuildProvider(esbuild_module_path=EsbuildSource.install())
+        )
+
+        # Use TypeScriptCodeCollection to build multiple functions
+        code_collection = TypeScriptCodeCollection(
+            self,
+            "MultiLambda",
+            entry_points={
+                "ts_handler": "test/fixtures/handlers/ts-handler.ts",
+                "js_handler": "test/fixtures/handlers/js-handler.js",
+            },
+        )
+
+        lambda_.Function(
+            self,
+            "TsHandlerFunction",
+            runtime=lambda_.Runtime.NODEJS_18_X,
+            handler="index.handler",
+            code=code_collection.get_code("ts_handler"),
+        )
+
+        lambda_.Function(
+            self,
+            "JsHandlerFunction",
+            runtime=lambda_.Runtime.NODEJS_18_X,
+            handler="index.handler",
+            code=code_collection.get_code("js_handler"),
+        )
+
+        # Use TypeScriptCodeCollection with build options
+        code_collection_with_options = TypeScriptCodeCollection(
+            self,
+            "MultiLambdaWithOptions",
+            entry_points={
+                "colors": "test/fixtures/handlers/colors.ts",
+            },
+            build_options=BuildOptions(
+                external=["aws-sdk"],
+            ),
+        )
+
+        lambda_.Function(
+            self,
+            "ColorsFunction",
+            runtime=lambda_.Runtime.NODEJS_18_X,
+            handler="index.handler",
+            code=code_collection_with_options.get_code("colors"),
+        )
+
+
+app = cdk.App()
+stack = MultiLambdaStack(app, "MultiLambda")
+
+integ.IntegTest(app, "MultiLambdaFunctions", test_cases=[stack])
+
+app.synth()