--[[
    This file is part of parser.lua - table based parsing
    Copyright (C) 2019  Soni L.

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
--]]

local parser = require "parser"

local caseno = 0
local function case()
    caseno = caseno + 1
    return caseno
end

do -- trim left spaces
    local defs = {}
    defs.self = defs
    defs[' '] = "whitespace"
    defs['\n'] = "whitespace"
    defs['\r'] = "whitespace"
    defs['\t'] = "whitespace"
    defs['\f'] = "whitespace"
    defs['\v'] = "whitespace"
    defs.whitespace = "self"
    defs[''] = function(state, token)
        state[#state + 1] = token
        if #state > 20 then
            state[1] = table.concat(state)
            for i=#state, 2, -1 do
                state[i] = nil
            end
        end
        return "start"
    end
    defs.start = {}
    defs.start.self = defs.start
    defs.start[''] = function(state, token)
        state[#state + 1] = token
        if #state > 20 then
            state[1] = table.concat(state)
            for i=#state, 2, -1 do
                state[i] = nil
            end
        end
        return "self"
    end
    for k,v in ipairs({"hello", "    hello", "\t \v \n\r hello"}) do
        local state, err = parser.parse(defs, v)
        local case = case()
        if not state then
            print(case, err)
        else
            assert(table.concat(state) == "hello")
        end
    end
end -- trim left spaces

do -- lua tokens
    local luatokens = require "luatokens"
    local tokens = luatokens.defs
    local state, err, etoken, estate = parser.parse(tokens, [["hello world"]])
    local case = case()
    if not state then
        print(case, "---- IN  TOKENS ----")
        print(case, err, etoken)
        for i,v in pairs(estate) do
            print(case, i, v)
        end
        print(case, "---- OUT TOKENS ----")
    else
        assert(state[1] == luatokens.tokens.TK_STRING)
        assert(state[2] == "hello world")
    end
end -- lua tokens

do -- more lua tokens
    local luatokens = require "luatokens"
    local tokens = luatokens.defs
    local state, err, etoken, estate = parser.parse(tokens, [["\a\b\f\n\r\t\v\\\"\'\z        \x41\65\
"]])
    local case = case()
    if not state then
        print(case, "---- IN  TOKENS ----")
        print(case, err, etoken)
        for i,v in pairs(estate) do
            print(case, i, v)
        end
        print(case, "---- OUT TOKENS ----")
    else
        assert(state[1] == luatokens.tokens.TK_STRING)
        assert(state[2] == "\7\8\12\10\13\9\11\92\34\39\65\65\10")
    end
end -- lua tokens

do -- even more lua tokens
    local luatokens = require "luatokens"
    local tokens = luatokens.defs
    local state, err, etoken, estate = parser.parse(tokens, [["\u{000000000000000000000000000000000000000000000000000000000000041}"]])
    local case = case()
    if not state then
        print(case, "---- IN  TOKENS ----")
        print(case, err, etoken)
        for i,v in pairs(estate) do
            print(case, i, v)
        end
        print(case, "---- OUT TOKENS ----")
    else
        assert(state[1] == luatokens.tokens.TK_STRING)
        assert(state[2] == "A")
    end
end -- lua tokens

do -- even more lua tokens
    local luatokens = require "luatokens"
    local tokens = luatokens.defs
    local state, err, etoken, estate = parser.parse(tokens, [["\u{7F}""\u{80}""\u{7FF}""\u{800}""\u{FFFF}""\u{10000}""\u{1FFFFF}""\u{200000}""\u{3FFFFFF}""\u{4000000}""\u{7FFFFFFF}"]])
    local case = case()
    if not state then
        print(case, "---- IN  TOKENS ----")
        print(case, err, etoken)
        for i,v in pairs(estate) do
            print(case, i, v)
        end
        print(case, "---- OUT TOKENS ----")
    else
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\127")
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\194\128")
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\223\191")
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\224\160\128")
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\239\191\191")
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\240\144\128\128")
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\247\191\191\191")
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\248\136\128\128\128")
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\251\191\191\191\191")
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\252\132\128\128\128\128")
        assert(table.remove(state, 1) == luatokens.tokens.TK_STRING)
        assert(table.remove(state, 1) == "\253\191\191\191\191\191")
    end
end -- lua tokens