commit 669129a6d8210e758ba94ea2786a370946572f7d
parent 46b063ef59c7b334f8f36a9ab5dfd76d094e70d7
Author: Roberto Ierusalimschy <roberto@inf.puc-rio.br>
Date: Fri, 11 Oct 2002 17:40:10 -0300
io.lines() iterate over the standard input file
Diffstat:
M | liolib.c | | | 35 | +++++++++++++++++++++++------------ |
1 file changed, 23 insertions(+), 12 deletions(-)
diff --git a/liolib.c b/liolib.c
@@ -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