fix: finding path of aws-cdk-lib module

Author: ServerlessLife

src/frameworks/cdkFramework.ts

@@ -12,6 +12,7 @@ import { Logger } from "../logger.js";
 import { Worker } from "node:worker_threads";
 import { getModuleDirname, getProjectDirname } from "../getDirname.js";
 import { Configuration } from "../configuration.js";
+import { findNpmPath } from "../utils/findNpmPath.js";
 
 /**
  * Support for AWS CDK framework
@@ -308,6 +309,12 @@ export class CdkFramework implements IFramework {
     process.env.CDK_CONTEXT_JSON = JSON.stringify(CDK_CONTEXT_JSON);
     Logger.verbose(`[CDK] Context:`, JSON.stringify(CDK_CONTEXT_JSON, null, 2));
 
+    const awsCdkLibPath = await findNpmPath(
+      path.join(getProjectDirname(), config.subfolder ?? "/"),
+      "aws-cdk-lib"
+    );
+    Logger.verbose(`[CDK] aws-cdk-lib path: ${awsCdkLibPath}`);
+
     const lambdas: any[] = await new Promise((resolve, reject) => {
       const worker = new Worker(
         path.resolve(
@@ -316,6 +323,7 @@ export class CdkFramework implements IFramework {
         {
           workerData: {
             verbose: Configuration.config.verbose,
+            awsCdkLibPath,
           },
         }
       );

src/utils/findNpmPath.ts

@@ -0,0 +1,42 @@
+import fs from "fs/promises";
+import path from "path";
+
+/**
+ * Function to find the path of a module in the directory and parent directories
+ * @param {*} dir
+ * @param {*} moduleName
+ */
+export async function findNpmPath(dir: string, moduleName: string) {
+  if (dir === "/") return undefined;
+
+  try {
+    await fs.access(path.join(dir, "package.json"));
+
+    const modulePath = await checkModuleInPackageJson(dir, moduleName);
+    if (modulePath) {
+      return modulePath;
+    }
+  } catch {
+    // ignore, no package.json in this directory
+  }
+
+  return await findNpmPath(path.resolve(path.join(dir, "..")), moduleName);
+}
+
+/**
+ * Function to check if a module exists in package.json and return the path
+ * @param {*} dir
+ * @param {*} moduleName
+ * @returns
+ */
+async function checkModuleInPackageJson(dir: string, moduleName: string) {
+  const packageJsonPath = path.join(dir, "package.json");
+  const packageJson = JSON.parse(await fs.readFile(packageJsonPath, "utf8"));
+  const dependencies = packageJson.dependencies || {};
+  const devDependencies = packageJson.devDependencies || {};
+  if (dependencies[moduleName] || devDependencies[moduleName]) {
+    const modulePath = path.join(dir, "node_modules", moduleName);
+    return modulePath;
+  }
+  return undefined;
+}