src/Files.App/Utils/Storage/Helpers/SyncRootHelpers.cs - main - githubmirror/Files

src/Files.App/Utils/Storage/Helpers/SyncRootHelpers.cs

58 строк2 KB

0x5bfa

Code Quality: Replace raw ABI pointers with ComObject instances for safer disposal (#18684)

16 июл 2026, 18:32

16 июл 2026, 18:3293dfb04

// Copyright (c) Files Community
// Licensed under the MIT License.

using Microsoft.Win32;
using Windows.Win32;
using Windows.Win32.System.Com;
using Windows.Win32.System.WinRT;

namespace Files.App.Utils.Storage
{
	internal static class SyncRootHelpers
	{
		private static (bool Success, ulong Capacity, ulong Used) GetSyncRootQuotaFromSyncRootId(string syncRootId)
		{
			using var key = Registry.LocalMachine.OpenSubKey($"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\SyncRootManager\\{syncRootId}");
			if (key?.GetValue("StorageProviderStatusUISourceFactory") is not string factoryClsidString ||
				!Guid.TryParse(factoryClsidString, out var factoryClsid))
				return (false, 0, 0);

			ulong ulTotalSize = 0ul, ulUsedSize = 0ul;

			if (PInvoke.CoCreateInstance(factoryClsid, null, CLSCTX.CLSCTX_LOCAL_SERVER, out IStorageProviderStatusUISourceFactory? pStorageProviderStatusUISourceFactory).ThrowIfFailedOnDebug().Failed ||
				pStorageProviderStatusUISourceFactory is null)
				return (false, 0, 0);

			if (pStorageProviderStatusUISourceFactory.GetStatusUISource(syncRootId, out IStorageProviderStatusUISource pStorageProviderStatusUISource).ThrowIfFailedOnDebug().Failed ||
				pStorageProviderStatusUISource.GetStatusUI(out IStorageProviderStatusUI pStorageProviderStatusUI).ThrowIfFailedOnDebug().Failed ||
				pStorageProviderStatusUI.GetQuotaUI(out IStorageProviderQuotaUI pStorageProviderQuotaUI).ThrowIfFailedOnDebug().Failed ||
				pStorageProviderQuotaUI.GetQuotaTotalInBytes(out ulTotalSize).ThrowIfFailedOnDebug().Failed ||
				pStorageProviderQuotaUI.GetQuotaUsedInBytes(out ulUsedSize).ThrowIfFailedOnDebug().Failed)
				return (false, 0, 0);

			return (true, ulTotalSize, ulUsedSize);
		}

		public static async Task<(bool Success, ulong Capacity, ulong Used)> GetSyncRootQuotaAsync(string path)
		{
			Windows.Storage.StorageFolder folder = await Windows.Storage.StorageFolder.GetFolderFromPathAsync(path);
			Windows.Storage.Provider.StorageProviderSyncRootInfo? syncRootInfo = null;

			try
			{
				syncRootInfo = Windows.Storage.Provider.StorageProviderSyncRootManager.GetSyncRootInformationForFolder(folder);
			}
			catch
			{
				return (false, 0, 0);
			}

			if (syncRootInfo is null || syncRootInfo.Id is null)
			{
				return (false, 0, 0);
			}

			return GetSyncRootQuotaFromSyncRootId(syncRootInfo.Id);
		}
	}
}