byGalax
37 lines
1.6 KiB
TypeScript
37 lines
1.6 KiB
TypeScript
// Window content-protection adapter. Enables / disables OS-level
|
|
// screenshot and screen-recording blocking on the host BrowserWindow.
|
|
//
|
|
// Windows: WDA_MONITOR (SetWindowDisplayAffinity) — the window surface
|
|
// appears black in any screen capture tool (OBS, Snipping Tool,
|
|
// Win+PrtScr, etc.) while protection is enabled.
|
|
// macOS: NSWindowSharingNone — equivalent coverage for QuickTime,
|
|
// Cmd+Shift+3/4, and external recorders.
|
|
// Linux: No-op. Electron exposes the API on all platforms but the
|
|
// X11/Wayland compositors don't honour it in Electron 33.
|
|
//
|
|
// Called by the renderer during view-once image reveals so the image
|
|
// cannot be captured by an OS-level screenshot while it is on screen.
|
|
|
|
import { BrowserWindow, ipcMain } from 'electron';
|
|
|
|
import { CHANNELS, type WindowSetContentProtectionArgs } from '../ipc-types';
|
|
|
|
export function register(mainWindow: BrowserWindow): void {
|
|
ipcMain.handle(
|
|
CHANNELS.WINDOW_SET_CONTENT_PROTECTION,
|
|
(_evt, args: WindowSetContentProtectionArgs) => {
|
|
// Electron's setContentProtection covers Windows (WDA_MONITOR) and
|
|
// macOS (NSWindowSharingNone) in one call. No-op on Linux X11.
|
|
// Wrapped in try/catch because the window can already be destroyed
|
|
// by the time this fires during a teardown.
|
|
try {
|
|
const win = BrowserWindow.fromWebContents(_evt.sender) ?? mainWindow;
|
|
if (!win || win.isDestroyed()) return;
|
|
win.setContentProtection(args.enabled);
|
|
} catch (err) {
|
|
console.warn('setContentProtection failed', err);
|
|
}
|
|
},
|
|
);
|
|
}
|