mirrors/lua
1
0
Fork 0
mirror of https://github.com/lua/lua synced 2025-03-19 12:12:52 +03:00
Code Issues Packages Projects Releases Wiki Activity

io.lines() iterate over the standard input file

Browse Source
This commit is contained in:
Roberto Ierusalimschy 2002-10-11 17:40:32 -03:00
parent 46b063ef59
commit 669129a6d8
1 changed files with 23 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
liolib.c
View File

@ -1,5 +1,5 @@
/*
** $Id: liolib.c,v 2.18 2002/09/17 20:35:54 roberto Exp roberto $
** $Id: liolib.c,v 2.19 2002/09/19 20:12:47 roberto Exp roberto $
** Standard I/O (and system) library
** See Copyright Notice in lua.h
*/
@ -194,28 +194,39 @@ static int io_output (lua_State *L) {
static int io_readline (lua_State *L);
static int io_lines (lua_State *L) {
FILE *f = fopen(luaL_check_string(L, 1), "r");
luaL_arg_check(L, f, 1, strerror(errno));
static void aux_lines (lua_State *L, int index, int close) {
lua_pushliteral(L, FILEHANDLE);
lua_rawget(L, LUA_REGISTRYINDEX);
newfile(L, f);
lua_pushboolean(L, 1); /* must close file when finished */
lua_pushvalue(L, index);
lua_pushboolean(L, close); /* close/not close file when finished */
lua_pushcclosure(L, io_readline, 3);
return 1;
}
static int f_lines (lua_State *L) {
tofile(L, 1); /* check that it's a valid file handle */
lua_pushliteral(L, FILEHANDLE);
lua_rawget(L, LUA_REGISTRYINDEX);
lua_pushvalue(L, 1);
lua_pushboolean(L, 0); /* does not close file when finished */
lua_pushcclosure(L, io_readline, 3);
aux_lines(L, 1, 0);
return 1;
}
static int io_lines (lua_State *L) {
if (lua_isnoneornil(L, 1)) { /* no arguments? */
lua_pushstring(L, IO_INPUT);
lua_rawget(L, lua_upvalueindex(1)); /* will iterate over default input */
return f_lines(L);
}
else {
FILE *f = fopen(luaL_check_string(L, 1), "r");
luaL_arg_check(L, f, 1, strerror(errno));
newfile(L, f);
aux_lines(L, lua_gettop(L), 1);
return 1;
}
}
/*
** {======================================================
** READ