TweetDuck/Core/Utils/WindowsUtils.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Diagnostics;
using System.IO;
using System.Net;
using System.Runtime.InteropServices;
using System.Text.RegularExpressions;
using System.Threading;
using System.Windows.Forms;
using Microsoft.Win32;

namespace TweetDuck.Core.Utils{
    static class WindowsUtils{
        private static readonly Lazy<Regex> RegexStripHtmlStyles = new Lazy<Regex>(() => new Regex(@"\s?(?:style|class)="".*?"""), false);
        private static readonly Lazy<Regex> RegexOffsetClipboardHtml = new Lazy<Regex>(() => new Regex(@"(?<=EndHTML:|EndFragment:)(\d+)"), false);

        private static readonly bool IsWindows8OrNewer;
        private static bool HasMicrosoftBeenBroughtTo2008Yet;

        public static int CurrentProcessID { get; }
        public static bool ShouldAvoidToolWindow { get; }
        public static bool IsAeroEnabled => IsWindows8OrNewer || (NativeMethods.DwmIsCompositionEnabled(out bool isCompositionEnabled) == 0 && isCompositionEnabled);

        static WindowsUtils(){
            using(Process me = Process.GetCurrentProcess()){
                CurrentProcessID = me.Id;
            }

            Version ver = Environment.OSVersion.Version;
            IsWindows8OrNewer = ver.Major == 6 && ver.Minor == 2; // windows 8/10

            ShouldAvoidToolWindow = IsWindows8OrNewer;
        }
        
        public static void EnsureTLS12(){
            if (!HasMicrosoftBeenBroughtTo2008Yet){
                ServicePointManager.SecurityProtocol |= SecurityProtocolType.Tls12;
                ServicePointManager.SecurityProtocol &= ~(SecurityProtocolType.Ssl3 | SecurityProtocolType.Tls | SecurityProtocolType.Tls11);
                HasMicrosoftBeenBroughtTo2008Yet = true;
            }
        }

        public static void CreateDirectoryForFile(string file){
            string dir = Path.GetDirectoryName(file);

            if (dir == null){
                throw new ArgumentException("Invalid file path: "+file);
            }
            else if (dir.Length > 0){
                Directory.CreateDirectory(dir);
            }
        }

        public static bool CheckFolderWritePermission(string path){
            string testFile = Path.Combine(path, ".test");

            try{
                Directory.CreateDirectory(path);

                using(File.Create(testFile)){}
                File.Delete(testFile);
                return true;
            }catch{
                return false;
            }
        }

        public static bool FileExistsAndNotEmpty(string path){
            try{
                return new FileInfo(path).Length > 0;
            }catch{
                return false;
            }
        }

        public static bool OpenAssociatedProgram(string file, string arguments = "", bool runElevated = false){
            try{
                using(Process.Start(new ProcessStartInfo{
                    FileName = file,
                    Arguments = arguments,
                    Verb = runElevated ? "runas" : string.Empty,
                    ErrorDialog = true
                })){
                    return true;
                }
            }catch(Win32Exception e) when (e.NativeErrorCode == 0x000004C7){ // operation canceled by the user
                return false;
            }catch(Exception e){
                Program.Reporter.HandleException("Error Opening Program", "Could not open the associated program for "+file, true, e);
                return false;
            }
        }

        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 void ClipboardStripHtmlStyles(){
            if (!Clipboard.ContainsText(TextDataFormat.Html) || !Clipboard.ContainsText(TextDataFormat.UnicodeText)){
                return;
            }

            string originalText = Clipboard.GetText(TextDataFormat.UnicodeText);
            string originalHtml = Clipboard.GetText(TextDataFormat.Html);

            string updatedHtml = RegexStripHtmlStyles.Value.Replace(originalHtml, string.Empty);

            int removed = originalHtml.Length-updatedHtml.Length;
            updatedHtml = RegexOffsetClipboardHtml.Value.Replace(updatedHtml, match => (int.Parse(match.Value)-removed).ToString().PadLeft(match.Value.Length, '0'));
            
            DataObject obj = new DataObject();
            obj.SetText(originalText, TextDataFormat.UnicodeText);
            obj.SetText(updatedHtml, TextDataFormat.Html);
            SetClipboardData(obj);
        }

        public static void SetClipboard(string text, TextDataFormat format){
            if (string.IsNullOrEmpty(text)){
                return;
            }

            DataObject obj = new DataObject();
            obj.SetText(text, format);
            SetClipboardData(obj);
        }

        private static void SetClipboardData(DataObject obj){
            try{
                Clipboard.SetDataObject(obj);
            }catch(ExternalException e){
                Program.Reporter.HandleException("Clipboard Error", "TweetDuck could not access the clipboard as it is currently used by another process.", true, e);
            }
        }

        public static IEnumerable<Browser> FindInstalledBrowsers(){
            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;
        }
    }
}