Glenn Maynard
115 lines
3.2 KiB
C++
115 lines
3.2 KiB
C++
#ifndef RAGE_UTIL_FILEDB
|
|
#define RAGE_UTIL_FILEDB 1
|
|
|
|
#include <set>
|
|
#include <map>
|
|
#include "RageTimer.h"
|
|
#include "RageFileManager.h"
|
|
|
|
struct FileSet;
|
|
struct File
|
|
{
|
|
CString name;
|
|
CString lname;
|
|
|
|
void SetName( const CString &fn )
|
|
{
|
|
name = fn;
|
|
lname = name;
|
|
lname.MakeLower();
|
|
}
|
|
|
|
bool dir;
|
|
int size;
|
|
/* Modification time of the file. The contents of this is undefined, except that
|
|
* when the file has been modified, this value will change. */
|
|
int hash;
|
|
|
|
/* Private data, for RageFileDrivers. */
|
|
void *priv;
|
|
|
|
/* If this is non-NULL, and dir is true, this is a pointer to the FileSet containing
|
|
* the directory contents. (This is a cache; it isn't always set.) */
|
|
FileSet *dirp;
|
|
|
|
File() { dir=false; dirp=NULL; size=-1; hash=-1; priv=NULL;}
|
|
File( const CString &fn )
|
|
{
|
|
SetName( fn );
|
|
dir=false; size=-1; hash=-1; priv=NULL;
|
|
}
|
|
|
|
bool operator== (const File &rhs) const { return lname==rhs.lname; }
|
|
bool operator< (const File &rhs) const { return lname<rhs.lname; }
|
|
|
|
bool equal(const File &rhs) const { return lname == rhs.lname; }
|
|
bool equal(const CString &rhs) const
|
|
{
|
|
CString l = rhs;
|
|
l.MakeLower();
|
|
return lname == l;
|
|
}
|
|
};
|
|
|
|
/* This represents a directory. */
|
|
struct FileSet
|
|
{
|
|
set<File> files;
|
|
RageTimer age;
|
|
void GetFilesMatching(
|
|
const CString &beginning, const CString &containing, const CString &ending,
|
|
vector<CString> &out, bool bOnlyDirs) const;
|
|
void GetFilesEqualTo(const CString &pat, vector<CString> &out, bool bOnlyDirs) const;
|
|
|
|
RageFileManager::FileType GetFileType( const CString &path ) const;
|
|
int GetFileSize(const CString &path) const;
|
|
int GetFileHash(const CString &path) const;
|
|
};
|
|
|
|
class FilenameDB
|
|
{
|
|
protected:
|
|
FileSet *GetFileSet( CString dir, bool create=true );
|
|
|
|
/* Directories we have cached: */
|
|
map<CString, FileSet *> dirs;
|
|
|
|
int ExpireSeconds;
|
|
|
|
void GetFilesEqualTo(const CString &dir, const CString &fn, vector<CString> &out, bool bOnlyDirs);
|
|
void GetFilesMatching(const CString &dir,
|
|
const CString &beginning, const CString &containing, const CString &ending,
|
|
vector<CString> &out, bool bOnlyDirs);
|
|
void AddFileSet( CString sPath, FileSet *fs );
|
|
|
|
/* The given path wasn't cached. Cache it. */
|
|
virtual void PopulateFileSet( FileSet &fs, const CString &sPath ) { }
|
|
|
|
public:
|
|
FilenameDB::FilenameDB():
|
|
ExpireSeconds( -1 ) { }
|
|
virtual FilenameDB::~FilenameDB() { FlushDirCache(); }
|
|
|
|
void AddFile( const CString &sPath, int size, int hash, void *priv=NULL );
|
|
void DelFile( const CString &sPath );
|
|
File *GetFile( const CString &path );
|
|
|
|
/* This handles at most two * wildcards. If we need anything more complicated,
|
|
* we'll need to use fnmatch or regex. */
|
|
void GetFilesSimpleMatch(const CString &dir, const CString &fn, vector<CString> &out, bool bOnlyDirs);
|
|
|
|
/* Search for "path" case-insensitively and replace it with the correct
|
|
* case. If only a portion of the path exists, resolve as much as possible.
|
|
* Return true if the entire path was matched. */
|
|
bool ResolvePath( CString &path );
|
|
|
|
RageFileManager::FileType GetFileType( const CString &path );
|
|
int GetFileSize(const CString &path);
|
|
int GetFileHash( const CString &sFilePath );
|
|
void GetDirListing( CString sPath, CStringArray &AddTo, bool bOnlyDirs, bool bReturnPathToo );
|
|
|
|
void FlushDirCache();
|
|
};
|
|
|
|
#endif
|