Merge pull request #454 from Modorganizer2/Develop

Release 2.1.4

2 files changed, 78 insertions, 0 deletions

diff --git a/src/shared/util.cpp b/src/shared/util.cpp
index 5491a9e6..102565f5 100644
--- a/src/shared/util.cpp
+++ b/src/shared/util.cpp
@@ -27,6 +27,7 @@ along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
 #include <DbgHelp.h>

 #include <set>

 #include <boost/scoped_array.hpp>

+#include <QApplication>

 

 namespace MOShared {

 

@@ -174,5 +175,78 @@ VS_FIXEDFILEINFO GetFileVersion(const std::wstring &fileName)
   }

 }

 

+std::wstring GetFileVersionString(const std::wstring &fileName)

+{

+  DWORD handle = 0UL;

+  DWORD size = ::GetFileVersionInfoSizeW(fileName.c_str(), &handle);

+  if (size == 0) {

+    throw windows_error("failed to determine file version info size");

+  }

+

+  boost::scoped_array<char> buffer(new char[size]);

+  try {

+    handle = 0UL;

+    if (!::GetFileVersionInfoW(fileName.c_str(), handle, size, buffer.get())) {

+      throw windows_error("failed to determine file version info");

+    }

+

+    LPVOID strBuffer = nullptr;

+    UINT strLength = 0;

+    if (!::VerQueryValue(buffer.get(), L"\\StringFileInfo\\040904B0\\ProductVersion", &strBuffer, &strLength)) {

+      throw windows_error("failed to determine file version");

+    }

+

+    return std::wstring((LPCTSTR)strBuffer);

+  }

+  catch (...) {

+    throw;

+  }

+}

+

+MOBase::VersionInfo createVersionInfo()

+{

+  VS_FIXEDFILEINFO version = GetFileVersion(QApplication::applicationFilePath().toStdWString());

+

+  if (version.dwFileFlags | VS_FF_PRERELEASE)

+  {

+    // Pre-release builds need annotating

+    QString versionString = QString::fromStdWString(GetFileVersionString(QApplication::applicationFilePath().toStdWString()));

+

+    // The pre-release flag can be set without the string specifying what type of pre-release

+    bool noLetters = true;

+    for (QChar character : versionString)

+    {

+      if (character.isLetter())

+      {

+        noLetters = false;

+        break;

+      }

+    }

+

+    if (noLetters)

+    {

+      // Default to pre-alpha when release type is unspecified

+      return MOBase::VersionInfo(version.dwFileVersionMS >> 16,

+                                 version.dwFileVersionMS & 0xFFFF,

+                                 version.dwFileVersionLS >> 16,

+                                 version.dwFileVersionLS & 0xFFFF,

+                                 MOBase::VersionInfo::RELEASE_PREALPHA);

+    }

+    else

+    {

+      // Trust the string to make sense

+      return MOBase::VersionInfo(versionString);

+    }

+  }

+  else

+  {

+    // Non-pre-release builds just need their version numbers reading

+    return MOBase::VersionInfo(version.dwFileVersionMS >> 16,

+                               version.dwFileVersionMS & 0xFFFF,

+                               version.dwFileVersionLS >> 16,

+                               version.dwFileVersionLS & 0xFFFF);

+  }

+}

+

 

 } // namespace MOShared

diff --git a/src/shared/util.h b/src/shared/util.h
index 1e498059..1fdfb089 100644
--- a/src/shared/util.h
+++ b/src/shared/util.h
@@ -25,6 +25,8 @@ along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
 #define WIN32_LEAN_AND_MEAN

 #include <Windows.h>

 

+#include <versioninfo.h>

+

 namespace MOShared {

 

 /// Test if a file (or directory) by the specified name exists

@@ -45,6 +47,8 @@ std::wstring ToLower(const std::wstring &text);
 bool CaseInsensitiveEqual(const std::wstring &lhs, const std::wstring &rhs);

 

 VS_FIXEDFILEINFO GetFileVersion(const std::wstring &fileName);

+std::wstring GetFileVersionString(const std::wstring &fileName);

+MOBase::VersionInfo createVersionInfo();

 

 } // namespace MOShared