seasonal builds

buildarchiveworker.cpp

@@ -5,6 +5,8 @@
 #include <QFile>
 #include <QFileInfo>
 
+#include <algorithm>
+
 #include <QtCore/private/qzipreader_p.h>
 #include <QtCore/private/qzipwriter_p.h>
 
@@ -25,6 +27,40 @@ bool isSafeRelativePath(const QString &path)
     return true;
 }
 
+// Последняя проверка перед записью: куда бы ни указывало имя из архива после
+// всех преобразований, файл обязан оказаться внутри папки игры. Правило имён
+// выше это уже гарантирует, но проверка по готовому пути не зависит от того,
+// как именно QZipReader разбирает записи.
+bool isInside(const QDir &root, const QString &absolutePath)
+{
+    const QString base = QDir::cleanPath(root.absolutePath()) + u'/';
+    return QDir::cleanPath(absolutePath).startsWith(base);
+}
+
+// Верхний сегмент относительного пути: по нему проверяется, не лезет ли пак в
+// общие каталоги лаунчера.
+QString topSegment(const QString &path)
+{
+    const int slash = path.indexOf(u'/');
+    return slash < 0 ? path : path.left(slash);
+}
+
+// После удаления файлов остаются пустые папки: пустой mods/ вместо удалённого
+// набора модов выглядит как незавершённая установка.
+void pruneEmptyDirs(const QDir &root, QStringList dirs)
+{
+    // Сначала самые глубокие: удаление вложенной может опустошить родителя.
+    std::sort(dirs.begin(), dirs.end(),
+              [](const QString &a, const QString &b) { return a.size() > b.size(); });
+    for (const QString &relative : std::as_const(dirs)) {
+        if (relative.isEmpty() || relative == QLatin1String("."))
+            continue;
+        QDir dir(root.absoluteFilePath(relative));
+        if (dir.exists() && dir.isEmpty())
+            root.rmdir(relative);
+    }
+}
+
 // Файлы, которые попадут в архив, — относительными путями от gameDir.
 QStringList collectFiles(const QString &gameDir, const QStringList &excludeTop)
 {
@@ -218,3 +254,119 @@ void BuildArchiveWorker::restore(const QString &zipPath, const QString &gameDir)
 
     emit finished(true, {});
 }
+
+void BuildArchiveWorker::applyPack(const QString &zipPath,
+                                   const QString &gameDir,
+                                   const QStringList &removeRelative,
+                                   const QStringList &forbiddenTop)
+{
+    QZipReader reader(zipPath);
+    if (!reader.isReadable()) {
+        emit finished(false,
+                      tr("Не удалось открыть архив %1").arg(QDir::toNativeSeparators(zipPath)));
+        return;
+    }
+
+    // Весь архив проверяется до первой записи на диск: наполовину раскатанный
+    // пак, отвергнутый на середине, оставил бы сборку в неизвестном состоянии.
+    const QList<QZipReader::FileInfo> entries = reader.fileInfoList();
+    QStringList applied;
+    applied.reserve(entries.size());
+    for (const QZipReader::FileInfo &entry : entries) {
+        if (!entry.isFile)
+            continue;
+        if (!isSafeRelativePath(entry.filePath)) {
+            emit finished(false,
+                          tr("Архив сборки повреждён: недопустимый путь «%1»").arg(entry.filePath));
+            return;
+        }
+        if (forbiddenTop.contains(topSegment(entry.filePath))) {
+            emit finished(false,
+                          tr("Архив сборки пытается изменить общие файлы лаунчера: «%1»")
+                              .arg(entry.filePath));
+            return;
+        }
+        applied << entry.filePath;
+    }
+
+    const QDir root(gameDir);
+    if (!QDir().mkpath(gameDir)) {
+        emit finished(false, tr("Не удалось создать %1").arg(QDir::toNativeSeparators(gameDir)));
+        return;
+    }
+
+    const int total = removeRelative.size() + applied.size();
+    emit progress(0, total, {});
+    int done = 0;
+
+    // ── Убираем то, чего в новой ревизии больше нет ──────────────────────
+    QStringList touchedDirs;
+    for (const QString &relative : removeRelative) {
+        if (canceled()) {
+            emit finished(false, tr("Установка сборки отменена"));
+            return;
+        }
+        // Список пришёл из нашего же файла состояния, но он лежит на диске
+        // рядом с игрой и мог быть отредактирован — правила те же, что у архива.
+        if (!isSafeRelativePath(relative) || forbiddenTop.contains(topSegment(relative))) {
+            emit progress(++done, total, relative);
+            continue;
+        }
+        // Файл мог быть удалён игроком или прийти заново в новой ревизии —
+        // и то и другое нормально.
+        const QString target = root.absoluteFilePath(relative);
+        if (!isInside(root, target)) {
+            emit progress(++done, total, relative);
+            continue;
+        }
+        if (QFileInfo::exists(target) && !QFile::remove(target)) {
+            emit finished(false,
+                          tr("Не удалось удалить %1").arg(QDir::toNativeSeparators(target)));
+            return;
+        }
+        const QString parent = QFileInfo(relative).path();
+        if (!touchedDirs.contains(parent))
+            touchedDirs << parent;
+        emit progress(++done, total, relative);
+    }
+    pruneEmptyDirs(root, touchedDirs);
+
+    // ── Раскладываем новую ревизию поверх ────────────────────────────────
+    for (const QString &relative : std::as_const(applied)) {
+        if (canceled()) {
+            emit finished(false, tr("Установка сборки отменена"));
+            return;
+        }
+
+        const QString target = root.absoluteFilePath(relative);
+        if (!isInside(root, target)) {
+            emit finished(false,
+                          tr("Архив сборки повреждён: путь «%1» ведёт за пределы папки игры")
+                              .arg(relative));
+            return;
+        }
+        if (!QDir().mkpath(QFileInfo(target).absolutePath())) {
+            emit finished(false,
+                          tr("Не удалось создать %1")
+                              .arg(QDir::toNativeSeparators(QFileInfo(target).absolutePath())));
+            return;
+        }
+
+        QFile file(target);
+        const QByteArray data = reader.fileData(relative);
+        // QIODevice::Truncate обязателен: поверх более длинного старого файла
+        // иначе остался бы его хвост.
+        if (!file.open(QIODevice::WriteOnly | QIODevice::Truncate)
+            || file.write(data) != data.size()) {
+            emit finished(false,
+                          tr("Не удалось записать %1: %2")
+                              .arg(QDir::toNativeSeparators(relative), file.errorString()));
+            return;
+        }
+        file.close();
+        emit progress(++done, total, relative);
+    }
+
+    emit packEntries(applied);
+    emit finished(true, {});
+}