#!/usr/bin/env lua

local parser = require "lua-parser.parser"

local pp = require "lua-parser.pp"

-- expected result, result, subject

local e, r, s

local filename = "test.lua"

local function parse (s)

local t,m = parser.parse(s,filename)

local r

if not t then

r = m

else

r = pp.tostring(t)

end

return r .. "\n"

end

local function fixint (s)

return _VERSION < "Lua 5.3" and s:gsub("%.0","") or s

end

print("> testing lexer...")

-- syntax ok

-- empty files

s = [=[

]=]

e = [=[

{ }

]=]

r = parse(s)

assert(r == e)

s = [=[

-- testing empty file

]=]

e = [=[

{ }

]=]

r = parse(s)

assert(r == e)

-- expressions

s = [=[

_nil,_false,_true,_dots = nil,false,true,...

]=]

e = [=[

{ `Set{ { `Id "_nil", `Id "_false", `Id "_true", `Id "_dots" }, { `Nil, `Boolean "false", `Boolean "true", `Dots } } }

]=]

r = parse(s)

assert(r == e)

-- floating points

s = [=[

f1 = 1.

f2 = 1.1

]=]

e = [=[

{ `Set{ { `Id "f1" }, { `Number "1.0" } }, `Set{ { `Id "f2" }, { `Number "1.1" } } }

]=]

r = parse(s)

assert(r == fixint(e))

s = [=[

f1 = 1.e-1

f2 = 1.e1

]=]

e = [=[

{ `Set{ { `Id "f1" }, { `Number "0.1" } }, `Set{ { `Id "f2" }, { `Number "10.0" } } }

]=]

r = parse(s)

assert(r == fixint(e))

s = [=[

f1 = 1.1e+1

f2 = 1.1e1

]=]

e = [=[

{ `Set{ { `Id "f1" }, { `Number "11.0" } }, `Set{ { `Id "f2" }, { `Number "11.0" } } }

]=]

r = parse(s)

assert(r == fixint(e))

s = [=[

f1 = .1

f2 = .1e1

]=]

e = [=[

{ `Set{ { `Id "f1" }, { `Number "0.1" } }, `Set{ { `Id "f2" }, { `Number "1.0" } } }

]=]

r = parse(s)

assert(r == fixint(e))

s = [=[

f1 = 1E1

f2 = 1e-1

]=]

e = [=[

{ `Set{ { `Id "f1" }, { `Number "10.0" } }, `Set{ { `Id "f2" }, { `Number "0.1" } } }

]=]

r = parse(s)

assert(r == fixint(e))

-- integers

s = [=[

i = 1

h = 0xff

]=]

e = [=[

{ `Set{ { `Id "i" }, { `Number "1" } }, `Set{ { `Id "h" }, { `Number "255" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

h = 0x76c

i = 4294967296 -- 2^32

]=]

e = [=[

{ `Set{ { `Id "h" }, { `Number "1900" } }, `Set{ { `Id "i" }, { `Number "4294967296" } } }

]=]

r = parse(s)

assert(r == e)

-- long comments

s = [=[

--[======[

testing

long

comment

[==[ one ]==]

[===[ more ]===]

[====[ time ]====]

bye

]======]

]=]

e = [=[

{ }

]=]

r = parse(s)

assert(r == e)

-- long strings

s = [=[

--[[

testing long string1 begin

]]

ls1 =

[[

testing long string

]]

--[[

testing long string1 end

]]

]=]

e = [=[

{ `Set{ { `Id "ls1" }, { `String "testing long string\n" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

--[==[

testing long string2 begin

]==]

ls2 = [==[ testing \n [[ long ]] \t [===[ string ]===]

\a ]==]

--[==[

[[ testing long string2 end ]]

]==]

]=]

e = [=[

{ `Set{ { `Id "ls2" }, { `String " testing \\n [[ long ]] \\t [===[ string ]===]\n\\a " } } }

]=]

r = parse(s)

assert(r == e)

-- short strings

s = [=[

-- short string test begin

ss1_a = "ola mundo\a"

ss1_b = 'ola mundo\a'

-- short string test end

]=]

e = [=[

{ `Set{ { `Id "ss1_a" }, { `String "ola mundo\a" } }, `Set{ { `Id "ss1_b" }, { `String "ola mundo\a" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

-- short string test begin

ss2_a = "testando,\tteste\n1\n2\n3 --> \"tchau\""

ss2_b = 'testando,\tteste\n1\n2\n3 --> \'tchau\''

-- short string test end

]=]

e = [=[

{ `Set{ { `Id "ss2_a" }, { `String "testando,\tteste\n1\n2\n3 --> \"tchau\"" } }, `Set{ { `Id "ss2_b" }, { `String "testando,\tteste\n1\n2\n3 --> 'tchau'" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

-- short string test begin

ss3_a = "ola \

'mundo'!"

ss3_b = 'ola \

"mundo"!'

-- short string test end

]=]

e = [=[

{ `Set{ { `Id "ss3_a" }, { `String "ola \n'mundo'!" } }, `Set{ { `Id "ss3_b" }, { `String "ola \n\"mundo\"!" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

-- short string test begin

ss4_a = "C:\\Temp/"

ss4_b = 'C:\\Temp/'

-- short string test end

]=]

e = [=[

{ `Set{ { `Id "ss4_a" }, { `String "C:\\Temp/" } }, `Set{ { `Id "ss4_b" }, { `String "C:\\Temp/" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

-- short string test begin

lf = "\\n"

-- short string test end

]=]

e = [=[

{ `Set{ { `Id "lf" }, { `String "\\n" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

-- short string test begin

ss5_a = "ola \

mundo \\ \

cruel"

ss5_b = 'ola \

mundo \\ \

cruel'

-- short string test end

]=]

e = [=[

{ `Set{ { `Id "ss5_a" }, { `String "ola \nmundo \\ \ncruel" } }, `Set{ { `Id "ss5_b" }, { `String "ola \nmundo \\ \ncruel" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

-- short string test begin

s1 = 'a \z b'

s2 = "adeus \z

mundo\

\z maravilhoso"

-- short string test end

]=]

e = [=[

{ `Set{ { `Id "s1" }, { `String "a b" } }, `Set{ { `Id "s2" }, { `String "adeus mundo\nmaravilhoso" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

-- short string test begin

deci = '\28'

hex = '\x1C'

uni = '\u{001C}'

-- short string test end

]=]

e = [=[

{ `Set{ { `Id "deci" }, { `String "\028" } }, `Set{ { `Id "hex" }, { `String "\028" } }, `Set{ { `Id "uni" }, { `String "\028" } } }

]=]

r = parse(s)

assert(r == e)

-- syntax error

-- floating points

s = [=[

f = 9e

]=]

e = [=[

test.lua:2:1: syntax error, expected one or more digits for the exponent

]=]

r = parse(s)

assert(r == e)

s = [=[

f = 5.e

]=]

e = [=[

test.lua:2:1: syntax error, expected one or more digits for the exponent

]=]

r = parse(s)

assert(r == e)

s = [=[

f = .9e-

]=]

e = [=[

test.lua:2:1: syntax error, expected one or more digits for the exponent

]=]

r = parse(s)

assert(r == e)

s = [=[

f = 5.9e+

]=]

e = [=[

test.lua:2:1: syntax error, expected one or more digits for the exponent

]=]

r = parse(s)

assert(r == e)

-- integers

s = [=[

-- invalid hexadecimal number

hex = 0xG

]=]

e = [=[

test.lua:3:9: syntax error, expected one or more hexadecimal digits after '0x'

]=]

r = parse(s)

assert(r == e)

-- long strings

s = [=[

--[==[

testing long string3 begin

]==]

ls3 = [===[

testing

unfinised

long string

]==]

--[==[

[[ testing long string3 end ]]

]==]

]=]

e = [=[

test.lua:14:1: syntax error, unclosed long string

]=]

r = parse(s)

assert(r == e)

-- short strings

s = [=[

-- short string test begin

ss6 = "testing unfinished string

-- short string test end

]=]

e = [=[

test.lua:4:1: syntax error, unclosed string

]=]

r = parse(s)

assert(r == e)

s = [=[

-- short string test begin

ss7 = 'testing \\

unfinished \\

string'

-- short string test end

]=]

e = [=[

test.lua:4:1: syntax error, unclosed string

]=]

r = parse(s)

assert(r == e)

-- unfinished comments

s = [=[

--[[ testing

unfinished

comment

]=]

e = [=[

test.lua:4:1: syntax error, unclosed long string

]=]

r = parse(s)

assert(r == e)

print("> testing parser...")

-- syntax ok

-- anonymous functions

s = [=[

local a,b,c = function () end

]=]

e = [=[

{ `Local{ { `Id "a", `Id "b", `Id "c" }, { `Function{ { }, { } } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

local test = function ( a , b , ... ) end

]=]

e = [=[

{ `Local{ { `Id "test" }, { `Function{ { `Id "a", `Id "b", `Dots }, { } } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

test = function (...) return ...,0 end

]=]

e = [=[

{ `Set{ { `Id "test" }, { `Function{ { `Dots }, { `Return{ `Dots, `Number "0" } } } } } }

]=]

r = parse(s)

assert(r == e)

-- arithmetic expressions

s = [=[

arithmetic = 1 - 2 * 3 + 4

]=]

e = [=[

{ `Set{ { `Id "arithmetic" }, { `Op{ "add", `Op{ "sub", `Number "1", `Op{ "mul", `Number "2", `Number "3" } }, `Number "4" } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

pow = -3^-2^2

]=]

e = [=[

{ `Set{ { `Id "pow" }, { `Op{ "unm", `Op{ "pow", `Number "3", `Op{ "unm", `Op{ "pow", `Number "2", `Number "2" } } } } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

q, r, f = 3//2, 3%2, 3/2

]=]

e = [=[

{ `Set{ { `Id "q", `Id "r", `Id "f" }, { `Op{ "idiv", `Number "3", `Number "2" }, `Op{ "mod", `Number "3", `Number "2" }, `Op{ "div", `Number "3", `Number "2" } } } }

]=]

r = parse(s)

assert(r == e)

-- assignments

s = [=[

a = f()[1]

]=]

e = [=[

{ `Set{ { `Id "a" }, { `Index{ `Call{ `Id "f" }, `Number "1" } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

a()[1] = 1;

]=]

e = [=[

{ `Set{ { `Index{ `Call{ `Id "a" }, `Number "1" } }, { `Number "1" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

i = a.f(1)

]=]

e = [=[

{ `Set{ { `Id "i" }, { `Call{ `Index{ `Id "a", `String "f" }, `Number "1" } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

i = a[f(1)]

]=]

e = [=[

{ `Set{ { `Id "i" }, { `Index{ `Id "a", `Call{ `Id "f", `Number "1" } } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

a[f()] = sub

i = i + 1

]=]

e = [=[

{ `Set{ { `Index{ `Id "a", `Call{ `Id "f" } } }, { `Id "sub" } }, `Set{ { `Id "i" }, { `Op{ "add", `Id "i", `Number "1" } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

a:b(1)._ = some_value

]=]

e = [=[

{ `Set{ { `Index{ `Invoke{ `Id "a", `String "b", `Number "1" }, `String "_" } }, { `Id "some_value" } } }

]=]

r = parse(s)

assert(r == e)

-- bitwise expressions

s = [=[

b = 1 & 0 | 1 ~ 1

]=]

e = [=[

{ `Set{ { `Id "b" }, { `Op{ "bor", `Op{ "band", `Number "1", `Number "0" }, `Op{ "bxor", `Number "1", `Number "1" } } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

b = 1 & 0 | 1 >> 1 ~ 1

]=]

e = [=[

{ `Set{ { `Id "b" }, { `Op{ "bor", `Op{ "band", `Number "1", `Number "0" }, `Op{ "bxor", `Op{ "shr", `Number "1", `Number "1" }, `Number "1" } } } } }

]=]

r = parse(s)

assert(r == e)

-- break

s = [=[

while 1 do

break

end

]=]

e = [=[

{ `While{ `Number "1", { `Break } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

while 1 do

while 1 do

break

end

break

end

]=]

e = [=[

{ `While{ `Number "1", { `While{ `Number "1", { `Break } }, `Break } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

repeat

if 2 > 1 then break end

until 1

]=]

e = [=[

{ `Repeat{ { `If{ `Op{ "gt", `Number "2", `Number "1" }, { `Break } } }, `Number "1" } }

]=]

r = parse(s)

assert(r == e)

s = [=[

for i=1,10 do

do

break

break

return

end

end

]=]

e = [=[

{ `Fornum{ `Id "i", `Number "1", `Number "10", { `Do{ `Break, `Break, `Return } } } }

]=]

r = parse(s)

assert(r == e)

-- block statements

s = [=[

do

var = 2+2;

return

end

]=]

e = [=[

{ `Do{ `Set{ { `Id "var" }, { `Op{ "add", `Number "2", `Number "2" } } }, `Return } }

]=]

r = parse(s)

assert(r == e)

-- concatenation expressions

s = [=[

concat1 = 1 .. 2^3

]=]

e = [=[

{ `Set{ { `Id "concat1" }, { `Op{ "concat", `Number "1", `Op{ "pow", `Number "2", `Number "3" } } } } }

]=]

r = parse(s)

assert(r == e)

-- empty files

s = [=[

;

]=]

e = [=[

{ }

]=]

r = parse(s)

assert(r == e)

-- for generic

s = [=[

for k,v in pairs(t) do print (k,v) end

]=]

e = [=[

{ `Forin{ { `Id "k", `Id "v" }, { `Call{ `Id "pairs", `Id "t" } }, { `Call{ `Id "print", `Id "k", `Id "v" } } } }

]=]

r = parse(s)

assert(r == e)

-- for numeric

s = [=[

for i = 1 , 10 , 2 do end

]=]

e = [=[

{ `Fornum{ `Id "i", `Number "1", `Number "10", `Number "2", { } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

for i=1,10 do end

]=]

e = [=[

{ `Fornum{ `Id "i", `Number "1", `Number "10", { } } }

]=]

r = parse(s)

assert(r == e)

-- global functions

s = [=[

function test(a , b , ...) end

]=]

e = [=[

{ `Set{ { `Id "test" }, { `Function{ { `Id "a", `Id "b", `Dots }, { } } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

function test (...) end

]=]

e = [=[

{ `Set{ { `Id "test" }, { `Function{ { `Dots }, { } } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

function t.a:b() end

]=]

e = [=[

{ `Set{ { `Index{ `Index{ `Id "t", `String "a" }, `String "b" } }, { `Function{ { `Id "self" }, { } } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

function t.a() end

]=]

e = [=[

{ `Set{ { `Index{ `Id "t", `String "a" } }, { `Function{ { }, { } } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

function testando . funcao . com : espcacos ( e, com , parametros, ... ) end

]=]

e = [=[

{ `Set{ { `Index{ `Index{ `Index{ `Id "testando", `String "funcao" }, `String "com" }, `String "espcacos" } }, { `Function{ { `Id "self", `Id "e", `Id "com", `Id "parametros", `Dots }, { } } } } }

]=]

r = parse(s)

assert(r == e)

-- goto

s = [=[

goto label

:: label :: return

]=]

e = [=[

{ `Goto{ "label" }, `Label{ "label" }, `Return }

]=]

r = parse(s)

assert(r == e)

s = [=[

::label::

goto label

]=]

e = [=[

{ `Label{ "label" }, `Goto{ "label" } }

]=]

r = parse(s)

assert(r == e)

s = [=[

goto label

::label::

]=]

e = [=[

{ `Goto{ "label" }, `Label{ "label" } }

]=]

r = parse(s)

assert(r == e)

s = [=[

::label::

do ::label:: goto label end

]=]

e = [=[

{ `Label{ "label" }, `Do{ `Label{ "label" }, `Goto{ "label" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

::label::

do goto label ; ::label:: end

]=]

e = [=[

{ `Label{ "label" }, `Do{ `Goto{ "label" }, `Label{ "label" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

::label::

do goto label end

]=]

e = [=[

{ `Label{ "label" }, `Do{ `Goto{ "label" } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

do goto label end

::label::

]=]

e = [=[

{ `Do{ `Goto{ "label" } }, `Label{ "label" } }

]=]

r = parse(s)

assert(r == e)

s = [=[

do do do do do goto label end end end end end

::label::

]=]

e = [=[

{ `Do{ `Do{ `Do{ `Do{ `Do{ `Goto{ "label" } } } } } }, `Label{ "label" } }

]=]

r = parse(s)

assert(r == e)

-- if-else

s = [=[

if a then end

]=]

e = [=[

{ `If{ `Id "a", { } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

if a then return a else return end

]=]

e = [=[

{ `If{ `Id "a", { `Return{ `Id "a" } }, { `Return } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

if a then

return a

else

local c = d

d = d + 1

return d

end

]=]

e = [=[

{ `If{ `Id "a", { `Return{ `Id "a" } }, { `Local{ { `Id "c" }, { `Id "d" } }, `Set{ { `Id "d" }, { `Op{ "add", `Id "d", `Number "1" } } }, `Return{ `Id "d" } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

if a then

return a

elseif b then

return b

elseif c then

return c

end

]=]

e = [=[

{ `If{ `Id "a", { `Return{ `Id "a" } }, `Id "b", { `Return{ `Id "b" } }, `Id "c", { `Return{ `Id "c" } } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

if a then return a

elseif b then return

else ;

end

]=]

e = [=[

{ `If{ `Id "a", { `Return{ `Id "a" } }, `Id "b", { `Return }, { } } }

]=]

r = parse(s)

assert(r == e)

s = [=[

if a then

return

elseif c then

end

]=]

e = [=[

{ `If{ `Id "a", { `Return }, `Id "c", { } } }

]=]

r = parse(s)

assert(r == e)

-- labels

s = [=[

::label::

do ::label:: end

::other_label::

]=]

e = [=[

{ `Label{ "label" }, `Do{ `Label{ "label" } }, `Label{ "other_label" } }

]=]

r = parse(s)

assert(r == e)

s = [=[

local x = glob

::label::

foo()

]=]

e = [=[

{ `Local{ { `Id "x" }, { `Id "glob" } }, `Label{ "label" }, `Call{ `Id "foo" } }

]=]

r = parse(s)

assert(r == e)

-- locals

s = [=[