@aws-sdk/find-v2 0.3.0 → 0.3.2

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,31 +1,40 @@
-import { Lambda, paginateListFunctions } from "@aws-sdk/client-lambda";
-import { cpus } from "node:os";
+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 functionCount = functions.length;
+    const concurrency = Math.min(functionCount, cpus().length || 1);
+    const codeSizeToDownload = functions.reduce((acc, fn) => acc + (fn.CodeSize || 0), 0);
+    const codeSizeToSaveOnDisk = functions
+        .map((fn) => fn.CodeSize || 0)
+        .sort((a, b) => b - a)
+        .slice(0, concurrency)
+        .reduce((acc, size) => acc + size, 0);
+    if (functionCount === 0) {
         console.log("No functions found.");
         process.exit(0);
     }
+    if (!yes) {
+        const confirmation = await getDownloadConfirmation(functionCount, codeSizeToDownload, codeSizeToSaveOnDisk);
+        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.`);
-    const limit = pLimit(Math.min(functionsLength, cpus().length || 1));
-    await Promise.all(functions.map((fn) => limit(() => scanLambdaFunction(client, fn))));
+    console.log(`Reading ${functionCount} function${functionCount > 1 ? "s" : ""} from "${clientRegion}" region.`);
+    const limit = pLimit(concurrency);
+    await Promise.all(functions.map((fn) => limit(() => scanLambdaFunction(client, fn.FunctionName))));
     console.log("\nDone.");
 };

package/dist/utils/getDownloadConfirmation.js

@@ -0,0 +1,26 @@
+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 codeSizeToDownload - Total size of all function code in bytes
+ * @param codeSizeToSaveOnDisk - Maximum disk space used at any point
+ * @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, codeSizeToDownload, codeSizeToSaveOnDisk) => {
+    const rl = createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    });
+    const answer = await rl.question(`This command will process ${functionCount} Lambda Node.js functions, and` +
+        `\ndownload ${getHumanReadableBytes(codeSizeToDownload)} of compressed archives over the network.` +
+        `\nIt'll store maximum of ${getHumanReadableBytes(codeSizeToSaveOnDisk)} on disk at any point.` +
+        `\n\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-1000 conversion for SI units
+ * - Rounds to specified number of decimal places
+ */
+export const getHumanReadableBytes = (bytes, decimals = 2) => {
+    if (bytes <= 0)
+        return "0 Bytes";
+    const k = 1000;
+    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.3.0",
+  "version": "0.3.2",
   "description": "CLI to find resources which call AWS using JavaScript SDK v2",
   "main": "dist/cli.js",
   "types": "dist/cli.d.ts",