amecs/src/amecs.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Net;
using System.Runtime.InteropServices;
using System.Threading.Tasks;
using Ameliorated.ConsoleUtils;

namespace amecs
{
    public class amecs
    {
        public static bool RunBasicAction(string status, string result, Action action, bool logoff = false, bool restart = false)
        {
            ConsoleTUI.OpenFrame.WriteCentered(status);

            try
            {
                using (new ConsoleUtils.LoadingIndicator(true))
                {
                    action.Invoke();
                }
            } catch (Exception e)
            {
                Console.WriteLine();
                ConsoleTUI.OpenFrame.Close("Error: " + e.Message.TrimEnd('\n').TrimEnd('\r'), ConsoleColor.Red, Console.BackgroundColor, new ChoicePrompt() { AnyKey = true, Text = "Press any key to return to the Menu: " });
                return false;
            }
            Console.WriteLine();
            if (logoff)
            {
                if ((int?)ConsoleTUI.OpenFrame.Close(result, ConsoleColor.Green, Console.BackgroundColor, new ChoicePrompt()
                    {
                        TextForeground = ConsoleColor.Yellow,
                        Text = "Logoff to apply changes? (Y/N): "
                    }) == 0) amecs.RestartWindows(true);
                return true;
            }
            if (restart)
            {
                if ((int?)ConsoleTUI.OpenFrame.Close(result, ConsoleColor.Green, Console.BackgroundColor, new ChoicePrompt()
                    {
                        TextForeground = ConsoleColor.Yellow,
                        Text = "Restart to apply changes? (Y/N): "
                    }) == 0) amecs.RestartWindows(false);
                return true;
            }

            ConsoleTUI.OpenFrame.Close(result, ConsoleColor.Green, Console.BackgroundColor, new ChoicePrompt() {AnyKey = true, Text = "Press any key to return to the Menu: "});
            return true;
        }

        public static bool RunBasicAction(string status, string result, Action<int> action, ChoicePrompt prompt = null, bool logoff = false, bool restart = false)
        {
            int choice = -1;
            if (prompt != null)
            {
                var choiceRes = prompt.Start();
                if (!choiceRes.HasValue)
                    return true;
                choice = choiceRes.Value;
            }

            ConsoleTUI.OpenFrame.WriteCentered(status);

            try
            {
                using (new ConsoleUtils.LoadingIndicator(true))
                {
                    action.Invoke(choice);
                }
            } catch (Exception e)
            {
                Console.WriteLine();
                ConsoleTUI.OpenFrame.Close("Error: " + e.Message.TrimEnd('\n').TrimEnd('\r'), ConsoleColor.Red, Console.BackgroundColor, new ChoicePrompt() { AnyKey = true, Text = "Press any key to return to the Menu: " });
                return false;
            }
            Console.WriteLine();
            if (logoff)
            {
                if ((int?)ConsoleTUI.OpenFrame.Close(result, ConsoleColor.Green, Console.BackgroundColor, new ChoicePrompt()
                    {
                        TextForeground = ConsoleColor.Yellow,
                        Text = "Logoff to apply changes? (Y/N): "
                    }) == 0) amecs.RestartWindows(true);
                return true;
            }
            if (restart)
            {
                if ((int?)ConsoleTUI.OpenFrame.Close(result, ConsoleColor.Green, Console.BackgroundColor, new ChoicePrompt()
                    {
                        TextForeground = ConsoleColor.Yellow,
                        Text = "Restart to apply changes? (Y/N): "
                    }) == 0) amecs.RestartWindows(false);
                return true;
            }

            ConsoleTUI.OpenFrame.Close(result, ConsoleColor.Green, Console.BackgroundColor, new ChoicePrompt() {AnyKey = true, Text = "Press any key to return to the Menu: "});
            return true;
        }
        
        [Flags]
        enum ExitWindows : uint
        {
            // ONE of the following five:
            LogOff = 0x00,
            ShutDown = 0x01,
            Reboot = 0x02,
            PowerOff = 0x08,
            RestartApps = 0x40,
            // plus AT MOST ONE of the following two:
            Force = 0x04,
            ForceIfHung = 0x10,
        }
        [Flags]
        enum ShutdownReason : uint
        {
            MajorApplication = 0x00040000,
            MajorHardware = 0x00010000,
            MajorLegacyApi = 0x00070000,
            MajorOperatingSystem = 0x00020000,
            MajorOther = 0x00000000,
            MajorPower = 0x00060000,
            MajorSoftware = 0x00030000,
            MajorSystem = 0x00050000,

