TweetDuck/Utils/WindowsUtils.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Threading;
using Microsoft.Win32;

namespace TweetDuck.Utils {
	static class WindowsUtils {
		public static bool ShouldAvoidToolWindow { get; } = OSVersionEquals(major: 6, minor: 2); // windows 8/10

		private static bool OSVersionEquals(int major, int minor) {
			System.Version ver = Environment.OSVersion.Version;
			return ver.Major == major && ver.Minor == minor;
		}

		public static bool TrySleepUntil(Func<bool> test, int timeoutMillis, int timeStepMillis) {
			for (int waited = 0; waited < timeoutMillis; waited += timeStepMillis) {
				if (test()) {
					return true;
				}

				Thread.Sleep(timeStepMillis);
			}

			return false;
		}

		public static void TryDeleteFolderWhenAble(string path, int timeout) {
			new Thread(() => {
				TrySleepUntil(() => {
					try {
						Directory.Delete(path, true);
						return true;
					} catch (DirectoryNotFoundException) {
						return true;
					} catch {
						return false;
					}
				}, timeout, 500);
			}).Start();
		}

		public static IEnumerable<Browser> FindInstalledBrowsers() {
			static IEnumerable<Browser> ReadBrowsersFromKey(RegistryHive hive) {
				using RegistryKey root = RegistryKey.OpenBaseKey(hive, RegistryView.Default);
				using RegistryKey browserList = root.OpenSubKey(@"SOFTWARE\Clients\StartMenuInternet", false);

				if (browserList == null) {
					yield break;
				}

				foreach (string sub in browserList.GetSubKeyNames()) {
					using RegistryKey browserKey = browserList.OpenSubKey(sub, false);
					using RegistryKey shellKey = browserKey?.OpenSubKey(@"shell\open\command");

					if (shellKey == null) {
						continue;
					}

					string browserName = browserKey.GetValue(null) as string;
					string browserPath = shellKey.GetValue(null) as string;

					if (string.IsNullOrEmpty(browserName) || string.IsNullOrEmpty(browserPath)) {
						continue;
					}

					if (browserPath[0] == '"' && browserPath[browserPath.Length - 1] == '"') {
						browserPath = browserPath.Substring(1, browserPath.Length - 2);
					}

					yield return new Browser(browserName, browserPath);
				}
			}

			HashSet<Browser> browsers = new HashSet<Browser>();

			try {
				browsers.UnionWith(ReadBrowsersFromKey(RegistryHive.CurrentUser));
				browsers.UnionWith(ReadBrowsersFromKey(RegistryHive.LocalMachine));
			} catch {
				// oops I guess
			}

			return browsers;
		}

		public sealed class Browser {
			public string Name { get; }
			public string Path { get; }

			public Browser(string name, string path) {
				this.Name = name;
				this.Path = path;
			}

			public override int GetHashCode() => Name.GetHashCode();
			public override bool Equals(object obj) => obj is Browser other && Name == other.Name;
			public override string ToString() => Name;
		}
	}
}