ame-migration-script/src/ame-upgrade-script/ConsoleTUI.cs

using System;
using System.Diagnostics;
using System.Runtime.InteropServices;
using System.Windows.Forms;
using Microsoft.Win32;

namespace Ameliorated.ConsoleUtils
{
    public static partial class ConsoleTUI
    {
        private const int MF_BYCOMMAND = 0x00000000;
        private const int SC_CLOSE = 0xF060;
        private const int SC_MINIMIZE = 0xF020;
        private const int SC_MAXIMIZE = 0xF030;
        private const int SC_SIZE = 0xF000; //resize

        private const uint CHECK_QUICK_EDIT = 0x0040;
        private const int ENABLE_QUICK_EDIT = 0x40 | 0x80;

        // STD_INPUT_HANDLE (DWORD): -10 is the standard input device.
        private const int STD_INPUT_HANDLE = -10;
        private static string PreviousTitle;
        private static int PreviousBufferHeight = 26;
        private static int PreviousBufferWidth = 80;
        private static int PreviousSizeHeight = 26;
        private static int PreviousSizeWidth = 80;

        private static bool IsInitialized;
        private static int InitializedWidth = 80;

        public static void ShowErrorBox(string message, string caption)
        {
            NativeWindow window = new NativeWindow();
            window.AssignHandle(Process.GetCurrentProcess().MainWindowHandle);
            MessageBox.Show(window, message, caption, MessageBoxButtons.OK, MessageBoxIcon.Error);
        }

        public enum BackdropType
        {
            None = 1,
            Mica = 2,
            Acrylic = 3,
            Tabbed = 4
        }
        public static readonly int WinVer = Int32.Parse(Registry.LocalMachine.OpenSubKey(@"SOFTWARE\Microsoft\Windows NT\CurrentVersion").GetValue("CurrentBuildNumber").ToString());
        [DllImport("dwmapi.dll")]
        private static extern int DwmSetWindowAttribute(IntPtr hwnd, int attribute, ref int pvAttribute, int cbAttribute);

        public static void Initialize(string title, int width = 80, int height = 26, bool resize = false, bool quickedit = false)
        {
            if (width < 2) throw new ArgumentException("Width must be greater than one.");

            IsInitialized = true;
            PreviousSizeHeight = Console.WindowHeight;
            PreviousSizeWidth = Console.WindowWidth;
            PreviousBufferHeight = Console.BufferHeight;
            PreviousBufferWidth = Console.BufferWidth;

            Console.SetWindowSize(width, height);
            Console.SetBufferSize(width, height);
            Console.SetWindowSize(width, height);

            InitializedWidth = width;

            Console.Clear();
            
            Console.CursorVisible = false;

            PreviousTitle = Console.Title;
            Console.Title = title;

            try
            {
                if ((Console.CursorLeft == 0 && Console.CursorTop == 0) || ParentProcess.ProcessName.Equals("Explorer", StringComparison.OrdinalIgnoreCase))
                {
                    var bd = (int)BackdropType.Mica;
                    var trueA = 0x01;

                    if (WinVer >= 22523)
                    {
                        var handle = Process.GetCurrentProcess().MainWindowHandle;
                        DwmSetWindowAttribute(handle, 38, ref bd, Marshal.SizeOf<int>());
                        DwmSetWindowAttribute(handle, 20, ref trueA, Marshal.SizeOf<int>());
                    }
                }
            } catch (Exception e) { }
            
            if (!resize)
                try
                {
                    DisableResize();
                } catch (Exception e)
                {
                    //ConsoleUtils.WriteError("Error disabling window resize - " + e.Message);
                }

            if (!quickedit)
                try
                {
                    DisableQuickEdit();
                } catch (Exception e)
                {
                    //ConsoleUtils.WriteError("Error disabling quickedit - " + e.Message);
                }
        }

        public static void Close()
        {
            if (!IsInitialized) throw new MethodAccessException("Console TUI must be initialized before calling other TUI functions.");
            IsInitialized = false;

            var parent = ParentProcess.ProcessName;
            if (parent.Equals("Explorer", StringComparison.CurrentCultureIgnoreCase)) return;

            try
            {
                EnableResize();
            } catch (Exception e)
            {
                //ConsoleUtils.WriteError("Error enabling window resize - " + e.Message);
            }

            try
            {
                EnableQuickEdit();
            } catch (Exception e)
            {
                //ConsoleUtils.WriteError("Error enabling quickedit - " + e.Message);
            }

            Console.CursorVisible = true;
            Console.Clear();
            Console.Title = PreviousTitle;

            Console.SetWindowSize(PreviousSizeWidth, PreviousSizeHeight);
            Console.SetBufferSize(PreviousBufferWidth, PreviousBufferHeight);
        }

