Added lua_obj.lua

1 files changed, 307 insertions, 0 deletions

diff --git a/lua_obj.lua b/lua_obj.lua
new file mode 100644
index 0000000..a0ff4db
--- /dev/null
+++ b/lua_obj.lua
@@ -0,0 +1,307 @@
+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, LuaValue, Location
+Location = immutable({
+  "filename",
+  "start",
+  "stop"
+}, {
+  name = "Location",
+  __new = function(self, filename, start, stop)
+    assert(type(filename) == 'string' and type(start) == 'number' and type(stop) == 'number')
+    return filename, start, stop or start
+  end,
+  __tostring = function(self)
+    return "Location(\"" .. tostring(self.filename) .. "\", " .. tostring(self.start) .. ", " .. tostring(self.stop) .. ")"
+  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 < other.stop
+    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 <= other.stop
+    else
+      return self.start <= other.start
+    end
+  end,
+  get_text = function(self)
+    return FILE_CACHE[self.filename]:sub(self.start, self.stop)
+  end,
+  get_line_number = function(self)
+    local line_starts = LINE_STARTS[FILE_CACHE[self.filename]]
+    local start_line = 1
+    while (line_starts[start_line + 1] or (#src + 1)) <= self.start do
+      start_line = start_line + 1
+    end
+    local stop_line = start_line
+    while (line_starts[stop_line + 1] or (#src + 1)) <= 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
+})
+do
+  local _class_0
+  local _base_0 = {
+    is_statement = true,
+    is_value = false,
+    add_free_vars = function(self, free_vars)
+      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 _index_0 = 1, #free_vars do
+        local var = free_vars[_index_0]
+        if not (seen[var]) then
+          self.free_vars[#self.free_vars + 1] = var
+          seen[var] = true
+        end
+      end
+    end,
+    as_statements = function(self)
+      return self
+    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 = { }
+      local _list_0 = self.bits
+      for _index_0 = 1, #_list_0 do
+        local b = _list_0[_index_0]
+        buff[#buff + 1] = tostring(b)
+      end
+      local ret = concat(buff, "")
+      assert(not ret:match(".*table: 0x.*"))
+      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 x = select(i, ...)
+        assert(type(x) ~= 'table' or getmetatable(x))
+        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
+        local x = select(i, ...)
+        assert(type(x) ~= 'table' or getmetatable(x))
+        bits[i] = bits[i - n]
+      end
+      for i = 1, n do
+        local x = select(i, ...)
+        assert(type(x) ~= 'table' or getmetatable(x))
+        bits[i] = select(i, ...)
+      end
+    end,
+    make_offset_table = function(self, lua_chunkname)
+      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, Location(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
+  }
+  _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 = Location(filename, tonumber(start), tonumber(stop))
+      end
+      for i = 1, select("#", ...) do
+        local x = select(i, ...)
+        assert(type(x) ~= 'table' or getmetatable(x))
+      end
+      self.bits = {
+        ...
+      }
+      self.free_vars = { }
+    end,
+    __base = _base_0,
+    __name = "Lua"
+  }, {
+    __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
+  Lua = _class_0
+end
+do
+  local _class_0
+  local _parent_0 = Lua
+  local _base_0 = {
+    is_statement = false,
+    is_value = true,
+    __tostring = function(self)
+      local buff = { }
+      local _list_0 = self.bits
+      for _index_0 = 1, #_list_0 do
+        local b = _list_0[_index_0]
+        buff[#buff + 1] = tostring(b)
+      end
+      local ret = concat(buff, "")
+      assert(not ret:match(".*table: 0x.*"))
+      return ret
+    end,
+    as_statements = function(self, prefix, suffix)
+      if prefix == nil then
+        prefix = ""
+      end
+      if suffix == nil then
+        suffix = ";"
+      end
+      local bits = {
+        prefix,
+        unpack(self.bits)
+      }
+      bits[#bits + 1] = suffix
+      return Lua(self.source, unpack(bits))
+    end,
+    parenthesize = function(self)
+      self:prepend("(")
+      return self:append(")")
+    end
+  }
+  _base_0.__index = _base_0
+  setmetatable(_base_0, _parent_0.__base)
+  _class_0 = setmetatable({
+    __init = function(self, source, ...)
+      self.source = source
+      self.bits = {
+        ...
+      }
+    end,
+    __base = _base_0,
+    __name = "LuaValue",
+    __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
+  LuaValue = _class_0
+end
+return {
+  Lua = Lua,
+  LuaValue = LuaValue,
+  Location = Location
+}