From b9d07132efcadde006a07eff7bd5ad27b4a6bd2e Mon Sep 17 00:00:00 2001
From: daurnimator <quae@daurnimator.com>
Date: Wed, 12 Apr 2017 19:13:39 +1000
Subject: Try CLI input with 'return' prepended

---
 tests/manual-tests/lua-cli.js | 24 +++++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/tests/manual-tests/lua-cli.js b/tests/manual-tests/lua-cli.js
index 6c7801c..6938a8b 100755
--- a/tests/manual-tests/lua-cli.js
+++ b/tests/manual-tests/lua-cli.js
@@ -12,6 +12,8 @@ readlineSync.setDefaultOptions({
     prompt: '> '
 });
 
+const stdin = lua.to_luastring("=stdin");
+
 const L = lauxlib.luaL_newstate();
 
 linit.luaL_openlibs(L);
@@ -24,10 +26,26 @@ for (;;) {
     if (input.length === 0)
         continue;
 
-    let buffer = lua.to_luastring(input);
-    if (lauxlib.luaL_loadbuffer(L, buffer, buffer.length, lua.to_luastring("=stdin"))
-        || lapi.lua_pcall(L, 0, 0, 0)) {
+    let status;
+    {
+        let buffer = lua.to_luastring("return " + input);
+        status = lauxlib.luaL_loadbuffer(L, buffer, buffer.length, stdin);
+    }
+    if (status !== lua.thread_status.LUA_OK) {
+        lapi.lua_pop(L, 1);
+        let buffer = lua.to_luastring(input);
+        if (lauxlib.luaL_loadbuffer(L, buffer, buffer.length, stdin) === lua.thread_status.LUA_OK) {
+            status = lua.thread_status.LUA_OK;
+        }
+    }
+    if (status !== lua.thread_status.LUA_OK) {
         lauxlib.lua_writestringerror(`${lapi.lua_tojsstring(L, -1)}\n`);
+        continue;
+    }
+    if (lapi.lua_pcall(L, 0, 0, 0) !== lua.thread_status.LUA_OK) {
+        lauxlib.lua_writestringerror(`${lapi.lua_tojsstring(L, -1)}\n`);
+        lapi.lua_settop(L, 0);
+        continue;
     }
     lapi.lua_settop(L, 0);  /* remove eventual returns */
 }
-- 
cgit v1.2.3-70-g09d2


From d2c29700b140c6656bc9cc4c8870f9f340b98945 Mon Sep 17 00:00:00 2001
From: daurnimator <quae@daurnimator.com>
Date: Wed, 12 Apr 2017 19:23:08 +1000
Subject: Print returned values from cli

---
 tests/manual-tests/lua-cli.js | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/tests/manual-tests/lua-cli.js b/tests/manual-tests/lua-cli.js
index 6938a8b..de64359 100755
--- a/tests/manual-tests/lua-cli.js
+++ b/tests/manual-tests/lua-cli.js
@@ -42,10 +42,20 @@ for (;;) {
         lauxlib.lua_writestringerror(`${lapi.lua_tojsstring(L, -1)}\n`);
         continue;
     }
-    if (lapi.lua_pcall(L, 0, 0, 0) !== lua.thread_status.LUA_OK) {
+    if (lapi.lua_pcall(L, 0, lua.LUA_MULTRET, 0) !== lua.thread_status.LUA_OK) {
         lauxlib.lua_writestringerror(`${lapi.lua_tojsstring(L, -1)}\n`);
         lapi.lua_settop(L, 0);
         continue;
     }
+    let n = lapi.lua_gettop(L);
+    if (n > 0) {  /* any result to be printed? */
+        lapi.lua_getglobal(L, lua.to_luastring("print"));
+        lapi.lua_insert(L, 1);
+        if (lapi.lua_pcall(L, n, 0, 0) != lua.thread_status.LUA_OK) {
+            lauxlib.lua_writestringerror(`error calling 'print' (${lapi.lua_tojsstring(L, -1)})\n`);
+            lapi.lua_settop(L, 0);
+            continue;
+        }
+    }
     lapi.lua_settop(L, 0);  /* remove eventual returns */
 }
-- 
cgit v1.2.3-70-g09d2