path: root/src/lua-script/script.lua

return function(config)
   local fmt = string.format

   --------------------------------
   --
   -- 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 set_tostring(set)
      local str = '[empty]'
      for element in pairs(set) do
	 if str == '[empty]' then
	    str = tostring(element)
	 else
	    str = str .. '; '.. tostring(element)
	 end
      end

      return str
   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 load_layouts(directory, parent, layouts)
      local directory = add_end_slash(directory) or ''
      local parent = parent or ''
      local layouts = layouts or {}

      local fullpath = parent..directory
      local dirs, files = argent.scanDirectory(argent.config.layout_directory..fullpath)

      for _, file in ipairs(files) do
	 if string.match(file, '%.lua$') then
	    local basename = string.gsub(file, '%.lua$', '')
	    local id = string.gsub(fullpath..basename, '/', '.')
	    argent.log('debug', fmt('loading layout %q', id))
	    success, result = pcall(loadfile(argent.config.layout_directory..fullpath..file), 0, 1)
	    if not success then
	       argent.log('error', result)
	       argent.log('warn', fmt('layout %q will not be available', id))
	    else
	       layouts[id] = result
	    end
	 end
      end

      for _, dir in ipairs(dirs) do
	 if dir ~= '.' and dir ~= '..' then
	    argent.log('debug', fmt('loading layouts from %q', parent..directory))
	    load_layouts(dir, parent..directory, layouts)
	 end
      end

      return layouts
   end


   function setup(config)
      argent.log('debug', 'begin setup')
      
      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 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 {},
	 noprocess = Set(config.noprocess, strip_end_slash) or {},
	 rss_include = Set(config.rss_include, strip_end_slash) or {},
      }

      argent.log('info', fmt('site directory: %s', argent.config.site_directory))
      argent.log('info', fmt('output directory: %s', argent.config.output_directory))
      argent.log('info', fmt('layout directory: %s', tostring(argent.config.layout_directory)))
      argent.log('info', fmt('plugin directory: %s', tostring(argent.config.plugin_directory)))

      argent.log('info', 'exclusions: '..set_tostring(argent.config.exclude))
      argent.log('info', 'inclusions: '..set_tostring(argent.config.include))
      argent.log('info', 'files to keep: '..set_tostring(argent.config.keep))
      argent.log('info', 'noprocess: '..set_tostring(argent.config.noprocess))
      argent.log('info', 'rss files: '..set_tostring(argent.config.rss_include))

      if argent.config.layout_directory then
	 argent.layouts = load_layouts()
      else
	 argent.layouts = {}
      end
      argent.log('info', fmt('available layouts: %s', set_tostring(argent.layouts)))
      
      if argent.config.plugin_directory then
	 package.path =
	    add_end_slash(argent.currentWorkingDirectory())
	    ..argent.config.plugin_directory..'?.lua;'..package.path
      end
      argent.log('debug', 'end setup')
   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
	 argent.log('debug', fmt('retaining file %q', parent..file))
	 return true
      end

      argent.log('debug', fmt('removing file %q', parent..file))
      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
	 argent.log('debug', fmt('retaining directory %q', parent..dir))
	 return true
      end

      argent.log('debug', fmt('obliterating files in %q', parent..dir))
      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 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 should_ignore(filename, parent) then
	 argent.log('debug', fmt('will not process file %q', parent..filename))
	 return
      end

      if string.match(filename, '%.lua$') and not argent.config.noprocess[filename] then
	 argent.log('debug', fmt('processing %q as lua file', parent..filename))
	 process_lua_file(filename, parent)
      else
	 argent.log('debug', fmt('processing %q as regular file', parent..filename))
	 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

      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 should_ignore(directory, parent) then
	 argent.log('debug', fmt('will not process directory %q', parent..directory))
	 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