fix tables, getDirectoryContent replaces directoyListFiles

This commit is contained in:
Syping 2019-10-24 09:07:32 +02:00
parent 1f0737a981
commit 96f0511b0f
4 changed files with 82 additions and 26 deletions

View file

@ -38,8 +38,11 @@ LuaEngineIO::LuaEngineIO(QObject *parent, bool loadBaseLibraries) : LuaEngine(pa
void LuaEngineIO::pushClass(lua_State *L_p)
{
// Directory
pushFunction(L_p, "directoryListFiles", directoryListFiles);
pushFunction(L_p, "getDirectoryContent", getDirectoryContent);
pushFunction(L_p, "getDirectoryPath", getDirectoryPath);
pushVariant(L_p, "Files", (int)1);
pushVariant(L_p, "Directories", (int)2);
pushVariant(L_p, "Subdirectories", (int)4);
// File
pushFunction(L_p, "linkFile", linkFile);
@ -52,16 +55,59 @@ void LuaEngineIO::pushClass(LuaEngine *luaEngine)
pushClass(luaEngine->luaState());
}
int LuaEngineIO::directoryListFiles(lua_State *L_p)
int LuaEngineIO::getDirectoryContent(lua_State *L_p)
{
if (getArgumentCount(L_p) >= 1) {
QStringList directories;
QStringList nameFilters;
QString directory = getVariant(L_p, 1).toString();
QDir::Filters dirFilter = QDir::Dirs | QDir::Files | QDir::NoDotAndDotDot;
QDirIterator::IteratorFlag dirIteratorFlag = QDirIterator::NoIteratorFlags;
if (getArgumentCount(L_p) >= 2) {
nameFilters << getVariant(L_p, 2).toString();
QVariant filterVariant = getVariant(L_p, 2);
if ((QMetaType::Type)filterVariant.type() == QMetaType::QVariantMap) {
QVariantMap filterMap = filterVariant.toMap();
QVariantMap::const_iterator it = filterMap.constBegin();
QVariantMap::const_iterator end = filterMap.constEnd();
while (it != end) {
nameFilters << it.value().toString();
it++;
}
}
else {
QString filterString = filterVariant.toString();
if (!filterString.isEmpty()) {
nameFilters << filterVariant.toString();
}
}
if (getArgumentCount(L_p) >= 3) {
bool ok;
int filterInt = getVariant(L_p, 3).toInt(&ok);
if (ok) {
switch (filterInt) {
case 1:
dirFilter = QDir::Files | QDir::NoDotAndDotDot;
break;
case 2:
dirFilter = QDir::Dirs | QDir::NoDotAndDotDot;
break;
case 4:
case 7:
dirIteratorFlag = QDirIterator::Subdirectories;
break;
case 5:
dirFilter = QDir::Files | QDir::NoDotAndDotDot;
dirIteratorFlag = QDirIterator::Subdirectories;
break;
case 6:
dirFilter = QDir::Dirs | QDir::NoDotAndDotDot;
dirIteratorFlag = QDirIterator::Subdirectories;
break;
}
}
}
}
QDirIterator dirIterator(directory, nameFilters, QDir::Files | QDir::NoDotAndDotDot, QDirIterator::Subdirectories);
QDirIterator dirIterator(directory, nameFilters, dirFilter, dirIteratorFlag);
while (dirIterator.hasNext()) {
directories << dirIterator.next();
}
@ -71,6 +117,18 @@ int LuaEngineIO::directoryListFiles(lua_State *L_p)
return 0;
}
int LuaEngineIO::getDirectoryPath(lua_State *L_p)
{
if (getArgumentCount(L_p) >= 1) {
QFileInfo fileInfo(getVariant(L_p, 1).toString());
if (fileInfo.exists()) {
pushVariant(L_p, fileInfo.absoluteDir().absolutePath());
return 1;
}
}
return 0;
}
int LuaEngineIO::linkFile(lua_State *L_p)
{
if (getArgumentCount(L_p) >= 2) {
@ -104,18 +162,6 @@ int LuaEngineIO::linkFile(lua_State *L_p)
return 1;
}
int LuaEngineIO::getDirectoryPath(lua_State *L_p)
{
if (getArgumentCount(L_p) >= 1) {
QFileInfo fileInfo(getVariant(L_p, 1).toString());
if (fileInfo.exists()) {
pushVariant(L_p, fileInfo.absoluteDir().absolutePath());
return 1;
}
}
return 0;
}
QString LuaEngineIO::nameForPointer(void *pointer)
{
QString nameStorage;

View file

@ -30,9 +30,9 @@ public:
LuaEngineIO(QObject *parent = nullptr, bool loadBaseLibraries = true);
static void pushClass(lua_State *L_p);
static void pushClass(LuaEngine *luaEngine);
static int directoryListFiles(lua_State *L_p);
static int linkFile(lua_State *L_p);
static int getDirectoryContent(lua_State *L_p);
static int getDirectoryPath(lua_State *L_p);
static int linkFile(lua_State *L_p);
private:
lua_State *L;