#include "pch.h"
#include "ExclModeHack.h"
#include <shellapi.h>
#include "App.h"
#include "Logger.h"
#include "Config.h"


// 模拟 D3D 独占全屏模式，以起到免打扰的效果
// SHQueryUserNotificationState 通常被用来检测是否有 D3D 游戏独占全屏，以确定是否应该向用户推送通知/弹窗
// 此函数内部使用名为 __DDrawExclMode__ 的 mutex 检测独占全屏，因此这里直接获取该 mutex 以模拟独占全屏
// 感谢 @codehz 提供的思路 https://github.com/Blinue/Magpie/issues/245
ExclModeHack::ExclModeHack() {
	if (!App::Get().GetConfig().IsSimulateExclusiveFullscreen()) {
		return;
	}

	QUERY_USER_NOTIFICATION_STATE state;
	HRESULT hr = SHQueryUserNotificationState(&state);
	if (FAILED(hr)) {
		Logger::Get().ComError("SHQueryUserNotificationState 失败", hr);
		return;
	}
	if (state != QUNS_ACCEPTS_NOTIFICATIONS) {
		Logger::Get().Info("已处于免打扰状态");
		return;
	}

	_exclModeMutex.reset(Utils::SafeHandle(
		OpenMutex(SYNCHRONIZE, FALSE, L"__DDrawExclMode__")));
	if (!_exclModeMutex) {
		Logger::Get().Win32Error("OpenMutex 失败");
		return;
	}

	DWORD result = WaitForSingleObject(_exclModeMutex.get(), 0);
	if (result != WAIT_OBJECT_0) {
		Logger::Get().Error("获取 __DDrawExclMode__ 失败");
		_exclModeMutex.reset();
		return;
	}

	hr = SHQueryUserNotificationState(&state);
	if (FAILED(hr)) {
		Logger::Get().ComError("SHQueryUserNotificationState 失败", hr);
		ReleaseMutex(_exclModeMutex.get());
		_exclModeMutex.reset();
		return;
	}
	if (state != QUNS_RUNNING_D3D_FULL_SCREEN) {
		Logger::Get().Error("模拟独占全屏失败");
		ReleaseMutex(_exclModeMutex.get());
		_exclModeMutex.reset();
		return;
	}

	Logger::Get().Info("模拟独占全屏成功");
}

ExclModeHack::~ExclModeHack() {
	if (_exclModeMutex) {
		ReleaseMutex(_exclModeMutex.get());
	}
}