#ifndef MODORGANIZER_LOOT_H
#define MODORGANIZER_LOOT_H

#include "envmodule.h"
#include <QWidget>
#include <log.h>
#include <lootcli/lootcli.h>
#include <windows.h>

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
  {
    bool okay = false;
    std::vector<QString> errors, warnings;
    std::vector<Message> messages;
    std::vector<Plugin> plugins;
    Stats stats;

    QString toMarkdown() const;

  private:
    QString successMarkdown() const;
    QString errorsMarkdown() const;
  };

  Loot(OrganizerCore& core);
  ~Loot();

  bool start(QWidget* parent, bool didUpdateMasterList);
  void cancel();
  bool result() const;

  const QString& reportPath() const;
  const QString& sortedPluginListPath() const;
  const Report& report() const;
  const QString getSortedPluginListMarkdown() const;
  const std::vector<QString>& errors() const;
  const std::vector<QString>& warnings() const;

signals:
  void output(const QString& s);
  void progress(const lootcli::Progress p);
  void log(MOBase::log::Levels level, const QString& s) const;
  void finished();

private:
  OrganizerCore& m_core;
  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;
  std::vector<QString> m_errors, m_warnings;
  Report m_report;

  bool spawnLootcli(QWidget* parent, bool didUpdateMasterList,
                    env::HandlePtr stdoutHandle);

  void lootThread();
  bool waitForCompletion();

  void processStdout(const std::string& lootOut);
  void processMessage(const lootcli::Message& m);

  Report createReport() const;
  void processReport(Report& r) const;
  void deleteReportFile();

  void deleteSortedLoadOrder();

  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