feature: support serverless.tf

Author: ServerlessLife

src/frameworks/terraformFramework.ts

@@ -12,19 +12,21 @@ import { LldConfigBase } from '../types/lldConfig.js';
 
 export const execAsync = promisify(exec);
 
-interface TerraformState {
-  resources: Array<{
-    type: string;
-    name: string;
-    values: {
-      function_name?: string;
-      handler?: string;
-      source_dir?: string;
-      source_file?: string;
+interface TerraformResource {
+  type: string;
+  name: string;
+  address: string;
+  values: {
+    function_name?: string;
+    handler?: string;
+    source_dir?: string;
+    source_file?: string;
+    query?: {
+      source_path?: string;
     };
-    //dependencies > depends_on
-    depends_on: Array<string>;
-  }>;
+  };
+  //dependencies > depends_on
+  depends_on: Array<string>;
 }
 
 /**
@@ -151,15 +153,15 @@ export class TerraformFramework implements IFramework {
     return lambdasDiscovered;
   }
 
-  protected extractLambdaInfo(state: TerraformState) {
+  protected extractLambdaInfo(resources: TerraformResource[]) {
     const lambdas: Array<{
       functionName: string;
       sourceDir?: string;
       sourceFilename?: string;
       handler: string;
     }> = [];
 
-    for (const resource of state.resources) {
+    for (const resource of resources) {
       if (resource.type === 'aws_lambda_function') {
         Logger.verbose(
           '[Terraform] Found Lambda:',
@@ -186,9 +188,7 @@ export class TerraformFramework implements IFramework {
         if (archiveFileResourceName) {
           // get the resource
           const name = archiveFileResourceName.split('.')[2];
-          const archiveFileResource = state.resources.find(
-            (r) => r.name === name,
-          );
+          const archiveFileResource = resources.find((r) => r.name === name);
 
           // get source_dir or source_filename
           if (archiveFileResource) {
@@ -197,6 +197,28 @@ export class TerraformFramework implements IFramework {
           }
         }
 
+        // get dependency "archive_prepare" = serverless.tf support
+        const archivePrepareResourceName = dependencies.find((dep) =>
+          dep.includes('.archive_prepare'),
+        );
+
+        if (archivePrepareResourceName) {
+          // get the resource
+          const name = archivePrepareResourceName;
+          const archivePrepareResource = resources.find((r) =>
+            r.address?.startsWith(name),
+          );
+
+          // get source_dir or source_filename
+          if (archivePrepareResource) {
+            sourceDir =
+              archivePrepareResource.values.query?.source_path?.replaceAll(
+                '"',
+                '',
+              );
+          }
+        }
+
         if (!sourceDir && !sourceFilename) {
           Logger.error(`Failed to find source code for Lambda ${functionName}`);
         } else {
@@ -213,7 +235,7 @@ export class TerraformFramework implements IFramework {
     return lambdas;
   }
 
-  protected async readTerraformState(): Promise<TerraformState> {
+  protected async readTerraformState(): Promise<TerraformResource[]> {
     // Is there a better way to get the Terraform state???
 
     let output: any;
@@ -254,7 +276,16 @@ export class TerraformFramework implements IFramework {
 
     try {
       const state = JSON.parse(jsonString);
-      return state.values.root_module as TerraformState;
+
+      const rootResources: TerraformResource[] =
+        state.values?.root_module?.resources ?? [];
+
+      const childResources: TerraformResource[] =
+        state.values?.root_module?.child_modules
+          ?.map((m: any) => m.resources)
+          .flat() ?? [];
+
+      return [...rootResources, ...childResources] as TerraformResource[];
     } catch (error: any) {
       //save state to file
       await fs.writeFile('terraform-state.json', jsonString);

test/terraform-basic.test.ts

@@ -103,6 +103,22 @@ describe('terraform-basic', async () => {
     }
   });
 
+  test('call Lambda - testJsCommonJs_3', async () => {
+    const lambdaName = await getFunctionName(
+      folder,
+      'lambda-test-js-commonjs_3_name',
+    );
+
+    const payload = getSamplePayload(lambdaName);
+    const response = await callLambda(lambdaName, payload);
+
+    expect(response.inputEvent).toEqual(payload);
+    expect(response.runningLocally).toEqual(!observableMode);
+    if (observableMode) {
+      await validateLocalResponse(lambdaName, payload);
+    }
+  });
+
   test('call Lambda - testJsEsModule', async () => {
     const lambdaName = await getFunctionName(
       folder,

test/terraform-basic/.gitignore

@@ -36,5 +36,6 @@ hello-world.zip
 response.json
 
 dist
+builds
 .terraform.lock.hcl
 terraform-outputs.json

test/terraform-basic/main.tf

@@ -87,6 +87,16 @@ resource "aws_lambda_function" "test-js-commonjs_2" {
   role             = aws_iam_role.lambda_role.arn
 }
 
+module "test-js-commonjs_3" {
+  source = "terraform-aws-modules/lambda/aws"
+
+  function_name = "lld-terraform-basic-test-js-commonjs_3"
+  handler       = "lambda.lambdaHandler"
+  runtime       = "nodejs22.x"
+
+  source_path = "services/testJsCommonJs"
+}
+
 // services/testJsEsModule/lambda.js
 data "archive_file" "test-js-esmodule_zip" {
   type        = "zip"
@@ -173,6 +183,10 @@ output "lambda-test-js-commonjs_2_name" {
   value = aws_lambda_function.test-js-commonjs_2.function_name
 }
 
+output "lambda-test-js-commonjs_3_name" {
+  value = module.test-js-commonjs_3.lambda_function_name
+}
+
 output "lambda-test-js-esmodule_name" {
   value = aws_lambda_function.test-js-esmodule.function_name
 }