Moving tree_to_lua into each of the Nomsu tree types, which are now in

their own file.

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