1 files changed, 138 insertions, 0 deletions
diff --git a/libs/game_bethesda/src/gamebryo/gamebryolocalsavegames.cpp b/libs/game_bethesda/src/gamebryo/gamebryolocalsavegames.cpp
new file mode 100644
index 0000000..e4df0ea
--- /dev/null
+++ b/libs/game_bethesda/src/gamebryo/gamebryolocalsavegames.cpp
@@ -0,0 +1,138 @@
+/*
+Copyright (C) 2015 Sebastian Herbord. All rights reserved.
+
+This library is free software; you can redistribute it and/or
+modify it under the terms of the GNU Lesser General Public
+License as published by the Free Software Foundation; either
+version 3 of the License, or (at your option) any later version.
+
+This library is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+Lesser General Public License for more details.
+
+You should have received a copy of the GNU Lesser General Public
+License along with this library; if not, write to the Free Software
+Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+*/
+
+#include "gamebryolocalsavegames.h"
+#include "registry.h"
+#include <QtDebug>
+#include <iprofile.h>
+#include <stddef.h>
+#include <string>
+
+#include <QSettings>
+
+#include "gamegamebryo.h"
+
+// Helper to remove a key from a Bethesda-style INI file
+static void removeIniValue(const QString& iniFile, const QString& section,
+                           const QString& key)
+{
+  QSettings settings(iniFile, QSettings::IniFormat);
+  settings.beginGroup(section);
+  settings.remove(key);
+  settings.endGroup();
+  settings.sync();
+}
+
+GamebryoLocalSavegames::GamebryoLocalSavegames(const GameGamebryo* game,
+                                               const QString& iniFileName)
+    : m_Game{game}, m_IniFileName(iniFileName)
+{}
+
+MappingType GamebryoLocalSavegames::mappings(const QDir& profileSaveDir) const
+{
+  return {{profileSaveDir.absolutePath(), localSavesDirectory().absolutePath(), true,
+           true}};
+}
+
+QString GamebryoLocalSavegames::localSavesDummy() const
+{
+  return "__MO_Saves\\";
+}
+
+QDir GamebryoLocalSavegames::localSavesDirectory() const
+{
+  return QDir(m_Game->myGamesPath()).absoluteFilePath(localSavesDummy());
+}
+
+QDir GamebryoLocalSavegames::localGameDirectory() const
+{
+  return QDir(m_Game->myGamesPath()).absolutePath();
+}
+
+bool GamebryoLocalSavegames::prepareProfile(MOBase::IProfile* profile)
+{
+  bool enable = profile->localSavesEnabled();
+
+  QString basePath    = profile->localSettingsEnabled()
+                            ? profile->absolutePath()
+                            : localGameDirectory().absolutePath();
+  QString iniFilePath = basePath + "/" + m_IniFileName;
+  QString saveIni     = profile->absolutePath() + "/" + "savepath.ini";
+
+  // Get the current sLocalSavePath
+  QString currentPath =
+      GameGamebryo::readIniValue(iniFilePath, "General", "sLocalSavePath", "SKIP_ME");
+  bool alreadyEnabled = (currentPath == localSavesDummy());
+
+  // Get the current bUseMyGamesDirectory
+  QString currentMyGames = GameGamebryo::readIniValue(
+      iniFilePath, "General", "bUseMyGamesDirectory", "SKIP_ME");
+
+  // Create the __MO_Saves directory if local saves are enabled and it doesn't exist
+  if (enable) {
+    QDir saves = localSavesDirectory();
+    if (!saves.exists()) {
+      saves.mkdir(".");
+    }
+  }
+
+  // Set the path to __MO_Saves if it's not already
+  if (enable && !alreadyEnabled) {
+    // If the path is not blank, save it to savepath.ini
+    if (currentPath != "SKIP_ME") {
+      MOBase::WriteRegistryValue("General", "sLocalSavePath", currentPath, saveIni);
+    }
+    if (currentMyGames != "SKIP_ME") {
+      MOBase::WriteRegistryValue("General", "bUseMyGamesDirectory", currentMyGames,
+                                 saveIni);
+    }
+    MOBase::WriteRegistryValue("General", "sLocalSavePath", localSavesDummy(),
+                               iniFilePath);
+    MOBase::WriteRegistryValue("General", "bUseMyGamesDirectory", "1", iniFilePath);
+  }
+
+  // Get rid of the local saves setting if it's still there
+  if (!enable && alreadyEnabled) {
+    // If savepath.ini exists, use it and delete it
+    if (QFile::exists(saveIni)) {
+      QString savedPath =
+          GameGamebryo::readIniValue(saveIni, "General", "sLocalSavePath", "DELETE_ME");
+      QString savedMyGames = GameGamebryo::readIniValue(
+          saveIni, "General", "bUseMyGamesDirectory", "DELETE_ME");
+      if (savedPath != "DELETE_ME") {
+        MOBase::WriteRegistryValue("General", "sLocalSavePath", savedPath, iniFilePath);
+      } else {
+        removeIniValue(iniFilePath, "General", "sLocalSavePath");
+      }
+      if (savedMyGames != "DELETE_ME") {
+        MOBase::WriteRegistryValue("General", "bUseMyGamesDirectory", savedMyGames,
+                                   iniFilePath);
+      } else {
+        removeIniValue(iniFilePath, "General", "bUseMyGamesDirectory");
+      }
+      QFile::remove(saveIni);
+    }
+    // Otherwise just delete the setting
+    else {
+      removeIniValue(iniFilePath, "General", "sLocalSavePath");
+      removeIniValue(iniFilePath, "General", "bUseMyGamesDirectory");
+    }
+  }
+
+  return enable != alreadyEnabled;
+}