1 files changed, 75 insertions, 0 deletions
diff --git a/libs/bsapacker/src/ArchiveNameService.cpp b/libs/bsapacker/src/ArchiveNameService.cpp
new file mode 100644
index 0000000..9cea91b
--- /dev/null
+++ b/libs/bsapacker/src/ArchiveNameService.cpp
@@ -0,0 +1,75 @@
+#include "ArchiveNameService.h"
+
+#include "NexusId.h"
+
+#include <QFileInfo>
+
+namespace BsaPacker
+{
+	ArchiveNameService::ArchiveNameService(const IModContext* modContext)
+		: m_ModContext(modContext)
+	{
+	}
+
+	QString ArchiveNameService::GetFileExtension() const
+	{
+		switch (this->m_ModContext->GetNexusId()) {
+		case NexusId::Morrowind:
+		case NexusId::Oblivion:
+		case NexusId::Fallout3:
+		case NexusId::NewVegas:
+		case NexusId::Skyrim:
+		case NexusId::SkyrimSE:
+		case NexusId::Enderal:
+		case NexusId::EnderalSE:
+			return QStringLiteral(".bsa");
+		case NexusId::Fallout4:
+		case  NexusId::Starfield:
+			return QStringLiteral(".ba2");
+		default:
+			return QString();
+		}
+	}
+
+	QString ArchiveNameService::GetArchiveFullPath(const bsa_archive_type_e type, const IModDto* modDto) const
+	{
+		const QString& pathNoExt(QDir::toNativeSeparators(modDto->Directory() + '/' + modDto->ArchiveName() + this->Infix(type)));
+		const QString& suffix = this->Suffix(pathNoExt);
+		return QDir::toNativeSeparators(pathNoExt + suffix + this->GetFileExtension());
+	}
+
+	QString ArchiveNameService::Infix(const bsa_archive_type_e type) const
+	{
+		switch (type) {
+		case baFO4:
+		case baSF:
+			return QStringLiteral(" - Main");
+		case baSFdds:
+		case baFO4dds:
+			return QStringLiteral(" - Textures");
+		case baTES3:
+		case baTES4:
+		case baFO3:
+		case baSSE:
+		case baNone:
+		default:
+			return QString();
+		};
+	}
+
+	// gets the number to append when there are multiple archives
+	// a way to avoid overwriting any existing files
+	QString ArchiveNameService::Suffix(const QString& pathNoExt) const {
+		int archiveIndex = 0;
+		const QString& fileExt = this->GetFileExtension();
+		QFileInfo fileInfo(pathNoExt + fileExt);
+		while (fileInfo.exists()) {
+			++archiveIndex;
+			fileInfo.setFile(pathNoExt + QString::number(archiveIndex) + fileExt);
+		}
+		if (archiveIndex != 0) {
+			return QString::number(archiveIndex);
+		}
+		return QString();
+	}
+}