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

No longer passing tree to every compile action. Now, you can just

Browse Source 
return a LuaCode object, and it will automatically get a source from
`tree` if it didn't already have a source. Plus some fixes/cleanup.
This commit is contained in:
Bruce Hill 2018-11-09 16:40:36 -08:00
parent a2f07415c5
commit 69aaea030e
22 changed files with 286 additions and 286 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
code_obj.lua
View File

@ -110,9 +110,23 @@ do
return self:text() return self:text()
end, end,
as_lua = function(self) as_lua = function(self)
return tostring(self.__class.__name) .. "(" .. tostring(concat({ if self.source then
tostring(self.source):as_lua(), return tostring(self.__class.__name) .. ":from(" .. tostring(concat({
unpack((function() tostring(self.source):as_lua(),
unpack((function()
local _accum_0 = { }
local _len_0 = 1
local _list_0 = self.bits
for _index_0 = 1, #_list_0 do
local b = _list_0[_index_0]
_accum_0[_len_0] = b:as_lua()
_len_0 = _len_0 + 1
end
return _accum_0
end)())
}, ", ")) .. ")"
else
return tostring(self.__class.__name) .. "(" .. tostring(concat((function()
local _accum_0 = { } local _accum_0 = { }
local _len_0 = 1 local _len_0 = 1
local _list_0 = self.bits local _list_0 = self.bits
@ -122,8 +136,8 @@ do
_len_0 = _len_0 + 1 _len_0 = _len_0 + 1
end end
return _accum_0 return _accum_0
end)()) end)(), ", ")) .. ")"
}, ", ")) .. ")" end
end, end,
__len = function(self) __len = function(self)
return #self:text() return #self:text()
@ -246,12 +260,8 @@ do
} }
_base_0.__index = _base_0 _base_0.__index = _base_0
_class_0 = setmetatable({ _class_0 = setmetatable({
__init = function(self, source, ...) __init = function(self, ...)
self.source = source
self.bits = { } self.bits = { }
if type(self.source) == 'string' then
self.source = Source:from_string(self.source)
end
return self:append(...) return self:append(...)
end, end,
__base = _base_0, __base = _base_0,
@ -266,6 +276,14 @@ do
}) })
_base_0.__class = _class_0 _base_0.__class = _class_0
local self = _class_0 local self = _class_0
self.from = function(self, source, ...)
local inst = self(...)
if type(source) == 'string' then
source = Source:from_string(source)
end
inst.source = source
return inst
end
self.is_instance = function(self, x) self.is_instance = function(self, x)
return type(x) == 'table' and x.__class == self return type(x) == 'table' and x.__class == self
end end
@ -370,27 +388,8 @@ do
end end
return to_declare return to_declare
end, end,
as_statements = function(self, prefix, suffix)
if prefix == nil then
prefix = ""
end
if suffix == nil then
suffix = ";"
end
if self:text():matches(";$") or self:text() == "" then
return self
end
local statements = LuaCode(self.source)
if prefix ~= "" then
statements:append(prefix)
end
statements:append(self)
if suffix ~= "" then
statements:append(suffix)
end
return statements
end,
make_offset_table = function(self) make_offset_table = function(self)
assert(self.source, "This code doesn't have a source")
local lua_to_nomsu, nomsu_to_lua = { }, { } local lua_to_nomsu, nomsu_to_lua = { }, { }
local walk local walk
walk = function(lua, pos) walk = function(lua, pos)

29
code_obj.moon
View File

@ -43,13 +43,17 @@ class Source
class Code class Code
is_code: true is_code: true
new: (@source, ...)=> new: (...)=>
@bits = {} @bits = {}
if type(@source) == 'string'
@source = Source\from_string(@source)
--assert(@source and Source\is_instance(@source), "Source has the wrong type")
@append(...) @append(...)
@from: (source, ...)=>
inst = self(...)
if type(source) == 'string'
source = Source\from_string(source)
inst.source = source
return inst
@is_instance: (x)=> type(x) == 'table' and x.__class == @ @is_instance: (x)=> type(x) == 'table' and x.__class == @
text: => text: =>
@ -71,7 +75,10 @@ class Code
__tostring: => @text! __tostring: => @text!
as_lua: => as_lua: =>
"#{@__class.__name}(#{concat {tostring(@source)\as_lua!, unpack([b\as_lua! for b in *@bits])}, ", "})" if @source
"#{@__class.__name}:from(#{concat {tostring(@source)\as_lua!, unpack([b\as_lua! for b in *@bits])}, ", "})"
else
"#{@__class.__name}(#{concat [b\as_lua! for b in *@bits], ", "})"
__len: => #@text! __len: => #@text!
@ -214,18 +221,8 @@ class LuaCode extends Code
@prepend "local #{concat to_declare, ", "};\n" @prepend "local #{concat to_declare, ", "};\n"
return to_declare return to_declare
as_statements: (prefix="", suffix=";")=>
if @text!\matches(";$") or @text! == ""
return self
statements = LuaCode(@source)
if prefix != ""
statements\append prefix
statements\append self
if suffix != ""
statements\append suffix
return statements
make_offset_table: => make_offset_table: =>
assert @source, "This code doesn't have a source"
-- Return a mapping from output (lua) character number to input (nomsu) character number -- Return a mapping from output (lua) character number to input (nomsu) character number
lua_to_nomsu, nomsu_to_lua = {}, {} lua_to_nomsu, nomsu_to_lua = {}, {}
walk = (lua, pos)-> walk = (lua, pos)->

4
compatibility/compatibility.nom
View File

@ -79,13 +79,13 @@ externally [..]
if ((%ver as list) > (%end_version as list)): stop %ver if ((%ver as list) > (%end_version as list)): stop %ver
if %ACTION_UPGRADES.%ver: if %ACTION_UPGRADES.%ver:
%tree = (..) %tree = (..)
%tree with % -> (..) %tree with % ->:
if ((% is "Action" syntax tree) and %ACTION_UPGRADES.%ver.(%.stub)): if ((% is "Action" syntax tree) and %ACTION_UPGRADES.%ver.(%.stub)):
%with_upgraded_args = (..) %with_upgraded_args = (..)
%k = (%v upgraded from %start_version to %end_version) for %k = %v in % %k = (%v upgraded from %start_version to %end_version) for %k = %v in %
set %with_upgraded_args 's metatable to (% 's metatable) set %with_upgraded_args 's metatable to (% 's metatable)
return (..) return (..)
call %ACTION_UPGRADES.%ver.(%.stub) with [%with_upgraded_args, %end_version] %ACTION_UPGRADES.%ver.(%.stub) %with_upgraded_args %end_version
if %UPGRADES.%ver: if %UPGRADES.%ver:
%with_upgraded_args = (..) %with_upgraded_args = (..)

20
containers.lua
View File

@ -210,19 +210,21 @@ local _list_mt = {
end, end,
from_1_to = function(self, start, stop) from_1_to = function(self, start, stop)
local n = #self local n = #self
if n < 0 then if start < 0 then
start = (n + 1 - start) start = (n + 1 - start)
end end
if n < 0 then if stop < 0 then
stop = (n + 1 - stop) stop = (n + 1 - stop)
end end
local _accum_0 = { } return List((function()
local _len_0 = 1 local _accum_0 = { }
for i = start, stop do local _len_0 = 1
_accum_0[_len_0] = self[i] for i = start, stop do
_len_0 = _len_0 + 1 _accum_0[_len_0] = self[i]
end _len_0 = _len_0 + 1
return _accum_0 end
return _accum_0
end)())
end end
}, },
__newindex = function(self, k, v) __newindex = function(self, k, v)

6
containers.moon
View File

@ -85,9 +85,9 @@ _list_mt =
return nil return nil
from_1_to: (start, stop)=> from_1_to: (start, stop)=>
n = #@ n = #@
start = (n+1-start) if n < 0 start = (n+1-start) if start < 0
stop = (n+1-stop) if n < 0 stop = (n+1-stop) if stop < 0
return [@[i] for i=start,stop] return List[@[i] for i=start,stop]
-- TODO: remove this safety check to get better performance? -- TODO: remove this safety check to get better performance?
__newindex: (k,v)=> __newindex: (k,v)=>
assert type(k) == 'number', "List indices must be numbers" assert type(k) == 'number', "List indices must be numbers"

1
core/control_flow.nom
View File

@ -4,7 +4,6 @@
like "if" statements and loops. like "if" statements and loops.
use "core/metaprogramming.nom" use "core/metaprogramming.nom"
use "core/text.nom"
use "core/operators.nom" use "core/operators.nom"
use "core/errors.nom" use "core/errors.nom"

8
core/io.nom
View File

@ -9,17 +9,17 @@ use "core/metaprogramming.nom"
(say %message) compiles to: (say %message) compiles to:
lua> "\ lua> "\
..if \%message.type == "Text" then ..if \%message.type == "Text" then
return LuaCode(tree.source, "print(", \(%message as lua expr), ");"); return LuaCode("print(", \(%message as lua expr), ");");
else else
return LuaCode(tree.source, "print(tostring(", \(%message as lua expr), "));"); return LuaCode("print(tostring(", \(%message as lua expr), "));");
end" end"
(ask %prompt) compiles to: (ask %prompt) compiles to:
lua> "\ lua> "\
..if \%prompt.type == "Text" then ..if \%prompt.type == "Text" then
return LuaCode(tree.source, "(io.write(", \(%prompt as lua expr), ") and io.read())"); return LuaCode("(io.write(", \(%prompt as lua expr), ") and io.read())");
else else
return LuaCode(tree.source, "(io.write(tostring(", \(..) return LuaCode("(io.write(tostring(", \(..)
%prompt as lua expr %prompt as lua expr
.., ")) and io.read())"); .., ")) and io.read())");
end" end"

8
core/math.nom
View File

