main.lua

local runtime = require 'zilscript.runtime'
local test_format = require 'zilscript.test_format'

local story_module = arg[1] or "infocom.zork1.zork1"

-- Create game environment
local env = runtime.create_game_env()

-- Load bootstrap
if not runtime.init(env) then
	os.exit(1)
end

-- Install ZIL support and load modules
env.require('zilscript')
if not runtime.load_modules(env, { story_module }, {save_lua = true}) then
	os.exit(1)
end

local esc = "\27["

local function highlight(text)
  if type(env.DESCS) == "table" then
    for _, dir in ipairs(env.DESCS) do
      local fmt = esc .. "1;32m%s" .. esc .. "0m"
      local cap = dir:sub(1,1):upper() .. dir:sub(2)
      text = text:gsub("(%f[%a]" .. dir .. "%f[%A])", function(m) return fmt:format(m) end)
      text = text:gsub("(%f[%a]" .. cap .. "%f[%A])", function(m) return fmt:format(m) end)
    end
  end
  if type(env.DIRS) == "table" then
    for _, dir in ipairs(env.DIRS) do
      local fmt = esc .. "1;36m%s" .. esc .. "0m"
      local cap = dir:sub(1,1):upper() .. dir:sub(2)
      text = text:gsub("(%f[%a]" .. dir .. "%f[%A])", function(m) return fmt:format(m) end)
      text = text:gsub("(%f[%a]" .. cap .. "%f[%A])", function(m) return fmt:format(m) end)
    end
  end
  return text
end

-- Create env as a coroutine
local game = runtime.create_game(env)
-- Start the game and get initial output
local res = game:start()
io.write(highlight(res))

local input
repeat
	input = io.read()
	if input then
		io.write("\n")
		res = game:resume(input)
		-- Check if result is a test response (table with status)
		if type(res) == "table" and res.status then
			io.write(test_format.format_test_result(res) .. "\n")
		else
			io.write(highlight(res))
		end
	end
until not input or not game:is_running()

-- local ast = parser.parse_file "infocom/zork1/actions.zil"
-- local res = compiler.compile(ast)

-- print(parser.view(ast, 0))
-- print(res.body)