From 1c50a09c2e7bfd9057b5562b6134590475325b72 Mon Sep 17 00:00:00 2001
From: Connor McLaughlin <stenzek@gmail.com>
Date: Sun, 17 May 2020 00:22:16 +1000
Subject: [PATCH] HostInterface: Set user directory to Documents\DuckStation

---
 README.md                   |  4 ++--
 src/core/host_interface.cpp | 47 ++++++++++++++++++++++++++++++-------
 2 files changed, 40 insertions(+), 11 deletions(-)

diff --git a/README.md b/README.md
index 4dfdd441f..74dd4cc4e 100644
--- a/README.md
+++ b/README.md
@@ -49,7 +49,7 @@ Prebuilt binaries of DuckStation for 64-bit Windows are available via AppVeyor C
 
 Once downloaded and extracted, you can launch the Qt frontend from `duckstation-qt-x64-ReleaseLTCG.exe`, or the SDL frontend from `duckstation-sdl-x64-ReleaseLTCG.exe`.
 To set up:
-1. Either configure the path to a BIOS image in the settings, or copy one or more PlayStation BIOS images to the bios/ subdirectory.
+1. Either configure the path to a BIOS image in the settings, or copy one or more PlayStation BIOS images to the bios/ subdirectory. On Windows, by default this will be located in `C:\Users\YOUR_USERNAME\Documents\DuckStation\bios`.
 2. If using the SDL frontend, add the directories containing your disc images by clicking `Settings->Add Game Directory`.
 2. Select a game from the list, or open a disc image file and enjoy.
 
@@ -131,7 +131,7 @@ The "User Directory" is where you should place your BIOS images, where settings
 
 This is located in the following places depending on the platform you're using:
 
-- Windows: Directory containing DuckStation executable.
+- Windows: My Documents\DuckStation
 - Linux: `$XDG_DATA_HOME/duckstation`, or `~/.local/share/duckstation`.
 - macOS: `~/Library/Application Support/DuckStation`.
 
diff --git a/src/core/host_interface.cpp b/src/core/host_interface.cpp
index 86a7cedec..ac01b1d0f 100644
--- a/src/core/host_interface.cpp
+++ b/src/core/host_interface.cpp
@@ -17,12 +17,15 @@
 #include "timers.h"
 #include <cmath>
 #include <cstring>
+#include <cwchar>
 #include <imgui.h>
 #include <stdlib.h>
 Log_SetChannel(HostInterface);
 
 #ifdef WIN32
 #include "common/windows_headers.h"
+#include <KnownFolders.h>
+#include <ShlObj.h>
 #include <mmsystem.h>
 #endif
 
@@ -666,16 +669,40 @@ void HostInterface::SetUserDirectory()
 
   if (FileSystem::FileExists(StringUtil::StdStringFromFormat("%s%c%s", m_program_directory.c_str(),
                                                              FS_OSPATH_SEPERATOR_CHARACTER, "portable.txt")
+                               .c_str()) ||
+      FileSystem::FileExists(StringUtil::StdStringFromFormat("%s%c%s", m_program_directory.c_str(),
+                                                             FS_OSPATH_SEPERATOR_CHARACTER, "settings.ini")
                                .c_str()))
   {
-    std::fprintf(stdout, "portable.txt found, using program directory as user directory.\n");
+    std::fprintf(stdout, "portable.txt or old settings.ini found, using program directory as user directory.\n");
     m_user_directory = m_program_directory;
   }
   else
   {
 #ifdef WIN32
-    // On Windows, use the path to the program. We might want to use My Documents in the future.
-    m_user_directory = m_program_directory;
+    // On Windows, use My Documents\DuckStation.
+    PWSTR documents_directory;
+    if (SUCCEEDED(SHGetKnownFolderPath(FOLDERID_Documents, 0, NULL, &documents_directory)))
+    {
+      const size_t documents_directory_len = std::wcslen(documents_directory);
+      int documents_directory_u8len = WideCharToMultiByte(
+        CP_UTF8, 0, documents_directory, static_cast<int>(documents_directory_len), nullptr, 0, nullptr, nullptr);
+      if (documents_directory_u8len > 0)
+      {
+        std::string documents_directory_str;
+        documents_directory_str.resize(documents_directory_u8len);
+        documents_directory_u8len = WideCharToMultiByte(
+          CP_UTF8, 0, documents_directory, static_cast<int>(documents_directory_len), documents_directory_str.data(),
+          static_cast<int>(documents_directory_str.size()), 0, nullptr);
+        if (documents_directory_u8len > 0)
+        {
+          documents_directory_str.resize(documents_directory_u8len);
+          m_user_directory = StringUtil::StdStringFromFormat("%s%c%s", documents_directory_str.c_str(),
+                                                             FS_OSPATH_SEPERATOR_CHARACTER, "DuckStation");
+        }
+      }
+      CoTaskMemFree(documents_directory);
+    }
 #elif __linux__
     // On Linux, use .local/share/duckstation as a user directory by default.
     const char* xdg_data_home = getenv("XDG_DATA_HOME");
@@ -686,19 +713,21 @@ void HostInterface::SetUserDirectory()
     else
     {
       const char* home_path = getenv("HOME");
-      if (!home_path)
-        m_user_directory = m_program_directory;
-      else
+      if (home_path)
         m_user_directory = StringUtil::StdStringFromFormat("%s/.local/share/duckstation", home_path);
     }
 #elif __APPLE__
     // On macOS, default to ~/Library/Application Support/DuckStation.
     const char* home_path = getenv("HOME");
-    if (!home_path)
-      m_user_directory = m_program_directory;
-    else
+    if (home_path)
       m_user_directory = StringUtil::StdStringFromFormat("%s/Library/Application Support/DuckStation", home_path);
 #endif
+
+    if (m_user_directory.empty())
+    {
+      std::fprintf(stderr, "User directory path could not be determined, falling back to program directory.");
+      m_user_directory = m_program_directory;
+    }
   }
 }