// The 15 second timeout allows enough time to cancel if someone's using the computer.
// The config class on line 7 can be used to customize what happens.

using System; using System.Text; using System.Windows.Forms; using System.Runtime.InteropServices; using System.Diagnostics;
public static class DisplayFusionFunction
{
	internal static class config {
		// number of seconds to wait
		internal const uint Timeout = 15;
		// The text in the messagebox
		internal const string Text = "The system will reboot in {0} seconds. After reboot, applications will be restarted. Press CANCEL or run \"shutdown.exe /a\" to cancel reboot.";
        // The title of the window.
		internal const string Title = "System rebooting in {0} seconds";
		// The icon. Valid icons are described at https://msdn.microsoft.com/en-us/library/system.windows.forms.messageboxicon(v=vs.110).aspx
		internal const MessageBoxIcon icon = MessageBoxIcon.Exclamation;
	}
	
	public static void Run(IntPtr windowHandle)
	{
        Process proc = new Process();
        proc.StartInfo.FileName = "CMD.exe";
        proc.StartInfo.CreateNoWindow = true;
        proc.StartInfo.Arguments = "/c shutdown.exe /g /t "+config.Timeout;
        proc.StartInfo.WindowStyle = ProcessWindowStyle.Hidden;
        proc.Start();
        proc.WaitForExit();
        string Title=String.Format(config.Title,config.Timeout);
        string Text=String.Format(config.Text,config.Timeout);
        DialogResult DoShutdown = MessageBoxEx.Show(Text,Title,MessageBoxButtons.OKCancel,config.icon,(config.Timeout-1));
		if(DoShutdown == DialogResult.Cancel) {
			Process proc2 = new Process();
            proc2.StartInfo.FileName = "CMD.exe";
            proc2.StartInfo.CreateNoWindow = true;
            proc2.StartInfo.Arguments = "/c shutdown.exe /a";
            proc2.StartInfo.WindowStyle = ProcessWindowStyle.Hidden;
            proc2.Start();
            proc2.WaitForExit();
		}
	}
}

//borrowed from http://www.codeproject.com/Articles/7968/MessageBox-with-a-timeout-for-NET
//unnecessary stuff removed
namespace System.Windows.Forms
{
	public class MessageBoxEx
	{
		public static DialogResult Show(string text, string caption, MessageBoxButtons buttons, MessageBoxIcon icon, uint uTimeout)
		{
			Setup(caption, uTimeout);
			return MessageBox.Show(text, caption, buttons, icon);
		}
		
		public delegate IntPtr HookProc(int nCode, IntPtr wParam, IntPtr lParam);
		public delegate void TimerProc(IntPtr hWnd, uint uMsg, UIntPtr nIDEvent, uint dwTime);
		
		public const int WH_CALLWNDPROCRET = 12;
		public const int WM_DESTROY = 0x0002;
		public const int WM_INITDIALOG = 0x0110;
		public const int WM_TIMER = 0x0113;
		public const int WM_USER = 0x400;
		public const int DM_GETDEFID = WM_USER + 0;

		[DllImport("User32.dll")]
		public static extern UIntPtr SetTimer(IntPtr hWnd, UIntPtr nIDEvent, uint uElapse, TimerProc lpTimerFunc);

		[DllImport("User32.dll")]
		public static extern IntPtr SendMessage(IntPtr hWnd, int Msg, IntPtr wParam, IntPtr lParam);

		[DllImport("user32.dll")]
		public static extern IntPtr SetWindowsHookEx(int idHook, HookProc lpfn, IntPtr hInstance, int threadId);

		[DllImport("user32.dll")]
		public static extern int UnhookWindowsHookEx(IntPtr idHook);
			
		[DllImport("user32.dll")]
		public static extern IntPtr CallNextHookEx(IntPtr idHook, int nCode, IntPtr wParam, IntPtr lParam);

		[DllImport("user32.dll")]
		public static extern int GetWindowTextLength(IntPtr hWnd);

		[DllImport("user32.dll")]
		public static extern int GetWindowText(IntPtr hWnd, StringBuilder text, int maxLength);

		[DllImport("user32.dll")]
		public static extern int EndDialog(IntPtr hDlg, IntPtr nResult);

		[StructLayout(LayoutKind.Sequential)]
		public struct CWPRETSTRUCT
		{
			public IntPtr lResult;
			public IntPtr lParam;
			public IntPtr wParam;
			public uint   message;
			public IntPtr hwnd;
		};

		private const int TimerID = 42;
		private static HookProc hookProc;
		private static TimerProc hookTimer;
		private static uint hookTimeout;
		private static string hookCaption;
		private static IntPtr hHook;
		
		static MessageBoxEx()
		{
			hookProc = new HookProc(MessageBoxHookProc);
			hookTimer = new TimerProc(MessageBoxTimerProc);
			hookTimeout = 0;
			hookCaption = null;
			hHook = IntPtr.Zero;
		}
		
		private static void Setup(string caption, uint uTimeout)
		{
			if (hHook != IntPtr.Zero)
				throw new NotSupportedException("multiple calls are not supported");
			
			hookTimeout = uTimeout;
			hookCaption = caption != null ? caption : "";
			hHook = SetWindowsHookEx(WH_CALLWNDPROCRET, hookProc, IntPtr.Zero, AppDomain.GetCurrentThreadId());
		}
		
		private static IntPtr MessageBoxHookProc(int nCode, IntPtr wParam, IntPtr lParam)
		{
			if (nCode < 0)
				return CallNextHookEx(hHook, nCode, wParam, lParam);

			CWPRETSTRUCT msg = (CWPRETSTRUCT)Marshal.PtrToStructure(lParam, typeof(CWPRETSTRUCT));
			IntPtr hook = hHook;

			if (hookCaption != null && msg.message == WM_INITDIALOG)
			{
				int nLength = GetWindowTextLength(msg.hwnd);
				StringBuilder text = new StringBuilder(nLength + 1);

				GetWindowText(msg.hwnd, text, text.Capacity);

				if (hookCaption == text.ToString())
				{
					hookCaption = null;
					SetTimer(msg.hwnd, (UIntPtr)TimerID, hookTimeout, hookTimer);
					UnhookWindowsHookEx(hHook);
					hHook = IntPtr.Zero;
				}
			}

			return CallNextHookEx(hook, nCode, wParam, lParam);
		}

		private static void MessageBoxTimerProc(IntPtr hWnd, uint uMsg, UIntPtr nIDEvent, uint dwTime)
		{
			if (nIDEvent == (UIntPtr)TimerID)
			{
				short dw = (short)SendMessage(hWnd, DM_GETDEFID, IntPtr.Zero, IntPtr.Zero);
			
				EndDialog(hWnd, (IntPtr)dw);
			}
		}
	}
}