@ -85,7 +85,7 @@ externally [all of %items, all %items] all mean:
return (yes) return (yes)
[all of %items, all %items] all compile to: [all of %items, all %items] all compile to:
unless (%items.type is "List"): unless (%items.type is "List"):
return %tree return \(all of %items)
if ((size of %items) == 0): return (Lua "true") if ((size of %items) == 0): return (Lua "true")
%lua = (Lua "(") %lua = (Lua "(")
%lua::add ((% as lua expr) for % in %items) joined with " and " %lua::add ((% as lua expr) for % in %items) joined with " and "
@ -99,7 +99,7 @@ externally [any of %items, any %items] all mean:
return (no) return (no)
[any of %items, any %items] all compile to: [any of %items, any %items] all compile to:
unless (%items.type is "List"): unless (%items.type is "List"):
return %tree return \(any of %items)
if ((size of %items) == 0): return (Lua "false") if ((size of %items) == 0): return (Lua "false")
%lua = (Lua "(") %lua = (Lua "(")
%lua::add ((% as lua expr) for % in %items) joined with " or " %lua::add ((% as lua expr) for % in %items) joined with " or "
@ -114,7 +114,7 @@ externally [sum of %items, sum %items] all mean:
return %total return %total
[sum of %items, sum %items] all compile to: [sum of %items, sum %items] all compile to:
unless (%items.type is "List"): unless (%items.type is "List"):
return %tree return \(sum of %items)
if ((size of %items) == 0): return (Lua "0") if ((size of %items) == 0): return (Lua "0")
%lua = (Lua "(") %lua = (Lua "(")
%lua::add ((% as lua expr) for % in %items) joined with " + " %lua::add ((% as lua expr) for % in %items) joined with " + "
@ -127,7 +127,7 @@ externally [product of %items, product %items] all mean:
return %prod return %prod
[product of %items, product %items] all compile to: [product of %items, product %items] all compile to:
unless (%items.type is "List"): unless (%items.type is "List"):
return %tree return \(product of %items)
if ((size of %items) == 0): return (Lua "1") if ((size of %items) == 0): return (Lua "1")
%lua = (Lua "(") %lua = (Lua "(")
%lua::add ((% as lua expr) for % in %items) joined with " * " %lua::add ((% as lua expr) for % in %items) joined with " * "

33
core/metaprogramming.nom
View File

@ -17,13 +17,13 @@ lua> "\
end end
end end
end end
compile.action["define mangler"] = function(compile, tree) compile.action["define mangler"] = function(compile)
return LuaCode(tree.source, "local mangle = mangler()") return LuaCode("local mangle = mangler()")
end" end"
lua> "\ lua> "\
..compile.action["1 ->"] = function(compile, tree, \%args, \%body) ..compile.action["1 ->"] = function(compile, \%args, \%body)
local lua = LuaCode(tree.source, "(function(") local lua = LuaCode("(function(")
if SyntaxTree:is_instance(\%args) and \%args.type == "Action" then \%args = \%args:get_args() end if SyntaxTree:is_instance(\%args) and \%args.type == "Action" then \%args = \%args:get_args() end
local lua_args = table.map(\%args, function(a) return SyntaxTree:is_instance(a) and compile(a):text() or a end) local lua_args = table.map(\%args, function(a) return SyntaxTree:is_instance(a) and compile(a):text() or a end)
lua:concat_append(lua_args, ", ") lua:concat_append(lua_args, ", ")
@ -36,11 +36,10 @@ lua> "\
end" end"
lua> "\ lua> "\
..compile.action["what 1 compiles to"] = function(compile, tree, \%action) ..compile.action["what 1 compiles to"] = function(compile, \%action)
local lua = LuaCode(tree.source, "compile.action[", \%action.stub:as_lua(), "](") local lua = LuaCode("compile.action[", \%action.stub:as_lua(), "](")
local lua_args = table.map(\%action:get_args(), function(a) return compile(a) end) local lua_args = table.map(\%action:get_args(), function(a) return compile(a) end)
table.insert(lua_args, 1, "compile") table.insert(lua_args, 1, "compile")
table.insert(lua_args, 2, "tree")
lua:concat_append(lua_args, ", ") lua:concat_append(lua_args, ", ")
lua:append(")") lua:append(")")
return lua return lua
@ -66,12 +65,12 @@ test:
asdf asdf
assume (%tmp is (nil)) or barf "compile to is leaking variables" assume (%tmp is (nil)) or barf "compile to is leaking variables"
lua> "\ lua> "\
..compile.action["1 compiles to"] = function(compile, tree, \%action, \%body) ..compile.action["1 compiles to"] = function(compile, \%action, \%body)
local \%args = List{\(\%compile), \(\%tree), unpack(\%action:get_args())} local \%args = List{\(\%compile), unpack(\%action:get_args())}
if \%body.type == "Text" then if \%body.type == "Text" then
\%body = SyntaxTree{source=\%body.source, type="Action", "Lua", \%body} \%body = SyntaxTree{source=\%body.source, type="Action", "Lua", \%body}
end end
return LuaCode(tree.source, "compile.action[", \%action.stub:as_lua(), return LuaCode("compile.action[", \%action.stub:as_lua(),
"] = ", \(what (%args -> %body) compiles to)) "] = ", \(what (%args -> %body) compiles to))
end" end"
@ -82,11 +81,11 @@ lua> "\
..if \%actions.type ~= "List" then ..if \%actions.type ~= "List" then
compile_error(\%actions, "This should be a list of actions.") compile_error(\%actions, "This should be a list of actions.")
end end
local lua = LuaCode(tree.source, \(what (%actions.1 compiles to %body) compiles to)) local lua = \(what (%actions.1 compiles to %body) compiles to)
local \%args = List{\(\%compile), \(\%tree), unpack(\%actions[1]:get_args())} local \%args = List{\(\%compile), unpack(\%actions[1]:get_args())}
for i=2,#\%actions do for i=2,#\%actions do
local alias = \%actions[i] local alias = \%actions[i]
local \%alias_args = List{\(\%compile), \(\%tree), unpack(alias:get_args())} local \%alias_args = List{\(\%compile), unpack(alias:get_args())}
lua:append("\\ncompile.action[", alias.stub:as_lua(), "] = ") lua:append("\\ncompile.action[", alias.stub:as_lua(), "] = ")
if \%alias_args == \%args then if \%alias_args == \%args then
lua:append("compile.action[", \%actions[1].stub:as_lua(), "]") lua:append("compile.action[", \%actions[1].stub:as_lua(), "]")
@ -100,7 +99,7 @@ lua> "\
(call %fn with %args) compiles to: (call %fn with %args) compiles to:
lua> "\ lua> "\
..local lua = LuaCode(tree.source, compile(\%fn), "(") ..local lua = LuaCode(compile(\%fn), "(")
if \%args.type == 'List' then if \%args.type == 'List' then
lua:concat_append(table.map(\%args, function(a) return compile(a) end), ", ") lua:concat_append(table.map(\%args, function(a) return compile(a) end), ", ")
else else
@ -126,14 +125,14 @@ test:
lua> "\ lua> "\
..local fn_name = \%action.stub:as_lua_id() ..local fn_name = \%action.stub:as_lua_id()
local \%args = \%action:get_args() local \%args = \%action:get_args()
local lua = LuaCode(tree.source, fn_name, " = ", \(what (%args -> %body) compiles to)) local lua = LuaCode(fn_name, " = ", \(what (%args -> %body) compiles to))
lua:add_free_vars({fn_name}) lua:add_free_vars({fn_name})
return lua" return lua"
(%actions all mean %body) compiles to: (%actions all mean %body) compiles to:
lua> "\ lua> "\
..local fn_name = \%actions[1].stub:as_lua_id() ..local fn_name = \%actions[1].stub:as_lua_id()
local \%args = List(\%actions[1]:get_args()) local \%args = List(\%actions[1]:get_args())
local lua = LuaCode(tree.source, \(what (%actions.1 means %body) compiles to)) local lua = \(what (%actions.1 means %body) compiles to)
for i=2,#\%actions do for i=2,#\%actions do
local alias = \%actions[i] local alias = \%actions[i]
local alias_name = alias.stub:as_lua_id() local alias_name = alias.stub:as_lua_id()
@ -222,7 +221,7 @@ test:
return t:as_lua() return t:as_lua()
end end
end end
local \%new_body = LuaCode(\%body.source, local \%new_body = LuaCode:from(\%body.source,
"local mangle = mangler()", "local mangle = mangler()",
"\\nreturn ", make_tree(\%body)) "\\nreturn ", make_tree(\%body))
local ret = \(what (%actions all compile to %new_body) compiles to) local ret = \(what (%actions all compile to %new_body) compiles to)

11
core/operators.nom
View File

