mirror of
https://github.com/leafo/moonscript.git
synced 2024-11-22 02:44:23 +00:00
117 lines
2.1 KiB
Plaintext
117 lines
2.1 KiB
Plaintext
|
|
alt_getopt = require "alt_getopt"
|
|
|
|
moonscript = require "moonscript.base"
|
|
util = require "moonscript.util"
|
|
errors = require "moonscript.errors"
|
|
|
|
unpack = util.unpack
|
|
|
|
opts, ind = alt_getopt.get_opts arg, "cvhd", {
|
|
version: "v"
|
|
help: "h"
|
|
}
|
|
|
|
help = [=[Usage: %s [options] [script [args]]
|
|
|
|
-h Print this message
|
|
-d Disable stack trace rewriting
|
|
-c Collect and print code coverage
|
|
-v Print version
|
|
]=]
|
|
|
|
|
|
print_err = (...) ->
|
|
msg = table.concat [tostring v for v in *{...}], "\t"
|
|
io.stderr\write msg .. "\n"
|
|
|
|
print_help = (err) ->
|
|
help = help\format arg[0]
|
|
|
|
if err
|
|
print_err err
|
|
print_err help
|
|
else
|
|
print help
|
|
|
|
os.exit!
|
|
|
|
run = ->
|
|
if opts.h
|
|
print_help!
|
|
|
|
if opts.v
|
|
require("moonscript.version").print_version!
|
|
os.exit!
|
|
|
|
script_fname = arg[ind]
|
|
|
|
unless script_fname
|
|
print_help "REPL not yet supported"
|
|
|
|
new_arg = {
|
|
[-1]: arg[0],
|
|
[0]: arg[ind],
|
|
select ind + 1, unpack arg
|
|
}
|
|
|
|
local moonscript_chunk, lua_parse_error
|
|
|
|
passed, err = pcall ->
|
|
moonscript_chunk, lua_parse_error = moonscript.loadfile script_fname, {
|
|
implicitly_return_root: false
|
|
}
|
|
|
|
unless passed
|
|
print_err err
|
|
os.exit 1
|
|
|
|
unless moonscript_chunk
|
|
if lua_parse_error
|
|
print_err lua_parse_error
|
|
else
|
|
print_err "Can't file file: #{script_fname}"
|
|
|
|
os.exit 1
|
|
|
|
util.getfenv(moonscript_chunk).arg = new_arg
|
|
|
|
run_chunk = ->
|
|
moonscript.insert_loader!
|
|
moonscript_chunk unpack new_arg
|
|
moonscript.remove_loader!
|
|
|
|
if opts.d
|
|
return run_chunk!
|
|
|
|
local err, trace, cov
|
|
|
|
if opts.c
|
|
print "starting coverage"
|
|
coverage = require "moonscript.cmd.coverage"
|
|
cov = coverage.CodeCoverage!
|
|
cov\start!
|
|
|
|
xpcall run_chunk, (_err) ->
|
|
err = _err
|
|
trace = debug.traceback "", 2
|
|
|
|
if err
|
|
truncated = errors.truncate_traceback util.trim trace
|
|
rewritten = errors.rewrite_traceback truncated, err
|
|
|
|
if rewritten
|
|
print_err rewritten
|
|
else
|
|
-- failed to rewrite, show original
|
|
print_err table.concat {
|
|
err,
|
|
util.trim trace
|
|
}, "\n"
|
|
else
|
|
if cov
|
|
cov\stop!
|
|
cov\print_results!
|
|
|
|
run!
|