437 lines
13 KiB
Lua
437 lines
13 KiB
Lua
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
|
|
methods.as_nomsu = function(self)
|
|
local leading_space = 0
|
|
local src_file = FILE_CACHE[self.source.filename]
|
|
while src_file:sub(self.source.start - leading_space - 1, self.source.start - leading_space - 1) == " " do
|
|
leading_space = leading_space + 1
|
|
end
|
|
if src_file:sub(self.source.start - leading_space - 1, self.source.start - leading_space - 1) ~= "\n" then
|
|
leading_space = 0
|
|
end
|
|
local ret = tostring(self.source:get_text()):gsub("\n" .. ((" "):rep(leading_space)), "\n")
|
|
return ret
|
|
end
|
|
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 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(Nomsu(", repr(self.value.source), ", ", repr(tostring(self.value.source:get_text())), ")).value[1]")
|
|
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_expression = 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
|
|
local ret = action(Lua(self.source), unpack(args))
|
|
return ret
|
|
end
|
|
local lua = Lua.Value(self.source)
|
|
if not metadata and math_expression: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, include_names)
|
|
if include_names == nil then
|
|
include_names = false
|
|
end
|
|
local bits
|
|
if include_names then
|
|
do
|
|
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 "%" .. tostring(t.value))
|
|
_len_0 = _len_0 + 1
|
|
end
|
|
bits = _accum_0
|
|
end
|
|
else
|
|
do
|
|
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
|
|
bits = _accum_0
|
|
end
|
|
end
|
|
return concat(bits, " ")
|
|
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 first_char = tostring(lua):sub(1, 1)
|
|
if first_char == "{" or first_char == '"' or first_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 = 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 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
|