/*
* POSIX library for Lua 5.1, 5.2 & 5.3.
* (c) Gary V. Vaughan <gary@vaughan.pe>, 2013-2015
* (c) Reuben Thomas <rrt@sc3d.org> 2010-2013
* (c) Natanael Copa <natanael.copa@gmail.com> 2008-2010
* Clean up and bug fixes by Leo Razoumov <slonik.az@gmail.com> 2006-10-11
* Luiz Henrique de Figueiredo <lhf@tecgraf.puc-rio.br> 07 Apr 2006 23:17:49
* Based on original by Claudio Terra for Lua 3.x.
* With contributions by Roberto Ierusalimschy.
* With documentation from Steve Donovan 2012
*/
/***
Directory Iterators.
@module posix.dirent
*/
#include <config.h>
#include <dirent.h>
#include "_helpers.c"
/***
Contents of directory.
@function dir
@string[opt="."] path directory to act on
@treturn table contents of *path*
@see dir.lua
*/
static int
Pdir(lua_State *L)
{
const char *path = optstring(L, 1, ".");
DIR *d;
checknargs(L, 1);
d = opendir(path);
/* Throw an argument error for consistency with eg. io.lines */
if (d == NULL)
{
const char *msg = strerror (errno);
msg = lua_pushfstring(L, "%s: %s", path, msg);
return luaL_argerror(L, 1, msg);
}
else
{
int i;
struct dirent *entry;
lua_newtable(L);
for (i=1; (entry = readdir(d)) != NULL; i++)
{
lua_pushstring(L, entry->d_name);
lua_rawseti(L, -2, i);
}
closedir(d);
lua_pushinteger(L, i-1);
return 2;
}
}
static int
aux_files(lua_State *L)
{
DIR **p = (DIR **)lua_touserdata(L, lua_upvalueindex(1));
DIR *d = *p;
struct dirent *entry;
if (d == NULL)
return 0;
entry = readdir(d);
if (entry == NULL)
{
closedir(d);
*p=NULL;
return 0;
}
return pushstringresult(entry->d_name);
}
static int
dir_gc (lua_State *L)
{
DIR *d = *(DIR **)lua_touserdata(L, 1);
if (d!=NULL)
closedir(d);
return 0;
}
/***
Iterator over all files in named directory.
@function files
@string[opt="."] path directory to act on
@return an iterator
*/
static int
Pfiles(lua_State *L)
{
const char *path = optstring(L, 1, ".");
DIR **d;
checknargs(L, 1);
d = (DIR **)lua_newuserdata(L, sizeof(DIR *));
*d = opendir(path);
/* Throw an argument error for consistency with eg. io.lines */
if (*d == NULL)
{
const char *msg = strerror (errno);
msg = lua_pushfstring(L, "%s: %s", path, msg);
return luaL_argerror(L, 1, msg);
}
if (luaL_newmetatable(L, PACKAGE_NAME " dir handle"))
{
lua_pushcfunction(L, dir_gc);
lua_setfield(L, -2, "__gc");
}
lua_setmetatable(L, -2);
lua_pushcclosure(L, aux_files, 1);
return 1;
}
static const luaL_Reg posix_dirent_fns[] =
{
LPOSIX_FUNC( Pdir ),
LPOSIX_FUNC( Pfiles ),
{NULL, NULL}
};
LUALIB_API int
luaopen_posix_dirent(lua_State *L)
{
luaL_register(L, "posix.dirent", posix_dirent_fns);
lua_pushliteral(L, "posix.dirent for " LUA_VERSION " / " PACKAGE_STRING);
lua_setfield(L, -2, "version");
return 1;
}