From 87868e1b22c27ebf10646269e89cc2848431c0b6 Mon Sep 17 00:00:00 2001
From: isanae <14251494+isanae@users.noreply.github.com>
Date: Sat, 14 Dec 2019 14:28:17 -0500
Subject: added a few missing consts removed incorrect assert about an empty
 game edition added FileTree to wrap a QTreeView and a FileTreeModel, moved
 some context menu actions over

---
 src/filetree.h | 46 +++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

(limited to 'src/filetree.h')

diff --git a/src/filetree.h b/src/filetree.h
index 108c5843..41592d82 100644
--- a/src/filetree.h
+++ b/src/filetree.h
@@ -1,8 +1,12 @@
+#ifndef MODORGANIZER_FILETREE_INCLUDED
+#define MODORGANIZER_FILETREE_INCLUDED
+
 #include "directoryentry.h"
 #include "iconfetcher.h"
 #include <QAbstractItemModel>
 
 class OrganizerCore;
+class PluginContainer;
 
 class FileTreeItem
 {
@@ -96,6 +100,7 @@ public:
   QVariant data(const QModelIndex& index, int role=Qt::DisplayRole) const override;
   QVariant headerData(int i, Qt::Orientation ori, int role=Qt::DisplayRole) const override;
   Qt::ItemFlags flags(const QModelIndex& index) const override;
+  FileTreeItem* itemFromIndex(const QModelIndex& index) const;
 
 private:
   enum class FillFlag
@@ -131,7 +136,6 @@ private:
 
   std::wstring makeModName(const MOShared::FileEntry& file, int originID) const;
 
-  FileTreeItem* itemFromIndex(const QModelIndex& index) const;
   void ensureLoaded(FileTreeItem* item) const;
   void updatePendingIcons();
 
@@ -143,3 +147,43 @@ private:
 
 Q_DECLARE_OPERATORS_FOR_FLAGS(FileTreeModel::Flags);
 Q_DECLARE_OPERATORS_FOR_FLAGS(FileTreeItem::Flags);
+
+
+class FileTree
+{
+public:
+  FileTree(OrganizerCore& core, PluginContainer& pc, QTreeView* tree);
+
+  void setFlags(FileTreeModel::Flags flags);
+  void refresh();
+
+  void open();
+  void openHooked();
+  void preview();
+
+  void addAsExecutable();
+  void exploreOrigin();
+  void openModInfo();
+
+  void toggleVisibility();
+  void hide();
+  void unhide();
+
+  void dumpToFile();
+
+private:
+  OrganizerCore& m_core;
+  PluginContainer& m_plugins;
+  QTreeView* m_tree;
+  FileTreeModel* m_model;
+
+  FileTreeItem* singleSelection();
+  void onContextMenu(const QPoint &pos);
+
+  void addDirectoryMenus(QMenu& menu, FileTreeItem& item);
+  void addFileMenus(QMenu& menu, const MOShared::FileEntry& file, int originID);
+  void addOpenMenus(QMenu& menu, const MOShared::FileEntry& file);
+  void addCommonMenus(QMenu& menu);
+};
+
+#endif // MODORGANIZER_FILETREE_INCLUDED
-- 
cgit v1.3.1