param([string]$Mode = "element")
$ErrorActionPreference = "Continue";
[Console]::OutputEncoding = [System.Text.Encoding]::UTF8;
$logPath = "D:/RPA/test-output/desktop-picker.log";
function Write-PickLog([string]$msg) {
  try { $logDirectory = Split-Path -Parent $logPath; if (-not (Test-Path -LiteralPath $logDirectory)) { New-Item -ItemType Directory -Path $logDirectory -Force | Out-Null }; Add-Content -Path $logPath -Value ((Get-Date -Format "HH:mm:ss.fff") + " | " + $msg) -Encoding UTF8 } catch {}
}
Write-PickLog ("=== start Mode=$Mode ===")
Add-Type -AssemblyName UIAutomationClient, UIAutomationTypes, WindowsBase
Add-Type -MemberDefinition '[DllImport("user32.dll")] public static extern bool SetProcessDpiAwarenessContext(IntPtr value);' -Name PickDpi -Namespace RpaPick | Out-Null
try { [RpaPick.PickDpi]::SetProcessDpiAwarenessContext([IntPtr](-4)) | Out-Null } catch {}
Write-PickLog "dpi aware set"
Add-Type -AssemblyName System.Windows.Forms
Write-PickLog "assemblies ok"
Add-Type -MemberDefinition '[DllImport("user32.dll")] public static extern short GetAsyncKeyState(int key); [DllImport("user32.dll", CharSet = CharSet.Unicode)] public static extern int GetWindowTextW(IntPtr h, System.Text.StringBuilder sb, int max); [DllImport("user32.dll")] public static extern int GetWindowLong(IntPtr hWnd, int nIndex); [DllImport("user32.dll")] public static extern int SetWindowLong(IntPtr hWnd, int nIndex, int dwNewLong);' -Name PickW32 -Namespace RpaPick | Out-Null
Write-PickLog "w32 ok"
if ($Mode -eq "windows") {
  $wins = @()
  foreach ($p in Get-Process) { if ($p.MainWindowHandle -ne [IntPtr]::Zero -and $p.MainWindowTitle) { $wins += @{ title = $p.MainWindowTitle; process = $p.ProcessName } } }
  Write-PickLog "windows: $($wins.Count)"
  Write-Output (ConvertTo-Json @($wins) -Compress)
  exit 0
}
$info = New-Object System.Windows.Forms.Form
$info.FormBorderStyle = [System.Windows.Forms.FormBorderStyle]::None
$info.TopMost = $true
$info.ShowInTaskbar = $false
$info.StartPosition = [System.Windows.Forms.FormStartPosition]::Manual
$sw2 = [System.Windows.Forms.Screen]::PrimaryScreen.Bounds.Width
$info.Location = New-Object System.Drawing.Point(([int]($sw2 / 2) - 280), 10)
$info.Size = New-Object System.Drawing.Size(560, 64)
$info.BackColor = [System.Drawing.Color]::FromArgb(15, 40, 90)
$info.Opacity = 0.92
$infoLabel = New-Object System.Windows.Forms.Label
$infoLabel.Dock = [System.Windows.Forms.DockStyle]::Fill
$infoLabel.ForeColor = [System.Drawing.Color]::White
$infoLabel.Font = New-Object System.Drawing.Font("Microsoft YaHei", 11)
$infoLabel.TextAlign = [System.Drawing.ContentAlignment]::MiddleCenter
$infoLabel.Text = "移动鼠标到目标元素上"
$info.Controls.Add($infoLabel)
$hl = New-Object System.Windows.Forms.Form
$hl.FormBorderStyle = [System.Windows.Forms.FormBorderStyle]::None
$hl.TopMost = $true
$hl.ShowInTaskbar = $false
$hl.StartPosition = [System.Windows.Forms.FormStartPosition]::Manual
$hl.BackColor = [System.Drawing.Color]::Fuchsia
$hl.TransparencyKey = [System.Drawing.Color]::Fuchsia
$hl.Add_Paint({
  param($sender, $e)
  $pen = New-Object System.Drawing.Pen([System.Drawing.Color]::FromArgb(255, 255, 32, 32), 3)
  try {
    $size = $sender.ClientSize
    $e.Graphics.DrawRectangle($pen, 1, 1, ($size.Width - 3), ($size.Height - 3))
  } finally {
    $pen.Dispose()
  }
})
function Set-ClickThrough($form) {
  if ($form.IsHandleCreated) {
    $GWL_EXSTYLE = -20
    $WS_EX_LAYERED = 0x80000
    $WS_EX_TRANSPARENT = 0x20
    $WS_EX_NOACTIVATE = 0x08000000
    $WS_EX_TOOLWINDOW = 0x80
    $style = [RpaPick.PickW32]::GetWindowLong($form.Handle, $GWL_EXSTYLE)
    [void][RpaPick.PickW32]::SetWindowLong($form.Handle, $GWL_EXSTYLE, $style -bor $WS_EX_LAYERED -bor $WS_EX_TRANSPARENT -bor $WS_EX_NOACTIVATE -bor $WS_EX_TOOLWINDOW)
  }
}

