make_squishy

#!/usr/bin/env lua

if not arg[1] or arg[1]:match("%-%-?help") or arg[1]:match("^[/-]%?$") then
	print "make_squishy - Generate squishy files for your projects"
	print ""
	print(" Usage: "..arg[0].." FILE1 FILE2 FILE...");
	print ""
	print "make_squishy will scan the given files for require() calls, and convert the "
	print "found modules into a squishy file, 'squishy.new'. make_squishy automatically scans "
	print "the modules it finds, looking for further dependencies. To disable this, use "
	print "the --no-recursion option."
	return;
end


local short_opts = { v = "verbose", vv = "very_verbose", o = "output", q = "quiet", qq = "very_quiet", f = "force" }
local files, opts = {}, {};
local scanned_files, modules = {}, {};

for _, opt in ipairs(arg) do
	if opt:match("^%-") then
		local name = opt:match("^%-%-?([^%s=]+)()")
		name = (short_opts[name] or name):gsub("%-+", "_");
		if name:match("^no_") then
			name = name:sub(4, -1);
			opts[name] = false;
		else
			opts[name] = opt:match("=(.*)$") or true;
		end
	else
		table.insert(files, opt);
	end
end

if opts.very_verbose then opts.verbose = true; end
if opts.very_quiet then opts.quiet = true; end

local noprint = function () end
local print_err, print_info, print_verbose, print_debug = noprint, noprint, noprint, noprint;

if not opts.very_quiet then print_err = print; end
if not opts.quiet then print_info = print; end
if opts.verbose or opts.very_verbose then print_verbose = print; end
if opts.very_verbose then print_debug = print; end

if not opts.force then
	local squishy = io.open("squishy.new", "r");
	if squishy then
		print_err("There is already a squishy file in this directory, use -f to force overwriting it");
		squishy:close();
		os.exit(1);
	end
end

local squishy, err = io.open("squishy.new", "w+");
if not squishy then
	print_err("Couldn't open squishy file for writing: "..tostring(err));
	os.exit(1);
end

local LUA_DIRSEP = package.config:sub(1,1);
local LUA_PATH_MARK = package.config:sub(5,5);

local base_path = (files[1]:match("^(.-)"..LUA_DIRSEP.."[^"..LUA_DIRSEP.."]*$") or ".").."/";

local package_path = package.path:gsub("[^;]+", function (path)
		if not path:match("^%"..LUA_DIRSEP) then
			return base_path..path;
		end
	end):gsub("/%./", "/");
local package_cpath = package.cpath:gsub("[^;]+", function (path)
		if not path:match("^%"..LUA_DIRSEP) then
			return base_path..path;
		end
	end):gsub("/%./", "/");


function scan_file(outfile, scanfile)
	for line in io.lines(scanfile) do
		for _, module in (" "..line):gmatch("[^%w_]require%s*%(?([\"'])(.-)%1") do
			if not modules[module] then
				local binary;
				modules[module] = true;
				local filename = resolve_module(module, package_path);
				if false and not filename then
					binary = true;
					filename = resolve_module(module, package_cpath);
				end
				if not filename then
					print_info("Couldn't resolve module '"..module.."' to a file (required in "..scanfile..")");
				elseif opts.recursion ~= false and not scanned_files[filename] then
					scanned_files[filename] = true;
					table.insert(files, filename);
				end
				if filename then
					outfile:write((binary and "Binary" or ""), string.format([[Module %q %q]], module, filename:gsub("^"..base_path:gsub("%p", "%%%1"), "")), "\n");
				end
			end
		end
	end
end


function resolve_module(name, path)
        name = name:gsub("%.", LUA_DIRSEP);
        for c in path:gmatch("[^;]+") do
                c = c:gsub("%"..LUA_PATH_MARK, name);
                print_debug("Testing: "..c)
                local f = io.open(c);
                if f then
                        f:close();
                        return c;
                end
        end
        return nil; -- not found
end

for _, filename in ipairs(files) do
	squishy:write(string.format([[Main %q]], filename:gsub("^"..base_path:gsub("%p", "%%%1"), "")), "\n");
end
squishy:write("\n");
for _, filename in ipairs(files) do
	scanned_files[filename] = true;
	print_verbose("Found:", filename);
	scan_file(squishy, filename)
end