Initial community commit

Src/Wasabi/bfc/file/readdir.cpp

@@ -0,0 +1,129 @@
+#include "precomp_wasabi_bfc.h"
+#include "readdir.h"
+
+#ifdef _WIN32
+#include <shlwapi.h>
+#endif
+
+#if !defined(WIN32) && !defined(LINUX)
+#error port me
+#endif
+
+//PORT
+ReadDir::ReadDir( const wchar_t *_path, const wchar_t *_match, bool _skipdots ) : skipdots( _skipdots ), first( 1 ), path( _path ), match( _match )
+{
+    files = INVALID_HANDLE_VALUE;
+    if ( match.isempty() )
+        match = MATCHALLFILES;
+    ZERO( data );
+
+}
+
+ReadDir::~ReadDir()
+{
+    //PORT
+#ifdef WIN32
+    if ( files != INVALID_HANDLE_VALUE ) FindClose( files );
+#endif
+#ifdef LINUX
+    if ( d != NULL ) closedir( d );
+#endif
+}
+
+int ReadDir::next()
+{
+    //PORT
+#ifdef WIN32
+    for ( ;;)
+    {
+        if ( first )
+        {
+            wchar_t fullpath[ MAX_PATH ];
+            PathCombineW( fullpath, path.getValue(), match.getValue() );
+            files = FindFirstFileW( fullpath, &data );
+        }
+        if ( files == INVALID_HANDLE_VALUE ) return 0;
+        if ( first )
+        {
+            first = 0;
+            if ( skipdots && ( isDotDir() || isDotDotDir() ) ) continue;
+            return 1;
+        }
+        if ( !FindNextFileW( files, &data ) ) return 0;
+
+        if ( skipdots && ( isDotDir() || isDotDotDir() ) ) continue;
+        return 1;
+    }
+#endif//WIN32
+#ifdef LINUX
+
+    path.AddBackslash();
+    if ( first || d == NULL )
+    {
+        if ( !( d = opendir( path ) ) ) return 0;
+        first = 0;
+    }
+
+    while ( 1 )
+    {
+        de = readdir( d );
+        if ( !de )
+        {
+            closedir( d );
+            d = NULL;
+            return 0;
+        }
+
+        StringW full;
+        full.printf( L"%s%s", path.v(), de->d_name );
+
+        if ( stat( full, &st ) == -1 )
+            continue;
+
+        if ( skipdots && ( isDotDir() || isDotDotDir() ) ) continue;
+
+        if ( !Std::match( match, de->d_name ) ) continue;
+
+        return 1;
+    }
+
+#endif
+}
+
+const wchar_t *ReadDir::getFilename()
+{
+    if ( first ) if ( !next() ) return NULL;
+    //PORT
+    return data.cFileName;
+
+}
+
+int ReadDir::isDir()
+{
+    //PORT
+    if ( files == INVALID_HANDLE_VALUE ) return 0;
+    return !!( data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY );
+}
+
+int ReadDir::isReadonly()
+{
+    //PORT
+    if ( files == INVALID_HANDLE_VALUE ) return 0;
+    return !!( data.dwFileAttributes & FILE_ATTRIBUTE_READONLY );
+
+}
+
+int ReadDir::isDotDir()
+{
+    //PORT
+    if ( files == INVALID_HANDLE_VALUE ) return 0;
+    return ( data.cFileName[ 0 ] == '.' && data.cFileName[ 1 ] == 0 );
+}
+
+int ReadDir::isDotDotDir()
+{
+    //PORT
+
+    if ( files == INVALID_HANDLE_VALUE ) return 0;
+    return ( data.cFileName[ 0 ] == '.' && data.cFileName[ 1 ] == '.' && data.cFileName[ 2 ] == 0 );
+}