summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/argent.c1
-rw-r--r--src/bindings.c11
-rw-r--r--src/bindings.h2
-rw-r--r--src/lua-script/script.h82
-rw-r--r--src/lua-script/script.lua82
5 files changed, 160 insertions, 18 deletions
diff --git a/src/argent.c b/src/argent.c
index 2c10472..f057380 100644
--- a/src/argent.c
+++ b/src/argent.c
@@ -45,6 +45,7 @@ int main(int argc, char **argv)
hs_create_table
(L,
hs_str_cfunc("markdown", markdown),
+ hs_str_cfunc("currentWorkingDirectory", current_working_directory),
hs_str_cfunc("scanDirectory", scan_directory),
hs_str_cfunc("createDirectory", create_directory),
hs_str_cfunc("copyFile", copy_file),
diff --git a/src/bindings.c b/src/bindings.c
index 7a94020..45abf6a 100644
--- a/src/bindings.c
+++ b/src/bindings.c
@@ -86,6 +86,17 @@ int markdown(lua_State *L)
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
+int current_working_directory(lua_State *L)
+{
+ char *cwd = getcwd(NULL, 0);
+ lua_pushstring(L, cwd);
+ free(cwd);
+ return 1;
+}
+
+
+/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
+
static void throw_directory_error(lua_State *L, char *dir_name);
static struct dirent *read_dir(lua_State *L, DIR *directory);
diff --git a/src/bindings.h b/src/bindings.h
index 02f33c6..2e4d26c 100644
--- a/src/bindings.h
+++ b/src/bindings.h
@@ -5,6 +5,8 @@
int markdown(lua_State *L);
+int current_working_directory(lua_State *L);
+
// return two arrays, one containing filenames for all directories in the dir
// and the other containing filenames for all regular files
int scan_directory(lua_State *L);
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