function Set-HighlightBox([int]$x, [int]$y, [int]$w, [int]$h) {
  $hl.Location = New-Object System.Drawing.Point(($x - 2), ($y - 2))
  $hl.Size = New-Object System.Drawing.Size(($w + 4), ($h + 4))
  $hl.Show()
  Set-ClickThrough $hl
}
function Hide-Highlight { $hl.Hide() }
function Get-DeepestAtPoint([double]$px, [double]$py) {
  $point = New-Object System.Windows.Point($px, $py)
  $cur = [System.Windows.Automation.AutomationElement]::FromPoint($point)
  if (-not $cur) { return $null }
  for ($depth = 0; $depth -lt 25; $depth++) {
    $children = $cur.FindAll([System.Windows.Automation.TreeScope]::Children, [System.Windows.Automation.Condition]::TrueCondition)
    $next = $null
    $nextArea = [double]::MaxValue
    foreach ($c in $children) {
      try {
        $r = $c.Current.BoundingRectangle
        if ($r.Width -gt 1 -and $r.Height -gt 1 -and $px -ge $r.X -and $px -le ($r.X + $r.Width) -and $py -ge $r.Y -and $py -le ($r.Y + $r.Height)) {
          $area = $r.Width * $r.Height
          if ($area -lt $nextArea) { $next = $c; $nextArea = $area }
        }
      } catch {}
    }
    if (-not $next) { break }
    $cur = $next
  }
  return $cur
}
$info.Show()
Set-ClickThrough $info
$stage = "hover"
$chain = New-Object System.Collections.ArrayList
$selIdx = -1
$prevSpace = $false
$prevPgUp = $false
$prevPgDn = $false
Write-PickLog "loop start"
while ($true) {
  try {
    $space = (([RpaPick.PickW32]::GetAsyncKeyState(32) -band 0x8001) -ne 0)
    $esc = (([RpaPick.PickW32]::GetAsyncKeyState(27) -band 0x8001) -ne 0)
    $pgUp = (([RpaPick.PickW32]::GetAsyncKeyState(33) -band 0x8001) -ne 0)
    $pgDn = (([RpaPick.PickW32]::GetAsyncKeyState(34) -band 0x8001) -ne 0)
    if ($esc) { Write-PickLog "esc"; break }
    if ($stage -eq "hover") {
      $ptX = [System.Windows.Forms.Cursor]::Position.X
      $ptY = [System.Windows.Forms.Cursor]::Position.Y
      $el = Get-DeepestAtPoint $ptX $ptY
      if ($el) {
        $c = $el.Current
        $ctrlName = $c.ControlType.ProgrammaticName.Replace("ControlType.", "")
        $r = $c.BoundingRectangle; Set-HighlightBox ([int]$r.X) ([int]$r.Y) ([int]$r.Width) ([int]$r.Height)
        $infoLabel.Text = "名称: " + $c.Name + "   类型: " + $ctrlName + "   [空格]锁定   [Esc]取消"
      } else { Hide-Highlight }
      if ($space -and -not $prevSpace -and $el) {
        Write-PickLog "space lock"
        $stage = "locked"
        $chain = New-Object System.Collections.ArrayList
        $walker = [System.Windows.Automation.TreeWalker]::ControlViewWalker
        $cur = $el
        while ($cur -and -not $cur.Equals([System.Windows.Automation.AutomationElement]::RootElement)) {
          $c = $cur.Current
          $r = $c.BoundingRectangle
          [void]$chain.Insert(0, @{ el = $cur; name = [string]$c.Name; automationId = [string]$c.AutomationId; controlType = $c.ControlType.ProgrammaticName.Replace("ControlType.", ""); x = [int]$r.X; y = [int]$r.Y; w = [int]$r.Width; h = [int]$r.Height; topLevel = $walker.GetParent($cur).Equals([System.Windows.Automation.AutomationElement]::RootElement) })
          $cur = $walker.GetParent($cur)
        }
        $selIdx = $chain.Count - 1
        Write-PickLog ("locked chain=" + $chain.Count)
        $infoLabel.Text = "已锁定 [" + ($selIdx + 1) + "/" + $chain.Count + "]  [PageUp/PageDown]调层级  [空格]确认回填  [Esc]取消"
      }
    }
    if ($stage -eq "locked") {
      $item = $chain[$selIdx]
      if ($item) {
        Set-HighlightBox $item.x $item.y $item.w $item.h
        $infoLabel.Text = "层级 " + ($selIdx + 1) + "/" + $chain.Count + "   名称: " + $item.name + "   类型: " + $item.controlType + "   [PageUp/PageDown]调层级  [空格]确认  [Esc]取消"
      }
      if ($pgUp -and -not $prevPgUp -and $selIdx -gt 0) { $selIdx -= 1 }
      if ($pgDn -and -not $prevPgDn -and $selIdx -lt ($chain.Count - 1)) { $selIdx += 1 }
      if ($space -and -not $prevSpace) {
        $item = $chain[$selIdx]
        $rootItem = $chain | Where-Object { $_.topLevel -and $_.name } | Select-Object -First 1
        if ($rootItem) { $rootTitle = $rootItem.name } else { foreach ($it in $chain) { if ($it.name) { $rootTitle = $it.name; break } } }
        $result = @{ selected = $selIdx; window = $rootTitle; name = $item.name; automationId = $item.automationId; controlType = $item.controlType; x = $item.x; y = $item.y; w = $item.w; h = $item.h }
        break
      }
    }
  } catch {
    Write-PickLog ("LOOP ERR: " + $_.Exception.Message)
  }
  [System.Windows.Forms.Application]::DoEvents()
  $prevSpace = $space
  $prevPgUp = $pgUp
  $prevPgDn = $pgDn
  Start-Sleep -Milliseconds 60
}
$info.Close()
$hl.Close()
if ($result) { Write-Output (ConvertTo-Json $result -Compress) } else { Write-Output (ConvertTo-Json @{ cancelled = $true } -Compress) }