Pit Lib Directory Stuff

wiki

Directory Code

-- dofile for every lua file in folder

function dodirectory(directory, mask)

  mask = mask or "*.lua" -- default extension is lua

  local list = External_ReadDirectory(directory.."\\"..mask)

  local i=1

  sort(list)

  while list[i] do

    dofile(directory.."\\"..list[i])

    i=i+1

  end

end



-- user function for every file in (current) folder

-- example: fordirectory(print) or fordirectory(dofile, "*.lua") 

--          or fordirectory(execute, "*.bat", "C:\\")

function fordirectory(f, mask, directory)

  local prefix = "" -- default folder is current folder

  if directory then prefix = directory.."\\" end

  mask = mask or "*.*"  -- default mask is "all files with extension"

  local list = External_ReadDirectory(prefix..mask)

  local i=1

  sort(list)

  while list[i] do

    f(prefix..list[i])

    i=i+1

  end

end

Here is Dirk Ringes implementation of the Read-Directory function that i used for these functions in the stdlib:

#include "io.h"



static int External_ReadDirectory(lua_State* pLuaState)

{

  const char *mask;

  struct _finddata_t c_file;

  long hFile;

  int i = 1;



  if(lua_isstring(pLuaState, 1))

    mask = lua_tostring(pLuaState, 1);

  else

    mask = "*.*";



  lua_newtable(pLuaState);



  hFile = _findfirst(mask, &c_file);

  if(hFile != -1)

  {

    lua_pushstring(pLuaState, c_file.name);

    lua_rawseti(pLuaState, -2, i);

 

    while(_findnext(hFile, &c_file) == 0)

    {

      i ++;



      lua_pushstring(pLuaState, c_file.name);

      lua_rawseti(pLuaState, -2, i);

    }



    _findclose(hFile);

  }



  return 1;

}

See also LuaRecipes for more detailed info on accessing file/directory listings from Lua.