@aws-sdk/find-v2 0.2.0 → 0.3.1

package/dist/cli.js

@@ -15,10 +15,11 @@ export const createProgram = () => {
         .version(packageJson.version, "-v, --version");
     program
         .command("lambda")
-        .description("Scans Lambda Node.js Functions for JavaScript SDK v2.")
+        .description("Scans Lambda Node.js Functions for JavaScript SDK v2")
         .option("-r, --region <region>", "AWS region to scan")
+        .option("-y, --yes", "answer yes for all prompts")
         .action(async (options) => {
-        await scanLambdaFunctions(options.region);
+        await scanLambdaFunctions(options);
     });
     return program;
 };

package/dist/scanLambdaFunctions.js

@@ -1,30 +1,34 @@
-import { Lambda, paginateListFunctions } from "@aws-sdk/client-lambda";
+import { Lambda } from "@aws-sdk/client-lambda";
+import pLimit from "p-limit";
+import { cpus } from "node:os";
 import { JS_SDK_V2_MARKER } from "./constants.js";
 import { scanLambdaFunction } from "./scanLambdaFunction.js";
-const getNodeJsFunctionNames = (functions) => (functions ?? [])
-    .filter((fn) => fn.Runtime?.startsWith("nodejs"))
-    .map((fn) => fn.FunctionName)
-    .filter((fnName) => fnName !== undefined);
-export const scanLambdaFunctions = async (region) => {
+import { getDownloadConfirmation } from "./utils/getDownloadConfirmation.js";
+import { getLambdaFunctions } from "./utils/getLambdaFunctions.js";
+export const scanLambdaFunctions = async ({ region, yes } = {}) => {
     const client = new Lambda({ region });
-    const functions = [];
-    const paginator = paginateListFunctions({ client }, {});
-    for await (const page of paginator) {
-        functions.push(...getNodeJsFunctionNames(page.Functions));
-    }
-    const functionsLength = functions.length;
-    if (functionsLength === 0) {
+    const functions = await getLambdaFunctions(client);
+    const totalCodeSize = functions.reduce((acc, fn) => acc + (fn.CodeSize || 0), 0);
+    const functionCount = functions.length;
+    if (functionCount === 0) {
         console.log("No functions found.");
         process.exit(0);
     }
+    if (!yes) {
+        const confirmation = await getDownloadConfirmation(functionCount, totalCodeSize);
+        console.log();
+        if (!confirmation) {
+            console.log("Exiting.");
+            process.exit(0);
+        }
+    }
     console.log(`Note about output:`);
     console.log(`- ${JS_SDK_V2_MARKER.Y} means "aws-sdk" is found in Lambda function, and migration is recommended.`);
     console.log(`- ${JS_SDK_V2_MARKER.N} means "aws-sdk" is not found in Lambda function.`);
     console.log(`- ${JS_SDK_V2_MARKER.UNKNOWN} means script was not able to proceed, and it emits reason.\n`);
     const clientRegion = await client.config.region();
-    console.log(`Reading ${functionsLength} function${functionsLength > 1 ? "s" : ""} from "${clientRegion}" region.`);
-    for (const functionName of functions) {
-        await scanLambdaFunction(client, functionName);
-    }
+    console.log(`Reading ${functionCount} function${functionCount > 1 ? "s" : ""} from "${clientRegion}" region.`);
+    const limit = pLimit(Math.min(functionCount, cpus().length || 1));
+    await Promise.all(functions.map((fn) => limit(() => scanLambdaFunction(client, fn.FunctionName))));
     console.log("\nDone.");
 };

package/dist/utils/getDownloadConfirmation.js

@@ -0,0 +1,24 @@
+import { createInterface } from "node:readline/promises";
+import { getHumanReadableBytes } from "./getHumanReadableBytes.js";
+/**
+ * Prompts user for confirmation before downloading Lambda function code
+ *
+ * @param functionCount - Number of Lambda functions to be processed
+ * @param totalCodeSize - Total size of all function code in bytes
+ * @returns Promise that resolves to boolean indicating user's choice
+ * @description
+ * - Creates interactive readline interface for user input
+ * - Displays summary of operations including function count and total size
+ * - Returns true if user confirms with 'y' or 'yes', false otherwise
+ */
+export const getDownloadConfirmation = async (functionCount, totalCodeSize) => {
+    const rl = createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    });
+    const answer = await rl.question(`This script will process ${functionCount} Lambda Node.js functions,` +
+        `\nand download ${getHumanReadableBytes(totalCodeSize)} of compressed archives over the network.` +
+        `\nDo you want to continue? (y/N): `);
+    rl.close();
+    return ["y", "yes"].includes(answer.trim().toLowerCase());
+};

package/dist/utils/getHumanReadableBytes.js

@@ -0,0 +1,20 @@
+/**
+ * Converts bytes to human readable format with appropriate units
+ *
+ * @param bytes - Size in bytes to convert
+ * @param decimals - Number of decimal places to show (default: 2)
+ * @returns String representation of size with unit (e.g., "1.50 MB")
+ * @description
+ * - Handles edge case where bytes is 0 or negative
+ * - Automatically selects appropriate unit (Bytes, KB, MB, GB, etc.)
+ * - Uses base-1024 conversion for binary units
+ * - Rounds to specified number of decimal places
+ */
+export const getHumanReadableBytes = (bytes, decimals = 2) => {
+    if (bytes <= 0)
+        return "0 Bytes";
+    const k = 1024;
+    const sizes = ["Bytes", "KB", "MB", "GB", "TB", "PB"];
+    const i = Math.min(Math.floor(Math.log(bytes) / Math.log(k)), sizes.length - 1);
+    return parseFloat((bytes / Math.pow(k, i)).toFixed(decimals)) + " " + sizes[i];
+};

package/dist/utils/getLambdaFunctions.js

@@ -0,0 +1,19 @@
+import { paginateListFunctions } from "@aws-sdk/client-lambda";
+/**
+ * Retrieves all Lambda functions with Node.js runtime from AWS
+ *
+ * @param client - AWS Lambda client instance
+ * @returns Promise that resolves to array of Lambda function configurations
+ * @description
+ * - Uses AWS SDK v3 pagination to handle large number of functions
+ * - Filters results to only include Node.js runtimes
+ * - Returns empty array if no functions found
+ */
+export const getLambdaFunctions = async (client) => {
+    const functions = [];
+    const paginator = paginateListFunctions({ client }, {});
+    for await (const page of paginator) {
+        functions.push(...(page.Functions ?? []).filter((fn) => fn.Runtime?.startsWith("nodejs")));
+    }
+    return functions;
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aws-sdk/find-v2",
-  "version": "0.2.0",
+  "version": "0.3.1",
   "description": "CLI to find resources which call AWS using JavaScript SDK v2",
   "main": "dist/cli.js",
   "types": "dist/cli.d.ts",
@@ -8,7 +8,8 @@
   "dependencies": {
     "@aws-sdk/client-lambda": "^3.942.0",
     "commander": "^14.0.2",
-    "node-stream-zip": "^1.15.0"
+    "node-stream-zip": "^1.15.0",
+    "p-limit": "^7.2.0"
   },
   "devDependencies": {
     "@aws-sdk/client-sts": "^3.946.0",