diff options
Diffstat (limited to 'nomsu_tree.lua')
| -rw-r--r-- | nomsu_tree.lua | 405 |
1 files changed, 405 insertions, 0 deletions
diff --git a/nomsu_tree.lua b/nomsu_tree.lua new file mode 100644 index 0000000..0f88ab4 --- /dev/null +++ b/nomsu_tree.lua @@ -0,0 +1,405 @@ +local utils = require('utils') +local re = require('re') +local repr, stringify, min, max, equivalent, set, is_list, sum +repr, stringify, min, max, equivalent, set, is_list, sum = utils.repr, utils.stringify, utils.min, utils.max, utils.equivalent, utils.set, utils.is_list, utils.sum +local immutable = require('immutable') +local insert, remove, concat +do + local _obj_0 = table + insert, remove, concat = _obj_0.insert, _obj_0.remove, _obj_0.concat +end +local Lua, Location +do + local _obj_0 = require("lua_obj") + Lua, Location = _obj_0.Lua, _obj_0.Location +end +local common_methods = { + __tostring = function(self) + return tostring(self.name) .. "(" .. tostring(repr(self.value)) .. ")" + end, + with_value = function(self, value) + return getmetatable(self)(value, self.source) + end +} +local fields = { + "value", + "source" +} +local Types = { } +Types.DictEntry = immutable({ + "key", + "value" +}, { + name = "DictEntry" +}) +Types.is_node = function(n) + return type(n) == 'userdata' and getmetatable(n) and Types[n.type] == getmetatable(n) +end +local Tree +Tree = function(name, methods) + do + methods.__tostring = function(self) + return tostring(self.name) .. "(" .. tostring(repr(self.value)) .. ")" + end + methods.with_value = function(self, value) + return getmetatable(self)(value, self.source) + end + methods.type = name + methods.name = name + end + Types[name] = immutable({ + "value", + "source" + }, methods) +end +Tree("File", { + as_lua = function(self, nomsu) + if #self.value == 1 then + return self.value[1]:as_lua(nomsu) + end + local declared_locals = { } + local lua = Lua(self.source) + for i, line in ipairs(self.value) do + local line_lua = line:as_lua(nomsu) + if not line_lua then + error("No lua produced by " .. tostring(repr(line)), 0) + end + if i < #self.value then + lua:append("\n") + end + lua:convert_to_statements() + lua:append(line_lua) + end + lua:declare_locals() + return lua + end +}) +Tree("Nomsu", { + as_lua = function(self, nomsu) + return Lua.Value(self.source, "nomsu:parse(", repr(self.source:get_text()), ", ", repr(self.source.filename), ")") + end +}) +Tree("Block", { + as_lua = function(self, nomsu) + if #self.value == 1 then + return self.value[1]:as_lua(nomsu) + end + local lua = Lua(self.source) + for i, line in ipairs(self.value) do + local line_lua = line:as_lua(nomsu) + if i < #self.value then + lua:append("\n") + end + line_lua:convert_to_statements() + lua:append(line_lua) + end + return lua + end +}) +local math_patt = re.compile([[ "%" (" " [*/^+-] " %")+ ]]) +Tree("Action", { + as_lua = function(self, nomsu) + local stub = self:get_stub() + local action = rawget(nomsu.environment.ACTIONS, stub) + local metadata = nomsu.action_metadata[action] + if metadata and metadata.compile_time then + local args + do + local _accum_0 = { } + local _len_0 = 1 + local _list_0 = self.value + for _index_0 = 1, #_list_0 do + local arg = _list_0[_index_0] + if arg.type ~= "Word" then + _accum_0[_len_0] = arg + _len_0 = _len_0 + 1 + end + end + args = _accum_0 + end + if metadata.arg_orders then + local new_args + do + local _accum_0 = { } + local _len_0 = 1 + local _list_0 = metadata.arg_orders[stub] + for _index_0 = 1, #_list_0 do + local p = _list_0[_index_0] + _accum_0[_len_0] = args[p - 1] + _len_0 = _len_0 + 1 + end + new_args = _accum_0 + end + args = new_args + end + return action(Lua(self.source), unpack(args)) + end + local lua = Lua.Value(self.source) + if not metadata and math_patt:match(stub) then + for i, tok in ipairs(self.value) do + if tok.type == "Word" then + lua:append(tok.value) + else + local tok_lua = tok:as_lua(nomsu) + if not (tok_lua.is_value) then + local src = tok.source:get_text() + error("non-expression value inside math expression: " .. tostring(colored.yellow(src))) + end + lua:append(tok_lua) + end + if i < #self.value then + lua:append(" ") + end + end + lua:parenthesize() + return lua + end + local args = { } + for i, tok in ipairs(self.value) do + local _continue_0 = false + repeat + if tok.type == "Word" then + _continue_0 = true + break + end + local arg_lua = tok:as_lua(nomsu) + if not (arg_lua.is_value) then + local line, src = tok.source:get_line(), tok.source:get_text() + error(tostring(line) .. ": Cannot use:\n" .. tostring(colored.yellow(src)) .. "\nas an argument to " .. tostring(stub) .. ", since it's not an expression, it produces: " .. tostring(repr(arg_lua)), 0) + end + insert(args, arg_lua) + _continue_0 = true + until true + if not _continue_0 then + break + end + end + if metadata and metadata.arg_orders then + do + local _accum_0 = { } + local _len_0 = 1 + local _list_0 = metadata.arg_orders[stub] + for _index_0 = 1, #_list_0 do + local p = _list_0[_index_0] + _accum_0[_len_0] = args[p] + _len_0 = _len_0 + 1 + end + args = _accum_0 + end + end + lua:append("ACTIONS[", repr(stub), "(") + for i, arg in ipairs(args) do + lua:append(arg) + if i < #args then + lua:append(", ") + end + end + lua:append(")") + return lua + end, + get_stub = function(self) + return concat((function() + local _accum_0 = { } + local _len_0 = 1 + local _list_0 = self.value + for _index_0 = 1, #_list_0 do + local t = _list_0[_index_0] + _accum_0[_len_0] = (t.type == "Word" and t.value or "%") + _len_0 = _len_0 + 1 + end + return _accum_0 + end)(), " ") + end +}) +Tree("Text", { + as_lua = function(self, nomsu) + local lua = Lua.Value(self.source) + local string_buffer = "" + local _list_0 = self.value + for _index_0 = 1, #_list_0 do + local _continue_0 = false + repeat + local bit = _list_0[_index_0] + if type(bit) == "string" then + string_buffer = string_buffer .. bit + _continue_0 = true + break + end + if string_buffer ~= "" then + if #lua.bits > 0 then + lua:append("..") + end + lua:append(repr(string_buffer)) + string_buffer = "" + end + local bit_lua = bit:as_lua(nomsu) + if not (bit_lua.is_value) then + local line, src = bit.source:get_line(), bit.source:get_text() + error(tostring(line) .. ": Cannot use " .. tostring(colored.yellow(bit)) .. " as a string interpolation value, since it's not an expression.", 0) + end + if #lua.bits > 0 then + lua:append("..") + end + if bit.type ~= "Text" then + bit_lua = Lua.Value(bit.source, "stringify(", bit_lua, ")") + end + lua:append(bit_lua) + _continue_0 = true + until true + if not _continue_0 then + break + end + end + if string_buffer ~= "" or #lua.bits == 0 then + if #lua.bits > 0 then + lua:append("..") + end + lua:append(repr(string_buffer)) + end + if #lua.bits > 1 then + lua:parenthesize() + end + return lua + end +}) +Tree("List", { + as_lua = function(self, nomsu) + local lua = Lua.Value(self.source, "{") + local line_length = 0 + for i, item in ipairs(self.value) do + local item_lua = item:as_lua(nomsu) + if not (item_lua.is_value) then + local line, src = item.source:get_line(), item.source:get_text() + error(tostring(line) .. ": Cannot use " .. tostring(colored.yellow(src)) .. " as a list item, since it's not an expression.", 0) + end + lua:append(item_lua) + local newlines, last_line = tostring(item_lua):match("^(.-)([^\n]*)$") + if #newlines > 0 then + line_length = #last_line + else + line_length = line_length + #last_line + end + if i < #self.value then + if line_length >= 80 then + lua:append(",\n") + line_length = 0 + else + lua:append(", ") + line_length = line_length + 2 + end + end + end + lua:append("}") + return lua + end +}) +Tree("Dict", { + as_lua = function(self, nomsu) + local lua = Lua.Value(self.source, "{") + local line_length = 0 + for i, entry in ipairs(self.value) do + local key_lua = entry.key:as_lua(nomsu) + if not (key_lua.is_value) then + local line, src = key.source:get_line(), key.source:get_text() + error(tostring(line) .. ": Cannot use " .. tostring(colored.yellow(src)) .. " as a dict key, since it's not an expression.", 0) + end + local value_lua = entry.value:as_lua(nomsu) + if not (value_lua.is_value) then + local line, src = value.source:get_line(), value.source:get_text() + error(tostring(line) .. ": Cannot use " .. tostring(colored.yellow(src)) .. " as a dict value, since it's not an expression.", 0) + end + local key_str = tostring(key_lua):match([=[["']([a-zA-Z_][a-zA-Z0-9_]*)['"]]=]) + if key_str then + lua:append(key_str, "=", value_lua) + elseif tostring(key_lua):sub(1, 1) == "[" then + lua:append("[ ", key_lua, "]=", value_lua) + else + lua:append("[", key_lua, "]=", value_lua) + end + local newlines, last_line = ("[" .. tostring(key_lua) .. "=" .. tostring(value_lua)):match("^(.-)([^\n]*)$") + if #newlines > 0 then + line_length = #last_line + else + line_length = line_length + #last_line + end + if i < #self.value then + if line_length >= 80 then + lua:append(",\n") + line_length = 0 + else + lua:append(", ") + line_length = line_length + 2 + end + end + end + lua:append("}") + return lua + end +}) +Tree("IndexChain", { + as_lua = function(self, nomsu) + local lua = self.value[1]:as_lua(nomsu) + if not (lua.is_value) then + local line, src = self.value[1].source:get_line(), self.value[1].source:get_text() + error(tostring(line) .. ": Cannot index " .. tostring(colored.yellow(src)) .. ", since it's not an expression.", 0) + end + local last_char = tostring(lua):sub(-1, -1) + if last_char == "}" or last_char == '"' or last_char == "]" then + lua:parenthesize() + end + for i = 2, #self.value do + local _continue_0 = false + repeat + local key = self.value[i] + if key.type == 'Text' and #key.value == 1 and type(key.value[1]) == 'string' and key.value[1]:match("^[a-zA-Z_][a-zA-Z0-9_]$") then + lua:append("." .. tostring(key.value[1])) + _continue_0 = true + break + end + local key_lua = self:tree_to_lua(key) + if not (key_lua.is_value) then + local line, src = key.source:get_line(), key.source:get_text() + error(tostring(line) .. ": Cannot use " .. tostring(colored.yellow(src)) .. " as an index, since it's not an expression.", 0) + end + if tostring(key_lua):sub(1, 1) == '[' then + lua:append("[ ", key_lua, "]") + else + lua:append("[", key_lua, "]") + end + _continue_0 = true + until true + if not _continue_0 then + break + end + end + return lua + end +}) +Tree("Number", { + as_lua = function(self, nomsu) + return Lua.Value(self.source, tostring(self.value)) + end +}) +Tree("Var", { + as_lua = function(self, nomsu) + local lua_id = "_" .. (self.value:gsub("%W", function(verboten) + if verboten == "_" then + return "__" + else + return ("_%x"):format(verboten:byte()) + end + end)) + return Lua.Value(self.source, lua_id) + end +}) +Tree("Word", { + as_lua = function(self, nomsu) + return error("Attempt to convert Word to lua") + end +}) +Tree("Comment", { + as_lua = function(self, nomsu) + return Lua(self.source, "--" .. self.value:gsub("\n", "\n--") .. "\n") + end +}) +return Types |