1 files changed, 180 insertions, 0 deletions
diff --git a/libs/installer_bsplugins/src/TESData/FormParser.cpp b/libs/installer_bsplugins/src/TESData/FormParser.cpp
new file mode 100644
index 0000000..08fc06b
--- /dev/null
+++ b/libs/installer_bsplugins/src/TESData/FormParser.cpp
@@ -0,0 +1,180 @@
+#include "FormParser.h"
+
+#include <bit>
+#include <ranges>
+#include <utility>
+
+namespace TESData
+{
+
+auto FormParserManager::registrationMap() -> RegistrationMap&
+{
+  static RegistrationMap registrationMap;
+  return registrationMap;
+}
+
+std::shared_ptr<const IFormParser> FormParserManager::getParser(Game game,
+                                                                TESFile::Type type)
+{
+  if (game == Game::SSE) {
+    if (const auto it = registrationMap().find(type); it != registrationMap().end()) {
+      return it->second;
+    }
+  }
+
+  return registrationMap()[TESFile::Type()];
+}
+
+QString readBytes(std::istream& stream, int size)
+{
+  QString data;
+  for (int i = 0; i < size; ++i) {
+    char ch;
+    stream.get(ch);
+    if (stream.eof()) {
+      break;
+    }
+    data += u"%1 "_s.arg(static_cast<std::uint8_t>(ch), 2, 16, QChar(u'0')).toUpper();
+  }
+  return data;
+}
+
+QString readLstring(bool localized, std::istream& stream)
+{
+  if (localized) {
+    const std::uint32_t index = TESFile::readType<std::uint32_t>(stream);
+    if (index == 0) {
+      return u""_s;
+    }
+    return u"<lstring:%1>"_s.arg(index);
+  } else {
+    std::string str;
+    std::getline(stream, str, '\0');
+    return QString::fromStdString(str);
+  }
+}
+
+QString readFormId(std::span<const std::string> masters,
+                          const std::string& plugin, std::istream& stream)
+{
+  const std::uint32_t formId = TESFile::readType<std::uint32_t>(stream);
+  if (!formId) {
+    return u"NONE"_s;
+  }
+
+  const std::uint8_t localIndex = formId >> 24U;
+  const std::string& file = localIndex < masters.size() ? masters[localIndex] : plugin;
+  return u"%2 | %1"_s.arg(formId & 0xFFFFFFU, 6, 16, QChar(u'0'))
+      .toUpper()
+      .arg(QString::fromStdString(file));
+}
+
+static void parseUnknown(DataItem* parent, int& index, int fileIndex,
+                         TESFile::Type signature, std::istream& stream)
+{
+  DataItem* item = nullptr;
+  for (int i = index; i < parent->numChildren(); ++i) {
+    const auto child = parent->childAt(i);
+    if (child->signature() == signature) {
+      item  = child;
+      index = i + 1;
+      break;
+    }
+  }
+
+  if (item == nullptr) {
+    item = parent->insertChild(index++, signature, u"Unknown"_s,
+                               DataItem::ConflictType::Override);
+  }
+
+  item->setData(fileIndex, readBytes(stream, 256));
+}
+
+static void
+pushItem(DataItem*& item, std::vector<int>& indexStack, const QString& name,
+         DataItem::ConflictType conflictType = DataItem::ConflictType::Override,
+         bool alignable                      = true)
+{
+  if (alignable) {
+    item = item->getOrInsertChild(indexStack.back()++, name, conflictType);
+  } else {
+    item = item->insertChild(indexStack.back()++, name, conflictType);
+  }
+  indexStack.push_back(0);
+}
+
+static void
+pushItem(DataItem*& item, std::vector<int>& indexStack, TESFile::Type signature,
+         const QString& name,
+         DataItem::ConflictType conflictType = DataItem::ConflictType::Override,
+         bool alignable                      = true)
+{
+  if (alignable) {
+    item = item->getOrInsertChild(indexStack.back()++, signature, name, conflictType);
+  } else {
+    item = item->insertChild(indexStack.back()++, signature, name, conflictType);
+  }
+  indexStack.push_back(0);
+}
+
+static void popItem(DataItem*& item, std::vector<int>& indexStack)
+{
+  item = item->parent();
+  indexStack.pop_back();
+}
+
+[[nodiscard]] static QString readZstring(std::istream& stream)
+{
+  return QString::fromStdString(TESFile::readZstring(stream));
+}
+
+template <std::integral T>
+[[nodiscard]] static QString readWstring(std::istream& stream)
+{
+  const T length = TESFile::readType<T>(stream);
+  std::string str;
+  str.resize(length);
+  stream.read(str.data(), length);
+  return QString::fromStdString(str);
+}
+
+template <>
+void FormParser<>::parseFlags(DataItem* root, int fileIndex, std::uint32_t flags) const
+{
+  DataItem* item = root->getOrInsertChild(0, u"Record Flags"_s);
+
+  item = item->getOrInsertChild(0, u"Compressed"_s);
+  if (flags & (TESFile::RecordFlags::Compressed)) {
+    item->setData(fileIndex, u"Compressed"_s);
+  }
+}
+
+template <>
+ParseTask FormParser<>::parseForm(DataItem* root, int fileIndex,
+                                  [[maybe_unused]] bool localized,
+                                  [[maybe_unused]] std::span<const std::string> masters,
+                                  [[maybe_unused]] const std::string& plugin,
+                                  const TESFile::Type& signature,
+                                  std::istream* const& stream) const
+{
+  int index = 1;
+  for (;;) {
+    if (signature == "EDID"_ts) {
+      std::string editorId;
+      std::getline(*stream, editorId, '\0');
+      root->getOrInsertChild(index++, "EDID"_ts, u"Editor ID"_s)
+          ->setData(fileIndex, QString::fromStdString(editorId));
+    } else {
+      parseUnknown(root, index, fileIndex, signature, *stream);
+    }
+
+    co_await std::suspend_always();
+  }
+}
+
+}  // namespace TESData
+
+#pragma warning(push)
+#pragma warning(disable : 4456)
+#include "FormParser.SSE.inl"
+#pragma warning(pop)