1 files changed, 133 insertions, 0 deletions
diff --git a/src/loot.h b/src/loot.h
new file mode 100644
index 00000000..7e5e01bd
--- /dev/null
+++ b/src/loot.h
@@ -0,0 +1,133 @@
+#ifndef MODORGANIZER_LOOT_H
+#define MODORGANIZER_LOOT_H
+
+#include "envmodule.h"
+#include <log.h>
+#include <lootcli/lootcli.h>
+#include <windows.h>
+#include <QWidget>
+
+Q_DECLARE_METATYPE(lootcli::Progress);
+Q_DECLARE_METATYPE(MOBase::log::Levels);
+
+class OrganizerCore;
+class AsyncPipe;
+
+class Loot : public QObject
+{
+  Q_OBJECT;
+
+public:
+  struct Message
+  {
+    MOBase::log::Levels type;
+    QString text;
+
+    QString toMarkdown() const;
+  };
+
+  struct File
+  {
+    QString name;
+    QString displayName;
+  };
+
+  struct Dirty
+  {
+    qint64 crc=0;
+    qint64 itm=0;
+    qint64 deletedReferences=0;
+    qint64 deletedNavmesh=0;
+    QString cleaningUtility;
+    QString info;
+
+    QString toString(bool isClean) const;
+    QString toMarkdown(bool isClean) const;
+    QString cleaningString() const;
+  };
+
+  struct Plugin
+  {
+    QString name;
+    std::vector<File> incompatibilities;
+    std::vector<Message> messages;
+    std::vector<Dirty> dirty, clean;
+    std::vector<QString> missingMasters;
+    bool loadsArchive = false;
+    bool isMaster = false;
+    bool isLightMaster = false;
+
+    QString toMarkdown() const;
+  };
+
+  struct Stats
+  {
+    qint64 time = 0;
+    QString lootcliVersion;
+    QString lootVersion;
+
+    QString toMarkdown() const;
+  };
+
+  struct Report
+  {
+    std::vector<Message> messages;
+    std::vector<Plugin> plugins;
+    Stats stats;
+
+    QString toMarkdown() const;
+  };
+
+
+  Loot();
+  ~Loot();
+
+  bool start(QWidget* parent, OrganizerCore& core, bool didUpdateMasterList);
+  void cancel();
+  bool result() const;
+  const QString& outPath() const;
+  const Report& report() const;
+
+signals:
+  void output(const QString& s);
+  void progress(const lootcli::Progress p);
+  void log(MOBase::log::Levels level, const QString& s);
+  void finished();
+
+private:
+  std::unique_ptr<QThread> m_thread;
+  std::atomic<bool> m_cancel;
+  std::atomic<bool> m_result;
+  env::HandlePtr m_lootProcess;
+  std::unique_ptr<AsyncPipe> m_pipe;
+  std::string m_outputBuffer;
+  Report m_report;
+
+  bool spawnLootcli(
+    QWidget* parent, OrganizerCore& core, bool didUpdateMasterList,
+    env::HandlePtr stdoutHandle);
+
+  void lootThread();
+  bool waitForCompletion();
+
+  void processStdout(const std::string &lootOut);
+  void processMessage(const lootcli::Message& m);
+
+  void processOutputFile();
+
+  Report createReport(const QJsonDocument& doc) const;
+  Message reportMessage(const QJsonObject& message) const;
+  std::vector<Plugin> reportPlugins(const QJsonArray& plugins) const;
+  Loot::Plugin reportPlugin(const QJsonObject& plugin) const;
+  Loot::Stats reportStats(const QJsonObject& stats) const;
+
+  std::vector<Message> reportMessages(const QJsonArray& array) const;
+  std::vector<Loot::File> reportFiles(const QJsonArray& array) const;
+  std::vector<Loot::Dirty> reportDirty(const QJsonArray& array) const;
+  std::vector<QString> reportStringArray(const QJsonArray& array) const;
+};
+
+
+bool runLoot(QWidget* parent, OrganizerCore& core, bool didUpdateMasterList);
+
+#endif // MODORGANIZER_LOOT_H