|
| 1 | +export function parseWindowHandleFromSourceId(sourceId?: string | null) { |
| 2 | + if (!sourceId?.startsWith("window:")) { |
| 3 | + return null; |
| 4 | + } |
| 5 | + |
| 6 | + const handlePart = sourceId.split(":")[1]; |
| 7 | + if (!handlePart || !/^\d+$/.test(handlePart)) { |
| 8 | + return null; |
| 9 | + } |
| 10 | + |
| 11 | + return handlePart; |
| 12 | +} |
| 13 | + |
| 14 | +export function buildPowerShellCommand(sampleIntervalMs: number, windowHandle?: string | null) { |
| 15 | + const script = String.raw` |
| 16 | +$ErrorActionPreference = 'Stop' |
| 17 | +Add-Type -AssemblyName System.Drawing |
| 18 | +
|
| 19 | +$targetWindowHandle = ${windowHandle ? `'${windowHandle}'` : '$null'} |
| 20 | +
|
| 21 | +$source = @" |
| 22 | +using System; |
| 23 | +using System.Runtime.InteropServices; |
| 24 | +
|
| 25 | +public static class OpenScreenCursorInterop { |
| 26 | + [StructLayout(LayoutKind.Sequential)] |
| 27 | + public struct POINT { |
| 28 | + public int X; |
| 29 | + public int Y; |
| 30 | + } |
| 31 | +
|
| 32 | + [StructLayout(LayoutKind.Sequential)] |
| 33 | + public struct CURSORINFO { |
| 34 | + public int cbSize; |
| 35 | + public int flags; |
| 36 | + public IntPtr hCursor; |
| 37 | + public POINT ptScreenPos; |
| 38 | + } |
| 39 | +
|
| 40 | + [StructLayout(LayoutKind.Sequential)] |
| 41 | + public struct ICONINFO { |
| 42 | + [MarshalAs(UnmanagedType.Bool)] |
| 43 | + public bool fIcon; |
| 44 | + public int xHotspot; |
| 45 | + public int yHotspot; |
| 46 | + public IntPtr hbmMask; |
| 47 | + public IntPtr hbmColor; |
| 48 | + } |
| 49 | +
|
| 50 | + [StructLayout(LayoutKind.Sequential)] |
| 51 | + public struct RECT { |
| 52 | + public int Left; |
| 53 | + public int Top; |
| 54 | + public int Right; |
| 55 | + public int Bottom; |
| 56 | + } |
| 57 | +
|
| 58 | + [DllImport("user32.dll", SetLastError = true)] |
| 59 | + [return: MarshalAs(UnmanagedType.Bool)] |
| 60 | + public static extern bool GetCursorInfo(ref CURSORINFO pci); |
| 61 | +
|
| 62 | + [DllImport("user32.dll", SetLastError = true)] |
| 63 | + [return: MarshalAs(UnmanagedType.Bool)] |
| 64 | + public static extern bool GetWindowRect(IntPtr hWnd, out RECT lpRect); |
| 65 | +
|
| 66 | + [DllImport("user32.dll", SetLastError = true)] |
| 67 | + [return: MarshalAs(UnmanagedType.Bool)] |
| 68 | + public static extern bool IsWindow(IntPtr hWnd); |
| 69 | +
|
| 70 | + [DllImport("user32.dll", SetLastError = true)] |
| 71 | + public static extern IntPtr CopyIcon(IntPtr hIcon); |
| 72 | +
|
| 73 | + [DllImport("user32.dll", SetLastError = true)] |
| 74 | + [return: MarshalAs(UnmanagedType.Bool)] |
| 75 | + public static extern bool DestroyIcon(IntPtr hIcon); |
| 76 | +
|
| 77 | + [DllImport("user32.dll", SetLastError = true)] |
| 78 | + [return: MarshalAs(UnmanagedType.Bool)] |
| 79 | + public static extern bool GetIconInfo(IntPtr hIcon, out ICONINFO piconinfo); |
| 80 | +
|
| 81 | + [DllImport("gdi32.dll", SetLastError = true)] |
| 82 | + [return: MarshalAs(UnmanagedType.Bool)] |
| 83 | + public static extern bool DeleteObject(IntPtr hObject); |
| 84 | +} |
| 85 | +"@ |
| 86 | +
|
| 87 | +Add-Type -TypeDefinition $source |
| 88 | +
|
| 89 | +function Write-JsonLine($payload) { |
| 90 | + [Console]::Out.WriteLine(($payload | ConvertTo-Json -Compress -Depth 6)) |
| 91 | +} |
| 92 | +
|
| 93 | +function Get-TargetBounds() { |
| 94 | + if ([string]::IsNullOrWhiteSpace($targetWindowHandle)) { |
| 95 | + return $null |
| 96 | + } |
| 97 | +
|
| 98 | + try { |
| 99 | + $handleValue = [int64]::Parse($targetWindowHandle) |
| 100 | + $windowHandle = [IntPtr]::new($handleValue) |
| 101 | + if (-not [OpenScreenCursorInterop]::IsWindow($windowHandle)) { |
| 102 | + return $null |
| 103 | + } |
| 104 | +
|
| 105 | + $rect = New-Object OpenScreenCursorInterop+RECT |
| 106 | + if (-not [OpenScreenCursorInterop]::GetWindowRect($windowHandle, [ref]$rect)) { |
| 107 | + return $null |
| 108 | + } |
| 109 | +
|
| 110 | + $width = $rect.Right - $rect.Left |
| 111 | + $height = $rect.Bottom - $rect.Top |
| 112 | + if ($width -le 0 -or $height -le 0) { |
| 113 | + return $null |
| 114 | + } |
| 115 | +
|
| 116 | + return @{ |
| 117 | + x = $rect.Left |
| 118 | + y = $rect.Top |
| 119 | + width = $width |
| 120 | + height = $height |
| 121 | + } |
| 122 | + } |
| 123 | + catch { |
| 124 | + return $null |
| 125 | + } |
| 126 | +} |
| 127 | +
|
| 128 | +function Get-CursorAsset($cursorHandle, $cursorId) { |
| 129 | + $copiedHandle = [OpenScreenCursorInterop]::CopyIcon($cursorHandle) |
| 130 | + if ($copiedHandle -eq [IntPtr]::Zero) { |
| 131 | + return $null |
| 132 | + } |
| 133 | +
|
| 134 | + $iconInfo = New-Object OpenScreenCursorInterop+ICONINFO |
| 135 | + $hasIconInfo = [OpenScreenCursorInterop]::GetIconInfo($copiedHandle, [ref]$iconInfo) |
| 136 | +
|
| 137 | + try { |
| 138 | + $icon = [System.Drawing.Icon]::FromHandle($copiedHandle) |
| 139 | + $bitmap = New-Object System.Drawing.Bitmap $icon.Width, $icon.Height, ([System.Drawing.Imaging.PixelFormat]::Format32bppArgb) |
| 140 | + $graphics = [System.Drawing.Graphics]::FromImage($bitmap) |
| 141 | + $memoryStream = New-Object System.IO.MemoryStream |
| 142 | +
|
| 143 | + try { |
| 144 | + $graphics.Clear([System.Drawing.Color]::Transparent) |
| 145 | + $graphics.DrawIcon($icon, 0, 0) |
| 146 | + $bitmap.Save($memoryStream, [System.Drawing.Imaging.ImageFormat]::Png) |
| 147 | + $base64 = [System.Convert]::ToBase64String($memoryStream.ToArray()) |
| 148 | +
|
| 149 | + return @{ |
| 150 | + id = $cursorId |
| 151 | + imageDataUrl = "data:image/png;base64,$base64" |
| 152 | + width = $bitmap.Width |
| 153 | + height = $bitmap.Height |
| 154 | + hotspotX = if ($hasIconInfo) { $iconInfo.xHotspot } else { 0 } |
| 155 | + hotspotY = if ($hasIconInfo) { $iconInfo.yHotspot } else { 0 } |
| 156 | + } |
| 157 | + } |
| 158 | + finally { |
| 159 | + $memoryStream.Dispose() |
| 160 | + $graphics.Dispose() |
| 161 | + $bitmap.Dispose() |
| 162 | + $icon.Dispose() |
| 163 | + } |
| 164 | + } |
| 165 | + finally { |
| 166 | + if ($hasIconInfo) { |
| 167 | + if ($iconInfo.hbmMask -ne [IntPtr]::Zero) { |
| 168 | + [OpenScreenCursorInterop]::DeleteObject($iconInfo.hbmMask) | Out-Null |
| 169 | + } |
| 170 | + if ($iconInfo.hbmColor -ne [IntPtr]::Zero) { |
| 171 | + [OpenScreenCursorInterop]::DeleteObject($iconInfo.hbmColor) | Out-Null |
| 172 | + } |
| 173 | + } |
| 174 | + [OpenScreenCursorInterop]::DestroyIcon($copiedHandle) | Out-Null |
| 175 | + } |
| 176 | +} |
| 177 | +
|
| 178 | +Write-JsonLine @{ type = 'ready'; timestampMs = [DateTimeOffset]::UtcNow.ToUnixTimeMilliseconds() } |
| 179 | +
|
| 180 | +$lastCursorId = $null |
| 181 | +while ($true) { |
| 182 | + $cursorInfo = New-Object OpenScreenCursorInterop+CURSORINFO |
| 183 | + $cursorInfo.cbSize = [Runtime.InteropServices.Marshal]::SizeOf([type][OpenScreenCursorInterop+CURSORINFO]) |
| 184 | +
|
| 185 | + if (-not [OpenScreenCursorInterop]::GetCursorInfo([ref]$cursorInfo)) { |
| 186 | + Write-JsonLine @{ type = 'error'; timestampMs = [DateTimeOffset]::UtcNow.ToUnixTimeMilliseconds(); message = 'GetCursorInfo failed' } |
| 187 | + Start-Sleep -Milliseconds ${sampleIntervalMs} |
| 188 | + continue |
| 189 | + } |
| 190 | +
|
| 191 | + $visible = ($cursorInfo.flags -band 1) -ne 0 |
| 192 | + $cursorId = if ($cursorInfo.hCursor -eq [IntPtr]::Zero) { $null } else { ('0x{0:X}' -f $cursorInfo.hCursor.ToInt64()) } |
| 193 | + $asset = $null |
| 194 | +
|
| 195 | + if ($visible -and $cursorId -and $cursorId -ne $lastCursorId) { |
| 196 | + $asset = Get-CursorAsset -cursorHandle $cursorInfo.hCursor -cursorId $cursorId |
| 197 | + $lastCursorId = $cursorId |
| 198 | + } |
| 199 | +
|
| 200 | + Write-JsonLine @{ |
| 201 | + type = 'sample' |
| 202 | + timestampMs = [DateTimeOffset]::UtcNow.ToUnixTimeMilliseconds() |
| 203 | + x = $cursorInfo.ptScreenPos.X |
| 204 | + y = $cursorInfo.ptScreenPos.Y |
| 205 | + visible = $visible |
| 206 | + handle = $cursorId |
| 207 | + bounds = Get-TargetBounds |
| 208 | + asset = $asset |
| 209 | + } |
| 210 | +
|
| 211 | + Start-Sleep -Milliseconds ${sampleIntervalMs} |
| 212 | +} |
| 213 | +`; |
| 214 | + |
| 215 | + return Buffer.from(script, "utf16le").toString("base64"); |
| 216 | +} |
0 commit comments