#include "starfieldunmanagedmods.h"

#include "log.h"

#include <QFile>
#include <QJsonArray>
#include <QJsonDocument>
#include <QJsonValue>

StarfieldUnmanagedMods::StarfieldUnmanagedMods(const GameStarfield* game,
                                               const QString& appDataFolder)
    : GamebryoUnmangedMods(game), m_AppDataFolder(appDataFolder)
{}

StarfieldUnmanagedMods::~StarfieldUnmanagedMods() {}

QStringList StarfieldUnmanagedMods::mods(bool onlyOfficial) const
{
  QStringList result;

  QStringList pluginList   = game()->primaryPlugins() + game()->enabledPlugins();
  QStringList otherPlugins = game()->DLCPlugins();
  otherPlugins.append(game()->CCPlugins());
  for (QString plugin : otherPlugins) {
    pluginList.removeAll(plugin);
  }
  QMap<QString, QDir> directories = {{"data", game()->dataDirectory()}};
  directories.insert(game()->secondaryDataDirectories());
  for (QDir directory : directories) {
    for (const QString& fileName : directory.entryList({"*.esp", "*.esl", "*.esm"})) {
      if (!pluginList.contains(fileName, Qt::CaseInsensitive)) {
        if (!onlyOfficial || pluginList.contains(fileName, Qt::CaseInsensitive)) {
          result.append(fileName.chopped(4));  // trims the extension off
        }
      }
    }
  }

  return result;
}

QFileInfo StarfieldUnmanagedMods::referenceFile(const QString& modName) const
{
  QFileInfoList files;
  QMap<QString, QDir> directories = {{"data", game()->dataDirectory()}};
  directories.insert(game()->secondaryDataDirectories());
  for (QDir directory : directories) {
    files += directory.entryInfoList(QStringList() << modName + ".es*");
  }
  if (files.size() > 0) {
    return files.at(0);
  } else {
    return QFileInfo();
  }
}

std::map<QString, StarfieldUnmanagedMods::ContentCatalog>
StarfieldUnmanagedMods::parseContentCatalog() const
{
  QFile content(m_AppDataFolder + "/" + game()->gameShortName() +
                "/ContentCatalog.txt");
  std::map<QString, StarfieldUnmanagedMods::ContentCatalog> contentCatalog;
  if (content.open(QIODevice::OpenModeFlag::ReadOnly)) {
    auto contentData      = content.readAll();
    QString convertedData = QString::fromLatin1(contentData);
    contentData           = convertedData.toUtf8();
    QJsonParseError jsonError;
    QJsonDocument contentDoc = QJsonDocument::fromJson(contentData, &jsonError);
    if (jsonError.error) {
      MOBase::log::warn(QObject::tr("ContentCatalog.txt appears to be corrupt: %1")
                            .arg(jsonError.errorString()));
    } else {
      QJsonObject contentObj = contentDoc.object();
      for (const auto& mod : contentObj.keys()) {
        if (mod == "ContentCatalog")
          continue;
        auto modInfo = contentObj.value(mod).toObject();
        QStringList pluginList;
        QStringList files;
        for (const auto& file : modInfo.value("Files").toArray()) {
          QString fileName = file.toString();
          files.append(fileName);
          if (fileName.endsWith(".esm", Qt::CaseInsensitive) ||
              fileName.endsWith(".esl", Qt::CaseInsensitive) ||
              fileName.endsWith(".esp", Qt::CaseInsensitive)) {
            pluginList.append(fileName);
          }
        }
        for (const auto& plugin : pluginList) {
          contentCatalog[plugin]       = ContentCatalog();
          contentCatalog[plugin].files = files;
          contentCatalog[plugin].name  = modInfo.value("Title").toString();
        }
      }
    }
  }
  return contentCatalog;
}

QStringList StarfieldUnmanagedMods::secondaryFiles(const QString& modName) const
{
  QStringList files;
  auto contentCatalog = parseContentCatalog();
  for (const auto& mod : contentCatalog) {
    if (mod.first.startsWith(modName, Qt::CaseInsensitive)) {
      files += mod.second.files;
      break;
    }
  }
  // file extension in FO4 is .ba2 instead of bsa
  QMap<QString, QDir> directories = {{"data", game()->dataDirectory()}};
  directories.insert(game()->secondaryDataDirectories());
  for (QDir directory : directories) {
    for (const QString& archiveName : directory.entryList({modName + "*.ba2"})) {
      files.append(directory.absoluteFilePath(archiveName));
    }
  }
  return files;
}

QString StarfieldUnmanagedMods::displayName(const QString& modName) const
{
  auto contentCatalog = parseContentCatalog();
  for (const auto& mod : contentCatalog) {
    if (mod.first.startsWith(modName, Qt::CaseInsensitive)) {
      return mod.second.name;
    }
  }
  // unlike in earlier games, in fallout 4 the file name doesn't correspond to
  // the public name
  return modName;
}