Bruce/nomsu
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplifications/streamlining the Make process, and cleaning up some of

Browse Source 
how command line interaction and `use %` work.
This commit is contained in:
Bruce Hill 2018-06-24 16:11:08 -07:00
parent 65addb9aa6
commit ad342b63b7
9 changed files with 308 additions and 261 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

109
Makefile
View File

@ -6,11 +6,7 @@
LUA= lua LUA= lua
LUA_BIN= $(shell which $(LUA)) LUA_BIN= $(shell which $(LUA))
PREFIX=/usr/local PREFIX=
NOMSU_BIN_DIR= $(PREFIX)/bin
NOMSU_LIB_DIR= $(PREFIX)/lib/nomsu
NOMSU_SHARE_DIR= $(PREFIX)/share/nomsu
# ========= You shouldn't need to mess with any of these variables below ================ # ========= You shouldn't need to mess with any of these variables below ================
MOON_FILES= code_obj.moon error_handling.moon files.moon nomsu.moon nomsu_compiler.moon nomsu_tree.moon parser.moon MOON_FILES= code_obj.moon error_handling.moon files.moon nomsu.moon nomsu_compiler.moon nomsu_tree.moon parser.moon
@ -36,25 +32,11 @@ test: build optimize
%.lua: %.nom %.lua: %.nom
@./nomsu_latest -c $< @./nomsu_latest -c $<
.PHONY: check_header .DELETE_ON_ERROR: version
check_header: $(PEG_FILE) nomsu.lua $(CORE_NOM_FILES) $(LIB_NOM_FILES) version: $(LUA_FILES) $(CORE_NOM_FILES) $(LIB_NOM_FILES) $(PEG_FILE)
@if [ -f nomsu_latest ]; then \ @$(LUA_BIN) nomsu.lua --version > version || exit
NOMSU_HEADER="#!$(LUA_BIN)\\nlocal NOMSU_VERSION, NOMSU_LIB, NOMSU_SHARE = [[`$(GET_VERSION)`]], [[$(NOMSU_LIB_DIR)]], [[$(NOMSU_SHARE_DIR)]]"; \
if [ "`head -n 3 nomsu_latest 2>/dev/null`" != "`echo $$NOMSU_HEADER`" ]; then \
rm -f nomsu_latest; \
fi; \
fi;
nomsu_latest: nomsu.lua build: $(LUA_FILES)
@rm -f nomsu_latest
@NOMSU_HEADER="#!$(LUA_BIN)\\nlocal NOMSU_VERSION, NOMSU_LIB, NOMSU_SHARE = [[`$(GET_VERSION)`]], [[$(NOMSU_LIB_DIR)]], [[$(NOMSU_SHARE_DIR)]]"; \
echo $$NOMSU_HEADER | cat - nomsu.lua > nomsu_latest
@chmod +x nomsu_latest
@mv -f nomsu_latest nomsu`$(GET_VERSION)`
@ln -s nomsu`$(GET_VERSION)` nomsu_latest
@echo "\033[1mBuilt nomsu binary\033[0m"
build: $(LUA_FILES) check_header nomsu_latest
.PHONY: optimize .PHONY: optimize
optimize: build $(CORE_LUA_FILES) $(LIB_LUA_FILES) optimize: build $(CORE_LUA_FILES) $(LIB_LUA_FILES)
@ -62,44 +44,71 @@ optimize: build $(CORE_LUA_FILES) $(LIB_LUA_FILES)
.PHONY: clean .PHONY: clean
clean: clean:
@echo "\033[1mDeleting...\033[0m" @echo "\033[1mDeleting...\033[0m"
@rm -rvf nomsu`$(GET_VERSION)` nomsu_latest core/*.lua lib/*.lua @rm -rvf version core/*.lua lib/*.lua
.PHONY: install .PHONY: install
install: all install: build version optimize
@echo "\033[1mNomsu will be installed to:\033[0m" @prefix="$(PREFIX)"; \
@echo " $(NOMSU_BIN_DIR)" if [[ ! $$prefix ]]; then \
@echo " $(NOMSU_LIB_DIR)" read -p $$'\033[1mWhere do you want to install Nomsu? (default: /usr/local) \033[0m' prefix; \
@echo " $(NOMSU_SHARE_DIR)" fi; \
@read -p $$'\033[1mis this okay? [Y/n]\033[0m ' ans; \ if [[ $$prefix != "`realpath $$prefix`" ]]; then \
if [[ ! $$ans =~ ^[Nn] ]]; then \ echo $$'\033[1;31mWarning: '$$prefix$$' is not an absolute path. This may cause problems.\033[0m'; \
mkdir -pv $(NOMSU_BIN_DIR) $(NOMSU_LIB_DIR)/`$(GET_VERSION)` $(NOMSU_SHARE_DIR)/`$(GET_VERSION)` \ read -p $$'\033[1mWould you rather use '`realpath $$prefix`$$' instead? (recommended)[Y/n]\033[0m ' use_real; \
&& cp -v nomsu nomsu`$(GET_VERSION)` $(NOMSU_BIN_DIR) \ if [[ ! $$use_real =~ ^[Nn] ]]; then \
&& cp -rv $(LUA_FILES) $(PEG_FILE) core lib tests $(NOMSU_SHARE_DIR)/`$(GET_VERSION)`; \ prefix="`realpath $$prefix`"; \
fi fi; \
fi; \
version="`cat version`"; \
echo "#!$(LUA_BIN)\\nlocal NOMSU_VERSION, NOMSU_PREFIX = [[$$version]], [[$$prefix]]" | cat - nomsu.lua > nomsu$$version; \
mkdir -pv $$prefix/bin $$prefix/lib/nomsu/$$version $$prefix/share/nomsu/$$version \
&& cp -v nomsu nomsu$$version $$prefix/bin \
&& cp -rv $(LUA_FILES) $(PEG_FILE) core lib tests $$prefix/share/nomsu/$$version;
.PHONY: uninstall .PHONY: uninstall
uninstall: all uninstall: version
@echo "\033[1mNomsu will be uninstalled from:\033[0m" @prefix="$(PREFIX)"; \
@echo " $(NOMSU_BIN_DIR)" if [[ ! $$prefix ]]; then \
@echo " $(NOMSU_LIB_DIR)" read -p $$'\033[1mWhere do you want to uninstall Nomsu from? (default: /usr/local) \033[0m' prefix; \
@echo " $(NOMSU_SHARE_DIR)" fi; \
@read -p $$'\033[1mis this okay? [Y/n]\033[0m ' ans; \ echo "\033[1mNomsu will be uninstalled from:\033[0m"; \
if [[ ! $$ans =~ ^[Nn] ]]; then \ echo " $$prefix/bin"; \
echo " $$prefix/lib"; \
echo " $$prefix/share"; \
read -p $$'\033[1mis this okay? [Y/n]\033[0m ' ans; \
if [[ $$ans =~ ^[Nn] ]]; then exit; fi; \
echo "\033[1mDeleting...\033[0m"; \ echo "\033[1mDeleting...\033[0m"; \
rm -rvf $(NOMSU_LIB_DIR)/`$(GET_VERSION)` $(NOMSU_SHARE_DIR)/`$(GET_VERSION)` $(NOMSU_BIN_DIR)/nomsu`$(GET_VERSION)`; \ version="`cat version`"; \
if [ "`ls $(NOMSU_BIN_DIR)/nomsu* 2> /dev/null`" == "nomsu" ]; then \ rm -rvf $$prefix/lib/nomsu/$$version $$prefix/share/nomsu/$$version $$prefix/bin/nomsu$$version; \
rm -vf $(NOMSU_BIN_DIR)/nomsu; \ if [[ "`find -E $$prefix/bin -type f -regex '.*/nomsu[0-9.]+\$$'`" == "" ]]; then \
rm -vf $$prefix/bin/nomsu; \
else \ else \
if [ "`ls $(NOMSU_BIN_DIR)/nomsu* 2> /dev/null`" != "" ]; then \ if [ -f $$prefix/bin/nomsu ]; then \
read -p $$'\033[1mIt looks like there are other versions of Nomsu installed. Is it okay to leave the "nomsu" cross-version launcher in place? (recommended) [Y/n]\033[0m ' ans; \ read -p $$'\033[1mIt looks like there are other versions of Nomsu installed. Is it okay to leave the "nomsu" cross-version launcher in place? (recommended) [Y/n]\033[0m ' ans; \
if [[ $$ans =~ ^[Nn] ]]; then \ if [[ $$ans =~ ^[Nn] ]]; then \
echo "\033[1mDeleting...\033[0m"; \ echo "\033[1mDeleting...\033[0m"; \
rm -vf $(NOMSU_BIN_DIR)/nomsu; \ rm -vf $$prefix/bin/nomsu; \
fi; \ fi; \
fi; \ fi; \
fi; \ fi; \
if [ "`ls $(NOMSU_LIB_DIR) 2>/dev/null`" == "" ]; then rm -rvf $(NOMSU_LIB_DIR); fi;\ if [ "`ls $$prefix/lib/nomsu 2>/dev/null`" == "" ]; then rm -rvf $$prefix/lib/nomsu; fi;\
if [ "`ls $(NOMSU_SHARE_DIR) 2>/dev/null`" == "" ]; then rm -rvf $(NOMSU_SHARE_DIR); fi;\ if [ "`ls $$prefix/share/nomsu 2>/dev/null`" == "" ]; then rm -rvf $$prefix/share/nomsu; fi;\
fi echo $$'\033[1mDone.\033[0m';
.PHONY: uninstall-all
uninstall-all:
@prefix="$(PREFIX)"; \
if [[ ! $$prefix ]]; then \
read -p $$'\033[1mWhere do you want to uninstall Nomsu from? (default: /usr/local) \033[0m' prefix; \
fi; \
echo "\033[1mEvery version of Nomsu will be uninstalled from:\033[0m"; \
echo " $$prefix/bin"; \
echo " $$prefix/lib"; \
echo " $$prefix/share"; \
read -p $$'\033[1mis this okay? [Y/n]\033[0m ' ans; \
if [[ ! $$ans =~ ^[Nn] ]]; then exit; fi; \
echo "\033[1mDeleting...\033[0m"; \
rm -rvf $$prefix/lib/nomsu $$prefix/share/nomsu $$prefix/bin/nomsu*;\
echo $$'\033[1mDone.\033[0m';
# eof # eof

2
core/metaprogramming.nom
View File

@ -206,7 +206,7 @@ compile [%lua <-write %code, to %lua write %code] to: Lua "\(%lua as lua expr):a
compile [quote %s] to compile [quote %s] to
Lua value ".." Lua value ".."
('"'..\(%s as lua expr):gsub("\\\\", "\\\\\\\\"):gsub("\n","\\\\n"):gsub('"', '\\\\"')..'"') repr(\(%s as lua expr))
compile [type of %obj] to: Lua value "type(\(%obj as lua expr))" compile [type of %obj] to: Lua value "type(\(%obj as lua expr))"
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

2
files.lua
View File

@ -60,7 +60,7 @@ if ok then
end end
elseif file_type == 'directory' then elseif file_type == 'directory' then
for subfile in lfs.dir(filename) do for subfile in lfs.dir(filename) do
if not (subfile == "." or subfile == "..") then if not (subfile == "." or subfile == ".." or not subfile:match("%.nom$")) then
browse(filename .. "/" .. subfile) browse(filename .. "/" .. subfile)
end end
end end

3
files.moon
View File

@ -39,7 +39,8 @@ if ok
return true return true
elseif file_type == 'directory' elseif file_type == 'directory'
for subfile in lfs.dir(filename) for subfile in lfs.dir(filename)
unless subfile == "." or subfile == ".." -- Only include .nom files unless directly specified
unless subfile == "." or subfile == ".." or not subfile\match("%.nom$")
browse(filename.."/"..subfile) browse(filename.."/"..subfile)
return true return true
elseif file_type == 'char device' elseif file_type == 'char device'

12
lib/consolecolor.nom
View File

@ -10,10 +10,8 @@ use "core"
for %name = %colornum in %colors for %name = %colornum in %colors
with {%escapecode: "\27[\(%colornum)m"} with {%escapecode: "\27[\(%colornum)m"}
lua> ".." run ".."
nomsu.COMPILE_ACTIONS[\%name] = function(nomsu, tree) compile [\%name] to: Lua value (quote \(quote %escapecode))
return LuaCode.Value(tree.source, repr(\%escapecode)) compile [\%name %text] to
end Lua value ".."
nomsu.COMPILE_ACTIONS[\%name.." %"] = function(nomsu, tree, text) (\\(quote \(quote %escapecode))..\\(%text as lua expr)..\(quote "\27[0m"))
return LuaCode.Value(tree.source, repr(\%escapecode), "..", nomsu:compile(text), "..'\\\\27[0m'")
end

109
nomsu.lua
View File

@ -117,7 +117,8 @@ local files = require("files")
local nomsu = NomsuCompiler local nomsu = NomsuCompiler
nomsu.arg = args.nomsu_args nomsu.arg = args.nomsu_args
if args.version then if args.version then
nomsu:run('use "core"\nsay (Nomsu version)') nomsu:run([[use "core"
say (Nomsu version)]])
os.exit(EXIT_SUCCESS) os.exit(EXIT_SUCCESS)
end end
FILE_CACHE = setmetatable({ }, { FILE_CACHE = setmetatable({ }, {
@ -192,54 +193,94 @@ run = function()
end end
return true return true
end end
if args.compile or args.verbose then
nomsu.on_compile = function(code, from_file)
if not (to_run[from_file]) then
return
end
if args.verbose then
io.write(tostring(code), "\n")
end
if args.compile and from_file:match("%.nom$") then
local output_filename = from_file:gsub("%.nom$", ".lua")
local output_file = io.open(output_filename, 'w')
output_file:write(tostring(code))
output_file:flush()
print(("Compiled %-25s -> %s"):format(from_file, output_filename))
return output_file:close()
end
end
end
local parse_errs = { } local parse_errs = { }
for _index_0 = 1, #input_files do local _list_1 = args.inputs
local filename = input_files[_index_0] for _index_0 = 1, #_list_1 do
local arg = _list_1[_index_0]
for filename in files.walk(arg) do
local _continue_0 = false
repeat
local file, source
if filename == STDIN then
file = io.input():read("*a")
files.spoof('stdin', file)
source = Source('stdin', 1, #file)
elseif filename:match("%.nom$") then
file = files.read(filename)
if not file then
error("File does not exist: " .. tostring(filename), 0)
end
source = Source(filename, 1, #file)
else
_continue_0 = true
break
end
source = Source(filename, 1, #file)
local output
if args.compile then
output = io.open(filename:gsub("%.nom$", ".lua"), "w")
else
output = nil
end
if args.syntax then if args.syntax then
local file_contents = io.open(filename):read('*a')
local err local err
ok, err = pcall(nomsu.parse, nomsu, file_contents, Source(filename, 1, #file_contents)) ok, err = pcall(nomsu.parse, nomsu, file, source)
if not ok then if not ok then
table.insert(parse_errs, err) table.insert(parse_errs, err)
elseif print_file then elseif print_file then
print_file:write("Parse succeeded: " .. tostring(filename) .. "\n") print_file:write("Parse succeeded: " .. tostring(filename) .. "\n")
print_file:flush() print_file:flush()
end end
elseif args.format then _continue_0 = true
local file = files.read(filename) break
if not file then
error("File does not exist: " .. tostring(filename), 0)
end end
local tree = nomsu:parse(file, Source(filename, 1, #file)) local tree = nomsu:parse(file, source)
local formatted = tostring(nomsu:tree_to_nomsu(tree)) if args.format then
local formatted = tree and tostring(nomsu:tree_to_nomsu(tree)) or ""
if print_file then if print_file then
print_file:write(formatted, "\n") print_file:write(formatted, "\n")
print_file:flush() print_file:flush()
end end
elseif filename == STDIN then _continue_0 = true
local file = io.input():read("*a") break
files.spoof('stdin', file) end
nomsu:run(file, Source('stdin', 1, #file)) if tree then
if tree.type == "FileChunks" then
for _index_1 = 1, #tree do
local chunk = tree[_index_1]
local lua = nomsu:compile(chunk):as_statements("return ")
lua:declare_locals()
lua:prepend("-- File: " .. tostring(source.filename:gsub("\n.*", "...")) .. "\n")
if args.compile then
output:write(tostring(lua), "\n")
end
if args.verbose then
print(tostring(lua))
end
nomsu:run_lua(lua)
end
else else
nomsu:run_file(filename) local lua = nomsu:compile(tree):as_statements("return ")
lua:declare_locals()
lua:prepend("-- File: " .. tostring(source.filename:gsub("\n.*", "...")) .. "\n")
if args.compile then
output:write(tostring(lua), "\n")
end
if args.verbose then
print(tostring(lua))
end
nomsu:run_lua(lua)
end
end
if args.compile then
print(("Compiled %-25s -> %s"):format(filename, filename:gsub("%.nom$", ".lua")))
output:close()
end
_continue_0 = true
until true
if not _continue_0 then
break
end
end end
end end
if #parse_errs > 0 then if #parse_errs > 0 then

82
nomsu.moon
View File

@ -70,7 +70,9 @@ nomsu = NomsuCompiler
nomsu.arg = args.nomsu_args nomsu.arg = args.nomsu_args
if args.version if args.version
nomsu\run 'use "core"\nsay (Nomsu version)' nomsu\run [[
use "core"
say (Nomsu version)]]
os.exit(EXIT_SUCCESS) os.exit(EXIT_SUCCESS)
export FILE_CACHE export FILE_CACHE
@ -125,46 +127,70 @@ run = ->
return false if to_run[f] return false if to_run[f]
return true return true
if args.compile or args.verbose
nomsu.on_compile = (code, from_file)->
return unless to_run[from_file]
if args.verbose
io.write(tostring(code), "\n")
if args.compile and from_file\match("%.nom$")
output_filename = from_file\gsub("%.nom$", ".lua")
output_file = io.open(output_filename, 'w')
output_file\write(tostring(code))
output_file\flush!
print ("Compiled %-25s -> %s")\format(from_file, output_filename)
output_file\close!
parse_errs = {} parse_errs = {}
for filename in *input_files for arg in *args.inputs
for filename in files.walk(arg)
local file, source
if filename == STDIN
file = io.input!\read("*a")
files.spoof('stdin', file)
source = Source('stdin', 1, #file)
elseif filename\match("%.nom$")
file = files.read(filename)
if not file
error("File does not exist: #{filename}", 0)
source = Source(filename, 1, #file)
else continue
source = Source(filename,1,#file)
output = if args.compile then io.open(filename\gsub("%.nom$", ".lua"), "w") else nil
if args.syntax if args.syntax
-- Check syntax: -- Check syntax:
file_contents = io.open(filename)\read('*a') ok,err = pcall nomsu.parse, nomsu, file, source
ok,err = pcall nomsu.parse, nomsu, file_contents, Source(filename, 1, #file_contents)
if not ok if not ok
table.insert parse_errs, err table.insert parse_errs, err
elseif print_file elseif print_file
print_file\write("Parse succeeded: #{filename}\n") print_file\write("Parse succeeded: #{filename}\n")
print_file\flush! print_file\flush!
elseif args.format continue
tree = nomsu\parse(file, source)
if args.format
-- Auto-format -- Auto-format
file = files.read(filename) formatted = tree and tostring(nomsu\tree_to_nomsu(tree)) or ""
if not file
error("File does not exist: #{filename}", 0)
tree = nomsu\parse(file, Source(filename,1,#file))
formatted = tostring(nomsu\tree_to_nomsu(tree))
if print_file if print_file
print_file\write(formatted, "\n") print_file\write(formatted, "\n")
print_file\flush! print_file\flush!
elseif filename == STDIN continue
file = io.input!\read("*a")
files.spoof('stdin', file) if tree
nomsu\run(file, Source('stdin',1,#file)) if tree.type == "FileChunks"
-- Each chunk's compilation is affected by the code in the previous chunks
-- (typically), so each chunk needs to compile and run before the next one
-- compiles.
for chunk in *tree
lua = nomsu\compile(chunk)\as_statements("return ")
lua\declare_locals!
lua\prepend "-- File: #{source.filename\gsub("\n.*", "...")}\n"
if args.compile
output\write(tostring(lua), "\n")
if args.verbose
print(tostring(lua))
nomsu\run_lua(lua)
else else
nomsu\run_file(filename) lua = nomsu\compile(tree)\as_statements("return ")
lua\declare_locals!
lua\prepend "-- File: #{source.filename\gsub("\n.*", "...")}\n"
if args.compile
output\write(tostring(lua), "\n")
if args.verbose
print(tostring(lua))
nomsu\run_lua(lua)
if args.compile
print ("Compiled %-25s -> %s")\format(filename, filename\gsub("%.nom$", ".lua"))
output\close!
if #parse_errs > 0 if #parse_errs > 0
io.stderr\write table.concat(parse_errs, "\n\n") io.stderr\write table.concat(parse_errs, "\n\n")

39
nomsu_compiler.lua
View File

@ -365,12 +365,13 @@ do
return add_lua_bits(self, LuaCode.Value(tree.source), _code) return add_lua_bits(self, LuaCode.Value(tree.source), _code)
end, end,
["use %"] = function(self, tree, _path) ["use %"] = function(self, tree, _path)
if not (_path.type == 'Text' and #_path == 1 and type(_path[1]) == 'string') then if _path.type == 'Text' and #_path == 1 and type(_path[1]) == 'string' then
return LuaCode(tree.source, "nomsu:run_file(" .. tostring(self:compile(_path)) .. ");")
end
local path = _path[1] local path = _path[1]
self:run_file(path) for f in files.walk(path) do
return LuaCode(tree.source, "nomsu:run_file(" .. tostring(repr(path)) .. ");") self:run_file(f)
end
end
return LuaCode(tree.source, "for f in files.walk(", self:compile(_path), ") do nomsu:run_file(f) end")
end end
}, { }, {
__index = function(self, stub) __index = function(self, stub)
@ -407,17 +408,11 @@ do
insert(all_lua, tostring(lua)) insert(all_lua, tostring(lua))
ret = self:run_lua(lua) ret = self:run_lua(lua)
end end
if self.on_compile then
self.on_compile(concat(all_lua, "\n"), source.filename)
end
return ret return ret
else else
local lua = self:compile(tree, compile_actions):as_statements("return ") local lua = self:compile(tree):as_statements("return ")
lua:declare_locals() lua:declare_locals()
lua:prepend("-- File: " .. tostring(source.filename:gsub("\n.*", "...")) .. "\n") lua:prepend("-- File: " .. tostring(source.filename:gsub("\n.*", "...")) .. "\n")
if self.on_compile then
self.on_compile(lua, source.filename)
end
return self:run_lua(lua) return self:run_lua(lua)
end end
end end
@ -426,17 +421,6 @@ do
if self.LOADED[filename] then if self.LOADED[filename] then
return self.LOADED[filename] return self.LOADED[filename]
end end
local ret = nil
for filename in files.walk(filename) do
local _continue_0 = false
repeat
do
ret = self.LOADED[filename]
if ret then
_continue_0 = true
break
end
end
for i, running in ipairs(_running_files) do for i, running in ipairs(_running_files) do
if running == filename then if running == filename then
local loop local loop
@ -454,6 +438,7 @@ do
end end
end end
insert(_running_files, filename) insert(_running_files, filename)
local ret = nil
if match(filename, "%.lua$") then if match(filename, "%.lua$") then
local file = assert(files.read(filename), "Could not find file: " .. tostring(filename)) local file = assert(files.read(filename), "Could not find file: " .. tostring(filename))
ret = self:run_lua(file, Source(filename, 1, #file)) ret = self:run_lua(file, Source(filename, 1, #file))
@ -470,7 +455,7 @@ do
end end
end end
if not (ran_lua) then if not (ran_lua) then
local file = file or files.read(filename) local file = files.read(filename)
if not file then if not file then
error("File does not exist: " .. tostring(filename), 0) error("File does not exist: " .. tostring(filename), 0)
end end
@ -481,12 +466,6 @@ do
end end
self.LOADED[filename] = ret or true self.LOADED[filename] = ret or true
remove(_running_files) remove(_running_files)
_continue_0 = true
until true
if not _continue_0 then
break
end
end
self.LOADED[filename] = ret or true self.LOADED[filename] = ret or true
return ret return ret
end end

23
nomsu_compiler.moon
View File

@ -246,11 +246,11 @@ with NomsuCompiler
return add_lua_bits(@, LuaCode.Value(tree.source), _code) return add_lua_bits(@, LuaCode.Value(tree.source), _code)
["use %"]: (tree, _path)=> ["use %"]: (tree, _path)=>
unless _path.type == 'Text' and #_path == 1 and type(_path[1]) == 'string' if _path.type == 'Text' and #_path == 1 and type(_path[1]) == 'string'
return LuaCode(tree.source, "nomsu:run_file(#{@compile(_path)});")
path = _path[1] path = _path[1]
@run_file(path) for f in files.walk(path)
return LuaCode(tree.source, "nomsu:run_file(#{repr path});") @run_file(f)
return LuaCode(tree.source, "for f in files.walk(", @compile(_path), ") do nomsu:run_file(f) end")
}, { }, {
__index: (stub)=> __index: (stub)=>
if math_expression\match(stub) if math_expression\match(stub)
@ -275,27 +275,19 @@ with NomsuCompiler
lua\prepend "-- File: #{source.filename\gsub("\n.*", "...")}\n" lua\prepend "-- File: #{source.filename\gsub("\n.*", "...")}\n"
insert all_lua, tostring(lua) insert all_lua, tostring(lua)
ret = @run_lua(lua) ret = @run_lua(lua)
if @on_compile
self.on_compile(concat(all_lua, "\n"), source.filename)
return ret return ret
else else
lua = @compile(tree, compile_actions)\as_statements("return ") lua = @compile(tree)\as_statements("return ")
lua\declare_locals! lua\declare_locals!
lua\prepend "-- File: #{source.filename\gsub("\n.*", "...")}\n" lua\prepend "-- File: #{source.filename\gsub("\n.*", "...")}\n"
if @on_compile
self.on_compile(lua, source.filename)
return @run_lua(lua) return @run_lua(lua)
_running_files = {} -- For detecting circular imports _running_files = {} -- For detecting circular imports
.run_file = (filename)=> .run_file = (filename)=>
if @LOADED[filename] if @LOADED[filename]
return @LOADED[filename] return @LOADED[filename]
ret = nil
for filename in files.walk(filename)
if ret = @LOADED[filename]
continue
-- Check for circular import -- Check for circular import
-- TODO: optimize?
for i,running in ipairs _running_files for i,running in ipairs _running_files
if running == filename if running == filename
loop = [_running_files[j] for j=i,#_running_files] loop = [_running_files[j] for j=i,#_running_files]
@ -303,6 +295,7 @@ with NomsuCompiler
error("Circular import, this loops forever: #{concat loop, " -> "}...") error("Circular import, this loops forever: #{concat loop, " -> "}...")
insert _running_files, filename insert _running_files, filename
ret = nil
if match(filename, "%.lua$") if match(filename, "%.lua$")
file = assert(files.read(filename), "Could not find file: #{filename}") file = assert(files.read(filename), "Could not find file: #{filename}")
ret = @run_lua file, Source(filename, 1, #file) ret = @run_lua file, Source(filename, 1, #file)
@ -313,7 +306,7 @@ with NomsuCompiler
ret = @run_lua file, Source(lua_filename, 1, #file) ret = @run_lua file, Source(lua_filename, 1, #file)
true true
unless ran_lua unless ran_lua
file = file or files.read(filename) file = files.read(filename)
if not file if not file
error("File does not exist: #{filename}", 0) error("File does not exist: #{filename}", 0)
ret = @run file, Source(filename,1,#file) ret = @run file, Source(filename,1,#file)