fix: sls stage parameter

.vscode/launch.json

@@ -19,7 +19,7 @@
     {
       "name": "LLDebugger - SLS Simple",
       "program": "${workspaceRoot}/node_modules/tsx/dist/cli.mjs",
-      "args": ["../../src/lldebugger.ts"],
+      "args": ["../../src/lldebugger.ts", "--stage=test"],
       "request": "launch",
       "skipFiles": [
         "<node_internals>/**"

package.json

@@ -48,6 +48,8 @@
   "workspaces": [
     "src/extension/*",
     "test",
-    "test/cdk-simple"
+    "test/cdk-simple",
+    "test/sls-simple",
+    "test/sls-esbuild"
   ]
 }

src/lldebugger.ts

@@ -33,6 +33,7 @@ export let argOptions: {
   remove?: boolean;
   verbose?: boolean;
   context?: string;
+  stage?: string;
   profile?: string;
   region?: string;
   role?: string;
@@ -47,6 +48,7 @@ async function run() {
   program.option("-r, --remove", "Remove Lambda Live Debugger infrastructure");
   program.option("-v, --verbose", "Verbose logs");
   program.option("-c, --context <context>", "AWS CDK context");
+  program.option("-s, --stage <stage>", "Serverless Framework stage");
   program.option("--profile <profile>", "AWS profile to use");
   program.option("--region <region>", "AWS region to use");
   program.option("--role <role>", "AWS role to use");

src/slsResourcesDiscovery.ts

@@ -13,10 +13,13 @@ import resolveConfigurationPath from "serverless/lib/cli/resolve-configuration-p
 import readConfiguration from "serverless/lib/configuration/read.js";
 import Serverless, { FunctionDefinitionHandler } from "serverless";
 import { EsBuildOptions } from "./functionProps.js";
+import { LlDebugger } from "./lldebugger.js";
 
 //check if there is serverless.yaml or serverless.js or serverless.ts
 export async function isSlsFramework() {
-  process.argv = []; // LLD arguments might conflict with serverless arguments
+  // LLD arguments might conflict with serverless arguments
+  process.argv = [];
+
   const serverlessFiles = [
     "serverless.yml",
     "serverless.yaml",
@@ -45,7 +48,12 @@ export const resourcesDiscovery: ResourcesDiscovery = async () => {
   const configurationFilename =
     configuration && configurationPath.slice(serviceDir.length + 1);
   const commands: string[] = [];
-  const options = {};
+  const options: any = {};
+
+  if (LlDebugger.argOptions.stage) {
+    options.stage = LlDebugger.argOptions.stage;
+  }
+
   const variablesMeta = {};
 
   const serverless = new Serverless({

test/sls-esbuild/package.json

@@ -1,5 +1,5 @@
 {
-  "name": "sls-simple",
+  "name": "sls-esbuild",
   "version": "1.0.0",
   "description": "",
   "main": "index.js",
@@ -15,4 +15,4 @@
     "serverless": "^3.38.0",
     "serverless-esbuild": "^1.52.1"
   }
-}
+}

test/sls-simple.test.ts

@@ -24,7 +24,7 @@ describe("sls-simple", () => {
       await execAsync("npm run deploy", {
         cwd: folder,
       });
-      lldProcess = await startDebugger(folder);
+      lldProcess = await startDebugger(folder, ["--stage=test"]);
     }
   });
 
@@ -64,7 +64,7 @@ describe("sls-simple", () => {
 
   test("remove infra", async () => {
     if (process.env.CI === "true" || process.env.RUN_TEST_FROM_CLI === "true") {
-      await removeInfra(lldProcess, folder);
+      await removeInfra(lldProcess, folder, ["--stage=test"]);
       const lambdaName = "lls-sls-simple-test-testJsCommonJs";
       await expectInfraRemoved(lambdaName);
     }

test/sls-simple/package.json

@@ -4,8 +4,8 @@
   "description": "",
   "main": "index.js",
   "scripts": {
-    "deploy": "serverless deploy",
-    "destroy": "serverless remove"
+    "deploy": "serverless deploy --stage=test",
+    "destroy": "serverless remove --stage=test"
   },
   "keywords": [],
   "author": "",

test/sls-simple/serverless.yml

@@ -5,7 +5,6 @@ provider:
   name: aws
   runtime: nodejs20.x
   region: eu-west-1
-  stage: test
 
 functions:
   testJsCommonJs:

test/utils/removeInfra.ts

@@ -3,11 +3,12 @@ import { setTimeout } from "timers/promises";
 
 export async function removeInfra(
   lldProcess: ChildProcess | undefined,
-  folder: string
+  folder: string,
+  args: string[] = []
 ) {
   lldProcess?.kill();
 
-  await execSync("node ../../dist/lldebugger.js --remove", {
+  await execSync(`node ../../dist/lldebugger.js --remove ${args?.join(" ")}`, {
     cwd: folder,
   });