From d8de99bbe452d000896d4ac75e036e854038dda9 Mon Sep 17 00:00:00 2001
From: Dean Herbert <pe@ppy.sh>
Date: Thu, 13 Oct 2022 18:39:52 +0900
Subject: [PATCH] Check for hard link support in first run overlay

---
 osu.Game/Database/LegacyImportManager.cs      | 21 +++++++++++++++++++
 osu.Game/Database/RealmFileStore.cs           |  2 +-
 .../FirstRunSetup/ScreenImportFromStable.cs   |  4 ++++
 3 files changed, 26 insertions(+), 1 deletion(-)

diff --git a/osu.Game/Database/LegacyImportManager.cs b/osu.Game/Database/LegacyImportManager.cs
index b5338fbe1f..80c8e9c19e 100644
--- a/osu.Game/Database/LegacyImportManager.cs
+++ b/osu.Game/Database/LegacyImportManager.cs
@@ -3,6 +3,7 @@
 
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 using osu.Framework;
@@ -54,6 +55,26 @@ namespace osu.Game.Database
 
         public void UpdateStorage(string stablePath) => cachedStorage = new StableStorage(stablePath, gameHost as DesktopGameHost);
 
+        public bool CheckHardLinkAvailability()
+        {
+            if (RuntimeInfo.OS != RuntimeInfo.Platform.Windows)
+                return false;
+
+            var stableStorage = GetCurrentStableStorage();
+
+            if (stableStorage == null || gameHost is not DesktopGameHost desktopGameHost)
+                return false;
+
+            const string test_filename = "_hard_link_test";
+
+            desktopGameHost.Storage.Delete(test_filename);
+
+            string testExistingPath = stableStorage.GetFullPath(stableStorage.GetFiles(string.Empty).First());
+            string testDestinationPath = desktopGameHost.Storage.GetFullPath(test_filename);
+
+            return RealmFileStore.CreateHardLink(testDestinationPath, testExistingPath, IntPtr.Zero);
+        }
+
         public virtual async Task<int> GetImportCount(StableContent content, CancellationToken cancellationToken)
         {
             var stableStorage = GetCurrentStableStorage();
diff --git a/osu.Game/Database/RealmFileStore.cs b/osu.Game/Database/RealmFileStore.cs
index 4aa65dd50a..99033e2c88 100644
--- a/osu.Game/Database/RealmFileStore.cs
+++ b/osu.Game/Database/RealmFileStore.cs
@@ -94,7 +94,7 @@ namespace osu.Game.Database
         }
 
         [DllImport("Kernel32.dll", CharSet = CharSet.Unicode)]
-        private static extern bool CreateHardLink(
+        public static extern bool CreateHardLink(
             string lpFileName,
             string lpExistingFileName,
             IntPtr lpSecurityAttributes
diff --git a/osu.Game/Overlays/FirstRunSetup/ScreenImportFromStable.cs b/osu.Game/Overlays/FirstRunSetup/ScreenImportFromStable.cs
index 29cf3824fd..8227106ac2 100644
--- a/osu.Game/Overlays/FirstRunSetup/ScreenImportFromStable.cs
+++ b/osu.Game/Overlays/FirstRunSetup/ScreenImportFromStable.cs
@@ -14,6 +14,7 @@ using osu.Framework.Graphics;
 using osu.Framework.Graphics.Containers;
 using osu.Framework.Graphics.UserInterface;
 using osu.Framework.Localisation;
+using osu.Framework.Logging;
 using osu.Game.Database;
 using osu.Game.Graphics;
 using osu.Game.Graphics.Containers;
@@ -105,6 +106,9 @@ namespace osu.Game.Overlays.FirstRunSetup
             toggleInteraction(true);
             stableLocatorTextBox.Current.Value = storage.GetFullPath(string.Empty);
             importButton.Enabled.Value = true;
+
+            bool available = legacyImportManager.CheckHardLinkAvailability();
+            Logger.Log($"Hard link support is {available}");
         }
 
         private void runImport()