using System;
using System.Drawing;
using System.Runtime.InteropServices;

// The 'windowHandle' parameter will contain the window handle for the:
//   - Active window when run by hotkey
//   - Trigger target when run by a Trigger rule
//   - TitleBar Button owner when run by a TitleBar Button
//   - Jump List owner when run from a Taskbar Jump List
//   - Currently focused window if none of these match
public static class DisplayFusionFunction
{
    [DllImport("User32.dll", CharSet = CharSet.Unicode, PreserveSig = true)]
    private static extern IntPtr FindWindow([MarshalAs(UnmanagedType.LPWStr)] string lpClassName, [MarshalAs(UnmanagedType.LPWStr)] string lpWindowName);

public static void Run(IntPtr windowHandle)
{
        //get the system tray window
        IntPtr trayWnd = FindWindow("Shell_TrayWnd", null);

        //focus the window
        BFS.Window.Focus(trayWnd);
        
        //mute the computer
        BFS.Input.SendKeys("{VK_173}");
        
        //send the nvidia surround hotkey
        BFS.Input.SendKeys("%(^(S))");
        
        //wait for surround to be applied
        BFS.General.ThreadWait(3500);
        
        //unmute the computer
        BFS.Input.SendKeys("{VK_173}");
        
        //load a monitor profile
        //NOTE: Change the profile names with profiles for your computer
        if(IsSurroundEnabled())
            BFS.DisplayFusion.LoadMonitorProfileHideErrors("Surround");
        else
            BFS.DisplayFusion.LoadMonitorProfileHideErrors("144hz y 60hz");
}

    private static bool IsSurroundEnabled()
    {
        return (BFS.Monitor.GetPrimaryMonitorBounds().Width / (double)BFS.Monitor.GetPrimaryMonitorBounds().Height > 1.8d);
    }
}

;==============================================================================================
; First-run configuration
;==============================================================================================
IniRead, FirstRunVar, %appdata%\NVSToggle\NVSToggle.ini, NVSToggle, FirstRun
IF (FirstRunVar = "Completed")
{
Goto, DetectNow
}

MsgBox, 65, Nvidia Surround Toggle, To allow this application to configure itself for first-time use, please enable Nvidia Surround using the Nvidia Control Panel now. `n`nClick "OK" when ready, or "Cancel" to quit.
IfMsgBox Ok
Goto, DoubleCheck
else
Exit

DoubleCheck:
IF (A_ScreenWidth/A_ScreenHeight > 1.8)
{
Goto, FirstRun
}

MsgBox, 21, Nvidia Surround Toggle, Error: Nvidia Surround is not enabled.
IfMsgBox Retry
Goto, DoubleCheck
else
Exit

FirstRun:
FileCreateDir, %appdata%\NVSToggle
IniWrite, Completed, %appdata%\NVSToggle\NVSToggle.ini, NVSToggle, FirstRun

;==============================================================================================
; Determine current display state
;==============================================================================================
DetectNow:
IF (A_ScreenWidth/A_ScreenHeight > 1.8)
{
Goto, ToggleOff
}

IF (A_ScreenWidth/A_ScreenHeight < 1.8)
{
Goto, ToggleOn
}

IF (A_ScreenWidth/A_ScreenHeight = 1.8)
{
Goto, ToggleOn
}

Exit

;==============================================================================================
; Toggle-off routine (switch to Extended Desktop mode)
;==============================================================================================
ToggleOff:
Send {Volume_Mute}
WinActivate ahk_class Shell_TrayWnd
Send {LCtrl down}
Send {LAlt down}
Send {s down}
Sleep 100
Send {LCtrl up}{LAlt up}{s up}
Sleep 3500
Send {Volume_Mute}
Run C:\Program Files (x86)\DisplayFusion\displayfusioncommand.exe -monitorloadprofile "144hz y 60hz"
Exit

;==============================================================================================
; Toggle-on routine (switch to Surround mode)
;==============================================================================================
ToggleOn:
WinActivate ahk_class Shell_TrayWnd
Send {LCtrl down}
Send {LAlt down}
Send {s down}
Sleep 100
Send {LCtrl up}{LAlt up}{s up}
Sleep 3500
IF (A_ScreenWidth/A_ScreenHeight < 1.8)
{
Goto, SurroundFail
}
IF (A_ScreenWidth/A_ScreenHeight = 1.8)
{
Goto, SurroundFail
}
Exit

SurroundFail:
MsgBox, 21, Nvidia Surround Toggle, Error: Nvidia Surround could not be enabled!
FileDelete, %appdata%\NVSToggle\NVSToggle.ini
Exit