1 files changed, 73 insertions, 0 deletions

diff --git a/lib/luacd.cpp b/lib/luacd.cpp
index 97dbbaf..a32ecaf 100644
--- a/lib/luacd.cpp
+++ b/lib/luacd.cpp
@@ -598,6 +598,11 @@ enum direntry_methods_t {
     DIRENTRY_ISXASTR,
     DIRENTRY_ISXAXA,
     DIRENTRY_ISXAFORM1,
+    DIRENTRY_ITERATE,
+};
+
+enum direntry_functions_t {
+    DIRENTRY_ITERATOR = 0,
 };
 
 struct lua_functypes_t direntry_methods[] = {
@@ -610,6 +615,11 @@ struct lua_functypes_t direntry_methods[] = {
     { DIRENTRY_ISXASTR,   "isxastr",   0, 0, 0 },
     { DIRENTRY_ISXAXA,    "isxaxa",    0, 0, 0 },
     { DIRENTRY_ISXAFORM1, "isxaform1", 0, 0, 0 },
+    { DIRENTRY_ITERATE,   "iterate",   1, 1, { BLUA_OBJECT } },
+};
+
+struct lua_functypes_t direntry_functions[] = {
+    { DIRENTRY_ITERATOR,  "iterator",  2, 2, { BLUA_TABLE, BLUA_ANY } },
     { -1, 0, 0, 0, 0 }
 };
 
@@ -624,8 +634,11 @@ class sLua_direntry : public Base {
     DECLARE_METHOD(direntry, DIRENTRY_ISXASTR);
     DECLARE_METHOD(direntry, DIRENTRY_ISXAXA);
     DECLARE_METHOD(direntry, DIRENTRY_ISXAFORM1);
+    DECLARE_METHOD(direntry, DIRENTRY_ITERATE);
+    DECLARE_FUNCTION(direntry, DIRENTRY_ITERATOR);
   private:
     static int direntry_proceed(Lua * L, int n, direntry * obj, int caller);
+    static int direntry_proceed_statics(Lua * L, int n, int caller);
 };
 
 void Luadirentry::pushmembers(Lua * L) {
@@ -640,6 +653,7 @@ void Luadirentry::pushmembers(Lua * L) {
     PUSH_METHOD(direntry, DIRENTRY_ISXASTR);
     PUSH_METHOD(direntry, DIRENTRY_ISXAXA);
     PUSH_METHOD(direntry, DIRENTRY_ISXAFORM1);
+    PUSH_METHOD(direntry, DIRENTRY_ITERATE);
 }
 
 int sLua_direntry::direntry_proceed(Lua * L, int n, direntry * dir, int caller) {
@@ -647,6 +661,7 @@ int sLua_direntry::direntry_proceed(Lua * L, int n, direntry * dir, int caller)
     String index;
     bool has_xa = false;
     Byte * p;
+    int cursor;
 
     s = 33 + dir->N;
     if (s & 1) {
@@ -740,11 +755,69 @@ int sLua_direntry::direntry_proceed(Lua * L, int n, direntry * dir, int caller)
         } else {
             L->push();
         }
+        break;
+    case DIRENTRY_ITERATE:
+        L->push(function_DIRENTRY_ITERATOR);
+        L->newtable();
+        L->push("direntry");
+        L->copy(1);
+        L->settable();
+        L->push("cdutil");
+        L->copy(2);
+        L->settable();
+        L->push("cursor");
+        {
+            int * cursor = (int *) L->newuser(sizeof(int));
+            *cursor = 0;
+        }
+        L->settable();
+        L->push();
+        r = 3;
+        break;
     }
 
     return r;
 }
 
+int sLua_direntry::direntry_proceed_statics(Lua * L, int n, int caller) {
+    int r = 0;
+    cdutils::DirEntry * dir = 0, rdir;
+    cdutils * cdutil = 0;
+    int * cursor = 0;
+    String name;
+    
+    switch (caller) {
+    case DIRENTRY_ITERATOR:
+        L->push("cursor");
+        L->gettable(1);
+        cursor = (int *) L->touserdata();
+        L->pop();
+        L->push("cdutil");
+        L->gettable(1);
+        cdutil = L->recast<cdutils>(3);
+        L->pop();
+        L->push("direntry");
+        L->gettable(1);
+        dir = (cdutils::DirEntry *) LuaObject::getme(L, 3);
+        L->pop();
+        rdir = cdutil->get_next_dir_entry(dir, cursor, name);
+        
+        if ((rdir.R == 0) && (rdir.NExt == 0) && (rdir.Sector == 0) && (rdir.BESector == 0) && (rdir.Size == 0)) {
+            L->push();
+            L->push();
+        } else {
+            L->push(name);
+            dir = (cdutils::DirEntry *) malloc(rdir.R);
+            memcpy(dir, &rdir, rdir.R);
+            Luadirentry ldir(dir);
+            ldir.pushdestruct(L);
+        }
+        r = 2;
+        break;
+    }
+    
+    return r;
+}
 
  /***************************\
 |**  class direntry exports **|