moonscript/bin/moon

#!/usr/bin/env lua

require "alt_getopt"
local moonscript = require "moonscript.base"

local util = require "moonscript.util"
local errors = require "moonscript.errors"

local unpack = util.unpack

-- moonloader and repl
local opts, ind = alt_getopt.get_opts(arg, "cvhd", { version = "v", help = "h" })

local help = [=[Usage: %s [options] [script [args]]

    -h          Print this message
    -d          Disable stack trace rewriting
    -c          Collect and print code coverage
    -v          Print version
]=]

local function print_err(...)
	local msg = table.concat({...}, "\t")
	io.stderr:write(msg .. "\n")
end

local function print_help(err)
	if err then print("Error: "..err) end
	print(help:format(arg[0]))
	os.exit()
end

if opts.h then print_help() end

if opts.v then
	local v = require "moonscript.version"
	v.print_version()
	os.exit()
end

local script_fname = arg[ind]
if not script_fname then
	print_help("repl not yet supported")
	return
end

local new_arg = {
	[-1] = arg[0],
	[0] = arg[ind],
	select(ind + 1, unpack(arg))
}

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 find file: " .. script_fname)
	end
	os.exit(1)
end

util.getfenv(moonscript_chunk).arg = new_arg

local function run_chunk()
	moonscript.insert_loader()
	moonscript_chunk(unpack(new_arg))
	moonscript.remove_loader()
end

if not opts.d then
	local err, trace
	local cov

	if opts.c then
		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
			-- faield to rewrite, show original
			print_err(table.concat({
				err,
				util.trim(trace)
			}, "\n"))
		end
	else
		if cov then
			cov:stop()
		end
	end
else
	run_chunk()
end