www

www

IDA/project/PreventScreenshot.cpp

@@ -0,0 +1,32 @@
+#include "PreventScreenshot.h"
+
+#include "helpers.h"
+
+PreventScreenshotModule::PreventScreenshotModule()
+    : BaseModule("PreventScreenshot", "Prevent Screenshot", ModuleCategory::Visual) {}
+
+void PreventScreenshotModule::RegisterTab() {
+}
+
+void PreventScreenshotModule::Render() {
+    if (!ImGui::CollapsingHeader(Tr("Prevent Screenshot"), ImGuiTreeNodeFlags_DefaultOpen)) {
+        return;
+    }
+
+    bool enabled = Enabled();
+    if (ImGui::Checkbox(Tr("ON / OFF"), &enabled)) {
+        SetEnabled(enabled);
+    }
+}
+
+void PreventScreenshotModule::Update() {
+    if (!g_Runtime.overlay_hwnd) {
+        return;
+    }
+
+    const DWORD affinity = Enabled()
+        ? (IsWindowsBuildAtLeast(19041) ? 0x11u : 0x1u)
+        : 0u;
+
+    SetWindowDisplayAffinity(g_Runtime.overlay_hwnd, affinity);
+}