Overlay.cpp

#include "Overlay.h"
#include <raylib.h>
#include "rlImGui.h"
#include "WinWrapper.h"
#include <iostream>

Overlay::Overlay(const std::string title, int updateRate) : title(title), isMenuOpen(false) {
	SetConfigFlags(
		ConfigFlags::FLAG_MSAA_4X_HINT |
		ConfigFlags::FLAG_WINDOW_UNDECORATED |
		ConfigFlags::FLAG_WINDOW_TOPMOST |
		ConfigFlags::FLAG_WINDOW_TRANSPARENT |
		ConfigFlags::FLAG_WINDOW_UNFOCUSED |
		ConfigFlags::FLAG_WINDOW_MOUSE_PASSTHROUGH
	);

	int length = GetRandomValue(16, 48);
	std::string overlayTitle{};
	overlayTitle.reserve(GetRandomValue(16, 48));
	for (int i = 0; i < length; i++) {
		overlayTitle.append(1,
			GetRandomValue(0, 1) == 0 ?
				GetRandomValue('a', 'z') :
				GetRandomValue('A', 'Z')
		);
	}

	InitWindow(0, 0, overlayTitle.c_str());
	SetWindowSize(GetScreenWidth(), GetScreenHeight() - 1);
	if (updateRate != 0) {
		SetTargetFPS(updateRate < 0 ? GetMonitorRefreshRate(GetCurrentMonitor()) : updateRate);
	}

	rlImGuiSetup(true);
}

Overlay::~Overlay() {
	rlImGuiShutdown();
	CloseWindow();
}

void Overlay::Update() {
	bool isHomePressed = IsWindowFocused() ?
		IsKeyPressed(KeyboardKey::KEY_INSERT) :
		WinWrapper::WGetAsyncKeyState(0x2D) & 1;
	if (isHomePressed) {
		this->isMenuOpen = !this->isMenuOpen;
		if (this->isMenuOpen) {
			ClearWindowState(ConfigFlags::FLAG_WINDOW_MOUSE_PASSTHROUGH);
			WinWrapper::WSetForegroundWindow(GetWindowHandle());
			this->OnFocusFound();
		}
		else {
			SetWindowState(ConfigFlags::FLAG_WINDOW_MOUSE_PASSTHROUGH);
			this->OnFocusLost();
		}
	}

	this->UpdateImGui();
}

void Overlay::Draw() {
	BeginDrawing();
	ClearBackground(BLANK);
	rlImGuiBegin();
	ImGui::GetIO().MouseDrawCursor = false;
	bool shouldShow = this->ShouldShowMenu();
	bool needsEnd = false;
	if (shouldShow) {
		ImGui::GetIO().MouseDrawCursor = true;
		needsEnd = ImGui::Begin(
			this->title.c_str(),
			nullptr,
			ImGuiWindowFlags_::ImGuiWindowFlags_NoCollapse
		);
	}

	this->DrawImGui();
	if (needsEnd) {
		ImGui::End();
	}

	rlImGuiEnd();
	EndDrawing();
}

bool Overlay::IsRunning() {
	return !WindowShouldClose();
}