param(
    [string]$ProgramRoot = "",
    [string]$DataRoot = "",
    [switch]$StartOnLaunch = $true,
    [switch]$CheckUpdateOnLaunch = $false,
    [switch]$AutoUpdate = $false
)

# Simple system-tray controller for NetX (Windows packaged installs).
# Start / Stop / Open UI / Check update / Exit

$ErrorActionPreference = "Stop"
. "$PSScriptRoot\_common.ps1"

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

$prog = Get-NetxProgramRoot -Override $ProgramRoot
$data = Get-NetxDataRoot -ProgramRoot $prog -Override $DataRoot
$hostBind = "127.0.0.1"
$port = 8890
$envPath = Join-Path $data ".env"
$map = @{}
if (Test-Path $envPath) {
    $map = Read-DotEnv -Path $envPath
    if ($map["NETX_HOST"]) { $hostBind = $map["NETX_HOST"] }
    if ($map["NETX_PORT"]) { try { $port = [int]$map["NETX_PORT"] } catch {} }
}
$autoVal = ""
if ($map["NETX_UPDATE_AUTO"]) { $autoVal = $map["NETX_UPDATE_AUTO"] }
elseif ($env:NETX_UPDATE_AUTO) { $autoVal = $env:NETX_UPDATE_AUTO }
if ($autoVal -match '^(1|true|yes|on)$') {
    $AutoUpdate = $true
    $CheckUpdateOnLaunch = $true
}
$uiUrl = "http://${hostBind}:${port}/"
$ver = Get-NetxVersion -ProgramRoot $prog
$zh = ([cultureinfo]::CurrentUICulture.Name -match '^(zh|zh-)')
function T([string]$En, [string]$Zh) {
    if ($zh) { return $Zh } else { return $En }
}

if (-not (Test-Path $envPath)) {
    [void][System.Windows.Forms.MessageBox]::Show(
        (T `
            "NetX is not configured yet (missing $envPath).`nRe-run the installer and choose built-in or external DB,`nor use Start Menu → NetX → Reconfigure database." `
            "尚未完成数据库配置（缺少 $envPath）。`n请重新运行安装向导并选择内置或外置数据库，`n或使用「开始菜单 → NetX → 重新配置数据库」。"),
        "NetX",
        [System.Windows.Forms.MessageBoxButtons]::OK,
        [System.Windows.Forms.MessageBoxIcon]::Warning
    )
    exit 1
}

# Only one tray icon per machine session (desktop + postinstall + autostart can race).
$script:netxTrayMutex = $null
try {
    $createdNew = $false
    $script:netxTrayMutex = New-Object System.Threading.Mutex($true, "Local\NetX.WinTray.SingleInstance", [ref]$createdNew)
    if (-not $createdNew) {
        if (Test-NetxApiHealthy -HostName $hostBind -Port $port -TimeoutSec 2) {
            try { Start-Process $uiUrl } catch {}
        } elseif ($StartOnLaunch) {
            Start-NetxPowerShell -File (Join-Path $PSScriptRoot "start_netx_app.ps1") `
                -Arguments @("-ProgramRoot", $prog, "-DataRoot", $data, "-SkipBrowser") `
                -WorkingDirectory $prog -WindowStyle Hidden | Out-Null
        }
        exit 0
    }
} catch {
    # If mutex fails, continue with a tray (better than no UI control).
}

$startPs1 = Join-Path $PSScriptRoot "start_netx_app.ps1"
$stopPs1 = Join-Path $PSScriptRoot "stop_netx_app.ps1"
$checkPs1 = Join-Path $PSScriptRoot "check_update.ps1"