        [DllImport("user32.dll")]
        public static extern int DeleteMenu(IntPtr hMenu, int nPosition, int wFlags);

        [DllImport("user32.dll")]
        private static extern IntPtr GetSystemMenu(IntPtr hWnd, bool bRevert);

        [DllImport("kernel32.dll", ExactSpelling = true)]
        private static extern IntPtr GetConsoleWindow();

        private static void DisableResize()
        {
            var handle = GetConsoleWindow();
            var sysMenu = GetSystemMenu(handle, false);

            if (handle != IntPtr.Zero)
            {
                //DeleteMenu(sysMenu, SC_CLOSE, MF_BYCOMMAND);
                //DeleteMenu(sysMenu, SC_MINIMIZE, MF_BYCOMMAND);
                DeleteMenu(sysMenu, SC_MAXIMIZE, MF_BYCOMMAND);
                DeleteMenu(sysMenu, SC_SIZE, MF_BYCOMMAND); //resize
            }
        }

        private static void EnableResize()
        {
            var handle = GetConsoleWindow();
            GetSystemMenu(handle, true);
        }

        [DllImport("kernel32.dll", SetLastError = true)]
        private static extern IntPtr GetStdHandle(int nStdHandle);

        [DllImport("kernel32.dll")]
        private static extern bool GetConsoleMode(IntPtr hConsoleHandle, out uint lpMode);

        [DllImport("kernel32.dll")]
        private static extern bool SetConsoleMode(IntPtr hConsoleHandle, uint dwMode);

        private static void DisableQuickEdit()
        {
            var consoleHandle = GetStdHandle(STD_INPUT_HANDLE);

            // get current console mode
            uint consoleMode;
            GetConsoleMode(consoleHandle, out consoleMode);

            // set the new mode
            SetConsoleMode(consoleHandle, consoleMode &= ~CHECK_QUICK_EDIT);
        }

        private static void EnableQuickEdit()
        {
            var consoleHandle = GetStdHandle(STD_INPUT_HANDLE);

            // get current console mode
            uint consoleMode;
            GetConsoleMode(consoleHandle, out consoleMode);

            // set the new mode
            SetConsoleMode(consoleHandle, consoleMode | ENABLE_QUICK_EDIT);
        }
    }
    
    
    
        public class ChoicePrompt : Prompt
        {
            public string Choices { get; set; } = "YN";
            public bool BeepSound { get; set; } = true;
            public bool CaseSensitive { get; set; } = false;
            public bool AllowEscape { get; set; } = true;
            public bool AnyKey { get; set; } = false;

            public ConsoleColor? TextForeground { get; set; }

            private bool _bindToOpenFrame;
            

            public new int? Start()
            {
                if (Choices.Length < 1 && !AnyKey) throw new ArgumentException("There must be at least 1 choice.");

                Console.Write(Text);
                
                var cursorVisibility = Console.CursorVisible;
                int? result;

                while (true)
                {
                    Console.CursorVisible = true;
                    var key = Console.ReadKey(true);
                    if (AnyKey)
                    {
                        Console.CursorVisible = cursorVisibility;
                        return key.KeyChar;
                    }
                    
                    if (key.Key == ConsoleKey.Escape && AllowEscape)
                    {
                        Console.CursorVisible = cursorVisibility;
                        return null;
                    }

                    if (CaseSensitive)
                        result = Choices.IndexOf(key.KeyChar.ToString(), StringComparison.Ordinal);
                    else
                        result = Choices.IndexOf(key.KeyChar.ToString(), StringComparison.OrdinalIgnoreCase);
                    
                    if (result >= 0)
                    {

                        
                        if (!CaseSensitive) Console.Write(key.KeyChar.ToString().ToUpper());
                        else Console.Write(key.KeyChar.ToString());

                        break;
                    }
                    else if (BeepSound) Console.Beep();
                } 

                Console.CursorVisible = cursorVisibility;
                Console.WriteLine();

                return result.Value;
            }
        }


        public abstract class Prompt
        {
            /// <summary>
            ///     Text to be displayed before the input.
            /// </summary>
           
            public string Text { get; set; } = "";
            
            public int? MaxLength { get; set; }

            /// <summary>
            ///     (Optional)
            /// </summary>
           
            public ConsoleColor? InputForeground { get; set; } = null;

            public ConsoleColor? InputBackground { get; set; } = null;

            public bool BindToOpenFrame { get; set; } = true;
            public bool AllowEscape { get; set; } = true;

        }
}