client.lua

#!/usr/bin/env luajit

local OUTPUT, REFNAME = ...
local REFNAME_VERSION = REFNAME and REFNAME:match("^refs/tags/(.*)$")
local MAIN_MODULE = "tptmp.client"
local ENV_DEFAULTS = {
	-- * Defaults for amalgamation mode.
	sim = { XRES = 612, YRES = 384, CELL = 4, PMAPBITS = 0, signs = {} },
	elem = {
		DEFAULT_PT_FIGH     = 0,
		DEFAULT_PT_LIGH     = 0,
		DEFAULT_PT_SPRK     = 0,
		DEFAULT_PT_STKM     = 0,
		DEFAULT_PT_STKM2    = 0,
		DEFAULT_PT_TESC     = 0,
		DEFAULT_UI_PROPERTY = 0,
		DEFAULT_UI_SAMPLE   = 0,
		DEFAULT_UI_SIGN     = 0,
		DEFAULT_UI_WIND     = 0,
		TPTMP_PT_UNKNOWN    = 0,
	},
	tpt = { version = { upstreamBuild = 356 } },
	http = {},
	ui = setmetatable({}, { __index = function(tbl, key)
		if key:find("^SDL_") then
			return 0
		end
		if key:find("^SDLK_") then
			return 0
		end
	end }),
	socket = { tcp = function() end },
}

if OUTPUT then
	for key, value in pairs(ENV_DEFAULTS) do
		rawset(getfenv(1), key, value)
	end
end

local env__ = setmetatable({}, { __index = function(_, key)
	error("__index on env: " .. tostring(key), 2)
end, __newindex = function(_, key)
	error("__newindex on env: " .. tostring(key), 2)
end })
for key, value in pairs(_G) do
	rawset(env__, key, value)
end
local _ENV = env__
if rawget(_G, "setfenv") then
	setfenv(1, env__)
end

math.randomseed(os.time())

local script_path = debug.getinfo(1).source
do
	assert(script_path:sub(1, 1) == "@", "something is fishy")
	script_path = script_path:sub(2)
	local slash_at
	for ix = #script_path, 1, -1 do
		if script_path:sub(ix, ix):find("[\\/]") then
			slash_at = ix
			break
		end
	end
	if slash_at then
		script_path = script_path:sub(1, slash_at - 1)
	else
		script_path = "."
	end
end

local unpack = rawget(_G, "unpack") or table.unpack
local function packn(...)
	return { [ 0 ] = select("#", ...), ... }
end
local function unpackn(tbl, from, to)
	return unpack(tbl, from or 1, to or tbl[0])
end
local function xpcall_wrap(func, handler)
	return function(...)
		local iargs = packn(...)
		local oargs
		xpcall(function()
			oargs = packn(func(unpackn(iargs)))
		end, function(err)
			if handler then
				handler(err)
			end
			print(debug.traceback(err, 2))
			return err
		end)
		if oargs then
			return unpackn(oargs)
		end
	end
end

local chunks = {}
local loaded = {}
local function require(modname)
	if not loaded[modname] then
		local try = {
			modname:gsub("%.", "/") .. ".lua",
			modname:gsub("%.", "/") .. "/init.lua",
		}
		local mod
		for i = 1, #try do
			local relative = try[i]
			local handle = io.open(script_path .. "/" .. relative, "r")
			if handle then
				local content = handle:read("*a")
				handle:close()
				local func, err
				if rawget(_G, "setfenv") then
					func, err = loadstring(content, "=" .. relative)
				else
					func, err = load(content, "=" .. relative, "bt", env__)
				end
				if not func then
					error(err, 0)
				end
				if rawget(_G, "setfenv") then
					setfenv(func, env__)
				end
				local ok = true
				local err_outer
				xpcall_wrap(function()
					mod = func()
				end, function(err)
					ok = false
					err_outer = err
				end)()
				if not ok then
					error(err_outer, 0)
				end
				chunks[modname] = content
				break
			end
		end
		if not mod then
			error("module " .. modname .. " not found", 2)
		end
		loaded[modname] = mod
	end
	return loaded[modname]
end
rawset(env__, "require", require)
rawset(env__, "xpcall_wrap", xpcall_wrap)

local main_module = require(MAIN_MODULE)
if main_module.loadtime_error then
	error(main_module.loadtime_error)
end
if OUTPUT then
	local handle = assert(io.open(OUTPUT, "w"))
	handle:write([[
local env__ = setmetatable({}, { __index = function(_, key)
	error("__index on env: " .. tostring(key), 2)
end, __newindex = function(_, key)
	error("__newindex on env: " .. tostring(key), 2)
end })
for key, value in pairs(_G) do
	rawset(env__, key, value)
end
local _ENV = env__
if rawget(_G, "setfenv") then
	setfenv(1, env__)
end

math.randomseed(os.time())

local require_preload__ = {}
local require_loaded__ = {}
local function require(modname)
	local mod = require_loaded__[modname]
	if not mod then
		mod = assert(assert(require_preload__[modname], "missing module " .. modname)())
		require_loaded__[modname] = mod
	end
	return mod
end
rawset(env__, "require", require)

local unpack = rawget(_G, "unpack") or table.unpack
local function packn(...)
	return { [ 0 ] = select("#", ...), ... }
end
local function unpackn(tbl, from, to)
	return unpack(tbl, from or 1, to or tbl[0])
end
local function xpcall_wrap(func, handler)
	return function(...)
		local iargs = packn(...)
		local oargs
		xpcall(function()
			oargs = packn(func(unpackn(iargs)))
		end, function(err)
			if handler then
				handler(err)
			end
			print(debug.traceback(err, 2))
			return err
		end)
		if oargs then
			return unpackn(oargs)
		end
	end
end
rawset(env__, "xpcall_wrap", xpcall_wrap)

]])
	local chunk_keys = {}
	for key in pairs(chunks) do
		table.insert(chunk_keys, key)
	end
	table.sort(chunk_keys)
	for i = 1, #chunk_keys do
		local chunk = chunks[chunk_keys[i]]:gsub("\n", "\n\t")
		if REFNAME_VERSION then
			chunk = chunk:gsub([[local versionstr = "v2%.[^"]+"]], ([[local versionstr = "%s"]]):format(REFNAME_VERSION))
		end
		handle:write(([[
require_preload__["%s"] = function()

	%s
end

]]):format(chunk_keys[i], chunk))
	end
	handle:write(([[
xpcall_wrap(function()
	require("%s").run()
end)()
]]):format(MAIN_MODULE))
	handle:close()
else
	xpcall_wrap(function()
		main_module.run()
	end)()
end