            MinorBlueScreen = 0x0000000F,
            MinorCordUnplugged = 0x0000000b,
            MinorDisk = 0x00000007,
            MinorEnvironment = 0x0000000c,
            MinorHardwareDriver = 0x0000000d,
            MinorHotfix = 0x00000011,
            MinorHung = 0x00000005,
            MinorInstallation = 0x00000002,
            MinorMaintenance = 0x00000001,
            MinorMMC = 0x00000019,
            MinorNetworkConnectivity = 0x00000014,
            MinorNetworkCard = 0x00000009,
            MinorOther = 0x00000000,
            MinorOtherDriver = 0x0000000e,
            MinorPowerSupply = 0x0000000a,
            MinorProcessor = 0x00000008,
            MinorReconfig = 0x00000004,
            MinorSecurity = 0x00000013,
            MinorSecurityFix = 0x00000012,
            MinorSecurityFixUninstall = 0x00000018,
            MinorServicePack = 0x00000010,
            MinorServicePackUninstall = 0x00000016,
            MinorTermSrv = 0x00000020,
            MinorUnstable = 0x00000006,
            MinorUpgrade = 0x00000003,
            MinorWMI = 0x00000015,

            FlagUserDefined = 0x40000000,
            FlagPlanned = 0x80000000
        }
        
        [DllImport("user32.dll")]
        static extern bool ExitWindowsEx(uint uFlags, uint dwReason);

        [DllImport("advapi32.dll", CharSet=CharSet.Auto, SetLastError=true)]
        private static extern bool InitiateSystemShutdownEx(
            string lpMachineName,
            string lpMessage,
            uint dwTimeout,
            bool bForceAppsClosed,
            bool bRebootAfterShutdown,
            ShutdownReason dwReason);

        public static void RestartWindows(bool logoff, bool cmd = false)
        {
            if (cmd)
            {
                var psi = new ProcessStartInfo("shutdown","/r /t 0");
                psi.CreateNoWindow = true;
                psi.UseShellExecute = false;
                Process.Start(psi);
                
                Environment.Exit(0);
            }
            
            if (!logoff)
            {
                NSudo.GetShutdownPrivilege();
                InitiateSystemShutdownEx(null, null, 0, false, true, ShutdownReason.MinorInstallation);
            } else
                ExitWindowsEx((uint)ExitWindows.LogOff | (uint)ExitWindows.ForceIfHung, (uint)ShutdownReason.MinorInstallation);

            Environment.Exit(0);
        }

        public static object ShowDefaultMenu(List<Menu.MenuItem> list)
        {
            list.AddRange(new [] {             
                Menu.MenuItem.BlankStatic,
                new Menu.MenuItem("Next Page", null) { IsNextButton = true },
                new Menu.MenuItem("Previous Page", null) { IsPreviousButton = true },
                new Menu.MenuItem("Return to Menu", new Func<bool>(() => true)) { IsStatic = true },
                new Menu.MenuItem("Exit", new Func<bool>(Globals.Exit)) { IsStatic = true },
            });
            
            var mainMenu = new Menu()
            {
                Choices = list,
                SelectionForeground = ConsoleColor.Green
            };
            mainMenu.Write();
            return mainMenu.Load();
        }


        [DllImport("wininet.dll", SetLastError = true)]
        public static extern bool InternetCheckConnection(string lpszUrl, int dwFlags, int dwReserved);

        [DllImport("wininet.dll", SetLastError=true)]
        private extern static bool InternetGetConnectedState(out int lpdwFlags, int dwReserved);

        public static bool IsInternetAvailable()
        {
            try
            {
                try
                {
                    if (!InternetCheckConnection("http://archlinux.org", 1, 0))
                    {
                        if (!InternetCheckConnection("http://google.com", 1, 0))
                            return false;
                    }
                    return true;
                }
                catch
                {
                    var request = (HttpWebRequest)WebRequest.Create("http://google.com");
                    request.KeepAlive = false;
                    request.Timeout = 5000;
                    using (var response = (HttpWebResponse)request.GetResponse())
                        return true;
                }
            }
            catch
            {
                return false;
            }
        }
        
        
        
        
        
        
        
        
        
        
        
        
        
        
        
        
    }
}