@ -31,7 +31,7 @@ test:
lua> "\ lua> "\
..local \%var_lua = \(%var as lua expr) ..local \%var_lua = \(%var as lua expr)
local \%value_lua = \(%value as lua expr) local \%value_lua = \(%value as lua expr)
local lua = LuaCode(tree.source, \%var_lua, ' = ', \%value_lua, ';') local lua = LuaCode(\%var_lua, ' = ', \%value_lua, ';')
if \%var.type == 'Var' then if \%var.type == 'Var' then
lua:add_free_vars({compile(\%var):text()}) lua:add_free_vars({compile(\%var):text()})
end end
@ -49,7 +49,7 @@ test:
assume (%assignments.type is "Dict") or barf "\ assume (%assignments.type is "Dict") or barf "\
..Expected a Dict for the assignments part of '<- %' statement, not \%assignments" ..Expected a Dict for the assignments part of '<- %' statement, not \%assignments"
lua> "\ lua> "\
..local lhs, rhs = LuaCode(tree.source), LuaCode(tree.source) ..local lhs, rhs = LuaCode(), LuaCode()
for i, item in ipairs(\%assignments) do for i, item in ipairs(\%assignments) do
local \%target, \%value = item[1], item[2] local \%target, \%value = item[1], item[2]
\%value = \%value:map(function(t) \%value = \%value:map(function(t)
@ -69,7 +69,7 @@ test:
lhs:append(target_lua) lhs:append(target_lua)
rhs:append(value_lua) rhs:append(value_lua)
end end
return LuaCode(tree.source, lhs, " = ", rhs, ";")" return LuaCode(lhs, " = ", rhs, ";")"
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@ -111,7 +111,7 @@ test:
(with %assignments %body) compiles to: (with %assignments %body) compiles to:
%lua = (%body as lua) %lua = (%body as lua)
lua> "\ lua> "\
..local lhs, rhs = LuaCode(tree.source), LuaCode(tree.source) ..local lhs, rhs = LuaCode(), LuaCode()
local vars = {} local vars = {}
for i, item in ipairs(\%assignments) do for i, item in ipairs(\%assignments) do
local \%target, \%value = item[1], item[2] local \%target, \%value = item[1], item[2]
@ -233,8 +233,7 @@ test:
test: test:
assume ((size of [1, 2, 3]) == 3) assume ((size of [1, 2, 3]) == 3)
[size of %list, size of %list, size of %list, size of %list] all compile to (..) (size of %list) compiles to "(#\(%list as lua expr))"
"(#\(%list as lua expr))"
(%list is empty) compiles to "(#\(%list as lua expr) == 0)" (%list is empty) compiles to "(#\(%list as lua expr) == 0)"

22
core/text.nom
View File

@ -4,6 +4,8 @@
color codes. color codes.
use "core/metaprogramming.nom" use "core/metaprogramming.nom"
use "core/operators.nom"
use "core/control_flow.nom"
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@ -47,17 +49,9 @@ test:
assume "\n" == (newline) assume "\n" == (newline)
# Text literals # Text literals
lua> "\ %escapes = {..}
..do nl:"\n", newline:"\n", tab:"\t", bell:"\a", cr:"\r", "carriage return":"\r",
local escapes = { backspace:"\b", "form feed":"\f", formfeed:"\f", "vertical tab":"\v"
nl="\\\\n", newline="\\\\n", tab="\\\\t", bell="\\\\a", cr="\\\\r", for %name = %str in %escapes:
["carriage return"]="\\\\r", backspace="\\\\b", ["form feed"]="\\\\f", with {%lua: Lua (quote %str)}:
formfeed="\\\\f", ["vertical tab"]="\\\\v", %compile.action.%name = ([]-> %lua)
};
for name, e in pairs(escapes) do
local lua = "'"..e.."'"
compile.action[name] = function(compile, tree)
return LuaCode(tree.source, lua)
end
end
end"

2
lib/consolecolor.nom
View File

@ -17,7 +17,7 @@ for %name = %colornum in %colors:
#(=lua "COMPILE_ACTIONS").%name = (..) #(=lua "COMPILE_ACTIONS").%name = (..)
[%nomsu, %tree] -> (Lua "'\\027[\(%colornum)m'") [%nomsu, %tree] -> (Lua "'\\027[\(%colornum)m'")
%compile.action.%name = (..) %compile.action.%name = (..)
[%nomsu, %tree, %text] ->: [%nomsu, %text] ->:
if %text: if %text:
return (Lua "('\\027[\(%colornum)m'..\(%text as lua expr)..'\\027[0m')") return (Lua "('\\027[\(%colornum)m'..\(%text as lua expr)..'\\027[0m')")
..else: return (Lua "'\\027[\(%colornum)m'") ..else: return (Lua "'\\027[\(%colornum)m'")

2
lib/object.nom
View File

@ -56,7 +56,7 @@ test:
..local fn_name = \%actions[1].stub:as_lua_id() ..local fn_name = \%actions[1].stub:as_lua_id()
local \%args = List(\%actions[1]:get_args()) local \%args = List(\%actions[1]:get_args())
table.insert(\%args, 1, \(\%me)) table.insert(\%args, 1, \(\%me))
local lua = LuaCode(tree.source, "class.", fn_name, " = ", \(..) local lua = LuaCode("class.", fn_name, " = ", \(..)
what (%args -> %body) compiles to what (%args -> %body) compiles to
..) ..)
for i=2,#\%actions do for i=2,#\%actions do

2
lib/things.nom
View File

@ -60,7 +60,7 @@ test:
lua> "\ lua> "\
..local fn_name = \%actions[1].stub:as_lua_id() ..local fn_name = \%actions[1].stub:as_lua_id()
local \%args = List{\(\%its), unpack(\%actions[1]:get_args())} local \%args = List{\(\%its), unpack(\%actions[1]:get_args())}
local lua = LuaCode(tree.source, "class.", fn_name, " = ", \(..) local lua = LuaCode("class.", fn_name, " = ", \(..)
what (%args -> %body) compiles to what (%args -> %body) compiles to
..) ..)
for i=2,#\%actions do for i=2,#\%actions do

6
nomsu.lua
View File

@ -181,7 +181,7 @@ run = function()
if args.check_syntax then if args.check_syntax then
local code = Files.read(filename) local code = Files.read(filename)
local source = Source(filename, 1, #code) local source = Source(filename, 1, #code)
nomsu_environment._1_parsed(NomsuCode(source, code)) nomsu_environment._1_parsed(NomsuCode:from(source, code))
print("Parse succeeded: " .. tostring(filename)) print("Parse succeeded: " .. tostring(filename))
elseif args.compile then elseif args.compile then
local output local output
@ -192,7 +192,7 @@ run = function()
end end
local code = Files.read(filename) local code = Files.read(filename)
local source = Source(filename, 1, #code) local source = Source(filename, 1, #code)
code = NomsuCode(source, code) code = NomsuCode:from(source, code)
local tree = nomsu_environment._1_parsed(code) local tree = nomsu_environment._1_parsed(code)
if not (tree.type == 'FileChunks') then if not (tree.type == 'FileChunks') then
tree = { tree = {
@ -214,7 +214,7 @@ run = function()
elseif args.verbose then elseif args.verbose then
local code = Files.read(filename) local code = Files.read(filename)
local source = Source(filename, 1, #code) local source = Source(filename, 1, #code)
code = NomsuCode(source, code) code = NomsuCode:from(source, code)
local tree = nomsu_environment._1_parsed(code) local tree = nomsu_environment._1_parsed(code)
if not (tree.type == 'FileChunks') then if not (tree.type == 'FileChunks') then
tree = { tree = {

6
nomsu.moon
View File

@ -117,7 +117,7 @@ run = ->
-- Check syntax -- Check syntax
code = Files.read(filename) code = Files.read(filename)
source = Source(filename, 1, #code) source = Source(filename, 1, #code)
nomsu_environment._1_parsed(NomsuCode(source, code)) nomsu_environment._1_parsed(NomsuCode\from(source, code))
print("Parse succeeded: #{filename}") print("Parse succeeded: #{filename}")
elseif args.compile elseif args.compile
-- Compile .nom files into .lua -- Compile .nom files into .lua
@ -125,7 +125,7 @@ run = ->
else io.open(filename\gsub("%.nom$", ".lua"), "w") else io.open(filename\gsub("%.nom$", ".lua"), "w")
code = Files.read(filename) code = Files.read(filename)
source = Source(filename, 1, #code) source = Source(filename, 1, #code)
code = NomsuCode(source, code) code = NomsuCode\from(source, code)
tree = nomsu_environment._1_parsed(code) tree = nomsu_environment._1_parsed(code)
tree = {tree} unless tree.type == 'FileChunks' tree = {tree} unless tree.type == 'FileChunks'
for chunk in *tree for chunk in *tree
@ -139,7 +139,7 @@ run = ->
elseif args.verbose elseif args.verbose
code = Files.read(filename) code = Files.read(filename)
source = Source(filename, 1, #code) source = Source(filename, 1, #code)
code = NomsuCode(source, code) code = NomsuCode\from(source, code)
tree = nomsu_environment._1_parsed(code) tree = nomsu_environment._1_parsed(code)
tree = {tree} unless tree.type == 'FileChunks' tree = {tree} unless tree.type == 'FileChunks'
for chunk in *tree for chunk in *tree

140
nomsu_compiler.lua
View File

@ -63,30 +63,11 @@ do
tree_to_nomsu, tree_to_inline_nomsu = _obj_0.tree_to_nomsu, _obj_0.tree_to_inline_nomsu tree_to_nomsu, tree_to_inline_nomsu = _obj_0.tree_to_nomsu, _obj_0.tree_to_inline_nomsu
end end
local math_expression = re.compile([[ (([*/^+-] / [0-9]+) " ")* [*/^+-] !. ]]) local math_expression = re.compile([[ (([*/^+-] / [0-9]+) " ")* [*/^+-] !. ]])
local compile_math_expression
compile_math_expression = function(compile, tree, ...)
local lua = LuaCode(tree.source)
for i, tok in ipairs(tree) do
if type(tok) == 'string' then
lua:append(tok)
else
local tok_lua = compile(tok)
if tok.type == "Action" then
tok_lua:parenthesize()
end
lua:append(tok_lua)
end
if i < #tree then
lua:append(" ")
end
end
return lua
end
local MAX_LINE = 80 local MAX_LINE = 80
local compile = setmetatable({ local compile = setmetatable({
action = Importer({ action = Importer({
[""] = function(compile, tree, fn, ...) [""] = function(compile, fn, ...)
local lua = LuaCode(tree.source) local lua = LuaCode()
local fn_lua = compile(fn) local fn_lua = compile(fn)
lua:append(fn_lua) lua:append(fn_lua)
if not (fn_lua:text():match("^%(.*%)$") or fn_lua:text():match("^[_a-zA-Z][_a-zA-Z0-9.]*$")) then if not (fn_lua:text():match("^%(.*%)$") or fn_lua:text():match("^[_a-zA-Z][_a-zA-Z0-9.]*$")) then
@ -102,9 +83,9 @@ local compile = setmetatable({
lua:append(")") lua:append(")")
return lua return lua
end, end,
["Lua"] = function(compile, tree, code) ["Lua"] = function(compile, code)
if code.type ~= "Text" then if code.type ~= "Text" then
return LuaCode(code.source, "LuaCode(", tostring(code.source):as_lua(), ", ", compile(code), ")") return LuaCode("LuaCode:from(", tostring(code.source):as_lua(), ", ", compile(code), ")")
end end
local add_bit_lua local add_bit_lua
add_bit_lua = function(lua, bit_lua) add_bit_lua = function(lua, bit_lua)
@ -114,7 +95,7 @@ local compile = setmetatable({
end end
local operate_on_text local operate_on_text
operate_on_text = function(text) operate_on_text = function(text)
local lua = LuaCode(text.source, "LuaCode(", tostring(text.source):as_lua()) local lua = LuaCode:from(text.source, "LuaCode:from(", tostring(text.source):as_lua())
for _index_0 = 1, #text do for _index_0 = 1, #text do
local bit = text[_index_0] local bit = text[_index_0]
if type(bit) == "string" then if type(bit) == "string" then
@ -130,13 +111,13 @@ local compile = setmetatable({
end end
return operate_on_text(code) return operate_on_text(code)
end, end,
["lua >"] = function(compile, tree, code) ["lua >"] = function(compile, code)
if code.type ~= "Text" then if code.type ~= "Text" then
return tree return code
end end
local operate_on_text local operate_on_text
operate_on_text = function(text) operate_on_text = function(text)
local lua = LuaCode(text.source) local lua = LuaCode:from(text.source)
for _index_0 = 1, #text do for _index_0 = 1, #text do
local bit = text[_index_0] local bit = text[_index_0]
if type(bit) == "string" then if type(bit) == "string" then
@ -151,18 +132,18 @@ local compile = setmetatable({
end end
return operate_on_text(code) return operate_on_text(code)
end, end,
["= lua"] = function(compile, tree, code) ["= lua"] = function(compile, code)
return compile.action["lua >"](compile, tree, code) return compile.action["lua >"](compile, code)
end, end,
["use"] = function(compile, tree, path) ["use"] = function(compile, path)
return LuaCode(tree.source, "run_file_1_in(" .. tostring(compile(path)) .. ", _ENV)") return LuaCode("run_file_1_in(" .. tostring(compile(path)) .. ", _ENV)")
end, end,
["tests"] = function(compile, tree) ["tests"] = function(compile)
return LuaCode(tree.source, "TESTS") return LuaCode("TESTS")
end, end,
["test"] = function(compile, tree, body) ["test"] = function(compile, body)
if not (body.type == 'Block') then if not (body.type == 'Block') then
compile_error(tree, "This should be a Block") compile_error(body, "This should be a Block")
end end
local test_nomsu = body:get_source_code():match(":[ ]*(.*)") local test_nomsu = body:get_source_code():match(":[ ]*(.*)")
do do
@ -171,16 +152,16 @@ local compile = setmetatable({
test_nomsu = test_nomsu:gsub("\n" .. indent, "\n") test_nomsu = test_nomsu:gsub("\n" .. indent, "\n")
end end
end end
return LuaCode(tree.source, "TESTS[" .. tostring(tostring(tree.source):as_lua()) .. "] = ", test_nomsu:as_lua()) return LuaCode("TESTS[" .. tostring(tostring(body.source):as_lua()) .. "] = ", test_nomsu:as_lua())
end, end,
["is jit"] = function(compile, tree, code) ["is jit"] = function(compile, code)
return LuaCode(tree.source, "jit") return LuaCode("jit")
end, end,
["Lua version"] = function(compile, tree, code) ["Lua version"] = function(compile, code)
return LuaCode(tree.source, "_VERSION") return LuaCode("_VERSION")
end, end,
["nomsu environment"] = function(compile, tree) ["nomsu environment"] = function(compile)
return LuaCode(tree.source, "_ENV") return LuaCode("_ENV")
end end
}) })
}, { }, {
@ -191,25 +172,27 @@ local compile = setmetatable({
if force_value == nil then if force_value == nil then
force_value = false force_value = false
end end
if tree.version then
do
local get_version = compile.action[("Nomsu version"):as_lua_id()]
if get_version then
do
local upgrade = compile.action[("1 upgraded from 2 to"):as_lua_id()]
if upgrade then
tree = upgrade(tree, tree.version, get_version())
end
end
end
end
end
local _exp_0 = tree.type local _exp_0 = tree.type
if "Action" == _exp_0 then if "Action" == _exp_0 then
local stub = tree.stub local stub = tree.stub
local compile_action = compile.action[stub] local compile_action = compile.action[stub]
if not compile_action and math_expression:match(stub) then if not compile_action and math_expression:match(stub) then
compile_action = compile_math_expression local lua = LuaCode:from(tree.source)
for i, tok in ipairs(tree) do
if type(tok) == 'string' then
lua:append(tok)
else
local tok_lua = compile(tok)
if tok.type == "Action" then
tok_lua:parenthesize()
end
lua:append(tok_lua)
end
if i < #tree then
lua:append(" ")
end
end
return lua
end end
if compile_action and not tree.target then if compile_action and not tree.target then
local args local args
@ -225,20 +208,21 @@ local compile = setmetatable({
end end
args = _accum_0 args = _accum_0
end end
local ret = compile_action(compile, tree, unpack(args)) local ret = compile_action(compile, unpack(args))
if ret == nil then if ret == nil then
local info = debug.getinfo(compile_action, "S") local info = debug.getinfo(compile_action, "S")
local filename = Source:from_string(info.source).filename local filename = Source:from_string(info.source).filename
compile_error(tree, "The compile-time action here (" .. tostring(stub) .. ") failed to return any value.", "Look at the implementation of (" .. tostring(stub) .. ") in " .. tostring(filename) .. ":" .. tostring(info.linedefined) .. " and make sure it's returning something.") compile_error(tree, "The compile-time action here (" .. tostring(stub) .. ") failed to return any value.", "Look at the implementation of (" .. tostring(stub) .. ") in " .. tostring(filename) .. ":" .. tostring(info.linedefined) .. " and make sure it's returning something.")
end end
if not (SyntaxTree:is_instance(ret)) then if not (SyntaxTree:is_instance(ret)) then
ret.source = ret.source or tree.source
return ret return ret
end end
if ret ~= tree then if ret ~= tree then
return compile(ret) return compile(ret)
end end
end end
local lua = LuaCode(tree.source) local lua = LuaCode:from(tree.source)
if tree.target then if tree.target then
local target_lua = compile(tree.target) local target_lua = compile(tree.target)
local target_text = target_lua:text() local target_text = target_lua:text()
@ -269,7 +253,7 @@ local compile = setmetatable({
lua:append(")") lua:append(")")
return lua return lua
elseif "EscapedNomsu" == _exp_0 then elseif "EscapedNomsu" == _exp_0 then
local lua = LuaCode(tree.source, "SyntaxTree{") local lua = LuaCode:from(tree.source, "SyntaxTree{")
local needs_comma, i = false, 1 local needs_comma, i = false, 1
local as_lua local as_lua
as_lua = function(x) as_lua = function(x)
@ -304,7 +288,7 @@ local compile = setmetatable({
return lua return lua
elseif "Block" == _exp_0 then elseif "Block" == _exp_0 then
if not force_value then if not force_value then
local lua = LuaCode(tree.source) local lua = LuaCode:from(tree.source)
lua:concat_append((function() lua:concat_append((function()
local _accum_0 = { } local _accum_0 = { }
local _len_0 = 1 local _len_0 = 1
@ -317,7 +301,7 @@ local compile = setmetatable({
end)(), "\n") end)(), "\n")
return lua return lua
else else
local lua = LuaCode(tree.source) local lua = LuaCode:from(tree.source)
lua:append("((function()") lua:append("((function()")
for i, line in ipairs(tree) do for i, line in ipairs(tree) do
lua:append("\n ", compile(line)) lua:append("\n ", compile(line))
@ -326,7 +310,7 @@ local compile = setmetatable({
return lua return lua
end end
elseif "Text" == _exp_0 then elseif "Text" == _exp_0 then
local lua = LuaCode(tree.source) local lua = LuaCode:from(tree.source)
local string_buffer = "" local string_buffer = ""
for i, bit in ipairs(tree) do for i, bit in ipairs(tree) do
local _continue_0 = false local _continue_0 = false
@ -348,7 +332,7 @@ local compile = setmetatable({
lua:append("..") lua:append("..")
end end
if bit.type ~= "Text" then if bit.type ~= "Text" then
bit_lua = LuaCode(bit.source, "tostring(", bit_lua, ")") bit_lua = LuaCode:from(bit.source, "tostring(", bit_lua, ")")
end end
lua:append(bit_lua) lua:append(bit_lua)
_continue_0 = true _continue_0 = true
@ -368,7 +352,7 @@ local compile = setmetatable({
end end
return lua return lua
elseif "List" == _exp_0 or "Dict" == _exp_0 then elseif "List" == _exp_0 or "Dict" == _exp_0 then
local lua = LuaCode(tree.source, tostring(tree.type) .. "{") local lua = LuaCode:from(tree.source, tostring(tree.type) .. "{")
local i = 1 local i = 1
local sep = '' local sep = ''
while i <= #tree do while i <= #tree do
@ -389,32 +373,38 @@ local compile = setmetatable({
end end
lua:append("}") lua:append("}")
if i <= #tree then if i <= #tree then
lua = LuaCode(tree.source, "(function()\n local it = ", lua) lua = LuaCode:from(tree.source, "(function()\n local comprehension = ", lua)
if tree.type == "List" then
lua:append("\n local function add(x) comprehension[#comprehension+1] = x end")
else
lua:append("\n local function " .. tostring(("add 1 ="):as_lua_id()) .. "(k, v) comprehension[k] = v end")
end
while i <= #tree do while i <= #tree do
lua:append("\n ") lua:append("\n ")
if tree[i].type == 'Block' or tree[i].type == 'Comment' then if tree[i].type == 'Block' or tree[i].type == 'Comment' then
lua:append(compile(tree[i])) lua:append(compile(tree[i]))
elseif tree[i].type == "DictEntry" then elseif tree[i].type == "DictEntry" then
lua:append("it[ ", compile(tree[i][1]), "] = ", (tree[i][2] and compile(tree[i][2]) or "true")) local entry_lua = compile(tree[i])
lua:append((entry_lua:text():sub(1, 1) == '[' and "comprehension" or "comprehension."), entry_lua)
else else
lua:append("it:add(", compile(tree[i]), ")") lua:append("comprehension[#comprehension+1] = ", compile(tree[i]))
end end
i = i + 1 i = i + 1
end end
lua:append("\n return it\nend)()") lua:append("\n return comprehension\nend)()")
end end
return lua return lua
elseif "DictEntry" == _exp_0 then elseif "DictEntry" == _exp_0 then
local key, value = tree[1], tree[2] local key, value = tree[1], tree[2]
local key_lua = compile(key) local key_lua = compile(key)
local value_lua = value and compile(value) or LuaCode(key.source, "true") local value_lua = value and compile(value) or LuaCode:from(key.source, "true")
local key_str = match(key_lua:text(), [=[^["']([a-zA-Z_][a-zA-Z0-9_]*)['"]$]=]) local key_str = match(key_lua:text(), [=[^["']([a-zA-Z_][a-zA-Z0-9_]*)['"]$]=])
if key_str and key_str:is_lua_id() then if key_str and key_str:is_lua_id() then
return LuaCode(tree.source, key_str, "=", value_lua) return LuaCode:from(tree.source, key_str, "=", value_lua)
elseif sub(key_lua:text(), 1, 1) == "[" then elseif sub(key_lua:text(), 1, 1) == "[" then
return LuaCode(tree.source, "[ ", key_lua, "]=", value_lua) return LuaCode:from(tree.source, "[ ", key_lua, "]=", value_lua)
else else
return LuaCode(tree.source, "[", key_lua, "]=", value_lua) return LuaCode:from(tree.source, "[", key_lua, "]=", value_lua)
end end
elseif "IndexChain" == _exp_0 then elseif "IndexChain" == _exp_0 then
local lua = compile(tree[1]) local lua = compile(tree[1])
@ -437,13 +427,13 @@ local compile = setmetatable({
end end
return lua return lua
elseif "Number" == _exp_0 then elseif "Number" == _exp_0 then
return LuaCode(tree.source, tostring(tree[1])) return LuaCode:from(tree.source, tostring(tree[1]))
elseif "Var" == _exp_0 then elseif "Var" == _exp_0 then
return LuaCode(tree.source, (tree[1]):as_lua_id()) return LuaCode:from(tree.source, (tree[1]):as_lua_id())
elseif "FileChunks" == _exp_0 then elseif "FileChunks" == _exp_0 then
return error("Can't convert FileChunks to a single block of lua, since each chunk's " .. "compilation depends on the earlier chunks") return error("Can't convert FileChunks to a single block of lua, since each chunk's " .. "compilation depends on the earlier chunks")
elseif "Comment" == _exp_0 then elseif "Comment" == _exp_0 then
return LuaCode(tree.source, "") return LuaCode:from(tree.source, "")
elseif "Error" == _exp_0 then elseif "Error" == _exp_0 then
return error("Can't compile errors") return error("Can't compile errors")
else else

110
nomsu_compiler.moon
View File

@ -40,23 +40,12 @@ compile_error = (tree, err_msg, hint=nil)->
-- math expressions like 2*x + 3^2 without having to define a single -- math expressions like 2*x + 3^2 without having to define a single
-- action for every possibility. -- action for every possibility.
math_expression = re.compile [[ (([*/^+-] / [0-9]+) " ")* [*/^+-] !. ]] math_expression = re.compile [[ (([*/^+-] / [0-9]+) " ")* [*/^+-] !. ]]
compile_math_expression = (compile, tree, ...)->
lua = LuaCode(tree.source)
for i,tok in ipairs tree
if type(tok) == 'string'
lua\append tok
else
tok_lua = compile(tok)
tok_lua\parenthesize! if tok.type == "Action"
lua\append tok_lua
lua\append " " if i < #tree
return lua
MAX_LINE = 80 -- For beautification purposes, try not to make lines much longer than this value MAX_LINE = 80 -- For beautification purposes, try not to make lines much longer than this value
compile = setmetatable({ compile = setmetatable({
action: Importer{ action: Importer{
[""]: (compile, tree, fn, ...)-> [""]: (compile, fn, ...)->
lua = LuaCode(tree.source) lua = LuaCode!
fn_lua = compile(fn) fn_lua = compile(fn)
lua\append fn_lua lua\append fn_lua
unless fn_lua\text!\match("^%(.*%)$") or fn_lua\text!\match("^[_a-zA-Z][_a-zA-Z0-9.]*$") unless fn_lua\text!\match("^%(.*%)$") or fn_lua\text!\match("^[_a-zA-Z][_a-zA-Z0-9.]*$")
@ -68,15 +57,15 @@ compile = setmetatable({
lua\append ")" lua\append ")"
return lua return lua
["Lua"]: (compile, tree, code)-> ["Lua"]: (compile, code)->
if code.type != "Text" if code.type != "Text"
return LuaCode(code.source, "LuaCode(", tostring(code.source)\as_lua!, ", ", compile(code), ")") return LuaCode("LuaCode:from(", tostring(code.source)\as_lua!, ", ", compile(code), ")")
add_bit_lua = (lua, bit_lua)-> add_bit_lua = (lua, bit_lua)->
bit_leading_len = #(bit_lua\match("^[^\n]*")) bit_leading_len = #(bit_lua\match("^[^\n]*"))
lua\append(lua\trailing_line_len! + bit_leading_len > MAX_LINE and ",\n " or ", ") lua\append(lua\trailing_line_len! + bit_leading_len > MAX_LINE and ",\n " or ", ")
lua\append(bit_lua) lua\append(bit_lua)
operate_on_text = (text)-> operate_on_text = (text)->
lua = LuaCode(text.source, "LuaCode(", tostring(text.source)\as_lua!) lua = LuaCode\from(text.source, "LuaCode:from(", tostring(text.source)\as_lua!)
for bit in *text for bit in *text
if type(bit) == "string" if type(bit) == "string"
add_bit_lua(lua, bit\as_lua!) add_bit_lua(lua, bit\as_lua!)
@ -88,11 +77,11 @@ compile = setmetatable({
return lua return lua
return operate_on_text code return operate_on_text code
["lua >"]: (compile, tree, code)-> ["lua >"]: (compile, code)->
if code.type != "Text" if code.type != "Text"
return tree return code
operate_on_text = (text)-> operate_on_text = (text)->
lua = LuaCode(text.source) lua = LuaCode\from(text.source)
for bit in *text for bit in *text
if type(bit) == "string" if type(bit) == "string"
lua\append bit lua\append bit
@ -103,47 +92,53 @@ compile = setmetatable({
return lua return lua
return operate_on_text code return operate_on_text code
["= lua"]: (compile, tree, code)-> compile.action["lua >"](compile, tree, code) ["= lua"]: (compile, code)-> compile.action["lua >"](compile, code)
["use"]: (compile, tree, path)-> ["use"]: (compile, path)->
--if path.type == 'Text' and #path == 1 and type(path[1]) == 'string' --if path.type == 'Text' and #path == 1 and type(path[1]) == 'string'
-- unless import_to_1_from(compile, path[1]) -- unless import_to_1_from(compile, path[1])
-- compile_error tree, "Could not find anything to import for #{path}" -- compile_error tree, "Could not find anything to import for #{path}"
return LuaCode(tree.source, "run_file_1_in(#{compile(path)}, _ENV)") return LuaCode("run_file_1_in(#{compile(path)}, _ENV)")
["tests"]: (compile, tree)-> LuaCode(tree.source, "TESTS") ["tests"]: (compile)-> LuaCode("TESTS")
["test"]: (compile, tree, body)-> ["test"]: (compile, body)->
unless body.type == 'Block' unless body.type == 'Block'
compile_error(tree, "This should be a Block") compile_error(body, "This should be a Block")
test_nomsu = body\get_source_code!\match(":[ ]*(.*)") test_nomsu = body\get_source_code!\match(":[ ]*(.*)")
if indent = test_nomsu\match("\n([ ]*)") if indent = test_nomsu\match("\n([ ]*)")
test_nomsu = test_nomsu\gsub("\n"..indent, "\n") test_nomsu = test_nomsu\gsub("\n"..indent, "\n")
LuaCode tree.source, "TESTS[#{tostring(tree.source)\as_lua!}] = ", test_nomsu\as_lua! return LuaCode "TESTS[#{tostring(body.source)\as_lua!}] = ", test_nomsu\as_lua!
["is jit"]: (compile, tree, code)-> LuaCode(tree.source, "jit") ["is jit"]: (compile, code)-> LuaCode("jit")
["Lua version"]: (compile, tree, code)-> LuaCode(tree.source, "_VERSION") ["Lua version"]: (compile, code)-> LuaCode("_VERSION")
["nomsu environment"]: (compile, tree)-> LuaCode(tree.source, "_ENV") ["nomsu environment"]: (compile)-> LuaCode("_ENV")
} }
}, { }, {
__import: (other)=> __import: (other)=>
import_to_1_from(@action, other.action) import_to_1_from(@action, other.action)
return return
__call: (compile, tree, force_value=false)-> __call: (compile, tree, force_value=false)->
if tree.version
if get_version = compile.action[("Nomsu version")\as_lua_id!]
if upgrade = compile.action[("1 upgraded from 2 to")\as_lua_id!]
tree = upgrade(tree, tree.version, get_version!)
switch tree.type switch tree.type
when "Action" when "Action"
stub = tree.stub stub = tree.stub
compile_action = compile.action[stub] compile_action = compile.action[stub]
if not compile_action and math_expression\match(stub) if not compile_action and math_expression\match(stub)
compile_action = compile_math_expression lua = LuaCode\from(tree.source)
for i,tok in ipairs tree
if type(tok) == 'string'
lua\append tok
else
tok_lua = compile(tok)
tok_lua\parenthesize! if tok.type == "Action"
lua\append tok_lua
lua\append " " if i < #tree
return lua
if compile_action and not tree.target if compile_action and not tree.target
args = [arg for arg in *tree when type(arg) != "string"] args = [arg for arg in *tree when type(arg) != "string"]
-- Force Lua to avoid tail call optimization for debugging purposes -- Force Lua to avoid tail call optimization for debugging purposes
-- TODO: use tail call? -- TODO: use tail call?
ret = compile_action(compile, tree, unpack(args)) ret = compile_action(compile, unpack(args))
if ret == nil if ret == nil
info = debug.getinfo(compile_action, "S") info = debug.getinfo(compile_action, "S")
filename = Source\from_string(info.source).filename filename = Source\from_string(info.source).filename
@ -151,11 +146,12 @@ compile = setmetatable({
"The compile-time action here (#{stub}) failed to return any value.", "The compile-time action here (#{stub}) failed to return any value.",
"Look at the implementation of (#{stub}) in #{filename}:#{info.linedefined} and make sure it's returning something." "Look at the implementation of (#{stub}) in #{filename}:#{info.linedefined} and make sure it's returning something."
unless SyntaxTree\is_instance(ret) unless SyntaxTree\is_instance(ret)
ret.source or= tree.source
return ret return ret
if ret != tree if ret != tree
return compile(ret) return compile(ret)
lua = LuaCode(tree.source) lua = LuaCode\from(tree.source)
if tree.target -- Method call if tree.target -- Method call
target_lua = compile tree.target target_lua = compile tree.target
target_text = target_lua\text! target_text = target_lua\text!
@ -174,7 +170,7 @@ compile = setmetatable({
return lua return lua
when "EscapedNomsu" when "EscapedNomsu"
lua = LuaCode tree.source, "SyntaxTree{" lua = LuaCode\from tree.source, "SyntaxTree{"
needs_comma, i = false, 1 needs_comma, i = false, 1
as_lua = (x)-> as_lua = (x)->
if type(x) == 'number' if type(x) == 'number'
@ -201,11 +197,11 @@ compile = setmetatable({
when "Block" when "Block"
if not force_value if not force_value
lua = LuaCode(tree.source) lua = LuaCode\from(tree.source)
lua\concat_append([compile(line) for line in *tree], "\n") lua\concat_append([compile(line) for line in *tree], "\n")
return lua return lua
else else
lua = LuaCode(tree.source) lua = LuaCode\from(tree.source)
lua\append("((function()") lua\append("((function()")
for i, line in ipairs(tree) for i, line in ipairs(tree)
lua\append "\n ", compile(line) lua\append "\n ", compile(line)
@ -213,7 +209,7 @@ compile = setmetatable({
return lua return lua
when "Text" when "Text"
lua = LuaCode(tree.source) lua = LuaCode\from(tree.source)
string_buffer = "" string_buffer = ""
for i, bit in ipairs tree for i, bit in ipairs tree
if type(bit) == "string" if type(bit) == "string"
@ -226,7 +222,7 @@ compile = setmetatable({
bit_lua = compile(bit) bit_lua = compile(bit)
if #lua.bits > 0 then lua\append ".." if #lua.bits > 0 then lua\append ".."
if bit.type != "Text" if bit.type != "Text"
bit_lua = LuaCode(bit.source, "tostring(",bit_lua,")") bit_lua = LuaCode\from(bit.source, "tostring(",bit_lua,")")
lua\append bit_lua lua\append bit_lua
if string_buffer ~= "" or #lua.bits == 0 if string_buffer ~= "" or #lua.bits == 0
@ -238,7 +234,7 @@ compile = setmetatable({
return lua return lua
when "List", "Dict" when "List", "Dict"
lua = LuaCode tree.source, "#{tree.type}{" lua = LuaCode\from tree.source, "#{tree.type}{"
i = 1 i = 1
sep = '' sep = ''
while i <= #tree while i <= #tree
@ -255,20 +251,26 @@ compile = setmetatable({
sep = ', ' sep = ', '
i += 1 i += 1
lua\append "}" lua\append "}"
-- List/dict comprehenstion
if i <= #tree if i <= #tree
lua = LuaCode tree.source, "(function()\n local it = ", lua lua = LuaCode\from tree.source, "(function()\n local comprehension = ", lua
if tree.type == "List"
lua\append "\n local function add(x) comprehension[#comprehension+1] = x end"
else
lua\append "\n local function #{("add 1 =")\as_lua_id!}(k, v) comprehension[k] = v end"
while i <= #tree while i <= #tree
lua\append "\n " lua\append "\n "
if tree[i].type == 'Block' or tree[i].type == 'Comment' if tree[i].type == 'Block' or tree[i].type == 'Comment'
lua\append compile(tree[i]) lua\append compile(tree[i])
elseif tree[i].type == "DictEntry" elseif tree[i].type == "DictEntry"
lua\append "it[ ", compile(tree[i][1]), "] = ", (tree[i][2] and compile(tree[i][2]) or "true") entry_lua = compile(tree[i])
lua\append (entry_lua\text!\sub(1,1) == '[' and "comprehension" or "comprehension."), entry_lua
else else
lua\append "it:add(", compile(tree[i]), ")" lua\append "comprehension[#comprehension+1] = ", compile(tree[i])
i += 1 i += 1
lua\append "\n return it\nend)()" lua\append "\n return comprehension\nend)()"
return lua return lua
--lua = LuaCode tree.source, "#{tree.type}{" --lua = LuaCode\from tree.source, "#{tree.type}{"
--lua\concat_append([compile(e) for e in *tree when e.type != 'Comment'], ", ", ",\n ") --lua\concat_append([compile(e) for e in *tree when e.type != 'Comment'], ", ", ",\n ")
--lua\append "}" --lua\append "}"
--return lua --return lua
@ -276,17 +278,17 @@ compile = setmetatable({
when "DictEntry" when "DictEntry"
key, value = tree[1], tree[2] key, value = tree[1], tree[2]
key_lua = compile(key) key_lua = compile(key)
value_lua = value and compile(value) or LuaCode(key.source, "true") value_lua = value and compile(value) or LuaCode\from(key.source, "true")
key_str = match(key_lua\text!, [=[^["']([a-zA-Z_][a-zA-Z0-9_]*)['"]$]=]) key_str = match(key_lua\text!, [=[^["']([a-zA-Z_][a-zA-Z0-9_]*)['"]$]=])
return if key_str and key_str\is_lua_id! return if key_str and key_str\is_lua_id!
LuaCode tree.source, key_str,"=",value_lua LuaCode\from tree.source, key_str,"=",value_lua
elseif sub(key_lua\text!,1,1) == "[" elseif sub(key_lua\text!,1,1) == "["
-- NOTE: this *must* use a space after the [ to avoid freaking out -- NOTE: this *must* use a space after the [ to avoid freaking out
-- Lua's parser if the inner expression is a long string. Lua -- Lua's parser if the inner expression is a long string. Lua
-- parses x[[[y]]] as x("[y]"), not as x["y"] -- parses x[[[y]]] as x("[y]"), not as x["y"]
LuaCode tree.source, "[ ",key_lua,"]=",value_lua LuaCode\from tree.source, "[ ",key_lua,"]=",value_lua
else else
LuaCode tree.source, "[",key_lua,"]=",value_lua LuaCode\from tree.source, "[",key_lua,"]=",value_lua
when "IndexChain" when "IndexChain"
lua = compile(tree[1]) lua = compile(tree[1])
@ -311,10 +313,10 @@ compile = setmetatable({
return lua return lua
when "Number" when "Number"
return LuaCode(tree.source, tostring(tree[1])) return LuaCode\from(tree.source, tostring(tree[1]))
when "Var" when "Var"
return LuaCode(tree.source, (tree[1])\as_lua_id!) return LuaCode\from(tree.source, (tree[1])\as_lua_id!)
when "FileChunks" when "FileChunks"
error("Can't convert FileChunks to a single block of lua, since each chunk's ".. error("Can't convert FileChunks to a single block of lua, since each chunk's "..
@ -322,7 +324,7 @@ compile = setmetatable({
when "Comment" when "Comment"
-- TODO: implement? -- TODO: implement?
return LuaCode(tree.source, "") return LuaCode\from(tree.source, "")
when "Error" when "Error"
error("Can't compile errors") error("Can't compile errors")

34
nomsu_decompiler.lua
View File

@ -47,7 +47,7 @@ local tree_to_inline_nomsu
tree_to_inline_nomsu = function(tree) tree_to_inline_nomsu = function(tree)
local _exp_0 = tree.type local _exp_0 = tree.type
if "Action" == _exp_0 then if "Action" == _exp_0 then
local nomsu = NomsuCode(tree.source) local nomsu = NomsuCode:from(tree.source)
if tree.target then if tree.target then
local inline_target = tree_to_inline_nomsu(tree.target) local inline_target = tree_to_inline_nomsu(tree.target)
if tree.target.type == "Action" then if tree.target.type == "Action" then
@ -88,9 +88,9 @@ tree_to_inline_nomsu = function(tree)
if not (tree[1].type == "List" or tree[1].type == "Dict" or tree[1].type == "Var") then if not (tree[1].type == "List" or tree[1].type == "Dict" or tree[1].type == "Var") then
inner_nomsu:parenthesize() inner_nomsu:parenthesize()
end end
return NomsuCode(tree.source, "\\", inner_nomsu) return NomsuCode:from(tree.source, "\\", inner_nomsu)
elseif "Block" == _exp_0 then elseif "Block" == _exp_0 then
local nomsu = NomsuCode(tree.source, ":") local nomsu = NomsuCode:from(tree.source, ":")
for i, line in ipairs(tree) do for i, line in ipairs(tree) do
nomsu:append(i == 1 and " " or "; ") nomsu:append(i == 1 and " " or "; ")
nomsu:append(tree_to_inline_nomsu(line)) nomsu:append(tree_to_inline_nomsu(line))
@ -119,11 +119,11 @@ tree_to_inline_nomsu = function(tree)
end end
end end
end end
local nomsu = NomsuCode(tree.source) local nomsu = NomsuCode:from(tree.source)
add_text(nomsu, tree) add_text(nomsu, tree)
return NomsuCode(tree.source, '"', nomsu, '"') return NomsuCode:from(tree.source, '"', nomsu, '"')
elseif "List" == _exp_0 or "Dict" == _exp_0 then elseif "List" == _exp_0 or "Dict" == _exp_0 then
local nomsu = NomsuCode(tree.source, (tree.type == "List" and "[" or "{")) local nomsu = NomsuCode:from(tree.source, (tree.type == "List" and "[" or "{"))
for i, item in ipairs(tree) do for i, item in ipairs(tree) do
if i > 1 then if i > 1 then
nomsu:append(", ") nomsu:append(", ")
@ -136,7 +136,7 @@ tree_to_inline_nomsu = function(tree)
local key, value = tree[1], tree[2] local key, value = tree[1], tree[2]
local nomsu local nomsu
if key.type == "Text" and #key == 1 and is_identifier(key[1]) then if key.type == "Text" and #key == 1 and is_identifier(key[1]) then
nomsu = NomsuCode(key.source, key[1]) nomsu = NomsuCode:from(key.source, key[1])
else else
nomsu = tree_to_inline_nomsu(key) nomsu = tree_to_inline_nomsu(key)
end end
@ -154,7 +154,7 @@ tree_to_inline_nomsu = function(tree)
end end
return nomsu return nomsu
elseif "IndexChain" == _exp_0 then elseif "IndexChain" == _exp_0 then
local nomsu = NomsuCode(tree.source) local nomsu = NomsuCode:from(tree.source)
for i, bit in ipairs(tree) do for i, bit in ipairs(tree) do
if i > 1 then if i > 1 then
nomsu:append(".") nomsu:append(".")
@ -173,9 +173,9 @@ tree_to_inline_nomsu = function(tree)
end end
return nomsu return nomsu
elseif "Number" == _exp_0 then elseif "Number" == _exp_0 then
return NomsuCode(tree.source, tostring(tree[1])) return NomsuCode:from(tree.source, tostring(tree[1]))
elseif "Var" == _exp_0 then elseif "Var" == _exp_0 then
return NomsuCode(tree.source, "%", tree[1]) return NomsuCode:from(tree.source, "%", tree[1])
elseif "FileChunks" == _exp_0 then elseif "FileChunks" == _exp_0 then
return error("Can't inline a FileChunks") return error("Can't inline a FileChunks")
elseif "Comment" == _exp_0 then elseif "Comment" == _exp_0 then
@ -188,7 +188,7 @@ tree_to_inline_nomsu = function(tree)
end end
local tree_to_nomsu local tree_to_nomsu
tree_to_nomsu = function(tree) tree_to_nomsu = function(tree)
local nomsu = NomsuCode(tree.source) local nomsu = NomsuCode:from(tree.source)
local recurse local recurse
recurse = function(t) recurse = function(t)
local space = MAX_LINE - nomsu:trailing_line_len() local space = MAX_LINE - nomsu:trailing_line_len()
@ -214,7 +214,7 @@ tree_to_nomsu = function(tree)
local indented = tree_to_nomsu(t) local indented = tree_to_nomsu(t)
if t.type == "Action" then if t.type == "Action" then
if indented:is_multiline() then if indented:is_multiline() then
return NomsuCode(t.source, "(..)\n ", indented) return NomsuCode:from(t.source, "(..)\n ", indented)
else else
indented:parenthesize() indented:parenthesize()
end end
@ -304,7 +304,7 @@ tree_to_nomsu = function(tree)
nomsu:append(line_nomsu:match('\n[^\n]*\n') and "\n\n" or "\n") nomsu:append(line_nomsu:match('\n[^\n]*\n') and "\n\n" or "\n")
end end
end end
return NomsuCode(tree.source, ":\n ", nomsu) return NomsuCode:from(tree.source, ":\n ", nomsu)
elseif "Text" == _exp_0 then elseif "Text" == _exp_0 then
local max_line = math.floor(1.25 * MAX_LINE) local max_line = math.floor(1.25 * MAX_LINE)
local add_text local add_text
@ -357,7 +357,7 @@ tree_to_nomsu = function(tree)
end end
end end
add_text(tree) add_text(tree)
return NomsuCode(tree.source, '"\\\n ..', nomsu, '"') return NomsuCode:from(tree.source, '"\\\n ..', nomsu, '"')
elseif "List" == _exp_0 or "Dict" == _exp_0 then elseif "List" == _exp_0 or "Dict" == _exp_0 then
if #tree == 0 then if #tree == 0 then
nomsu:append(tree.type == "List" and "[]" or "{}") nomsu:append(tree.type == "List" and "[]" or "{}")
@ -376,14 +376,14 @@ tree_to_nomsu = function(tree)
end end
end end
if tree.type == "List" then if tree.type == "List" then
return NomsuCode(tree.source, "[..]\n ", nomsu) return NomsuCode:from(tree.source, "[..]\n ", nomsu)
else else
return NomsuCode(tree.source, "{..}\n ", nomsu) return NomsuCode:from(tree.source, "{..}\n ", nomsu)
end end
elseif "DictEntry" == _exp_0 then elseif "DictEntry" == _exp_0 then
local key, value = tree[1], tree[2] local key, value = tree[1], tree[2]
if key.type == "Text" and #key == 1 and is_identifier(key[1]) then if key.type == "Text" and #key == 1 and is_identifier(key[1]) then
nomsu = NomsuCode(key.source, key[1]) nomsu = NomsuCode:from(key.source, key[1])
else else
nomsu = tree_to_inline_nomsu(key) nomsu = tree_to_inline_nomsu(key)
end end

34
nomsu_decompiler.moon
View File

@ -31,7 +31,7 @@ escape = (s)->
tree_to_inline_nomsu = (tree)-> tree_to_inline_nomsu = (tree)->
switch tree.type switch tree.type
when "Action" when "Action"
nomsu = NomsuCode(tree.source) nomsu = NomsuCode\from(tree.source)
if tree.target if tree.target
inline_target = tree_to_inline_nomsu(tree.target) inline_target = tree_to_inline_nomsu(tree.target)
if tree.target.type == "Action" if tree.target.type == "Action"
@ -61,10 +61,10 @@ tree_to_inline_nomsu = (tree)->
inner_nomsu = tree_to_inline_nomsu(tree[1]) inner_nomsu = tree_to_inline_nomsu(tree[1])
unless tree[1].type == "List" or tree[1].type == "Dict" or tree[1].type == "Var" unless tree[1].type == "List" or tree[1].type == "Dict" or tree[1].type == "Var"
inner_nomsu\parenthesize! inner_nomsu\parenthesize!
return NomsuCode(tree.source, "\\", inner_nomsu) return NomsuCode\from(tree.source, "\\", inner_nomsu)
when "Block" when "Block"
nomsu = NomsuCode(tree.source, ":") nomsu = NomsuCode\from(tree.source, ":")
for i,line in ipairs tree for i,line in ipairs tree
nomsu\append(i == 1 and " " or "; ") nomsu\append(i == 1 and " " or "; ")
nomsu\append tree_to_inline_nomsu(line) nomsu\append tree_to_inline_nomsu(line)
@ -86,12 +86,12 @@ tree_to_inline_nomsu = (tree)->
elseif bit.type == "Var" and type(tree[i+1]) == 'string' and not match(tree[i+1], "^[ \n\t,.:;#(){}[%]]") elseif bit.type == "Var" and type(tree[i+1]) == 'string' and not match(tree[i+1], "^[ \n\t,.:;#(){}[%]]")
interp_nomsu\parenthesize! interp_nomsu\parenthesize!
nomsu\append "\\", interp_nomsu nomsu\append "\\", interp_nomsu
nomsu = NomsuCode(tree.source) nomsu = NomsuCode\from(tree.source)
add_text(nomsu, tree) add_text(nomsu, tree)
return NomsuCode(tree.source, '"', nomsu, '"') return NomsuCode\from(tree.source, '"', nomsu, '"')
when "List", "Dict" when "List", "Dict"
nomsu = NomsuCode(tree.source, (tree.type == "List" and "[" or "{")) nomsu = NomsuCode\from(tree.source, (tree.type == "List" and "[" or "{"))
for i, item in ipairs tree for i, item in ipairs tree
nomsu\append ", " if i > 1 nomsu\append ", " if i > 1
nomsu\append tree_to_inline_nomsu(item) nomsu\append tree_to_inline_nomsu(item)
@ -101,7 +101,7 @@ tree_to_inline_nomsu = (tree)->
when "DictEntry" when "DictEntry"
key, value = tree[1], tree[2] key, value = tree[1], tree[2]
nomsu = if key.type == "Text" and #key == 1 and is_identifier(key[1]) nomsu = if key.type == "Text" and #key == 1 and is_identifier(key[1])
NomsuCode(key.source, key[1]) NomsuCode\from(key.source, key[1])
else tree_to_inline_nomsu(key) else tree_to_inline_nomsu(key)
nomsu\parenthesize! if key.type == "Action" or key.type == "Block" nomsu\parenthesize! if key.type == "Action" or key.type == "Block"
assert(value.type != "Block", "Didn't expect to find a Block as a value in a dict") assert(value.type != "Block", "Didn't expect to find a Block as a value in a dict")
@ -113,7 +113,7 @@ tree_to_inline_nomsu = (tree)->
return nomsu return nomsu
when "IndexChain" when "IndexChain"
nomsu = NomsuCode(tree.source) nomsu = NomsuCode\from(tree.source)
for i, bit in ipairs tree for i, bit in ipairs tree
nomsu\append "." if i > 1 nomsu\append "." if i > 1
local bit_nomsu local bit_nomsu
@ -127,10 +127,10 @@ tree_to_inline_nomsu = (tree)->
return nomsu return nomsu
when "Number" when "Number"
return NomsuCode(tree.source, tostring(tree[1])) return NomsuCode\from(tree.source, tostring(tree[1]))
when "Var" when "Var"
return NomsuCode(tree.source, "%", tree[1]) return NomsuCode\from(tree.source, "%", tree[1])
when "FileChunks" when "FileChunks"
error("Can't inline a FileChunks") error("Can't inline a FileChunks")
@ -146,7 +146,7 @@ tree_to_inline_nomsu = (tree)->
error("Unknown type: #{tree.type}") error("Unknown type: #{tree.type}")
tree_to_nomsu = (tree)-> tree_to_nomsu = (tree)->
nomsu = NomsuCode(tree.source) nomsu = NomsuCode\from(tree.source)
-- For concision: -- For concision:
recurse = (t)-> recurse = (t)->
@ -166,7 +166,7 @@ tree_to_nomsu = (tree)->
indented = tree_to_nomsu(t) indented = tree_to_nomsu(t)
if t.type == "Action" if t.type == "Action"
if indented\is_multiline! if indented\is_multiline!
return NomsuCode(t.source, "(..)\n ", indented) return NomsuCode\from(t.source, "(..)\n ", indented)
else indented\parenthesize! else indented\parenthesize!
return indented return indented
@ -234,7 +234,7 @@ tree_to_nomsu = (tree)->
if i < #tree if i < #tree
-- number of lines > 2 (TODO: improve this) -- number of lines > 2 (TODO: improve this)
nomsu\append(line_nomsu\match('\n[^\n]*\n') and "\n\n" or "\n") nomsu\append(line_nomsu\match('\n[^\n]*\n') and "\n\n" or "\n")
return NomsuCode(tree.source, ":\n ", nomsu) return NomsuCode\from(tree.source, ":\n ", nomsu)
when "Text" when "Text"
-- Multi-line text has more generous wrap margins -- Multi-line text has more generous wrap margins
@ -274,7 +274,7 @@ tree_to_nomsu = (tree)->
if interp_nomsu\is_multiline! if interp_nomsu\is_multiline!
nomsu\append "\n.." nomsu\append "\n.."
add_text(tree) add_text(tree)
return NomsuCode(tree.source, '"\\\n ..', nomsu, '"') return NomsuCode\from(tree.source, '"\\\n ..', nomsu, '"')
when "List", "Dict" when "List", "Dict"
if #tree == 0 if #tree == 0
@ -290,14 +290,14 @@ tree_to_nomsu = (tree)->
if i < #tree if i < #tree
nomsu\append((item_nomsu\is_multiline! or nomsu\trailing_line_len! + #tostring(item_nomsu) >= MAX_LINE) and '\n' or ', ') nomsu\append((item_nomsu\is_multiline! or nomsu\trailing_line_len! + #tostring(item_nomsu) >= MAX_LINE) and '\n' or ', ')
return if tree.type == "List" then return if tree.type == "List" then
NomsuCode(tree.source, "[..]\n ", nomsu) NomsuCode\from(tree.source, "[..]\n ", nomsu)
else else
NomsuCode(tree.source, "{..}\n ", nomsu) NomsuCode\from(tree.source, "{..}\n ", nomsu)
when "DictEntry" when "DictEntry"
key, value = tree[1], tree[2] key, value = tree[1], tree[2]
nomsu = if key.type == "Text" and #key == 1 and is_identifier(key[1]) nomsu = if key.type == "Text" and #key == 1 and is_identifier(key[1])
NomsuCode(key.source, key[1]) NomsuCode\from(key.source, key[1])
else tree_to_inline_nomsu(key) else tree_to_inline_nomsu(key)
nomsu\parenthesize! if key.type == "Block" nomsu\parenthesize! if key.type == "Block"
value_nomsu = tree_to_nomsu(value) value_nomsu = tree_to_nomsu(value)

18
nomsu_environment.lua
View File

@ -52,6 +52,7 @@ do
tree_to_nomsu, tree_to_inline_nomsu = _obj_0.tree_to_nomsu, _obj_0.tree_to_inline_nomsu tree_to_nomsu, tree_to_inline_nomsu = _obj_0.tree_to_nomsu, _obj_0.tree_to_inline_nomsu
end end
local compile = require('nomsu_compiler') local compile = require('nomsu_compiler')
local _currently_running_files = List({ })
local nomsu_environment = Importer({ local nomsu_environment = Importer({
NOMSU_COMPILER_VERSION = 12, NOMSU_COMPILER_VERSION = 12,
NOMSU_SYNTAX_VERSION = max_parser_version, NOMSU_SYNTAX_VERSION = max_parser_version,
@ -113,7 +114,7 @@ local nomsu_environment = Importer({
_1_parsed = function(nomsu_code) _1_parsed = function(nomsu_code)
if type(nomsu_code) == 'string' then if type(nomsu_code) == 'string' then
local filename = Files.spoof(nomsu_code) local filename = Files.spoof(nomsu_code)
nomsu_code = NomsuCode(Source(filename, 1, #nomsu_code), nomsu_code) nomsu_code = NomsuCode:from(Source(filename, 1, #nomsu_code), nomsu_code)
end end
local source = nomsu_code.source local source = nomsu_code.source
nomsu_code = tostring(nomsu_code) nomsu_code = tostring(nomsu_code)
@ -186,7 +187,7 @@ local nomsu_environment = Importer({
run_1_in = function(to_run, environment) run_1_in = function(to_run, environment)
if type(to_run) == 'string' then if type(to_run) == 'string' then
local filename = Files.spoof(to_run) local filename = Files.spoof(to_run)
to_run = NomsuCode(Source(filename, 1, #to_run), to_run) to_run = NomsuCode:from(Source(filename, 1, #to_run), to_run)
local ret = environment.run_1_in(to_run, environment) local ret = environment.run_1_in(to_run, environment)
return ret return ret
elseif NomsuCode:is_instance(to_run) then elseif NomsuCode:is_instance(to_run) then
@ -277,6 +278,14 @@ local nomsu_environment = Importer({
import_to_1_from(environment, environment.FILE_CACHE[path]) import_to_1_from(environment, environment.FILE_CACHE[path])
return return
end end
if _currently_running_files:has(path) then
local i = _currently_running_files:index_of(path)
_currently_running_files:add(path)
local circle = _currently_running_files:from_1_to(i, -1)
print(_currently_running_files, path)
error("Circular import detected:\n " .. circle:joined_with("\n..imports "))
end
_currently_running_files:add(path)
local mod = _1_forked(environment) local mod = _1_forked(environment)
assert(mod._1_parsed) assert(mod._1_parsed)
mod._ENV = mod mod._ENV = mod
@ -291,10 +300,10 @@ local nomsu_environment = Importer({
local code local code
if optimization ~= 0 and Files.read(lua_filename) then if optimization ~= 0 and Files.read(lua_filename) then
local file = Files.read(lua_filename) local file = Files.read(lua_filename)
code = LuaCode(Source(filename, 1, #file), file) code = LuaCode:from(Source(filename, 1, #file), file)
else else
local file = Files.read(filename) local file = Files.read(filename)
code = NomsuCode(Source(filename, 1, #file), file) code = NomsuCode:from(Source(filename, 1, #file), file)
end end
environment.run_1_in(code, mod) environment.run_1_in(code, mod)
_continue_0 = true _continue_0 = true
@ -305,6 +314,7 @@ local nomsu_environment = Importer({
end end
import_to_1_from(environment, mod) import_to_1_from(environment, mod)
environment.FILE_CACHE[path] = mod environment.FILE_CACHE[path] = mod
return _currently_running_files:remove()
end, end,
compile_error_at = function(tree, err_msg, hint) compile_error_at = function(tree, err_msg, hint)
if hint == nil then if hint == nil then

17
nomsu_environment.moon
View File

@ -32,6 +32,7 @@ for version=1,999
{:tree_to_nomsu, :tree_to_inline_nomsu} = require "nomsu_decompiler" {:tree_to_nomsu, :tree_to_inline_nomsu} = require "nomsu_decompiler"
compile = require('nomsu_compiler') compile = require('nomsu_compiler')
_currently_running_files = List{} -- Used to check for circular imports in run_file_1_in
nomsu_environment = Importer{ nomsu_environment = Importer{
NOMSU_COMPILER_VERSION: 12, NOMSU_SYNTAX_VERSION: max_parser_version NOMSU_COMPILER_VERSION: 12, NOMSU_SYNTAX_VERSION: max_parser_version
-- Lua stuff: -- Lua stuff:
@ -57,7 +58,7 @@ nomsu_environment = Importer{
_1_parsed: (nomsu_code)-> _1_parsed: (nomsu_code)->
if type(nomsu_code) == 'string' if type(nomsu_code) == 'string'
filename = Files.spoof(nomsu_code) filename = Files.spoof(nomsu_code)
nomsu_code = NomsuCode(Source(filename, 1, #nomsu_code), nomsu_code) nomsu_code = NomsuCode\from(Source(filename, 1, #nomsu_code), nomsu_code)
source = nomsu_code.source source = nomsu_code.source
nomsu_code = tostring(nomsu_code) nomsu_code = tostring(nomsu_code)
version = nomsu_code\match("^#![^\n]*nomsu[ ]+-V[ ]*([0-9.]+)") version = nomsu_code\match("^#![^\n]*nomsu[ ]+-V[ ]*([0-9.]+)")
@ -88,7 +89,7 @@ nomsu_environment = Importer{
run_1_in: (to_run, environment)-> run_1_in: (to_run, environment)->
if type(to_run) == 'string' if type(to_run) == 'string'
filename = Files.spoof(to_run) filename = Files.spoof(to_run)
to_run = NomsuCode(Source(filename, 1, #to_run), to_run) to_run = NomsuCode\from(Source(filename, 1, #to_run), to_run)
ret = environment.run_1_in(to_run, environment) ret = environment.run_1_in(to_run, environment)
return ret return ret
elseif NomsuCode\is_instance(to_run) elseif NomsuCode\is_instance(to_run)
@ -151,6 +152,13 @@ nomsu_environment = Importer{
if environment.FILE_CACHE[path] if environment.FILE_CACHE[path]
import_to_1_from(environment, environment.FILE_CACHE[path]) import_to_1_from(environment, environment.FILE_CACHE[path])
return return
if _currently_running_files\has(path)
i = _currently_running_files\index_of(path)
_currently_running_files\add path
circle = _currently_running_files\from_1_to(i, -1)
print(_currently_running_files, path)
error("Circular import detected:\n "..circle\joined_with("\n..imports "))
_currently_running_files\add path
mod = _1_forked(environment) mod = _1_forked(environment)
assert mod._1_parsed assert mod._1_parsed
mod._ENV = mod mod._ENV = mod
@ -160,13 +168,14 @@ nomsu_environment = Importer{
-- TODO: don't automatically use precompiled version? -- TODO: don't automatically use precompiled version?
code = if optimization != 0 and Files.read(lua_filename) code = if optimization != 0 and Files.read(lua_filename)
file = Files.read(lua_filename) file = Files.read(lua_filename)
LuaCode(Source(filename, 1, #file), file) LuaCode\from(Source(filename, 1, #file), file)
else else
file = Files.read(filename) file = Files.read(filename)
NomsuCode(Source(filename, 1, #file), file) NomsuCode\from(Source(filename, 1, #file), file)
environment.run_1_in(code, mod) environment.run_1_in(code, mod)
import_to_1_from(environment, mod) import_to_1_from(environment, mod)
environment.FILE_CACHE[path] = mod environment.FILE_CACHE[path] = mod
_currently_running_files\remove!
compile_error_at: (tree, err_msg, hint=nil)-> compile_error_at: (tree, err_msg, hint=nil)->
err_str = pretty_error{ err_str = pretty_error{