From ef8289ff73fa3503466f76e0902e11aa7b5792c7 Mon Sep 17 00:00:00 2001
From: Ta-Ching Chen <contact@tachingchen.com>
Date: Fri, 30 Aug 2019 15:31:04 +0800
Subject: [PATCH] Support nodejs function with 0 argument (#1295)

---
 environments/nodejs/server.js | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/environments/nodejs/server.js b/environments/nodejs/server.js
index f2ab9c0e..8c042320 100644
--- a/environments/nodejs/server.js
+++ b/environments/nodejs/server.js
@@ -13,6 +13,12 @@ if (!argv.port) {
     argv.port = 8888;
 }
 
+// To catch unhandled exceptions thrown by user code async callbacks,
+// these exceptions cannot be catched by try-catch in user function invocation code below
+process.on('uncaughtException', (err) => {
+    console.error(`Caught exception: ${err}`);
+});
+
 // User function.  Starts out undefined.
 let userFunction;
 
@@ -91,7 +97,7 @@ function specialize(req, res) {
 
 
 // Request logger
-app.use(morgan('combined'))
+app.use(morgan('combined'));
 
 app.use(bodyParser.urlencoded({ extended: false, limit: '1mb' }));
 app.use(bodyParser.json({limit: '1mb'}));
@@ -132,10 +138,15 @@ app.all('/', function (req, res) {
     // you can do that here by adding properties to the context.
     //
 
-    let functionProm;
-    if (userFunction.length === 1) { // One argument (context)
+    if (userFunction.length <= 1) { // One or zero argument (context)
+        let result;
         // Make sure their function returns a promise
-        Promise.resolve(userFunction(context)).then(function({ status, body, headers }) {
+        if (userFunction.length === 0) {
+            result = Promise.resolve(userFunction())
+        } else {
+            result = Promise.resolve(userFunction(context))
+        }
+        result.then(function({status, body, headers}) {
             callback(status, body, headers);
         }).catch(function(err) {
             console.log(`Function error: ${err}`);