moonscript support fixes

2020-06-13 12:19:13 -06:00
parent 1a166bdb22
commit 27c7d2dd18
8 changed files with 17 additions and 195 deletions
--- a/moonscript/moon.works
+++ b/moonscript/moon.works
@@ -0,0 +1,116 @@
+#!/usr/bin/env lua
+local argparse = require("argparse")
+local moonscript = require("moonscript.base")
+local util = require("moonscript.util")
+local errors = require("moonscript.errors")
+local unpack = util.unpack
+local argparser = argparse()({
+	name = "moon"
+})
+argparser:argument("script")
+argparser:argument("args"):args("*")
+argparser:option("-c --coverage", "Collect and print code coverage")
+argparser:option("-d", "Disable stack trace rewriting")
+argparser:option("-v --version", "Print version information")
+local base = 0
+local _list_0 = arg
+for _index_0 = 1, #_list_0 do
+	local flag = _list_0[_index_0]
+	base = base + 1
+	if flag:sub(1, 1) ~= "-" then
+		break
+	end
+end
+local args = {
+	unpack(arg, 1, base)
+}
+local opts = argparser:parse(args)
+local print_err
+print_err = function(...)
+	local msg = table.concat((function(...)
+		local _accum_0 = { }
+		local _len_0 = 1
+		local _list_1 = {
+			...
+		}
+		for _index_0 = 1, #_list_1 do
+			local v = _list_1[_index_0]
+			_accum_0[_len_0] = tostring(v)
+			_len_0 = _len_0 + 1
+		end
+		return _accum_0
+	end)(...), "\t")
+	return io.stderr:write(msg .. "\n")
+end
+local run
+run = function()
+	if opts.version then
+		require("moonscript.version").print_version()
+		os.exit()
+	end
+	local script_fname = shell.resolveProgram(opts.script)
+	args = {
+		unpack(arg, base + 1)
+	}
+	args[-1] = arg[0]
+	args[0] = opts.script
+	local moonscript_chunk, lua_parse_error
+	local passed, err = pcall(function()
+		moonscript_chunk, lua_parse_error = moonscript.loadfile(script_fname, {
+			implicitly_return_root = false
+		})
+	end)
+	if not (passed) then
+		print_err(err)
+		os.exit(1)
+	end
+	if not (moonscript_chunk) then
+		if lua_parse_error then
+			print_err(lua_parse_error)
+		else
+			print_err("Can't file file: " .. tostring(script_fname))
+		end
+		os.exit(1)
+	end
+	util.getfenv(moonscript_chunk).arg = args
+	local run_chunk
+	run_chunk = function()
+		moonscript.insert_loader()
+		moonscript_chunk(unpack(args))
+		return moonscript.remove_loader()
+	end
+	if opts.d then
+		return run_chunk()
+	end
+	local err, trace, cov
+	if opts.coverage then
+		print("starting coverage")
+		local coverage = require("moonscript.cmd.coverage")
+		cov = coverage.CodeCoverage()
+		cov:start()
+	end
+	xpcall(run_chunk, function(_err)
+		err = _err
+		trace = debug.traceback("", 2)
+	end)
+	if err then
+		local truncated = errors.truncate_traceback(util.trim(trace))
+		local rewritten = errors.rewrite_traceback(truncated, err)
+		if rewritten then
+			print_err(rewritten)
+		else
+			print_err(table.concat({
+				err,
+				util.trim(trace)
+			}, "\n"))
+		end
+		return os.exit(1)
+	else
+		if cov then
+			cov:stop()
+			return cov:print_results()
+		end
+	end
+end
+return run()
+-- vim: set filetype=lua: