Support function switch for sls manage command

logandk · Oct 22, 2024 · 2d71715 · 2d71715
2 parents f6eee36 + 8c80d58
commit 2d71715
Show file tree

Hide file tree

Showing 3 changed files with 150 additions and 69 deletions.
diff --git a/.eslintrc.js b/.eslintrc.js
@@ -2,6 +2,7 @@ module.exports = {
   env: {
     es6: true,
     node: true,
+    mocha: true,
   },
   extends: "eslint:recommended",
   parserOptions: {

diff --git a/index.js b/index.js
@@ -434,13 +434,29 @@ class ServerlessWSGI {
   }
 
   findHandler() {
-    return _.findKey(this.serverless.service.functions, (fun) =>
-      _.includes(fun.handler, "wsgi_handler.handler")
-    );
+    const functionName = this.options.function || this.options.f;
+
+    if (functionName) {
+      // If the function name is specified, return it directly
+      if (this.serverless.service.functions[functionName]) {
+        return functionName;
+      } else {
+        throw new Error(`Function "${functionName}" not found.`);
+      }
+    } else {
+      return _.findKey(this.serverless.service.functions, (fun) =>
+        _.includes(fun.handler, "wsgi_handler.handler")
+      );
+    }
   }
 
   invokeHandler(command, data, local) {
-    const handlerFunction = this.findHandler();
+    let handlerFunction;
+    try {
+      handlerFunction = this.findHandler();
+    } catch (error) {
+      return BbPromise.reject(error.message);
+    }
 
     if (!handlerFunction) {
       return BbPromise.reject(