1 files changed, 102 insertions, 0 deletions
diff --git a/libs/bsapacker/src/ArchiveAutoService.cpp b/libs/bsapacker/src/ArchiveAutoService.cpp
new file mode 100644
index 0000000..4c89c69
--- /dev/null
+++ b/libs/bsapacker/src/ArchiveAutoService.cpp
@@ -0,0 +1,102 @@
+#include <bsapacker/ArchiveAutoService.h>
+#include <uibase/utility.h>
+#include <QProgressDialog>
+#include <QtConcurrent>
+#include <QLabel>
+#include <QDebug>
+
+#include "NexusId.h"
+
+#ifdef __linux__
+#include <bsa_ffi.h>
+#endif
+
+namespace BsaPacker
+{
+	static const char* gameIdFromNexusId(const int nexusId)
+	{
+		switch (nexusId) {
+		case NexusId::Morrowind: return "morrowind";
+		case NexusId::Oblivion: return "oblivion";
+		case NexusId::Fallout3: return "fo3";
+		case NexusId::NewVegas: return "fonv";
+		case NexusId::Skyrim:
+		case NexusId::Enderal: return "skyrimle";
+		case NexusId::SkyrimSE:
+		case NexusId::EnderalSE: return "skyrimse";
+		case NexusId::Fallout4: return "fo4-fo76";
+		case NexusId::Starfield: return "starfield-v3";
+		default: return "fo4-fo76";
+		}
+	}
+
+	static int includeModeFromArchiveType(const bsa_archive_type_e type)
+	{
+		switch (type) {
+		case bsa_archive_type_e::baFO4: return 1;     // non-dds
+		case bsa_archive_type_e::baFO4dds: return 2;  // dds only
+		case bsa_archive_type_e::baSF: return 1;      // non-dds
+		case bsa_archive_type_e::baSFdds: return 2;   // dds only
+		default: return 0;                            // all files
+		}
+	}
+
+	bool ArchiveAutoService::CreateBSA(libbsarch::bs_archive_auto* archive,
+									   const QString& archiveName,
+									   const bsa_archive_type_e type,
+									   const QString& sourceDir,
+									   const int nexusId) const
+	{
+		const QString hostArchiveName = MOBase::normalizePathForHost(archiveName);
+		const QString hostSourceDir = MOBase::normalizePathForHost(sourceDir);
+		const char* gameId = gameIdFromNexusId(nexusId);
+		const int includeMode = includeModeFromArchiveType(type);
+
+		QProgressDialog savingDialog;
+		savingDialog.setWindowFlags(savingDialog.windowFlags() & ~Qt::WindowCloseButtonHint);
+		savingDialog.setWindowTitle(QObject::tr("Writing Archive"));
+		savingDialog.setCancelButton(0);
+		QLabel text;
+		text.setText(QObject::tr("Writing %1").arg(hostArchiveName));
+		savingDialog.setLabel(&text);
+		savingDialog.setRange(0, 0);
+		savingDialog.show();
+		auto future = QtConcurrent::run([=]() -> bool {
+#ifdef __linux__
+			char* err = bsa_ffi_pack_dir_filtered(
+				hostSourceDir.toUtf8().constData(),
+				hostArchiveName.toUtf8().constData(),
+				gameId,
+				includeMode,
+				nullptr,
+				nullptr);
+			if (err == nullptr) {
+				qDebug() << "packed archive via bsa_ffi for" << hostArchiveName;
+				return true;
+			}
+
+			qWarning() << "bsa_ffi primary pack failed for" << hostArchiveName
+					   << ":" << QString::fromUtf8(err)
+					   << "- falling back to libbsarch";
+			bsa_ffi_string_free(err);
+#endif
+			try {
+				archive->save_to_disk(hostArchiveName.toStdString());
+			} catch (std::exception&) {
+				return false;
+			}
+			return true;
+			});
+		while (!future.isFinished())
+		{
+			QCoreApplication::processEvents();
+		}
+		savingDialog.hide();
+
+		if (future.result()) {
+			return true;
+		}
+
+		return false;
+	}
+} // namespace BsaPacker