directory-tree.lua

tex = require "tex";

function _print(...)
  tex.print(...);
  print(...);
end

local directoryTree = {
  exclude = "^%.%l";
}

directoryTree.label = function (attr, name)
  local label = "\\textit{WARNING: NO LABEL}";
  label = name:gsub("_","\\_");
  if attr.mode == "directory" then
    label = "\\dtDirLabel{" .. label .. '/}';
  else
    label = "\\dtFileLabel{" .. label .. '}';
  end
  return label;
end

directoryTree.style = function (attr, path)
  local style = "";
  if attr.mode == "directory" then
    style = "directory";
  else
    style = "file";
  end
  return style;
end

directoryTree.printNode = function (path, level, listoffiles)
  local node_cnt = 0;
  if not level then level = 0; end
  if not listoffiles then listoffiles = {} end
  local attr, msg = lfs.attributes(path);
  if (not attr) and msg then
    tex.error(msg);
  else
    local name = path:gsub(".*/","");
    if name:find(directoryTree.exclude) then
      return node_cnt, listoffiles;
    end
    local label = directoryTree.label(attr, name);
    local style = directoryTree.style(attr, path);
    if level == 0 then
      _print("\\node[" .. style .. "] {" .. label .. "}");
    else
      _print("child{ node[" .. style .. "] {" .. label .. "}");
    end
    if attr.mode == "directory" then
      for entry in lfs.dir(path) do
        if entry ~= "." and entry ~= ".." then
          local n_cnt, listoffiles = directoryTree.printNode(path .. '/' .. entry, level+1, listoffiles);
          node_cnt = node_cnt + n_cnt;
        end
      end
    else
      table.insert(listoffiles, path);
    end
    if level == 0 then
      _print(";");
    else
      _print("}");
      for _=1,node_cnt do
        _print("child[missing] {}");
      end
    end
  end
  return node_cnt+1, listoffiles;
end

directoryTree.createTree = function(path, name)
  local _, listoffiles = directoryTree.printNode(path);
  local files = table.concat(listoffiles, ", ");
  if name then
    _print('\\makeatletter\\gdef\\dt@listoffiles@'..name..'{'..files..'}\\makeatother');
  end
end

return directoryTree;