130 lines
2.6 KiB
C++
130 lines
2.6 KiB
C++
#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 );
|
|
}
|