-- This file contains objects that are used to track code positions and incrementally
-- build up generated code, while keeping track of where it came from, and managing
-- indentation levels.
{:insert, :remove, :concat} = table
unpack or= table.unpack
local LuaCode, NomsuCode, Source
class Source
new: (@filename, @start, @stop)=>
@from_string: (str)=>
filename,start,stop = str\match("^@(.-)%[(%d+):(%d+)%]$")
unless filename
filename,start = str\match("^@(.-)%[(%d+)%]$")
return @(filename or str, tonumber(start or 1), tonumber(stop))
@is_instance: (x)=> type(x) == 'table' and x.__class == @
__tostring: => "@#{@filename}[#{@start}#{@stop and ':'..@stop or ''}]"
as_lua: => "Source(#{@filename\as_lua!}, #{@start}#{@stop and ', '..@stop or ''})"
__eq: (other)=>
getmetatable(@) == getmetatable(other) and @filename == other.filename and @start == other.start and @stop == other.stop
__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 if type(offset) != 'number' then error("Cannot add Source and #{type(offset)}")
return Source(@filename, @start+offset, @stop)
class Code
is_code: true
new: (@source, ...)=>
@bits = {}
if type(@source) == 'string'
@source = Source\from_string(@source)
--assert(@source and Source\is_instance(@source), "Source has the wrong type")
@append(...)
as_smext: =>
if @__str == nil
buff, indent = {}, 0
{:match, :gsub, :rep} = string
for i,b in ipairs @bits
if type(b) == 'string'
if spaces = match(b, "\n([ ]*)[^\n]*$")
indent = #spaces
else
b = b\as_smext!
if indent > 0
b = gsub(b, "\n", "\n"..rep(" ", indent))
buff[#buff+1] = b
@__str = concat(buff, "")
return @__str
__tostring: => @as_smext!
as_lua: =>
"#{@__class.__name}(#{concat {tostring(@source)\as_lua!, unpack([b\as_lua! for b in *@bits])}, ", "})"
__len: => #@as_smext!
match: (...)=> @as_smext!\match(...)
gmatch: (...)=> @as_smext!\gmatch(...)
dirty: =>
@__str = nil
@_trailing_line_len = nil
-- Multi-line only goes from false->true, since there is no API for removing bits
@_is_multiline = nil if @_is_multiline == false
append: (...)=>
n = select("#",...)
match = string.match
bits = @bits
for i=1,n
b = select(i, ...)
assert(b, "code bit is nil")
assert(not Source\is_instance(b), "code bit is a Source")
if b == '' then continue
b.dirty = error if b.is_code
--if type(b) != 'string' and not (type(b) == 'table' and b.is_code)
-- b = b\as_lua!
bits[#bits+1] = b
@dirty!
trailing_line_len: =>
if @_trailing_line_len == nil
@_trailing_line_len = #@as_smext!\match("[^\n]*$")
return @_trailing_line_len
is_multiline: =>
if @_is_multiline == nil
match = string.match
@_is_multiline = false
for b in *@bits
if type(b) == 'string'
if match(b, '\n')
@_is_multiline = true
break
elseif b\is_multiline!
@_is_multiline = true
break
return @_is_multiline
concat_append: (values, joiner, wrapping_joiner)=>
wrapping_joiner or= joiner
match = string.match
bits = @bits
line_len = 0
for i=1,#values
b = values[i]
if i > 1
if line_len > 80
bits[#bits+1] = wrapping_joiner
line_len = 0
else
bits[#bits+1] = joiner
bits[#bits+1] = b
b.dirty = error if b.is_code
unless type(b) == 'string'
b = b\as_smext!
line = match(b, "\n([^\n]*)$")
if line
line_len = #line
else
line_len += #b
@dirty!
prepend: (...)=>
n = select("#",...)
bits = @bits
for i=#bits+n,n+1,-1
bits[i] = bits[i-n]
for i=1,n
b = select(i, ...)
b.dirty = error if b.is_code
--if type(b) != 'string' and not (type(b) == 'table' and b.is_code)
-- b = b\as_lua!
bits[i] = b
@dirty!
parenthesize: =>
@prepend "("
@append ")"
class LuaCode extends Code
__tostring: Code.__tostring
as_lua: Code.as_lua
__len: Code.__len
new: (...)=>
super ...
@free_vars = {}
@is_value = false
@Value = (...)->
lua = LuaCode(...)
lua.is_value = true
return lua
add_free_vars: (vars)=>
return unless #vars > 0
seen = {[v]:true for v in *@free_vars}
for var in *vars
assert type(var) == 'string'
unless seen[var]
@free_vars[#@free_vars+1] = var
seen[var] = true
@dirty!
remove_free_vars: (vars)=>
return unless #vars > 0
removals = {}
for var in *vars
assert type(var) == 'string'
removals[var] = true
stack = {self}
while #stack > 0
lua, stack[#stack] = stack[#stack], nil
for i=#lua.free_vars,1,-1
free_var = lua.free_vars[i]
if removals[free_var]
remove lua.free_vars, i
for b in *lua.bits
if type(b) != 'string'
stack[#stack+1] = b
@dirty!
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
unless type(bit) == 'string'
gather_from bit
gather_from self
if #to_declare > 0
@remove_free_vars to_declare
@prepend "local #{concat to_declare, ", "};\n"
return to_declare
as_statements: (prefix="", suffix=";")=>
unless @is_value
return self
statements = LuaCode(@source)
if prefix != ""
statements\append prefix
statements\append self
if suffix != ""
statements\append suffix
return statements
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
b = b\as_smext!
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: =>
assert @is_value, "Cannot parenthesize lua statements"
@prepend "("
@append ")"
class NomsuCode extends Code
__tostring: Code.__tostring
as_lua: Code.as_lua
__len: Code.__len
Code.__base.append_1 = assert Code.__base.append
Code.__base.append_1_joined_by_2 = assert Code.__base.concat_append
Code.__base.prepend_1 = assert Code.__base.prepend
LuaCode.__base.declare_locals_1 = assert LuaCode.__base.declare_locals
LuaCode.__base.remove_free_vars_1 = assert LuaCode.__base.remove_free_vars
LuaCode.__base.add_free_vars_1 = assert LuaCode.__base.add_free_vars
return {:Code, :NomsuCode, :LuaCode, :Source}