diff options
Diffstat (limited to 'src/lua-script')
-rw-r--r-- | src/lua-script/script.h | 82 | ||||
-rw-r--r-- | src/lua-script/script.lua | 82 |
2 files changed, 146 insertions, 18 deletions
diff --git a/src/lua-script/script.h b/src/lua-script/script.h index 0bb6451..e29003f 100644 --- a/src/lua-script/script.h +++ b/src/lua-script/script.h @@ -88,7 +88,7 @@ const char *argent_script = " site_directory = add_end_slash(config.site_directory) or 'site/',\n" " output_directory = add_end_slash(config.output_directory) or 'public/',\n" " layout_directory = add_end_slash(config.layout_directory) or nil,\n" - " plugin_directory = add_end_slash(config.layout_directory) or nil,\n" + " plugin_directory = add_end_slash(config.plugin_directory) or nil,\n" " exclude = Set(config.exclude, strip_end_slash) or {},\n" " include = Set(config.include, strip_end_slash) or {},\n" " keep = Set(config.keep, strip_end_slash) or {},\n" @@ -115,7 +115,9 @@ const char *argent_script = " argent.log('info', fmt('available layouts: %s', set_tostring(argent.layouts)))\n" " \n" " if argent.config.plugin_directory then\n" - " package.path = argent.config.plugin_directory..'?.lua;'..package.path\n" + " package.path =\n" + " add_end_slash(argent.currentWorkingDirectory())\n" + " ..argent.config.plugin_directory..'?.lua;'..package.path\n" " end\n" " argent.log('debug', 'end setup')\n" " end\n" @@ -211,10 +213,32 @@ const char *argent_script = " --\n" " --------------------------------\n" "\n" + " function should_include(filename, parent)\n" + " for pattern in pairs(argent.config.include) do\n" + " if string.match(filename, pattern) or string.match(parent..filename, pattern) then\n" + " return true\n" + " end\n" + " end\n" + " return false\n" + " end\n" + "\n" + " function should_ignore(filename, parent)\n" + " if is_dotfile(filename) and not should_include(filename, parent) then\n" + " return false\n" + " end\n" + "\n" + " for pattern in pairs(argent.config.exclude) do\n" + " if string.match(filename, pattern)\n" + " or string.match(parent..filename, pattern) then\n" + " return true\n" + " end\n" + " end\n" + "\n" + " return false\n" + " end\n" + "\n" " function process_file(filename, parent)\n" - " if (is_dotfile(filename) and not argent.config.include[filename])\n" - " or argent.config.exclude[filename]\n" - " then\n" + " if should_ignore(filename, parent) then\n" " argent.log('debug', fmt('will not process file %q', parent..filename))\n" " return\n" " end\n" @@ -239,14 +263,54 @@ const char *argent_script = " return\n" " end\n" "\n" - " print(filename, result)\n" + " if not type(result) == 'table' then\n" + " argent.log('error', fmt('%q returned %q instead of %q', filename, type(result), 'table'))\n" + " argent.log('warn', fmt('%q will not result in file output!', filename))\n" + " return\n" + " end\n" + "\n" + " local html\n" + " if result.html then\n" + " html = result.html\n" + " elseif result.markdown then\n" + " html = argent.markdown(result.markdown)\n" + " else\n" + " argent.log('error', fmt('%q did not specify any content!', filename))\n" + " argent.log('warn', fmt('%q will not result in file output!', filename))\n" + " return\n" + " end\n" + "\n" + " if not result.title then\n" + " result.title = string.gsub(filename, '%.lua$', '')\n" + " argent.log('warn', fmt('%q did not specify a title; using default title of %q',\n" + " filename, result.title))\n" + " end\n" + "\n" + " local layout = function(html, tbl) return html end\n" + " if result.layout then\n" + " if not argent.layouts[result.layout] then\n" + " argent.log('error', fmt('%q requested nonexistent layout %q', filename, result.layout))\n" + " argent.log('warn', fmt('%q will not result in file output!', filename))\n" + " return\n" + " else\n" + " layout = argent.layouts[result.layout]\n" + " end\n" + " else -- no layout specified\n" + " argent.log('warn', fmt(\n" + " '%q did not specify a layout; output will be naked content!',\n" + " filename))\n" + " end\n" + "\n" + " local output_data = layout(html, result)\n" + " local output_name = string.gsub(filename, '%.lua$', '.html')\n" + " local output_file = io.open(argent.config.output_directory..parent..output_name, 'w')\n" + " output_file:write(output_data)\n" + " output_file:close()\n" " end\n" "\n" "\n" " function process_dir(directory, parent)\n" - " if (is_dotfile(directory) and not argent.config.include[strip_end_slash(directory)])\n" - " or argent.config.exclude[strip_end_slash(directory)]\n" - " then\n" + " if should_ignore(directory, parent) then\n" " argent.log('debug', fmt('will not process directory %q', parent..directory))\n" " return\n" " end\n" diff --git a/src/lua-script/script.lua b/src/lua-script/script.lua index e5da91f..c6c2ecb 100644 --- a/src/lua-script/script.lua +++ b/src/lua-script/script.lua @@ -87,7 +87,7 @@ return function(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 nil, - plugin_directory = add_end_slash(config.layout_directory) or nil, + plugin_directory = add_end_slash(config.plugin_directory) or nil, exclude = Set(config.exclude, strip_end_slash) or {}, include = Set(config.include, strip_end_slash) or {}, keep = Set(config.keep, strip_end_slash) or {}, @@ -114,7 +114,9 @@ return function(config) argent.log('info', fmt('available layouts: %s', set_tostring(argent.layouts))) if argent.config.plugin_directory then - package.path = argent.config.plugin_directory..'?.lua;'..package.path + package.path = + add_end_slash(argent.currentWorkingDirectory()) + ..argent.config.plugin_directory..'?.lua;'..package.path end argent.log('debug', 'end setup') end @@ -210,10 +212,32 @@ return function(config) -- -------------------------------- + function should_include(filename, parent) + for pattern in pairs(argent.config.include) do + if string.match(filename, pattern) or string.match(parent..filename, pattern) then + return true + end + end + return false + end + + function should_ignore(filename, parent) + if is_dotfile(filename) and not should_include(filename, parent) then + return false + end + + for pattern in pairs(argent.config.exclude) do + if string.match(filename, pattern) + or string.match(parent..filename, pattern) then + return true + end + end + + return false + end + function process_file(filename, parent) - if (is_dotfile(filename) and not argent.config.include[filename]) - or argent.config.exclude[filename] - then + if should_ignore(filename, parent) then argent.log('debug', fmt('will not process file %q', parent..filename)) return end @@ -238,14 +262,54 @@ return function(config) return end - print(filename, result) + if not type(result) == 'table' then + argent.log('error', fmt('%q returned %q instead of %q', filename, type(result), 'table')) + argent.log('warn', fmt('%q will not result in file output!', filename)) + return + end + + local html + if result.html then + html = result.html + elseif result.markdown then + html = argent.markdown(result.markdown) + else + argent.log('error', fmt('%q did not specify any content!', filename)) + argent.log('warn', fmt('%q will not result in file output!', filename)) + return + end + + if not result.title then + result.title = string.gsub(filename, '%.lua$', '') + argent.log('warn', fmt('%q did not specify a title; using default title of %q', + filename, result.title)) + end + + local layout = function(html, tbl) return html end + if result.layout then + if not argent.layouts[result.layout] then + argent.log('error', fmt('%q requested nonexistent layout %q', filename, result.layout)) + argent.log('warn', fmt('%q will not result in file output!', filename)) + return + else + layout = argent.layouts[result.layout] + end + else -- no layout specified + argent.log('warn', fmt( + '%q did not specify a layout; output will be naked content!', + filename)) + end + + local output_data = layout(html, result) + local output_name = string.gsub(filename, '%.lua$', '.html') + local output_file = io.open(argent.config.output_directory..parent..output_name, 'w') + output_file:write(output_data) + output_file:close() 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 + if should_ignore(directory, parent) then argent.log('debug', fmt('will not process directory %q', parent..directory)) return end |