diff options
Diffstat (limited to 'code_obj.moon')
| -rw-r--r-- | code_obj.moon | 254 |
1 files changed, 254 insertions, 0 deletions
diff --git a/code_obj.moon b/code_obj.moon new file mode 100644 index 0000000..9df218c --- /dev/null +++ b/code_obj.moon @@ -0,0 +1,254 @@ +{:insert, :remove, :concat} = table +immutable = require 'immutable' +local Lua, Source +export LINE_STARTS + +Source = immutable {"filename","start","stop"}, { + name:"Source" + __new: (filename, start, stop)=> + if not start + start, stop = 1, #FILE_CACHE[filename] + if stop then assert(start <= stop, "Invalid range: #{start}, #{stop}") + return filename, start, stop + __tostring: => + if @stop + "\"#{@filename}[#{@start}:#{@stop}]\"" + else + "\"#{@filename}[#{@start}]\"" + __lt: (other)=> + assert(@filename == other.filename, "Cannot compare sources from different files") + return if @start == other.start + (@stop or @start) < (other.stop or other.start) + else @start < other.start + __le: (other)=> + assert(@filename == other.filename, "Cannot compare sources from different files") + return if @start == other.start + (@stop or @start) <= (other.stop or other.start) + else @start <= other.start + __add: (offset)=> + if type(self) == 'number' + offset, self = self, offset + else assert(type(offset) == 'number', "Cannot add Source and #{type(offset)}") + return Source(@filename, @start+offset, @stop) + sub: (start, stop)=> + start or= 1 + assert(start > 0 and (stop == nil or stop > 0), "Negative subscripts not supported") + if not @stop + assert(not stop, "cannot subscript non-range with range") + return Source(@filename, @start + start - 1) + else + stop or= @stop + return Source(@filename, @start + start - 1, @start + stop - 1) + get_text: => + FILE_CACHE[@filename]\sub(@start,@stop) + get_line_number: => + -- TODO: do a binary search if this is actually slow, which I doubt + src = FILE_CACHE[@filename] + line_starts = LINE_STARTS[src] + start_line = 1 + while (line_starts[start_line+1] or math.huge) <= @start + start_line += 1 + stop_line = start_line + while (line_starts[stop_line+1] or math.huge) <= @stop + stop_line += 1 + return start_line, stop_line + get_line: => "#{@filename}:#{@get_line_number!}" + get_line_range: => + start_line, stop_line = @get_line_number! + return if stop_line == start_line + "#{@filename}:#{start_line}" + else "#{@filename}:#{start_line}-#{stop_line}" +} + +class Code + new: (@source, ...)=> + @indents = {} + @bits = {...} + if type(@source) == 'string' + filename,start,stop = @source\match("^(.-)%[(%d+):(%d+)%]$") + unless filename + filename,start = @source\match("^(.-)%[(%d+)%]$") + if start or stop + @source = Source(filename, tonumber(start), tonumber(stop)) + else + @source = Source(@source, 1, #self+1) + assert(@source == nil or Source\is_instance(@source)) + indent = 0 + for i,b in ipairs @bits + assert(not Source\is_instance(b)) + if type(b) == 'string' + if spaces = b\match("\n([ ]*)[^\n]*$") + indent = #spaces + elseif indent != 0 + @indents[i] = indent + @current_indent = indent + @__str = nil + + sub: (start,stop)=> + -- TODO: implement this better + str = tostring(self)\sub(start,stop) + cls = @__class + return cls(@source\sub(start,stop), str) + + append: (...)=> + n = select("#",...) + bits = @bits + for i=1,n + b = select(i, ...) + assert(b != self, "No recursion please.") + bits[#bits+1] = b + if type(b) == 'string' + if spaces = b\match("\n([ ]*)[^\n]*$") + @current_indent = #spaces + elseif @current_indent != 0 + @indents[#bits] = @current_indent + @__str = nil + + prepend: (...)=> + n = select("#",...) + bits, indents = @bits, @indents + for i=#bits+n,n+1,-1 + bits[i] = bits[i-n] + for i=1,n + bits[i] = select(i, ...) + @current_indent = 0 + for i,b in ipairs(bits) + assert(b != self, "No recursion please.") + if type(b) == 'string' + if spaces = b\match("\n([ ]*)[^\n]*$") + @current_indent = #spaces + elseif @current_indent != 0 + indents[i] = @current_indent + else indents[i] = nil + @__str = nil + +class Lua extends Code + new: (...)=> + super ... + @free_vars = {} + @is_value = false + @__str = nil + + @Value = (...)-> + lua = Lua(...) + lua.is_value = true + return lua + + add_free_vars: (...)=> + seen = {[v]:true for v in *@free_vars} + for i=1,select("#",...) + var = select(i, ...) + if type(var) == 'userdata' and var.type == "Var" + var = tostring(var\as_lua!) + elseif type(var) != 'string' + var = tostring(var) + unless seen[var] + @free_vars[#@free_vars+1] = var + seen[var] = true + @__str = nil + + remove_free_vars: (...)=> + removals = {} + for i=1,select("#",...) + var = select(i, ...) + if type(var) == 'userdata' and var.type == "Var" + var = tostring(var\as_lua!) + elseif type(var) != 'string' + var = tostring(var) + removals[var] = true + + stack = {self} + while #stack > 0 + lua, stack[#stack] = stack[#stack], nil + for i=#lua.free_vars,1,-1 + if removals[lua.free_vars[i]] + remove lua.free_vars, i + for b in *lua.bits + if type(b) != 'string' + stack[#stack+1] = b + @__str = nil + + convert_to_statements: (prefix="", suffix=";")=> + unless @is_value + return + if prefix != "" + @prepend prefix + if suffix != "" + @append suffix + + declare_locals: (to_declare=nil)=> + if to_declare == nil + to_declare, seen = {}, {} + gather_from = => + for var in *@free_vars + unless seen[var] + seen[var] = true + to_declare[#to_declare+1] = var + for bit in *@bits + if bit.__class == Lua + gather_from bit + gather_from self + if #to_declare > 0 + @remove_free_vars unpack(to_declare) + @prepend "local #{concat to_declare, ", "};\n" + + __tostring: => + if @__str == nil + buff, indents = {}, @indents + for i,b in ipairs @bits + b = tostring(b) + if indents[i] + b = b\gsub("\n", "\n"..((" ")\rep(indents[i]))) + buff[#buff+1] = b + @__str = concat(buff, "") + return @__str + + __len: => + #tostring(self) + + make_offset_table: => + -- Return a mapping from output (lua) character number to input (nomsu) character number + lua_to_nomsu, nomsu_to_lua = {}, {} + walk = (lua, pos)-> + for b in *lua.bits + if type(b) == 'string' + if lua.source + lua_to_nomsu[pos] = lua.source.start + nomsu_to_lua[lua.source.start] = pos + else + walk b, pos + pos += #b + walk self, 1 + return { + nomsu_filename:@source.filename + lua_filename:tostring(@source)..".lua", lua_file:@stringify! + :lua_to_nomsu, :nomsu_to_lua + } + + parenthesize: => + if @is_value + @prepend "(" + @append ")" + else + error "Cannot parenthesize lua statements" + +class Nomsu extends Code + __tostring: => + if @__str == nil + buff, indents = {}, @indents + for i,b in ipairs @bits + b = tostring(b) + if indents[i] + b = b\gsub("\n", "\n"..((" ")\rep(indents[i]))) + buff[#buff+1] = b + @__str = concat(buff, "") + return @__str + + __len: => + #tostring(self) + + parenthesize: => + @prepend "(" + @append ")" + +return {:Code, :Nomsu, :Lua, :Source} |