// ==WindhawkMod==
// @id              classic-theme-windows
// @name            Classic Theme Windows
// @description     Forces Classic Theme on all Windows
// @version         0.3
// @author          Travis, Cynosphere
// @include         *
// @exclude         wininit.exe
// @exclude         winlogon.exe
// @exclude         taskmgr.exe
// @exclude         dwm.exe
// @exclude         C:\Windows\System32\*.scr
// @exclude         svchost.exe
// @exclude         taskhostw.exe
// @exclude         dllhost.exe
// @exclude         rundll32.exe
// @exclude         sihost.exe
// @exclude         lsass.exe
// @exclude         C:\Program Files (x86)\Steam\*
// @exclude         msedge.exe
// @exclude         vmware.exe
// @exclude         vmware-vmx.exe
// @exclude         Spotify.exe
// @exclude         smartscreen.exe
// @exclude         RuntimeBroker.exe
// @exclude         ApplicationFrameHost.exe
// @exclude         SystemSettings.exe
// @exclude         SecHealthUI.exe
// @exclude         SecurityHealthHost.exe
// @exclude         PhoneExperienceHost.exe
// @exclude         SecurityHealthTray.exe
// @exclude         Window Detective.exe
// @compilerOptions -luxtheme -ldwmapi -lole32 -lcomdlg32
// ==/WindhawkMod==

// ==WindhawkModReadme==
/*
# Classic Theme Windows
Forces Classic Theme on all Windows
*/
// ==/WindhawkModReadme==

#include <windows.h>
#include <uxtheme.h>
#include <dwmapi.h>

DWORD thisPid = -1;

// BasicThemer2 reimplementation to render classic titlebars
static const int DisableDWM = DWMNCRP_DISABLED;
static const int EnableDWM = DWMNCRP_ENABLED;

void BasicThemerEnable(HWND hwnd) {
    DwmSetWindowAttribute(hwnd, DWMWA_NCRENDERING_POLICY, &DisableDWM, sizeof(int));
}
void BasicThemerDisable(HWND hwnd) {
    DwmSetWindowAttribute(hwnd, DWMWA_NCRENDERING_POLICY, &EnableDWM, sizeof(int));
}

using CreateWindowExW_t = decltype(&CreateWindowExW);
CreateWindowExW_t CreateWindowExW_Orig;
HWND WINAPI CreateWindowExW_Hook(DWORD dwExStyle,LPCWSTR lpClassName,LPCWSTR lpWindowName,DWORD dwStyle,int X,int Y,int nWidth,int nHeight,HWND hWndParent,HMENU hMenu,HINSTANCE hInstance,LPVOID lpParam) {
    HWND res = CreateWindowExW_Orig(dwExStyle,lpClassName,lpWindowName,dwStyle,X,Y,nWidth,nHeight,hWndParent,hMenu,hInstance,lpParam);

    if (thisPid != -1) {
        DWORD targetPid;
        GetWindowThreadProcessId(res, &targetPid);
        if (targetPid == thisPid) {
            BasicThemerEnable(res);
        }
    }

    return res;
}
using CreateWindowExA_t = decltype(&CreateWindowExA);
CreateWindowExA_t CreateWindowExA_Orig;
HWND WINAPI CreateWindowExA_Hook(DWORD dwExStyle,LPCSTR lpClassName,LPCSTR lpWindowName,DWORD dwStyle,int X,int Y,int nWidth,int nHeight,HWND hWndParent,HMENU hMenu,HINSTANCE hInstance,LPVOID lpParam) {
    HWND res = CreateWindowExA_Orig(dwExStyle,lpClassName,lpWindowName,dwStyle,X,Y,nWidth,nHeight,hWndParent,hMenu,hInstance,lpParam);

    if (thisPid != -1) {
        DWORD targetPid;
        GetWindowThreadProcessId(res, &targetPid);
        if (targetPid == thisPid) {
            BasicThemerEnable(res);
        }
    }

    return res;
}

BOOL CALLBACK EnableBasicThemerForAll(HWND hWnd, LPARAM lParam) {
    if (IsWindow(hWnd))
        BasicThemerEnable(hWnd);

    return TRUE;
}
BOOL CALLBACK DisableBasicThemerForAll(HWND hWnd, LPARAM lParam) {
    if (IsWindow(hWnd))
        BasicThemerDisable(hWnd);

    return TRUE;
}

// File picker
template<class S>
CLSID CreateGUID(const S& hexString)
{
    CLSID clsid;
    CLSIDFromString(hexString, &clsid);

    return clsid;
}

const CLSID CLSID_FileOpenDialog = CreateGUID(L"{DC1C5A9C-E88A-4dde-A5A1-60F82A20AEF7}");
const CLSID CLSID_FileOpenDialogLegacy = CreateGUID(L"{725F645B-EAED-4fc5-B1C5-D9AD0ACCBA5E}");
const CLSID CLSID_FileSaveDialog = CreateGUID(L"{c0b4e2f3-ba21-4773-8dba-335ec946eb8b}");
const CLSID CLSID_FileSaveDialogLegacy = CreateGUID(L"{AF02484C-A0A9-4669-9051-058AB12B9195}");

using CoCreateInstance_t = decltype(&CoCreateInstance);
CoCreateInstance_t CoCreateInstance_Original;
HRESULT WINAPI CoCreateInstance_Hook(REFCLSID rclsid, LPUNKNOWN pUnkOuter, DWORD dwClsContext, REFIID riid, LPVOID *ppv) {
    if (rclsid == CLSID_FileOpenDialog) {
        return CoCreateInstance_Original(CLSID_FileOpenDialogLegacy, pUnkOuter, dwClsContext, riid, ppv);
    } else if (rclsid == CLSID_FileSaveDialog) {
        return CoCreateInstance_Original(CLSID_FileSaveDialogLegacy, pUnkOuter, dwClsContext, riid, ppv);
    }

    return CoCreateInstance_Original(rclsid, pUnkOuter, dwClsContext, riid, ppv);
}

// Windhawk functions
BOOL Wh_ModInit() {
    // Enable classic theme
    SetThemeAppProperties(0);

    // BasicThemer hooks
    Wh_SetFunctionHook((void*)CreateWindowExW,
                       (void*)CreateWindowExW_Hook,
                       (void**)&CreateWindowExW_Orig);
    Wh_SetFunctionHook((void*)CreateWindowExA,
                       (void*)CreateWindowExA_Hook,
                       (void**)&CreateWindowExA_Orig);

    // Iterate every window to enable BasicThemer
    thisPid = GetProcessIdOfThread(GetCurrentThread());
    EnumWindows(EnableBasicThemerForAll, thisPid);

    // File picker
    Wh_SetFunctionHook((void*)CoCreateInstance, (void*)CoCreateInstance_Hook,
                       (void**)&CoCreateInstance_Original);

    return TRUE;
}

void Wh_ModUninit() {
    // Disable classic theme
    if (GetThemeAppProperties() == 0) {
        SetThemeAppProperties(3);
    }

    // Iterate every window to disable BasicThemer
    thisPid = GetProcessIdOfThread(GetCurrentThread());
    EnumWindows(DisableBasicThemerForAll, thisPid);
}