path: root/src/bin/printtokens.lua

--[[
    This file is part of Cratera Compiler
    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/>.
--]]

-- this isn't supposed to be installed or anything it's just a basic CLI to
-- print a lua or cratera file as a token stream.

local parser = require "cratera.parser"

-- CLI argument rules
local defs = parser.selfify({})
defs['-'] = function(state, token)
    if state.filename then
        error("Must specify only one filename")
    end
    state.filename = true
    state.file = io.stdin
    return "self"
end
defs[parser.FALLBACK] = function(state, token)
    if state.filename then
        error("Must specify only one filename")
    end
    state.filename = token
    state.file, state.err = io.open(state.filename, "r")
    return "self"
end
defs[parser.EOZ] = function(state, token)
    if not state.file then
        error((state.filename and (state.err or "") or "No file specified") )
    end
    return {}
end
defs[-1] = function(state, token, rule)
    if token ~= parser.EOZ and token:sub(1,1) == "-" and not rule then
        error("Unknown option: " .. token)
    end
end
defs['-l'] = function(state, token, rule)
    state.print_lineno = true
    return "self"
end
defs['--'] = parser.selfify({[parser.FALLBACK] = defs[parser.FALLBACK], [parser.EOZ] = defs[parser.EOZ]})

local state = parser.parse(defs, arg)
local luatokens = require "cratera.luatokens"
local file = state.file
local tokens = luatokens.defs
local print_lineno = state.print_lineno
--local state, err, etoken, estate = parser.parse(tokens, function() return file:read(8192) end)
local f, s, i = parser.stream(tokens, function() return file:read(8192) end)
local lastline
local lasttoken = 0
for pos, state, transemsg, etoken, estate in f, s, i do
    if not state then
        print("Parse error")
        break
    end
    if (state.line or 1) ~= lastline then
        lastline = state.line or 1
        print("line", lastline)
    end
    for i,v in ipairs(state) do
        lasttoken = lasttoken + 1
        v = luatokens.reverse_keywords[v] or luatokens.reverse_tokens[v] or v
        print(lasttoken, v) -- TODO formatting
        state[i] = nil
    end
    if not transemsg then
        break
    end
end