From 23e68dc785d274327efeecb22cfc712c71cfbfc0 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sat, 7 May 2022 18:08:43 +0800
Subject: ...

---
 src/common/io/Resource.cpp | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

(limited to 'src/common/io/Resource.cpp')

diff --git a/src/common/io/Resource.cpp b/src/common/io/Resource.cpp
index 0ee7f753..b847e1cf 100644
--- a/src/common/io/Resource.cpp
+++ b/src/common/io/Resource.cpp
@@ -1,13 +1,19 @@
 #include "cru/common/io/Resource.h"
 #include "cru/common/Exception.h"
+#include "cru/common/log/Logger.h"
 
 #if defined(CRU_PLATFORM_OSX)
 #include <CoreFoundation/CoreFoundation.h>
 #elif defined(CRU_PLATFORM_WINDOWS)
+#include <Windows.h>
 #endif
 
+#include <filesystem>
+
 namespace cru::io {
 std::filesystem::path GetResourceDir() {
+  constexpr auto kLogTag = u"GetResourceDir";
+
 #if defined(CRU_PLATFORM_OSX)
   CFBundleRef main_bundle = CFBundleGetMainBundle();
   CFURLRef bundle_url = CFBundleCopyBundleURL(main_bundle);
@@ -21,7 +27,20 @@ std::filesystem::path GetResourceDir() {
 
   return bundle_path / "Contents/Resources";
 #elif defined(CRU_PLATFORM_WINDOWS)
-  throw Exception(u"Not implemented.");
+  wchar_t buffer[MAX_PATH];
+  DWORD size = ::GetModuleFileNameW(nullptr, buffer, MAX_PATH);
+  std::filesystem::path module_path(buffer, buffer + size);
+  auto p = module_path;
+  while (p.has_parent_path()) {
+    p = p.parent_path();
+    auto resource_dir_path = p / "assets";
+    if (std::filesystem::exists(resource_dir_path) &&
+        std::filesystem::is_directory(resource_dir_path)) {
+      return resource_dir_path;
+    }
+  }
+
+  throw Exception(u"Failed to find resource directory.");
 #else
   throw Exception(u"Not implemented.");
 #endif
-- 
cgit v1.2.3