Mercurial > squish / changeset

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/compile/squish.compile.lua	Sat Jul 25 18:46:47 2009 +0100
@@ -0,0 +1,45 @@
+
+-- Not entirely sure that this is correct
+-- (produces files twice the size of luac)
+-- but it appears to work...
+
+function compile_string(str, name)
+	-- Strips debug info, if you're wondering :)
+	local b=string.dump(assert(loadstring(str,name)))
+	local x,y=string.find(b,str.."\0")
+	if not (x and y) then return b; end -- No debug info sometimes?
+	return string.sub(b,1,x-5).."\0\0\0\0"..string.sub(b, y+1, -1)
+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(data, 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

--- a/squishy	Sat Jul 25 16:44:47 2009 +0100
+++ b/squishy	Sat Jul 25 18:46:47 2009 +0100
@@ -26,3 +26,8 @@
 	
 	Main "uglify/squish.uglify.lua"
 end
+
+-- Compile output files to Lua bytecode
+if opts.with_compile then
+	Main "compile/squish.compile.lua"
+end