From 1ca8e7d2542f948d9ed8cca8ad4a1337e087d416 Mon Sep 17 00:00:00 2001
From: 0x5BFA <62196528+0x5bfa@users.noreply.github.com>
Date: Mon, 18 Nov 2024 14:37:20 +0000
Subject: [PATCH] Init

---
 .../Data/Contracts/ICommonDialogService.cs    |  2 +-
 .../Services/Windows/WindowsDialogService.cs  | 64 ++++++++-----------
 2 files changed, 26 insertions(+), 40 deletions(-)

diff --git a/src/Files.App/Data/Contracts/ICommonDialogService.cs b/src/Files.App/Data/Contracts/ICommonDialogService.cs
index 7362a786e495..db95c296a70c 100644
--- a/src/Files.App/Data/Contracts/ICommonDialogService.cs
+++ b/src/Files.App/Data/Contracts/ICommonDialogService.cs
@@ -43,6 +43,6 @@ public interface ICommonDialogService
 		/// <param name="remoteNetworkName">The name of the remote network.</param>
 		/// <param name="useMostRecentPath">The value indicating whether to enter the most recently used paths into the combination box.</param>
 		/// <returns>True if the 'OK' button was clicked; otherwise, false.</returns>
-		bool Open_NetworkConnectionDialog(nint hWind, bool hideRestoreConnectionCheckBox = false, bool persistConnectionAtLogon = false, bool readOnlyPath = false, string? remoteNetworkName = null, bool useMostRecentPath = false);
+		bool Open_NetworkConnectionDialog(nint hWnd, bool hideRestoreConnectionCheckBox = false, bool persistConnectionAtLogon = false, bool readOnlyPath = false, string? remoteNetworkName = null, bool useMostRecentPath = false);
 	}
 }
diff --git a/src/Files.App/Services/Windows/WindowsDialogService.cs b/src/Files.App/Services/Windows/WindowsDialogService.cs
index c4a5c2ed4324..c50e43932cbd 100644
--- a/src/Files.App/Services/Windows/WindowsDialogService.cs
+++ b/src/Files.App/Services/Windows/WindowsDialogService.cs
@@ -178,57 +178,43 @@ public unsafe bool Open_FileSaveDialog(nint hWnd, bool pickFoldersOnly, string[]
 		}
 
 		/// <inheritdoc/>
-		public unsafe bool Open_NetworkConnectionDialog(nint hWind, bool hideRestoreConnectionCheckBox = false, bool persistConnectionAtLogon = false, bool readOnlyPath = false, string? remoteNetworkName = null, bool useMostRecentPath = false)
+		public unsafe bool Open_NetworkConnectionDialog(nint hWnd, bool hideRestoreConnectionCheckBox = false, bool persistConnectionAtLogon = false, bool readOnlyPath = false, string? remoteNetworkName = null, bool useMostRecentPath = false)
 		{
-			NETRESOURCEW netRes = default;
-			CONNECTDLGSTRUCTW dialogOptions = default;
+			if (useMostRecentPath && !string.IsNullOrEmpty(remoteNetworkName))
+				throw new ArgumentException($"{nameof(useMostRecentPath)} cannot be set to true if {nameof(remoteNetworkName)} has a value.");
 
-			dialogOptions.cbStructure = (uint)Marshal.SizeOf(typeof(CONNECTDLGSTRUCTW));
-			netRes.dwType = NET_RESOURCE_TYPE.RESOURCETYPE_DISK;
+			NETRESOURCEW netResource = default;
+			CONNECTDLGSTRUCTW connectDlgOptions = default;
+			WIN32_ERROR res = default;
 
 			if (hideRestoreConnectionCheckBox)
-				dialogOptions.dwFlags |= CONNECTDLGSTRUCT_FLAGS.CONNDLG_HIDE_BOX;
-			else
-				dialogOptions.dwFlags &= ~CONNECTDLGSTRUCT_FLAGS.CONNDLG_HIDE_BOX;
-
+				connectDlgOptions.dwFlags |= CONNECTDLGSTRUCT_FLAGS.CONNDLG_HIDE_BOX;
 			if (persistConnectionAtLogon)
-			{
-				dialogOptions.dwFlags |= CONNECTDLGSTRUCT_FLAGS.CONNDLG_PERSIST;
-				dialogOptions.dwFlags |= CONNECTDLGSTRUCT_FLAGS.CONNDLG_NOT_PERSIST;
-			}
-			else
-			{
-				dialogOptions.dwFlags &= ~CONNECTDLGSTRUCT_FLAGS.CONNDLG_PERSIST;
-				dialogOptions.dwFlags &= ~CONNECTDLGSTRUCT_FLAGS.CONNDLG_NOT_PERSIST;
-			}
-
-			fixed (char* lpcRemoteName = remoteNetworkName)
-				netRes.lpRemoteName = lpcRemoteName;
-
-			if (useMostRecentPath && !string.IsNullOrEmpty(remoteNetworkName))
-				throw new InvalidOperationException($"{nameof(useMostRecentPath)} cannot be set to true if {nameof(remoteNetworkName)} has a value.");
-
+				connectDlgOptions.dwFlags |= (CONNECTDLGSTRUCT_FLAGS.CONNDLG_PERSIST & CONNECTDLGSTRUCT_FLAGS.CONNDLG_NOT_PERSIST);
 			if (useMostRecentPath)
-				dialogOptions.dwFlags |= CONNECTDLGSTRUCT_FLAGS.CONNDLG_USE_MRU;
-			else
-				dialogOptions.dwFlags &= ~CONNECTDLGSTRUCT_FLAGS.CONNDLG_USE_MRU;
-
-			dialogOptions.hwndOwner = new(hWind);
+				connectDlgOptions.dwFlags |= CONNECTDLGSTRUCT_FLAGS.CONNDLG_USE_MRU;
+			if (readOnlyPath && !string.IsNullOrEmpty(remoteNetworkName))
+				connectDlgOptions.dwFlags |= CONNECTDLGSTRUCT_FLAGS.CONNDLG_RO_PATH;
 
-			dialogOptions.lpConnRes = &netRes;
-
-			if (readOnlyPath && !string.IsNullOrEmpty(netRes.lpRemoteName.ToString()))
-				dialogOptions.dwFlags |= CONNECTDLGSTRUCT_FLAGS.CONNDLG_RO_PATH;
+			fixed (char* pszRemoteName = remoteNetworkName)
+			{
+				netResource.dwType = NET_RESOURCE_TYPE.RESOURCETYPE_DISK;
+				netResource.lpRemoteName = pszRemoteName;
 
-			var result = PInvoke.WNetConnectionDialog1W(ref dialogOptions);
+				connectDlgOptions.cbStructure = (uint)sizeof(CONNECTDLGSTRUCTW);
+				connectDlgOptions.hwndOwner = new(hWnd);
+				connectDlgOptions.lpConnRes = &netResource;
 
-			dialogOptions.lpConnRes = null;
+				res = PInvoke.WNetConnectionDialog1W(ref connectDlgOptions);
+			}
 
-			if ((uint)result == unchecked((uint)-1))
+			// User canceled
+			if ((uint)res == unchecked((uint)-1))
 				return false;
 
-			if (result == 0)
-				throw new Win32Exception("Cannot display dialog");
+			// Unexpected error happened
+			if (res is not WIN32_ERROR.NO_ERROR)
+				throw new Win32Exception("Failed to process the network connection dialog successfully.");
 
 			return true;
 		}