From 649c49dcc675a26dd2b7598c09fc3e3539b5ef88 Mon Sep 17 00:00:00 2001
From: Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>
Date: Sat, 21 Feb 2015 01:47:36 +0900
Subject: [PATCH] Mingw: Use _wgetenv to get user's home directory

Fixes GH-342
---
 src/util.cc | 53 ++++++++++++++++++++++++++++++++++++-----------------
 src/util.h  |  3 +++
 2 files changed, 39 insertions(+), 17 deletions(-)

diff --git a/src/util.cc b/src/util.cc
index bdfd9fce..68310aa8 100644
--- a/src/util.cc
+++ b/src/util.cc
@@ -146,6 +146,13 @@ std::string wCharToUtf8(const std::wstring& wsrc)
   }
 }
 
+std::string toForwardSlash(const std::string &src) {
+  auto dst = src;
+  std::transform(std::begin(dst), std::end(dst), std::begin(dst),
+                 [](char c) { return c == '\\' ? '/' : c; });
+  return dst;
+}
+
 #endif // __MINGW32__
 
 namespace util {
@@ -1288,35 +1295,47 @@ void setGlobalSignalHandler(int sig, sigset_t* mask, signal_handler_t handler,
 #endif // HAVE_SIGACTION
 }
 
+#ifndef __MINGW32__
 std::string getHomeDir()
 {
   const char* p = getenv("HOME");
   if (p) {
     return p;
   }
-#ifdef __MINGW32__
-  p = getenv("USERPROFILE");
-  if (p) {
-    return p;
-  }
-  p = getenv("HOMEDRIVE");
-  if (p) {
-    std::string homeDir = p;
-    p = getenv("HOMEPATH");
-    if (p) {
-      homeDir += p;
-      return homeDir;
-    }
-  }
-#elif HAVE_PWD_H
-  passwd* pw = getpwuid(geteuid());
-  if(pw && pw->pw_dir) {
+#ifdef HAVE_PWD_H
+  auto pw = getpwuid(geteuid());
+  if (pw && pw->pw_dir) {
     return pw->pw_dir;
   }
 #endif // HAVE_PWD_H
   return A2STR::NIL;
 }
 
+#else // __MINGW32__
+
+std::string getHomeDir()
+{
+  auto p = _wgetenv(L"HOME");
+  if (p) {
+    return toForwardSlash(wCharToUtf8(p));
+  }
+  p = _wgetenv(L"USERPROFILE");
+  if (p) {
+    return toForwardSlash(wCharToUtf8(p));
+  }
+  p = _wgetenv(L"HOMEDRIVE");
+  if (p) {
+    std::wstring homeDir = p;
+    p = _wgetenv(L"HOMEPATH");
+    if (p) {
+      homeDir += p;
+      return toForwardSlash(wCharToUtf8(homeDir));
+    }
+  }
+  return A2STR::NIL;
+}
+#endif // __MINGW32__
+
 int64_t getRealSize(const std::string& sizeWithUnit)
 {
   std::string::size_type p = sizeWithUnit.find_first_of("KMkm");
diff --git a/src/util.h b/src/util.h
index 04589acf..6e5eb8c8 100644
--- a/src/util.h
+++ b/src/util.h
@@ -108,6 +108,9 @@ std::wstring utf8ToWChar(const std::string& src);
 std::wstring utf8ToWChar(const char* str);
 
 std::string wCharToUtf8(const std::wstring& wsrc);
+
+// replace any backslash '\' in |src| with '/' and returns it.
+std::string toForwardSlash(const std::string &src);
 #else // !__MINGW32__
 # define utf8ToWChar(src) src
 # define utf8ToNative(src) src