1 files changed, 90 insertions, 0 deletions
diff --git a/libs/bsapacker/src/ModDtoFactory.cpp b/libs/bsapacker/src/ModDtoFactory.cpp
new file mode 100644
index 0000000..3faed36
--- /dev/null
+++ b/libs/bsapacker/src/ModDtoFactory.cpp
@@ -0,0 +1,90 @@
+#include <bsapacker/ModDtoFactory.h>
+
+#include "ModDto.h"
+#include "NullModDto.h"
+#include "PackerDialog.h"
+#include <QInputDialog>
+#include <QMessageBox>
+
+namespace BsaPacker
+{
+	const uint16_t FALLOUT_4_NEXUS_ID = 1151;
+	const uint16_t STARFIELD_NEXUS_ID = 4187;
+
+	ModDtoFactory::ModDtoFactory(
+		const IModContext* modContext,
+		IPackerDialog* packerDialog) :
+		m_ModContext(modContext),
+		m_PackerDialog(packerDialog)
+	{
+	}
+
+	std::unique_ptr<IModDto> ModDtoFactory::Create() const
+	{
+		this->m_PackerDialog->RefreshModList();
+		int result = m_PackerDialog->Exec();
+		if (result != QDialog::DialogCode::Accepted)
+		{
+			return std::make_unique<NullModDto>();
+		}
+
+		const int nexusId = this->m_ModContext->GetNexusId();
+		const QString& modName = this->m_PackerDialog->SelectedMod();
+		const QString& modDir = this->m_ModContext->GetAbsoluteModPath(modName);
+		const QString& pluginName = this->m_PackerDialog->SelectedPluginItem();
+		const bool needsNewName = this->m_PackerDialog->IsNewFilename();
+		const QString& archiveName = ModDtoFactory::ArchiveNameValidator(modName, pluginName, needsNewName);
+		if (archiveName == nullptr) {
+			return std::make_unique<NullModDto>();
+		}
+
+		const QString& archiveExtension = (nexusId == FALLOUT_4_NEXUS_ID || nexusId == STARFIELD_NEXUS_ID)
+				? QStringLiteral(".ba2")
+				: QStringLiteral(".bsa");
+
+		return std::make_unique<ModDto>(nexusId, modDir, archiveName, archiveExtension);
+	}
+
+	QString ModDtoFactory::ArchiveNameValidator(
+		const QString& modName,
+		const QString& pluginName,
+		const bool needsNewName)
+	{
+		QString archive_name_base;
+		if (needsNewName) {
+			bool ok = false;
+			const QString& name = QInputDialog::getText(nullptr,
+														QStringLiteral("BSA Packer"),
+														QObject::tr("Archive name (no file extension):"),
+														QLineEdit::Normal,
+														modName,
+														&ok).simplified();
+			if (!ok) {
+				return nullptr;
+			}
+			// Nameless plugins are not loaded in the games I tested. Nameless archives can be loaded in
+			// some games using the SArchiveList INI setting or equivalent. It is simpler to just prevent nameless archives
+			else if (name.isEmpty()) {
+				qWarning("Archive name cannot be empty. Cancelling archive creation.");
+				return nullptr;
+			}
+			archive_name_base = name;
+		} else {
+			archive_name_base = pluginName.chopped(4); // trims the file extension off
+		}
+		return archive_name_base;
+	}
+
+	bool ModDtoFactory::CanOverwriteFile(const QString& filePath,
+										 const QString& fileName)
+	{
+		const QString& absoluteFileName = filePath + '/' + fileName;
+		const QFileInfo fileInfo(absoluteFileName);
+		if (!fileInfo.exists() || !fileInfo.isFile()) {
+			return true;
+		}
+
+		const QString& message = QObject::tr("File \"") + absoluteFileName + QObject::tr("\" already exists. Overwrite?");
+		return QMessageBox::question(nullptr, QStringLiteral("BSA Packer"), message, QMessageBox::Ok | QMessageBox::Cancel) != QMessageBox::Cancel;
+	}
+} // namespace BsaPacker