435 lines
12 KiB
Lua
435 lines
12 KiB
Lua
local insert, remove, concat
|
|
do
|
|
local _obj_0 = table
|
|
insert, remove, concat = _obj_0.insert, _obj_0.remove, _obj_0.concat
|
|
end
|
|
local immutable = require('immutable')
|
|
local Lua, Source
|
|
Source = immutable({
|
|
"filename",
|
|
"start",
|
|
"stop"
|
|
}, {
|
|
name = "Source",
|
|
__new = function(self, filename, start, stop)
|
|
if stop then
|
|
assert(start <= stop, "Invalid range: " .. tostring(start) .. ", " .. tostring(stop))
|
|
end
|
|
return filename, start, stop
|
|
end,
|
|
__tostring = function(self)
|
|
if self.stop then
|
|
return "Source(\"" .. tostring(self.filename) .. "\", " .. tostring(self.start) .. ", " .. tostring(self.stop) .. ")"
|
|
else
|
|
return "Source(\"" .. tostring(self.filename) .. "\", " .. tostring(self.start) .. ")"
|
|
end
|
|
end,
|
|
__lt = function(self, other)
|
|
assert(self.filename == other.filename, "Cannot compare sources from different files")
|
|
if self.start == other.start then
|
|
return (self.stop or self.start) < (other.stop or other.start)
|
|
else
|
|
return self.start < other.start
|
|
end
|
|
end,
|
|
__le = function(self, other)
|
|
assert(self.filename == other.filename, "Cannot compare sources from different files")
|
|
if self.start == other.start then
|
|
return (self.stop or self.start) <= (other.stop or other.start)
|
|
else
|
|
return self.start <= other.start
|
|
end
|
|
end,
|
|
__add = function(self, offset)
|
|
if type(self) == 'number' then
|
|
offset, self = self, offset
|
|
else
|
|
assert(type(offset) == 'number', "Cannot add Source and " .. tostring(type(offset)))
|
|
end
|
|
return Source(self.filename, self.start + offset, self.stop)
|
|
end,
|
|
sub = function(self, start, stop)
|
|
if not self.stop then
|
|
assert(not stop, "cannot subscript non-range with range")
|
|
assert(start > 0, "cannot subscript non-range with negative index")
|
|
return Source(self.filename, self.start + (start or 0))
|
|
else
|
|
start = start or 1
|
|
if start < 0 then
|
|
start = self.stop + start + 1
|
|
end
|
|
stop = stop or -1
|
|
if stop < 0 then
|
|
stop = self.stop + stop + 1
|
|
end
|
|
return Source(self.filename, self.start + start - 1, self.start + stop - 1)
|
|
end
|
|
end,
|
|
get_text = function(self)
|
|
return FILE_CACHE[self.filename]:sub(self.start, self.stop)
|
|
end,
|
|
get_line_number = function(self)
|
|
local src = FILE_CACHE[self.filename]
|
|
local line_starts = LINE_STARTS[src]
|
|
local start_line = 1
|
|
while (line_starts[start_line + 1] or math.huge) <= self.start do
|
|
start_line = start_line + 1
|
|
end
|
|
local stop_line = start_line
|
|
while (line_starts[stop_line + 1] or math.huge) <= self.stop do
|
|
stop_line = stop_line + 1
|
|
end
|
|
return start_line, stop_line
|
|
end,
|
|
get_line = function(self)
|
|
return tostring(self.filename) .. ":" .. tostring(self:get_line_number())
|
|
end,
|
|
get_line_range = function(self)
|
|
local start_line, stop_line = self:get_line_number()
|
|
if stop_line == start_line then
|
|
return tostring(self.filename) .. ":" .. tostring(start_line)
|
|
else
|
|
return tostring(self.filename) .. ":" .. tostring(start_line) .. "-" .. tostring(stop_line)
|
|
end
|
|
end
|
|
})
|
|
local Code
|
|
do
|
|
local _class_0
|
|
local _base_0 = {
|
|
clone = function(self)
|
|
local cls = self.__class
|
|
local copy = cls(self.source, unpack(self.bits))
|
|
copy.is_value = self.is_value
|
|
for k, v in pairs(self.free_vars) do
|
|
copy.free_vars[k] = v
|
|
end
|
|
return copy
|
|
end,
|
|
__tostring = function(self)
|
|
local buff = { }
|
|
for i, b in ipairs(self.bits) do
|
|
buff[#buff + 1] = tostring(b)
|
|
end
|
|
local ret = concat(buff, "")
|
|
return ret
|
|
end,
|
|
__len = function(self)
|
|
local len = 0
|
|
local _list_0 = self.bits
|
|
for _index_0 = 1, #_list_0 do
|
|
local b = _list_0[_index_0]
|
|
len = len + #b
|
|
end
|
|
return len
|
|
end,
|
|
sub = function(self, start, stop)
|
|
local str = tostring(self):sub(start, stop)
|
|
local cls = self.__class
|
|
return cls(self.source:sub(start - self.source.start + 1, stop - self.source.stop + 1), str)
|
|
end,
|
|
append = function(self, ...)
|
|
local n = select("#", ...)
|
|
local bits = self.bits
|
|
for i = 1, n do
|
|
bits[#bits + 1] = select(i, ...)
|
|
end
|
|
end,
|
|
prepend = function(self, ...)
|
|
local n = select("#", ...)
|
|
local bits = self.bits
|
|
for i = #bits + n, n + 1, -1 do
|
|
bits[i] = bits[i - n]
|
|
end
|
|
for i = 1, n do
|
|
bits[i] = select(i, ...)
|
|
end
|
|
end
|
|
}
|
|
_base_0.__index = _base_0
|
|
_class_0 = setmetatable({
|
|
__init = function(self, source, ...)
|
|
self.source = source
|
|
if type(self.source) == 'string' then
|
|
local filename, start, stop = self.source:match("^(.-)[(%d+):(%d+)]$")
|
|
self.source = Source(filename, tonumber(start), tonumber(stop))
|
|
end
|
|
self.bits = {
|
|
...
|
|
}
|
|
end,
|
|
__base = _base_0,
|
|
__name = "Code"
|
|
}, {
|
|
__index = _base_0,
|
|
__call = function(cls, ...)
|
|
local _self_0 = setmetatable({}, _base_0)
|
|
cls.__init(_self_0, ...)
|
|
return _self_0
|
|
end
|
|
})
|
|
_base_0.__class = _class_0
|
|
Code = _class_0
|
|
end
|
|
do
|
|
local _class_0
|
|
local _parent_0 = Code
|
|
local _base_0 = {
|
|
add_free_vars = function(self, ...)
|
|
local seen
|
|
do
|
|
local _tbl_0 = { }
|
|
local _list_0 = self.free_vars
|
|
for _index_0 = 1, #_list_0 do
|
|
local v = _list_0[_index_0]
|
|
local _key_0, _val_0 = {
|
|
[v] = true
|
|
}
|
|
_tbl_0[_key_0] = _val_0
|
|
end
|
|
seen = _tbl_0
|
|
end
|
|
for i = 1, select("#", ...) do
|
|
local var = select(i, ...)
|
|
if not (seen[var]) then
|
|
self.free_vars[#self.free_vars + 1] = var
|
|
seen[var] = true
|
|
end
|
|
end
|
|
end,
|
|
convert_to_statements = function(self, prefix, suffix)
|
|
if prefix == nil then
|
|
prefix = ""
|
|
end
|
|
if suffix == nil then
|
|
suffix = ";"
|
|
end
|
|
if not (self.is_value) then
|
|
return
|
|
end
|
|
if prefix ~= "" then
|
|
self:prepend(prefix)
|
|
end
|
|
if suffix ~= "" then
|
|
return self:append(suffix)
|
|
end
|
|
end,
|
|
declare_locals = function(self, skip)
|
|
if skip == nil then
|
|
skip = { }
|
|
end
|
|
if next(skip) == 1 then
|
|
do
|
|
local _tbl_0 = { }
|
|
for _index_0 = 1, #skip do
|
|
local s = skip[_index_0]
|
|
local _key_0, _val_0 = {
|
|
[s] = true
|
|
}
|
|
_tbl_0[_key_0] = _val_0
|
|
end
|
|
skip = _tbl_0
|
|
end
|
|
end
|
|
if #self.free_vars > 0 then
|
|
self:prepend("local " .. tostring(concat(self.free_vars, ", ")) .. ";\n")
|
|
end
|
|
local _list_0 = self.free_vars
|
|
for _index_0 = 1, #_list_0 do
|
|
local var = _list_0[_index_0]
|
|
skip[var] = true
|
|
end
|
|
local _list_1 = self.bits
|
|
for _index_0 = 1, #_list_1 do
|
|
local bit = _list_1[_index_0]
|
|
if type(bit) == Lua then
|
|
bit:declare_locals(skip)
|
|
end
|
|
end
|
|
end,
|
|
__tostring = function(self)
|
|
local buff = { }
|
|
for i, b in ipairs(self.bits) do
|
|
buff[#buff + 1] = tostring(b)
|
|
end
|
|
local ret = concat(buff, "")
|
|
return ret
|
|
end,
|
|
__len = function(self)
|
|
local len = 0
|
|
local _list_0 = self.bits
|
|
for _index_0 = 1, #_list_0 do
|
|
local b = _list_0[_index_0]
|
|
len = len + #b
|
|
end
|
|
return len
|
|
end,
|
|
append = function(self, ...)
|
|
local n = select("#", ...)
|
|
local bits = self.bits
|
|
for i = 1, n do
|
|
local bit = select(i, ...)
|
|
bits[#bits + 1] = bit
|
|
if type(bit) ~= 'string' and not bit.is_value and #self.bits > 0 then
|
|
bits[#bits + 1] = "\n"
|
|
end
|
|
end
|
|
end,
|
|
prepend = function(self, ...)
|
|
local n = select("#", ...)
|
|
local bits = self.bits
|
|
local insert_index = 1
|
|
for i = 1, n do
|
|
local bit = select(i, ...)
|
|
insert(bits, insert_index, bit)
|
|
insert_index = insert_index + 1
|
|
if type(bit) ~= 'string' and not bit.is_value and insert_index < #self.bits + 1 then
|
|
insert(bits, insert_index, "\n")
|
|
insert_index = insert_index + 1
|
|
end
|
|
end
|
|
end,
|
|
make_offset_table = function(self)
|
|
local lua_chunkname = tostring(self.source) .. ".lua"
|
|
local lua_str = tostring(self)
|
|
local metadata = {
|
|
nomsu_filename = self.source.filename,
|
|
lua_filename = lua_chunkname,
|
|
lua_file = lua_str,
|
|
lua_to_nomsu = { },
|
|
nomsu_to_lua = { }
|
|
}
|
|
local lua_offset = 1
|
|
local walk
|
|
walk = function(lua)
|
|
if type(lua) == 'string' then
|
|
lua_offset = lua_offset + #lua
|
|
else
|
|
local lua_start = lua_offset
|
|
local _list_0 = lua.bits
|
|
for _index_0 = 1, #_list_0 do
|
|
local b = _list_0[_index_0]
|
|
walk(b)
|
|
end
|
|
local lua_stop = lua_offset
|
|
local nomsu_src, lua_src = lua.source, Source(lua_chunkname, lua_start, lua_stop)
|
|
metadata.lua_to_nomsu[lua_src] = nomsu_src
|
|
metadata.nomsu_to_lua[nomsu_src] = lua_src
|
|
end
|
|
end
|
|
walk(self)
|
|
return lua_str, metadata
|
|
end,
|
|
parenthesize = function(self)
|
|
if self.is_value then
|
|
self:prepend("(")
|
|
return self:append(")")
|
|
else
|
|
return error("Cannot parenthesize lua statements")
|
|
end
|
|
end
|
|
}
|
|
_base_0.__index = _base_0
|
|
setmetatable(_base_0, _parent_0.__base)
|
|
_class_0 = setmetatable({
|
|
__init = function(self, ...)
|
|
_class_0.__parent.__init(self, ...)
|
|
self.free_vars = { }
|
|
self.is_value = false
|
|
end,
|
|
__base = _base_0,
|
|
__name = "Lua",
|
|
__parent = _parent_0
|
|
}, {
|
|
__index = function(cls, name)
|
|
local val = rawget(_base_0, name)
|
|
if val == nil then
|
|
local parent = rawget(cls, "__parent")
|
|
if parent then
|
|
return parent[name]
|
|
end
|
|
else
|
|
return val
|
|
end
|
|
end,
|
|
__call = function(cls, ...)
|
|
local _self_0 = setmetatable({}, _base_0)
|
|
cls.__init(_self_0, ...)
|
|
return _self_0
|
|
end
|
|
})
|
|
_base_0.__class = _class_0
|
|
local self = _class_0
|
|
self.Value = function(...)
|
|
local lua = Lua(...)
|
|
lua.is_value = true
|
|
return lua
|
|
end
|
|
if _parent_0.__inherited then
|
|
_parent_0.__inherited(_parent_0, _class_0)
|
|
end
|
|
Lua = _class_0
|
|
end
|
|
local Nomsu
|
|
do
|
|
local _class_0
|
|
local _parent_0 = Code
|
|
local _base_0 = {
|
|
__tostring = function(self)
|
|
local buff = { }
|
|
for i, b in ipairs(self.bits) do
|
|
buff[#buff + 1] = tostring(b)
|
|
end
|
|
local ret = concat(buff, "")
|
|
return ret
|
|
end,
|
|
__len = function(self)
|
|
local len = 0
|
|
local _list_0 = self.bits
|
|
for _index_0 = 1, #_list_0 do
|
|
local b = _list_0[_index_0]
|
|
len = len + #b
|
|
end
|
|
return len
|
|
end
|
|
}
|
|
_base_0.__index = _base_0
|
|
setmetatable(_base_0, _parent_0.__base)
|
|
_class_0 = setmetatable({
|
|
__init = function(self, ...)
|
|
return _class_0.__parent.__init(self, ...)
|
|
end,
|
|
__base = _base_0,
|
|
__name = "Nomsu",
|
|
__parent = _parent_0
|
|
}, {
|
|
__index = function(cls, name)
|
|
local val = rawget(_base_0, name)
|
|
if val == nil then
|
|
local parent = rawget(cls, "__parent")
|
|
if parent then
|
|
return parent[name]
|
|
end
|
|
else
|
|
return val
|
|
end
|
|
end,
|
|
__call = function(cls, ...)
|
|
local _self_0 = setmetatable({}, _base_0)
|
|
cls.__init(_self_0, ...)
|
|
return _self_0
|
|
end
|
|
})
|
|
_base_0.__class = _class_0
|
|
if _parent_0.__inherited then
|
|
_parent_0.__inherited(_parent_0, _class_0)
|
|
end
|
|
Nomsu = _class_0
|
|
end
|
|
return {
|
|
Code = Code,
|
|
Nomsu = Nomsu,
|
|
Lua = Lua,
|
|
Source = Source
|
|
}
|