compile/squish.compile.lua


local cs = require "minichunkspy"

function compile_string(str, name)
	-- Strips debug info, if you're wondering :)
	local chunk = string.dump(loadstring(str, name));
	if ((not opts.debug) or opts.compile_strip) and opts.compile_strip ~= false then
		local c = cs.disassemble(chunk);
		local function strip_debug(c)
			c.source_lines, c.locals, c.upvalues = {}, {}, {};
			
			for i, f in ipairs(c.prototypes) do
				strip_debug(f);
			end
		end
		print_verbose("Stripping debug info...");
		strip_debug(c.body);
		return cs.assemble(c);
	end
	return chunk;
end

function compile_file(infile_fn, outfile_fn)
	local infile, err = io.open(infile_fn);
	if not infile then
		print_err("Can't open input file for reading: "..tostring(err));
		return;
	end
	
	local outfile, err = io.open(outfile_fn..".compiled", "w+");
	if not outfile then
		print_err("Can't open output file for writing: "..tostring(err));
		return;
	end
	
	local data = infile:read("*a");
	infile:close();
	
	local shebang, newdata = data:match("^(#.-\n)(.+)$");
	local code = newdata or data;
	if shebang then
		outfile:write(shebang)
	end

	outfile:write(compile_string(code, outfile_fn));
	
	os.rename(outfile_fn..".compiled", outfile_fn);
end

if opts.compile then
	print_info("Compiling "..out_fn.."...");
	compile_file(out_fn, out_fn);
	print_info("OK!");
end