1 files changed, 278 insertions, 0 deletions
diff --git a/src/shared/skyriminfo.cpp b/src/shared/skyriminfo.cpp
new file mode 100644
index 00000000..f3043c28
--- /dev/null
+++ b/src/shared/skyriminfo.cpp
@@ -0,0 +1,278 @@
+/*
+Copyright (C) 2012 Sebastian Herbord. All rights reserved.
+
+This file is part of Mod Organizer.
+
+Mod Organizer is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+Mod Organizer 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 General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "skyriminfo.h"
+
+#include "util.h"
+#include <tchar.h>
+#include <ShlObj.h>
+#include <sstream>
+#include "windows_error.h"
+#include "error_report.h"
+#define WIN32_LEAN_AND_MEAN
+#include <Windows.h>
+#include <boost/assign.hpp>
+
+
+SkyrimInfo::SkyrimInfo(const std::wstring &omoDirectory, const std::wstring &gameDirectory)
+  : GameInfo(omoDirectory, gameDirectory)
+{
+  identifyMyGamesDirectory(L"skyrim");
+}
+
+bool SkyrimInfo::identifyGame(const std::wstring &searchPath)
+{
+  return FileExists(searchPath, L"TESV.exe") &&
+         FileExists(searchPath, L"SkyrimLauncher.exe");
+}
+
+unsigned long SkyrimInfo::getBSAVersion()
+{
+  return 0x68;
+}
+
+
+std::wstring SkyrimInfo::getRegPathStatic()
+{
+  HKEY key;
+  LONG errorcode = ::RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Bethesda Softworks\\Skyrim",
+                                  0, KEY_QUERY_VALUE, &key);
+
+  if (errorcode != ERROR_SUCCESS) {
+    return L"";
+  }
+
+  WCHAR temp[MAX_PATH];
+  DWORD bufferSize = MAX_PATH;
+
+  errorcode = ::RegQueryValueExW(key, L"Installed Path", NULL, NULL, (LPBYTE)temp, &bufferSize);
+
+  return std::wstring(temp);
+}
+
+
+std::wstring SkyrimInfo::getInvalidationBSA()
+{
+  return L"Skyrim - Invalidation.bsa";
+}
+
+bool SkyrimInfo::isInvalidationBSA(const std::wstring &bsaName)
+{
+  static LPCWSTR invalidation[] = { L"Skyrim - Invalidation.bsa", NULL };
+
+  for (int i = 0; invalidation[i] != NULL; ++i) {
+    if (wcscmp(bsaName.c_str(), invalidation[i]) == 0) {
+      return true;
+    }
+  }
+  return false;
+}
+
+
+GameInfo::LoadOrderMechanism SkyrimInfo::getLoadOrderMechanism() const
+{
+  std::wstring fileName = getGameDirectory().append(L"\\TESV.exe");
+
+  try {
+    VS_FIXEDFILEINFO versionInfo = GetFileVersion(fileName);
+    if ((versionInfo.dwFileVersionMS > 0x10004) || // version >= 1.5.x?
+        ((versionInfo.dwFileVersionMS == 0x10004) && (versionInfo.dwFileVersionLS >= 0x1A0000))) { // version >= ?.4.26
+      return TYPE_PLUGINSTXT;
+    } else {
+      return TYPE_FILETIME;
+    }
+  } catch (const std::exception &e) {
+    reportError("TESV.exe is invalid: %s", e.what());
+    return TYPE_FILETIME;
+  }
+}
+
+std::wstring SkyrimInfo::getDocumentsDir()
+{
+  std::wostringstream temp;
+  temp << getMyGamesDirectory() << L"\\Skyrim";
+
+  return temp.str();
+}
+
+std::wstring SkyrimInfo::getSaveGameDir()
+{
+  std::wostringstream temp;
+  temp << getDocumentsDir() << L"\\Saves";
+  return temp.str();
+}
+
+std::vector<std::wstring> SkyrimInfo::getPrimaryPlugins()
+{
+  return boost::assign::list_of(L"skyrim.esm")(L"update.esm");
+}
+
+std::vector<std::wstring> SkyrimInfo::getIniFileNames()
+{
+  return boost::assign::list_of(L"skyrim.ini")(L"skyrimprefs.ini");
+}
+
+std::wstring SkyrimInfo::getSaveGameExtension()
+{
+  return L"*.ess";
+}
+
+std::wstring SkyrimInfo::getReferenceDataFile()
+{
+  return L"Skyrim - Meshes.bsa";
+}
+
+
+std::wstring SkyrimInfo::getOMODExt()
+{
+  return L"fomod";
+}
+
+
+std::wstring SkyrimInfo::getSteamAPPId()
+{
+  return L"72850";
+}
+
+
+std::wstring SkyrimInfo::getSEName()
+{
+  return L"skse";
+}
+
+
+std::wstring SkyrimInfo::getNexusPage()
+{
+  return L"http://skyrim.nexusmods.com";
+}
+
+
+std::wstring SkyrimInfo::getNexusInfoUrlStatic()
+{
+  return L"http://skyrim.nexusmods.com";
+}
+
+
+int SkyrimInfo::getNexusModIDStatic()
+{
+  return 1334;
+}
+
+
+void SkyrimInfo::createProfile(const std::wstring &directory, bool useDefaults)
+{
+  { // copy plugins.txt
+    std::wstring target = directory.substr().append(L"\\plugins.txt");
+    if (!FileExists(target)) {
+      std::wostringstream source;
+      source << getLocalAppFolder() << "\\Skyrim\\plugins.txt";
+      if (!::CopyFileW(source.str().c_str(), target.c_str(), true)) {
+        HANDLE file = ::CreateFileW(target.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
+        ::CloseHandle(file);
+      }
+    }
+    target = directory.substr().append(L"\\loadorder.txt");
+    if (!FileExists(target)) {
+      std::wostringstream source;
+      source << getLocalAppFolder() << "\\Skyrim\\loadorder.txt";
+      if (!::CopyFileW(source.str().c_str(), target.c_str(), true)) {
+        HANDLE file = ::CreateFileW(target.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
+        ::CloseHandle(file);
+      }
+    }
+  }
+
+  { // copy skyrim.ini-file
+    std::wstring target = directory.substr().append(L"\\skyrim.ini");
+    if (!FileExists(target)) {
+      std::wostringstream source;
+      if (useDefaults) {
+        source << getGameDirectory() << L"\\skyrim_default.ini";
+      } else {
+        source << getMyGamesDirectory() << L"\\Skyrim";
+        if (::FileExists(source.str(), L"skyrim.ini")) {
+          source << L"\\skyrim.ini";
+        } else {
+          source.str(L"");
+          source << getGameDirectory() << L"\\skyrim_default.ini";
+        }
+      }
+      if (!::CopyFileW(source.str().c_str(), target.c_str(), true)) {
+        if (::GetLastError() != ERROR_FILE_EXISTS) {
+          std::ostringstream stream;
+          stream << "failed to copy ini file: " << ToString(source.str(), false);
+          throw windows_error(stream.str());
+        }
+      }
+    }
+  }
+
+  { // copy skyrimprefs.ini-file
+    std::wstring target = directory.substr().append(L"\\skyrimprefs.ini");
+    if (!FileExists(target)) {
+      std::wostringstream source;
+      source << getMyGamesDirectory() << L"\\Skyrim\\skyrimprefs.ini";
+      if (!::CopyFileW(source.str().c_str(), target.c_str(), true)) {
+        if (::GetLastError() != ERROR_FILE_EXISTS) {
+          std::ostringstream stream;
+          stream << "failed to copy ini file: " << ToString(source.str(), false);
+          throw windows_error(stream.str());
+        }
+      }
+    }
+  }
+}
+
+
+void SkyrimInfo::repairProfile(const std::wstring &directory)
+{
+  createProfile(directory, false);
+}
+
+
+bool SkyrimInfo::rerouteToProfile(const wchar_t *fileName, const wchar_t *fullPath)
+{
+  static LPCWSTR profileFiles[] = { L"skyrim.ini", L"skyrimprefs.ini", L"loadorder.txt", NULL };
+
+  for (int i = 0; profileFiles[i] != NULL; ++i) {
+    if (_wcsicmp(fileName, profileFiles[i]) == 0) {
+      return true;
+    }
+  }
+
+  if ((_wcsicmp(fileName, L"plugins.txt") == 0) &&
+      (wcsstr(fullPath, L"AppData") != NULL)){
+    return true;
+  }
+
+  return false;
+}
+
+
+std::vector<ExecutableInfo> SkyrimInfo::getExecutables()
+{
+  std::vector<ExecutableInfo> result;
+  result.push_back(ExecutableInfo(L"SKSE", L"skse_loader.exe", L"", L"", DEFAULT_CLOSE));
+  result.push_back(ExecutableInfo(L"SBW", L"SBW.exe", L"", L"", DEFAULT_CLOSE));
+  result.push_back(ExecutableInfo(L"Skyrim", L"TESV.exe", L"", L"", DEFAULT_CLOSE));
+  result.push_back(ExecutableInfo(L"BOSS", L"BOSS/BOSS.exe", L"", L"", DEFAULT_STAY));
+  result.push_back(ExecutableInfo(L"Creation Kit", L"CreationKit.exe", L"", L"", DEFAULT_STAY, L"202480"));
+
+  return result;
+}