function Invoke-NetxScript {
    param(
        [string]$File,
        [string[]]$ExtraArgs = @(),
        [switch]$Wait = $false
    )
    return Start-NetxPowerShell -File $File -Arguments (@("-ProgramRoot", $prog, "-DataRoot", $data) + $ExtraArgs) `
        -WorkingDirectory $prog -WindowStyle Hidden -PassThru -Wait:$Wait
}

function Open-NetxUiWhenReady {
    param([int]$TimeoutSec = 180)
    $notify.ShowBalloonTip(
        5000,
        "NetX",
        (T "Starting… first run may take a minute." "正在启动…首次迁移可能需要一分钟。"),
        [System.Windows.Forms.ToolTipIcon]::Info
    )
    if (Wait-NetxApiHealthy -HostName $hostBind -Port $port -TimeoutSec $TimeoutSec) {
        try { Start-Process $uiUrl } catch {}
        return
    }
    $notify.ShowBalloonTip(
        8000,
        "NetX",
        (T "UI not ready yet. Use tray → Open UI when the icon appears, or check data\runtime\netx.err.log." "界面尚未就绪。就绪后请用托盘「打开界面」，或查看 data\runtime\netx.err.log。"),
        [System.Windows.Forms.ToolTipIcon]::Warning
    )
}

$form = New-Object System.Windows.Forms.Form
$form.Text = "NetX"
$form.ShowInTaskbar = $false
$form.WindowState = "Minimized"
$form.Visible = $false
$form.Size = New-Object System.Drawing.Size(0, 0)

$notify = New-Object System.Windows.Forms.NotifyIcon
$notify.Text = "NetX $ver"
$notify.Visible = $true
$iconPath = Join-Path $PSScriptRoot "assets\netx.ico"
try {
    if (Test-Path $iconPath) {
        $notify.Icon = New-Object System.Drawing.Icon $iconPath
    } else {
        $notify.Icon = [System.Drawing.SystemIcons]::Application
    }
} catch {
    try { $notify.Icon = [System.Drawing.SystemIcons]::Application } catch {}
}

$menu = New-Object System.Windows.Forms.ContextMenuStrip

$miOpen = $menu.Items.Add((T "Open UI ($uiUrl)" "打开界面 ($uiUrl)"))
$miOpen.add_Click({ Start-Process $uiUrl })

$miStart = $menu.Items.Add((T "Start NetX" "启动 NetX"))
$miStart.add_Click({
    Invoke-NetxScript -File $startPs1 -ExtraArgs @("-SkipBrowser") | Out-Null
    Open-NetxUiWhenReady
})

$miStop = $menu.Items.Add((T "Stop NetX (keep tray)" "停止 NetX（保留托盘）"))
$miStop.add_Click({
    $notify.ShowBalloonTip(3000, "NetX", (T "Stopping…" "正在停止…"), [System.Windows.Forms.ToolTipIcon]::Info)
    Invoke-NetxScript -File $stopPs1 | Out-Null
})

$miStopExit = $menu.Items.Add((T "Stop NetX and exit tray" "停止 NetX 并退出托盘"))
$miStopExit.add_Click({
    $notify.ShowBalloonTip(3000, "NetX", (T "Stopping…" "正在停止…"), [System.Windows.Forms.ToolTipIcon]::Info)
    Invoke-NetxScript -File $stopPs1 -Wait | Out-Null
    $notify.Visible = $false
    $form.Close()
    [System.Windows.Forms.Application]::Exit()
})

$miUpdate = $menu.Items.Add((T "Check for updates" "检查更新"))
$miUpdate.add_Click({
    try {
        $p = Start-NetxPowerShell -File $checkPs1 -Arguments @("-ProgramRoot", $prog, "-DataRoot", $data) `
            -WorkingDirectory $prog -WindowStyle Normal -Wait -PassThru
        if ($p.ExitCode -eq 10) {
            $ans = [System.Windows.Forms.MessageBox]::Show(
                (T "A newer NetX is available. Download and apply now?" "发现新版本，是否立即下载并更新？"),
                (T "NetX update" "NetX 更新"),
                [System.Windows.Forms.MessageBoxButtons]::YesNo,
                [System.Windows.Forms.MessageBoxIcon]::Question
            )
            if ($ans -eq [System.Windows.Forms.DialogResult]::Yes) {
                Start-NetxPowerShell -File $checkPs1 -Arguments @("-ProgramRoot", $prog, "-DataRoot", $data, "-Apply") `
                    -WorkingDirectory $prog -WindowStyle Normal | Out-Null
            }
        } elseif ($p.ExitCode -eq 0) {
            [System.Windows.Forms.MessageBox]::Show((T "NetX is up to date ($ver)." "已是最新版本 ($ver)。"), (T "NetX update" "NetX 更新"))
        }
    } catch {
        [System.Windows.Forms.MessageBox]::Show((T "Update check failed: $($_.Exception.Message)" "检查更新失败：$($_.Exception.Message)"), "NetX")
    }
})

[void]$menu.Items.Add("-")

$miExit = $menu.Items.Add((T "Exit tray (services keep running)" "退出托盘（服务继续运行）"))
$miExit.add_Click({
    $notify.Visible = $false
    $form.Close()
    [System.Windows.Forms.Application]::Exit()
})

$notify.ContextMenuStrip = $menu
$notify.add_DoubleClick({ Start-Process $uiUrl })

$form.add_Shown({
    if ($AutoUpdate) {
        $notify.ShowBalloonTip(4000, "NetX", (T "Checking for updates…" "正在检查更新…"), [System.Windows.Forms.ToolTipIcon]::Info)
        Start-NetxPowerShell -File $checkPs1 `
            -Arguments @("-ProgramRoot", $prog, "-DataRoot", $data, "-Apply", "-Quiet", "-AutoOnly") `
            -WorkingDirectory $prog -WindowStyle Hidden -Wait | Out-Null
    } elseif ($CheckUpdateOnLaunch) {
        Start-NetxPowerShell -File $checkPs1 `
            -Arguments @("-ProgramRoot", $prog, "-DataRoot", $data, "-Quiet") `
            -WorkingDirectory $prog -WindowStyle Hidden | Out-Null
    }
    if ($StartOnLaunch) {
        # Detach start so the tray message pump stays alive; poll /health before browser.
        Invoke-NetxScript -File $startPs1 -ExtraArgs @("-SkipBrowser") | Out-Null
        $script:netxUiWaitTicks = 0
        $script:netxUiWaitMax = 360  # 360 * 500ms = 180s
        $script:netxUiTimer = New-Object System.Windows.Forms.Timer
        $script:netxUiTimer.Interval = 500
        $script:netxUiTimer.add_Tick({
            $script:netxUiWaitTicks++
            if (Test-NetxApiHealthy -HostName $hostBind -Port $port -TimeoutSec 2) {
                $script:netxUiTimer.Stop()
                $script:netxUiTimer.Dispose()
                try { Start-Process $uiUrl } catch {}
                return
            }
            if ($script:netxUiWaitTicks -eq 1) {
                $notify.ShowBalloonTip(
                    5000,
                    "NetX",
                    (T "Starting… first run may take a minute." "正在启动…首次迁移可能需要一分钟。"),
                    [System.Windows.Forms.ToolTipIcon]::Info
                )
            }
            if ($script:netxUiWaitTicks -ge $script:netxUiWaitMax) {
                $script:netxUiTimer.Stop()
                $script:netxUiTimer.Dispose()
                $notify.ShowBalloonTip(
                    8000,
                    "NetX",
                    (T "UI not ready yet. Use tray → Open UI later, or check data\runtime\netx.err.log." "界面尚未就绪。请稍后用托盘「打开界面」，或查看 data\runtime\netx.err.log。"),
                    [System.Windows.Forms.ToolTipIcon]::Warning
                )
            }
        })
        $script:netxUiTimer.Start()
    }
})

$form.add_FormClosing({
    $notify.Visible = $false
    $notify.Dispose()
})

[System.Windows.Forms.Application]::Run($form)