moonscript/bin/splat.moon
2011-12-11 17:12:17 -08:00

76 lines
1.7 KiB
Plaintext
Executable File

#!/usr/bin/env moon
-- concatenate a collection of lua modules into one
require "lfs"
import insert, concat from table
import dump from require "moonscript.util"
if not arg[1]
print "usage: splat directory [directories...]"
os.exit!
dirs = [a for a in *arg[,]]
normalize = (path) ->
path\match("(.-)/*$").."/"
scan_directory = (root, patt, collected={}) ->
root = normalize root
for fname in lfs.dir root
if not fname\match "^%."
full_path = root..fname
if lfs.attributes(full_path, "mode") == "directory"
scan_directory full_path, patt, collected
else
if full_path\match patt
insert collected, full_path
collected
path_to_module_name = (path) ->
(path\match("(.-)%.lua")\gsub("/", "."))
each_line = (text) ->
import yield from coroutine
coroutine.wrap ->
start = 1
while true
pos, after = text\find "\n", start, true
break if not pos
yield text\sub start, pos - 1
start = after + 1
yield text\sub start, #text
nil
write_module = (name, text) ->
print "package.preload['"..name.."'] = function()"
for line in each_line text
print " "..line
print "end"
modules = {}
for dir in *dirs
files = scan_directory dir, "%.lua$"
chunks = for path in *files
module_name = path_to_module_name path
content = io.open(path)\read"*a"
modules[module_name] = true
{module_name, content}
for chunk in *chunks
name, content = unpack chunk
base = name\match"(.-)%.init"
if base and not modules[base] then
modules[base] = true
name = base
write_module name, content
for dir in *dirs
module_name = dir\gsub("/", ".")\gsub("%.$", "")
if modules[module_name]
print ([[package.preload["%s"]()]])\format module_name