diff options
Diffstat (limited to 'src/lua-script/script.lua')
| -rw-r--r-- | src/lua-script/script.lua | 207 |
1 files changed, 207 insertions, 0 deletions
diff --git a/src/lua-script/script.lua b/src/lua-script/script.lua new file mode 100644 index 0000000..733c494 --- /dev/null +++ b/src/lua-script/script.lua @@ -0,0 +1,207 @@ +return function(config) + + -------------------------------- + -- + -- basics + -- + -------------------------------- + + function Set(array, transform) + if type(array) ~= 'table' then return nil end + + local transform = transform or function(x) return x end + + local set = {} + for _, element in ipairs(array) do + set[transform(element)] = true + end + return set + end + + + function add_end_slash(str) + if type(str) ~= 'string' then return nil end + if string.match(str, '/$') then return str end + return str .. '/' + end + + + function strip_end_slash(str) + if string.match(str, '/$') then return string.sub(str, 1, #str-1) end + return str + end + + + function setup(config) + argent.config = { + site_directory = add_end_slash(config.site_directory) or 'site/', + output_directory = add_end_slash(config.output_directory) or 'public/', + layout_directory = add_end_slash(config.layout_directory) or 'layouts/', + plugin_directory = add_end_slash(config.layout_directory) or 'plugins/', + exclude = Set(config.exclude, strip_end_slash) or {}, + include = Set(config.include, strip_end_slash) or {}, + keep = Set(config.keep, strip_end_slash) or {}, + noprocess = Set(config.noprocess, strip_end_slash) or {}, + rss_include = Set(config.rss_include, strip_end_slash) or {}, + } + + package.path = argent.config.plugin_directory..'?.lua;'..package.path + end + + + function is_dotfile(filename) + if string.match(filename, '^%.') then return true end + return false + end + + + function directory_exists(directory, parent) + local directory = directory or '' + local parent = parent or '' + local dirs = argent.scanDirectory(argent.config.output_directory..parent) + for _, dir in ipairs(dirs) do + if strip_end_slash(directory) == dir then return true end + end + return false + end + + + function output_directory_exists() + local dirs = argent.scanDirectory('./') + for _, dir in ipairs(dirs) do + if add_end_slash(dir) == argent.config.output_directory then + return true + end + end + return false + end + + + -------------------------------- + -- + -- obliterating + -- + -------------------------------- + + function obliterate_file(file, parent) + if argent.config.keep[file] then + return true + end + + os.remove(argent.config.output_directory..parent..file) + return false + end + + + function obliterate_dir(dir, parent) + if argent.config.keep[strip_end_slash(dir)] then + return true + end + + return obliterate_directory(dir, parent) + end + + + function obliterate_directory(dirname, parent) + local dirname = add_end_slash(dirname) or '' + local parent = parent or '' + + if argent.config.keep[dirname] then return true end + + local keep_self = (dirname == '') + + local dirs, files = argent.scanDirectory(argent.config.output_directory..parent..dirname) + for _, file in ipairs(files) do + keep_self = obliterate_file(file, parent..dirname) or keep_self + end + for _, dir in ipairs(dirs) do + if dir ~= '.' and dir ~= '..' then + keep_self = obliterate_dir(dir, parent..dirname) or keep_self + end + end + + if not keep_self then + os.remove(argent.config.output_directory..parent..dirname) + return false + else + return true + end + end + + + -------------------------------- + -- + -- processing + -- + -------------------------------- + + function process_file(filename, parent) + if (is_dotfile(filename) and not argent.config.include[filename]) + or argent.config.exclude[filename] + then return end + + if string.match(filename, '%.lua$') and not argent.config.noprocess[filename] then + process_lua_file(filename, parent) + else + argent.copyFile( + argent.config.site_directory..parent..filename, + argent.config.output_directory..parent..filename + ) + end + end + + + function process_lua_file(filename, parent) + local success, result = pcall(loadfile(argent.config.site_directory..parent..filename)) + if not success then + print('WARNING: '..result) + return + end + + print(filename, result) + end + + + function process_dir(directory, parent) + if (is_dotfile(directory) and not argent.config.include[strip_end_slash(directory)]) + or argent.config.exclude[strip_end_slash(directory)] + then return end + + if not directory_exists(directory, parent) then + argent.createDirectory(argent.config.output_directory..parent..directory) + end + process_directory(directory, parent) + end + + + function process_directory(directory, parent) + local directory = add_end_slash(directory) or '' + local parent = add_end_slash(parent) or '' + local dirs, files = argent.scanDirectory(argent.config.site_directory..parent..directory) + for _, file in ipairs(files) do + process_file(file, parent..directory) + end + + for _, dir in ipairs(dirs) do + if dir ~= '.' and dir ~= '..' then + process_dir(dir, parent..directory) + end + end + end + + + -------------------------------- + -- + -- putting it all together + -- + -------------------------------- + + setup(config) + if not output_directory_exists() then + argent.createDirectory(argent.config.output_directory) + else + obliterate_directory() + end + process_directory() +end + |