From 75c14fadc22366a6165f593fb1a41e2a32de4354 Mon Sep 17 00:00:00 2001 From: Benoit Giannangeli Date: Tue, 18 Apr 2017 13:30:22 +0200 Subject: Fixed table comparison (was comparing TValue instead of values) --- tests/manual-tests/lua-cli.js | 291 +++++++++++++++++++++++++++++++++++++----- 1 file changed, 256 insertions(+), 35 deletions(-) (limited to 'tests') diff --git a/tests/manual-tests/lua-cli.js b/tests/manual-tests/lua-cli.js index de64359..3a3b386 100755 --- a/tests/manual-tests/lua-cli.js +++ b/tests/manual-tests/lua-cli.js @@ -8,54 +8,275 @@ const linit = require('../../src/linit.js'); const fs = require('fs'); const readlineSync = require('readline-sync'); -readlineSync.setDefaultOptions({ - prompt: '> ' -}); - const stdin = lua.to_luastring("=stdin"); +const _PROMPT = lua.to_luastring("_PROMPT"); +const _PROMPT2 = lua.to_luastring("_PROMPT2"); + +const report = function(L, status) { + if (status !== lua.thread_status.LUA_OK) { + lauxlib.lua_writestringerror(`${lapi.lua_tojsstring(L, -1)}\n`); + lapi.lua_pop(L, 1); + } + return status; +}; + +const docall = function(L, narg, nres) { + let status = lapi.lua_pcall(L, narg, nres, 0); + return status; +}; + +const dochunk = function(L, status) { + if (status === lua.thread_status.LUA_OK) { + status = docall(L, 0, 0); + } + return report(L, status); +}; + +const dofile = function(L, name) { + return dochunk(L, lauxlib.luaL_loadfile(L, lua.to_luastring(name))); +}; + +const dostring = function(L, s, name) { + let buffer = lua.to_luastring(s); + return dochunk(L, lauxlib.luaL_loadbuffer(L, buffer, buffer.length, lua.to_luastring(name))); +}; + +const dolibrary = function(L, name) { + lapi.lua_getglobal(L, "require"); + lapi.lua_pushliteral(L, name); + let status = docall(L, 1, 1); /* call 'require(name)' */ + if (status === lua.thread_status.LUA_OK) + lapi.lua_setglobal(L, name); /* global[name] = require return */ + return report(L, status); +}; + +let progname = process.argv[1]; + +const print_usage = function(badoption) { + lauxlib.lua_writestringerror(`${progname}: `); + if (badoption[1] === "e" || badoption[1] === "l") + lauxlib.lua_writestringerror(`'${badoption}' needs argument\n`); + else + lauxlib.lua_writestringerror(`'unrecognized option '${badoption}'\n`); + lauxlib.lua_writestringerror( + `usage: ${progname} [options] [script [args]]\n` + + "Available options are:\n" + + " -e stat execute string 'stat'\n" + + " -i enter interactive mode after executing 'script'\n" + + " -l name require library 'name'\n" + + " -v show version information\n" + + " -E ignore environment variables\n" + + " -- stop handling options\n" + + " - stop handling options and execute stdin\n" + ); +}; const L = lauxlib.luaL_newstate(); -linit.luaL_openlibs(L); +let script = 2; // Where to start args from +let has_E = false; +let has_i = false; +let has_v = false; +let has_e = false; -console.log(lua.FENGARI_COPYRIGHT); +for (let i = 2; i 2) { + /* invalid option */ + print_usage(process.argv[script]); + return process.exit(1); + } + has_v = true; + break; + case 'e': + has_e = true; + /* falls through */ + case 'l': /* both options need an argument */ + if (process.argv[i].length < 2) { /* no concatenated argument? */ + i++; /* try next 'process.argv' */ + if (process.argv.length <= i || process.argv[i][0] === '-') { + /* no next argument or it is another option */ + print_usage(process.argv[script]); + return process.exit(1); + } + } + break; + default: /* invalid option */ + print_usage(process.argv[script]); + return process.exit(1); + } +} -for (;;) { - let input = readlineSync.prompt(); +if (has_v) + console.log(lua.FENGARI_COPYRIGHT); - if (input.length === 0) - continue; +if (has_E) { + /* signal for libraries to ignore env. vars. */ + lapi.lua_pushboolean(L, 1); + lapi.lua_setfield(L, lua.LUA_REGISTRYINDEX, lua.to_luastring("LUA_NOENV")); +} - let status; - { - let buffer = lua.to_luastring("return " + input); - status = lauxlib.luaL_loadbuffer(L, buffer, buffer.length, stdin); +/* open standard libraries */ +linit.luaL_openlibs(L); + +/* create table 'arg' */ +lapi.lua_createtable(L, process.argv.length - (script + 1), script + 1); +for (let i = 0; i < process.argv.length; i++) { + lapi.lua_pushliteral(L, process.argv[i]); + lapi.lua_seti(L, -2, i - script); /* TODO: rawseti */ +} +lapi.lua_setglobal(L, lua.to_luastring("arg")); + +if (!has_E) { + /* run LUA_INIT */ + let name = "LUA_INIT"+lua.LUA_VERSUFFIX; + let init = process.env[name]; + if (!init) { + name = "LUA_INIT"; + init = process.env[name]; } - 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 (init) { + let status; + if (init[0] === '@') { + status = dofile(L, init.substring(1)); + } else { + status = dostring(L, init, name); + } + if (status !== lua.thread_status.LUA_OK) { + return process.exit(1); } } - if (status !== lua.thread_status.LUA_OK) { - lauxlib.lua_writestringerror(`${lapi.lua_tojsstring(L, -1)}\n`); - continue; +} + +/* execute arguments -e and -l */ +for (let i = 1; i < script; i++) { + let option = process.argv[i][1]; + if (option == 'e' || option == 'l') { + let extra = process.argv[i].substring(2); /* both options need an argument */ + if (extra.length === 0) + extra = process.argv[++i]; + let status; + if (option == 'e') { + status = dostring(L, extra, "=(command line)"); + } else { + status = dolibrary(L, extra); + } + if (status !== lua.thread_status.LUA_OK) { + return process.exit(1); + } } - 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); +} + +const pushargs = function(L) { + if (lapi.lua_getglobal(L, lua.to_luastring("arg")) !== lua.constant_types.LUA_TTABLE) + lauxlib.luaL_error(L, lua.to_luastring("'arg' is not a table")); + let n = lauxlib.luaL_len(L, -1); + lauxlib.luaL_checkstack(L, n+3, lua.to_luastring("too many arguments to script")); + let i; + for (i=1; i<=n; i++) + lapi.lua_rawgeti(L, -i, i); + lapi.lua_remove(L, -i); + return n; +}; + +const handle_script = function(L, argv) { + let fname = argv[0]; + let status; + if (fname === "-" && argv[-1] !== "--") + fname = void 0; /* stdin */ + status = lauxlib.luaL_loadfile(L, lua.to_luastring(fname)); + if (status === lua.thread_status.LUA_OK) { + let n = pushargs(L); /* push arguments to script */ + status = docall(L, n, lua.LUA_MULTRET); + } + return report(L, status); +}; + +const doREPL = function(L) { + for (;;) { + lapi.lua_getglobal(L, _PROMPT); + let input = readlineSync.prompt({ + prompt: lapi.lua_tojsstring(L, -1) || '> ' + }); + lapi.lua_pop(L, 1); + + if (input.length === 0) continue; + + 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; + } + } + while (status === lua.thread_status.LUA_ERRSYNTAX && lapi.lua_tojsstring(L, -1).endsWith("")) { + /* continuation */ + lapi.lua_pop(L, 1); + lapi.lua_getglobal(L, _PROMPT2); + input += "\n" + readlineSync.prompt({ + prompt: lapi.lua_tojsstring(L, -1) || '>> ' + }); + lapi.lua_pop(L, 1); + let buffer = lua.to_luastring(input); + status = lauxlib.luaL_loadbuffer(L, buffer, buffer.length, stdin); + } + if (status === lua.thread_status.LUA_OK) { + status = docall(L, 0, lua.LUA_MULTRET); + } + if (status === lua.thread_status.LUA_OK) { + 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`); + } + } + } else { + report(L, status); + } + lapi.lua_settop(L, 0); /* remove eventual returns */ + } +}; + +if (script < process.argv.length && /* execute main script (if there is one) */ + handle_script(L, process.argv.slice(script)) !== lua.thread_status.LUA_OK) { + /* success */ +} else if (has_i) { + doREPL(L); +} else if (script == process.argv.length && !has_e && !has_v) { /* no arguments? */ + if (process.stdin.isTTY) { /* running in interactive mode? */ + console.log(lua.FENGARI_COPYRIGHT); + doREPL(L); /* do read-eval-print loop */ + } else { + dofile(L, null); } - lapi.lua_settop(L, 0); /* remove eventual returns */ } -- cgit v1.2.3-70-g09d2