From c0035ada020fe90085c698c0829f771c8d542031 Mon Sep 17 00:00:00 2001
From: Mikaël Capelle <capelle.mikael@gmail.com>
Date: Sat, 16 May 2020 21:54:01 +0200
Subject: Add QDirFileTree implementation.

---
 src/CMakeLists.txt   |  1 +
 src/qdirfiletree.cpp | 46 ++++++++++++++++++++++++++++++++++++++
 src/qdirfiletree.h   | 63 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 110 insertions(+)
 create mode 100644 src/qdirfiletree.cpp
 create mode 100644 src/qdirfiletree.h

(limited to 'src')

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 99d31541..6625e4b6 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -36,6 +36,7 @@ add_filter(NAME src/core GROUPS
 	organizerproxy
 	apiuseraccount
 	processrunner
+	qdirfiletree
 	uilocker
 )
 
diff --git a/src/qdirfiletree.cpp b/src/qdirfiletree.cpp
new file mode 100644
index 00000000..5570060a
--- /dev/null
+++ b/src/qdirfiletree.cpp
@@ -0,0 +1,46 @@
+#include "qdirfiletree.h"
+
+#include <QDirIterator>
+
+using namespace MOBase;
+
+/**
+ *
+ */
+std::shared_ptr<const QDirFileTree> QDirFileTree::makeTree(QDir directory) {
+  return std::shared_ptr<const QDirFileTree>(new QDirFileTree(nullptr, directory));
+}
+
+/**
+ *
+ */
+QDirFileTree::QDirFileTree(std::shared_ptr<const IFileTree> parent, QDir directory) : FileTreeEntry(parent, directory.dirName()), IFileTree(), qDir(directory) {
+  qDir.setFilter(qDir.filter() | QDir::NoDotAndDotDot);
+}
+
+/**
+ * No mutable operations allowed.
+ */
+bool QDirFileTree::beforeReplace(IFileTree const* dstTree, FileTreeEntry const* destination, FileTreeEntry const* source) { return false; }
+bool QDirFileTree::beforeInsert(IFileTree const* entry, FileTreeEntry const* name) { return false; }
+bool QDirFileTree::beforeRemove(IFileTree const* entry, FileTreeEntry const* name) { return false; }
+std::shared_ptr<FileTreeEntry> QDirFileTree::makeFile(std::shared_ptr<const IFileTree> parent, QString name, QDateTime time) const { return nullptr; }
+std::shared_ptr<IFileTree> QDirFileTree::makeDirectory(std::shared_ptr<const IFileTree> parent, QString name) const { return nullptr; }
+
+void QDirFileTree::doPopulate(std::shared_ptr<const IFileTree> parent, std::vector<std::shared_ptr<FileTreeEntry>>& entries) const {
+  QDirIterator iter(qDir);
+  while (iter.hasNext()) {
+    QString name = iter.next();
+    QFileInfo info = iter.fileInfo();
+    if (info.isDir()) {
+      entries.push_back(std::shared_ptr<QDirFileTree>(new QDirFileTree(parent, QDir(info.absoluteFilePath()))));
+    }
+    else {
+      entries.push_back(createFileEntry(parent, info.fileName(), info.fileTime(QFileDevice::FileModificationTime)));
+    }
+  }
+}
+
+std::shared_ptr<IFileTree> QDirFileTree::doClone() const {
+  return std::shared_ptr<QDirFileTree>(new QDirFileTree(nullptr, qDir));
+}
diff --git a/src/qdirfiletree.h b/src/qdirfiletree.h
new file mode 100644
index 00000000..000ef69a
--- /dev/null
+++ b/src/qdirfiletree.h
@@ -0,0 +1,63 @@
+/*
+Copyright (C) MO2 Team. All rights reserved.
+
+This file is part of Mod Organizer.
+
+Mod Organizer is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+Mod Organizer is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef ARCHIVEFILENETRY_H
+#define ARCHIVEFILENTRY_H
+
+#include <QDir>
+
+#include "ifiletree.h"
+
+
+/**
+ * Class that expose a directory on the drive, using QDir, as a `MOBase::IFileTree`.
+ *
+ * This class does not expose mutable operations, so any mutable operations will
+ * fail.
+ */
+class QDirFileTree : public MOBase::IFileTree {
+public:
+
+  /**
+   * @brief Create a new file tree representing the given directory.
+   *
+   * @param directory Directory to represent.
+   *
+   * @return a file tree representing the given directory.
+   */
+  static std::shared_ptr<const QDirFileTree> makeTree(QDir directory);
+
+protected:
+
+  QDirFileTree(std::shared_ptr<const IFileTree> parent, QDir directory);
+
+  virtual bool beforeReplace(IFileTree const* dstTree, FileTreeEntry const* destination, FileTreeEntry const* source) override;
+  virtual bool beforeInsert(IFileTree const* entry, FileTreeEntry const* name) override;
+  virtual bool beforeRemove(IFileTree const* entry, FileTreeEntry const* name) override;
+
+  virtual std::shared_ptr<FileTreeEntry> makeFile(std::shared_ptr<const IFileTree> parent, QString name, QDateTime time) const override;
+  virtual std::shared_ptr<IFileTree> makeDirectory(std::shared_ptr<const IFileTree> parent, QString name) const override;
+  virtual void doPopulate(std::shared_ptr<const IFileTree> parent, std::vector<std::shared_ptr<FileTreeEntry>>& entries) const override;
+  virtual std::shared_ptr<IFileTree> doClone() const override;
+
+  QDir qDir;
+
+};
+
+#endif
-- 
cgit v1.3.1