io.lines() iterate over the standard input file

1 files changed, 23 insertions, 12 deletions

diff --git a/liolib.c b/liolib.c
index d0799122..13d63d33 100644
--- 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