@bifravst/aws-cdk-lambda-helpers 3.2.0 → 3.3.1

package/README.md

@@ -17,6 +17,14 @@ Helper functions which simplify working with TypeScript lambdas for AWS CDK.
 
 See [the end-to-end test stack](./cdk/e2e.ts).
 
+## `updateLambdaCode()` helper
+
+[`updateLambdaCode()`](./src/updateLambdaCode.ts) is a helper that will update
+the function code of a lambda directly without a CloudFormation deployment,
+which is many times faster. This is useful during development.
+
+See [this example](./cdk/update-lambdas.ts) on how to use it.
+
 ## Example migrations to `@bifravst/aws-cdk-lambda-helpers`
 
 - [world.thingy.rocks backend](https://github.com/NordicPlayground/thingy-rocks-cloud-aws-js/commit/3ca6e267917db4d8cb09ca63ed54384c0e23f163)

package/dist/src/updateLambdaCode.d.ts

@@ -0,0 +1,7 @@
+import { type CloudFormationClient } from '@aws-sdk/client-cloudformation';
+import { type LambdaClient } from '@aws-sdk/client-lambda';
+import type { PackedLambda } from './packLambda.ts';
+export declare const updateLambdaCode: ({ cf, lambda }: {
+    cf: CloudFormationClient;
+    lambda: LambdaClient;
+}) => (stackName: string, packedLambdas: Record<string, PackedLambda>, debug?: (...args: Array<any>) => void) => Promise<void>;

package/dist/src/updateLambdaCode.js

@@ -0,0 +1,73 @@
+import { paginateListStackResources } from '@aws-sdk/client-cloudformation';
+import { GetFunctionCommand, LastUpdateStatus, TagResourceCommand, UpdateFunctionCodeCommand, UpdateFunctionConfigurationCommand } from '@aws-sdk/client-lambda';
+import assert from 'node:assert';
+import { readFile } from 'node:fs/promises';
+import pRetry from 'p-retry';
+const updateLambda = ({ lambda, packs, debug })=>async (lambdaResource)=>{
+        const info = await lambda.send(new GetFunctionCommand({
+            FunctionName: lambdaResource.PhysicalResourceId
+        }));
+        const packed = Object.values(packs).find((p)=>p.id === info.Tags?.['packedLambda:id']);
+        if (packed === undefined) {
+            debug?.(`[${info.Tags?.['packedLambda:id']}]`, 'No pack found for lambda');
+            return;
+        }
+        debug?.(`[${packed.id}]`, 'found pack');
+        if (packed.hash === info.Tags?.['packedLambda:hash']) {
+            debug?.(`[${packed.id}]`, 'No update needed');
+            return;
+        }
+        if (packed.hash !== info.Tags?.['packedLambda:hash']) {
+            debug?.(`[${packed.id}]`, 'Updating lambda');
+        }
+        const updateResult = await lambda.send(new UpdateFunctionCodeCommand({
+            FunctionName: lambdaResource.PhysicalResourceId,
+            ZipFile: new Uint8Array(await readFile(packed.zipFilePath))
+        }));
+        debug?.(`[${packed.id}]`, 'RevisionId', updateResult.RevisionId);
+        debug?.(`[${packed.id}]`, 'CodeSha256', updateResult.CodeSha256);
+        await pRetry(async ()=>{
+            const updateStatus = (await lambda.send(new GetFunctionCommand({
+                FunctionName: lambdaResource.PhysicalResourceId
+            }))).Configuration?.LastUpdateStatus;
+            assert.equal(updateStatus, LastUpdateStatus.Successful, `Lambda update should have succeeded, got ${updateStatus}`);
+        }, {
+            minTimeout: 1000,
+            maxTimeout: 1000,
+            retries: 10
+        });
+        debug?.(`[${packed.id}]`, `updating config to trigger reload ...`);
+        await Promise.all([
+            await lambda.send(new UpdateFunctionConfigurationCommand({
+                FunctionName: lambdaResource.PhysicalResourceId,
+                Environment: {
+                    Variables: {
+                        ...info.Configuration?.Environment?.Variables,
+                        PACKED_LAMBDA_HASH: packed.hash
+                    }
+                }
+            })),
+            await lambda.send(new TagResourceCommand({
+                Resource: info.Configuration.FunctionArn,
+                Tags: {
+                    'packedLambda:hash': packed.hash
+                }
+            }))
+        ]);
+    };
+export const updateLambdaCode = ({ cf, lambda })=>async (stackName, packedLambdas, debug)=>{
+        const u = updateLambda({
+            lambda,
+            packs: packedLambdas,
+            debug
+        });
+        const p = [];
+        for await (const page of paginateListStackResources({
+            client: cf
+        }, {
+            StackName: stackName
+        })){
+            p.push(...(page.StackResourceSummaries ?? []).filter((resource)=>resource.ResourceType === 'AWS::Lambda::Function').map(u) ?? []);
+        }
+        await Promise.all(p);
+    };

package/dist/src/util.d.ts

@@ -2,3 +2,4 @@ export * from './checksumOfFiles.ts';
 export * from './commonParent.ts';
 export * from './findDependencies.ts';
 export * from './isTest.ts';
+export * from './updateLambdaCode.ts';

package/dist/src/util.js

@@ -2,3 +2,4 @@ export * from './checksumOfFiles.js';
 export * from './commonParent.js';
 export * from './findDependencies.js';
 export * from './isTest.js';
+export * from './updateLambdaCode.js';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bifravst/aws-cdk-lambda-helpers",
-  "version": "3.2.0",
+  "version": "3.3.1",
   "description": "Helper functions which simplify working with TypeScript lambdas for AWS CDK.",
   "exports": {
     ".": {
@@ -32,6 +32,7 @@
   "scripts": {
     "test": "node --experimental-strip-types --no-warnings --test ./src/*.spec.ts",
     "test:e2e": "node --experimental-strip-types --test e2e.spec.ts",
+    "test:e2e:updated": "node --experimental-strip-types --test e2e-updated.spec.ts",
     "prepare": "husky",
     "prepublishOnly": "node --experimental-strip-types npm-compile.ts && npx tsc -P tsconfig.npm.json --outDir ./dist/src"
   },
@@ -52,8 +53,8 @@
   "author": "Nordic Semiconductor ASA | nordicsemi.no",
   "license": "BSD-3-Clause",
   "devDependencies": {
-    "@aws-sdk/client-cloudformation": "3.797.0",
-    "@aws-sdk/client-dynamodb": "3.797.0",
+    "@aws-sdk/client-cloudformation": "3.798.0",
+    "@aws-sdk/client-dynamodb": "3.798.0",
     "@bifravst/cloudformation-helpers": "9.1.1",
     "@bifravst/eslint-config-typescript": "6.1.22",
     "@bifravst/from-env": "3.0.2",
@@ -61,7 +62,7 @@
     "@commitlint/config-conventional": "19.8.0",
     "@swc/cli": "0.7.3",
     "@types/aws-lambda": "8.10.149",
-    "@types/node": "22.15.2",
+    "@types/node": "22.15.3",
     "@types/unzip-stream": "0.3.4",
     "@types/yazl": "2.4.6",
     "cdk": "2.1012.0",
@@ -115,10 +116,12 @@
     "@swc/core": "1.11.22",
     "fp-ts": "2.16.10",
     "glob": "11.0.2",
+    "p-retry": "6.2.1",
     "typescript": "5.8.3",
     "yazl": "3.3.1"
   },
   "peerDependencies": {
+    "@aws-sdk/client-lambda": "^3.798.0",
     "@bifravst/aws-ssm-settings-helpers": "^1.2.134",
     "aws-cdk-lib": "^2.192.0",
     "constructs": "^10.4.2"