New method Directory::findFiles to search files inside directory recu…

…rsively (#5) New method Directory::findFiles to search files inside directory recursively
kusharami · Aug 18, 2022 · 74f1b66 · 74f1b66
1 parent ed1dde1
commit 74f1b66
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/BananaCore/Directory.cpp b/BananaCore/Directory.cpp
@@ -82,6 +82,35 @@ QString Directory::getAbsoluteFilePathFor(
 	return QDir::cleanPath(path);
 }
 
+QObjectList Directory::findFiles(const QStringList &filters) const
+{
+	QObjectList result;
+	auto projDir = dynamic_cast<AbstractProjectDirectory *>(getTopDirectory());
+	if (projDir)
+		findFilesRecursive(result, projDir, QDir(getFilePath()), filters);
+	return result;
+}
+
+void Directory::findFilesRecursive(QObjectList &result,
+	AbstractProjectDirectory *projectDirectory, QDir currentDir,
+	const QStringList &filters) const
+{
+	for (auto &entry : currentDir.entryInfoList(
+			 QDir::Files | QDir::Dirs | QDir::NoDotAndDotDot))
+	{
+		if (entry.isDir())
+		{
+			findFilesRecursive(result, projectDirectory,
+				QDir(entry.absoluteFilePath()), filters);
+		} else
+		{
+			if (filters.contains(entry.suffix()))
+				result.append(projectDirectory->addExistingFile(
+					entry.absoluteFilePath(), false));
+		}
+	}
+}
+
 Directory::Directory(const QString &name)
 	: AbstractDirectory(this)
 	, searched(false)

diff --git a/BananaCore/Directory.h b/BananaCore/Directory.h
@@ -32,10 +32,12 @@ SOFTWARE.
 
 #include <memory>
 #include <set>
+#include <QDir>
 
 namespace Banana
 {
 class AbstractFile;
+class AbstractProjectDirectory;
 class Directory
 	: public ObjectGroup
 	, public AbstractDirectory
@@ -58,6 +60,7 @@ class Directory
 		const QString &path, bool search = false) const;
 	Q_INVOKABLE inline QString getRelativeFilePathFor(
 		const QString &path) const;
+	Q_INVOKABLE QObjectList findFiles(const QStringList &filters) const;
 
 	enum class Error
 	{
@@ -155,6 +158,9 @@ class Directory
 		Qt::CaseSensitivity sensitivity) const override;
 	virtual void sortChildren(QObjectList &children) override;
 	virtual void childEvent(QChildEvent *event) override;
+	void findFilesRecursive(QObjectList &result,
+		AbstractProjectDirectory *projectDirectory, QDir currentDir,
+		const QStringList &filters) const;
 
 private:
 	int searchOrder;