UnrealEngineUWP/Engine/Source/Programs/Shared/EpicGames.Core/RegistryConfigurationSource.cs

// Copyright Epic Games, Inc. All Rights Reserved.

using System.Collections.Generic;
using System.Runtime.InteropServices;
using System.Runtime.Versioning;
using Microsoft.Extensions.Configuration;
using Microsoft.Win32;

namespace EpicGames.Core
{
	/// <summary>
	/// Configuration source which reads values from the Registry on Windows
	/// </summary>
	[SupportedOSPlatform("windows")]
	public class RegistryConfigurationSource : IConfigurationSource
	{
		readonly RegistryKey _baseKey;
		readonly string _keyPath;
		readonly string _baseConfigName;

		/// <summary>
		/// Constructor
		/// </summary>
		/// <param name="baseKey">Hive to resolve keyPath relative to</param>
		/// <param name="keyPath">Path within the registry to enumerate</param>
		/// <param name="baseConfigName">Prefix for returned configuration values</param>
		public RegistryConfigurationSource(RegistryKey baseKey, string keyPath, string baseConfigName)
		{
			_baseKey = baseKey;
			_keyPath = keyPath;
			_baseConfigName = baseConfigName;
		}
		
		/// <inheritdoc/>
		public IConfigurationProvider Build(IConfigurationBuilder builder)
			=> new RegistryConfigProvider(_baseKey, _keyPath, _baseConfigName);
	}

	class RegistryConfigProvider : ConfigurationProvider
	{
		readonly RegistryKey _baseKey;
		readonly string _keyPath;
		readonly string _baseConfigName;

		public RegistryConfigProvider(RegistryKey baseKey, string keyPath, string baseConfigName)
		{
			_baseKey = baseKey;
			_keyPath = keyPath;
			_baseConfigName = baseConfigName;
		}

		public override void Load()
		{
			if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
			{
				Dictionary<string, string?> data = new Dictionary<string, string?>();
				GetValues(_baseKey, _keyPath, _baseConfigName, data);
				Data = data;
			}
		}

		[SupportedOSPlatform("windows")]
		static void GetValues(RegistryKey baseKey, string keyPath, string baseConfigName, Dictionary<string, string?> data)
		{
			using RegistryKey? registryKey = baseKey.OpenSubKey(keyPath);
			if (registryKey != null)
			{
				string[] subKeyNames = registryKey.GetSubKeyNames();
				foreach (string subKeyName in subKeyNames)
				{
					GetValues(registryKey, subKeyName, $"{baseConfigName}:{subKeyName}", data);
				}

				string[] valueNames = registryKey.GetValueNames();
				foreach (string valueName in valueNames)
				{
					object? value = registryKey.GetValue(valueName);
					if (value != null)
					{
						data[$"{baseConfigName}:{valueName}"] = value.ToString();
					}
				}
			}
		}
	}
}