OmniRoute/bin/cli/tray/tray.ps1

# OmniRoute tray icon for Windows using NotifyIcon (zero binary, AV-safe)
# IPC: stdin JSON commands, stdout JSON events
param([string]$IconPath, [string]$Tooltip)

Add-Type -AssemblyName System.Windows.Forms
Add-Type -AssemblyName System.Drawing

$ErrorActionPreference = "Stop"
[Console]::OutputEncoding = [System.Text.Encoding]::UTF8
[Console]::InputEncoding = [System.Text.Encoding]::UTF8
$OutputEncoding = [System.Text.Encoding]::UTF8

$script:notifyIcon = New-Object System.Windows.Forms.NotifyIcon
$script:notifyIcon.Icon = New-Object System.Drawing.Icon($IconPath)
$script:notifyIcon.Text = $Tooltip
$script:notifyIcon.Visible = $true

$script:menu = New-Object System.Windows.Forms.ContextMenuStrip
$script:notifyIcon.ContextMenuStrip = $script:menu
$script:items = @()

function Write-Event($obj) {
  $json = $obj | ConvertTo-Json -Compress
  [Console]::Out.WriteLine($json)
  [Console]::Out.Flush()
}

function Add-MenuItem($index, $title, $enabled) {
  $item = New-Object System.Windows.Forms.ToolStripMenuItem
  $item.Text = $title
  $item.Enabled = $enabled
  $idx = $index
  $item.Add_Click({ Write-Event @{ type = "click"; index = $idx } }.GetNewClosure())
  $script:menu.Items.Add($item) | Out-Null
  $script:items += $item
}

function Update-MenuItem($index, $title, $enabled) {
  if ($index -lt $script:items.Count) {
    $script:items[$index].Text = $title
    $script:items[$index].Enabled = $enabled
  }
}

function Set-Tooltip($text) {
  if ($text.Length -gt 63) { $text = $text.Substring(0, 63) }
  $script:notifyIcon.Text = $text
}

# Read commands from stdin on a timer to keep the Windows message loop alive
$reader = [Console]::In
$script:running = $true
$timer = New-Object System.Windows.Forms.Timer
$timer.Interval = 100
$timer.Add_Tick({
  while ($reader.Peek() -ge 0) {
    $line = $reader.ReadLine()
    if ($null -eq $line) {
      $script:notifyIcon.Visible = $false
      [System.Windows.Forms.Application]::Exit()
      return
    }
    try {
      $cmd = $line | ConvertFrom-Json
      switch ($cmd.type) {
        "setMenu" {
          $script:menu.Items.Clear()
          $script:items = @()
          for ($i = 0; $i -lt $cmd.items.Count; $i++) {
            $it = $cmd.items[$i]
            Add-MenuItem $i $it.title $it.enabled
          }
        }
        "updateItem" { Update-MenuItem $cmd.index $cmd.title $cmd.enabled }
        "setTooltip" { Set-Tooltip $cmd.text }
        "quit" {
          $script:notifyIcon.Visible = $false
          [System.Windows.Forms.Application]::Exit()
        }
      }
    } catch {}
  }
})
$timer.Start()

Write-Event @{ type = "ready" }
[System.Windows.Forms.Application]::Run()