@xevy/heny-connect 0.1.0 → 0.2.0

package/README.md

@@ -23,6 +23,34 @@ npx @xevy/heny-connect run
 
 The process sends a heartbeat every 30 seconds. Keep it running for the device to remain available in Heny.
 
+## Windows system tray
+
+Install the tray companion after pairing:
+
+```powershell
+heny-connect install
+```
+
+Heny Connect starts at Windows sign-in and runs without an open terminal. The tray menu shows the connection state and provides these controls:
+
+- Open Heny
+- Pause or resume the connection
+- Reconnect now
+- Turn start-at-sign-in on or off
+- Quit until the next sign-in
+
+To start the tray for the current session without installing start-at-sign-in:
+
+```powershell
+heny-connect tray
+```
+
+To remove start-at-sign-in:
+
+```powershell
+heny-connect uninstall
+```
+
 ## Check status
 
 ```bash

package/bin/heny-connect.mjs

@@ -9,11 +9,15 @@
  *
  * Pairing state is stored in ~/.heny-connect.json (mode 600).
  */
+import { spawn, spawnSync } from "node:child_process";
 import { chmod, readFile, writeFile } from "node:fs/promises";
 import { homedir, hostname, platform, release } from "node:os";
-import { join } from "node:path";
+import { dirname, join } from "node:path";
+import { fileURLToPath } from "node:url";
 
 const STATE_FILE = join(process.env.HENY_CONNECT_HOME || homedir(), ".heny-connect.json");
+const CLI_FILE = fileURLToPath(import.meta.url);
+const TRAY_SCRIPT = join(dirname(CLI_FILE), "..", "windows", "heny-connect-tray.ps1");
 const args = process.argv.slice(2);
 const command = args[0] || "help";
 const opt = (name, fallback) => { const i = args.indexOf(`--${name}`); return i >= 0 ? args[i + 1] : fallback; };
@@ -46,7 +50,15 @@ async function pair() {
 }
 
 async function heartbeat(state, detail) {
-  return call(state.server, "/api/devices/heartbeat", { state: "available", detail }, state.token);
+  return call(state.server, "/api/devices/heartbeat", { state: detail.state, detail: detail.message }, state.token);
+}
+
+async function sendHeartbeat(stateName = "available", message = `${hostname()}: browser ready`) {
+  const state = await loadState();
+  if (!state) { console.error("Not paired. Run: heny-connect pair --code … --server …"); process.exitCode = 2; return false; }
+  if (!["online", "available", "offline"].includes(stateName)) { console.error("State must be online, available, or offline."); process.exitCode = 2; return false; }
+  await heartbeat(state, { state: stateName, message });
+  return true;
 }
 
 async function run() {
@@ -55,7 +67,7 @@ async function run() {
   const every = Number(opt("every", 30)) * 1000;
   const beats = Number(opt("beats", 0)); let count = 0;
   const tick = async () => {
-    try { await heartbeat(state, `${hostname()}: browser ready`); count += 1; console.log(`${new Date().toISOString()} heartbeat ok (${count})`); }
+    try { await heartbeat(state, { state: "available", message: `${hostname()}: browser ready` }); count += 1; console.log(`${new Date().toISOString()} heartbeat ok (${count})`); }
     catch (err) { console.error(`${new Date().toISOString()} heartbeat failed: ${err.message}`); if (/Unknown device token/.test(err.message)) process.exit(3); }
     if (beats && count >= beats) process.exit(0);
   };
@@ -63,6 +75,12 @@ async function run() {
   if (!beats || count < beats) setInterval(tick, every);
 }
 
+async function heartbeatOnce() {
+  const stateName = opt("state", "available");
+  const message = opt("detail", stateName === "offline" ? `${hostname()}: paused locally` : `${hostname()}: browser ready`);
+  if (await sendHeartbeat(stateName, message)) console.log(`Heartbeat reported ${stateName}.`);
+}
+
 async function status() {
   const state = await loadState();
   if (!state) { console.log("Not paired."); return; }
@@ -70,5 +88,47 @@ async function status() {
   console.log(res.ok ? JSON.stringify(await res.json(), null, 2) : `Server answered HTTP ${res.status}`);
 }
 
-const commands = { pair, run, status, help: async () => console.log("Commands: pair --code <6 digits> --server <url> [--run] | run [--every 30] [--beats N] | status") };
-(commands[command] || commands.help)().catch((err) => { console.error(err.message); process.exit(1); });
+function requireWindows() {
+  if (platform() === "win32") return true;
+  console.error("The system tray is available on Windows.");
+  process.exitCode = 2;
+  return false;
+}
+
+function trayArgs(mode) {
+  const values = ["-NoLogo", "-NoProfile", "-STA", "-ExecutionPolicy", "Bypass", "-WindowStyle", "Hidden", "-File", TRAY_SCRIPT, "-NodePath", process.execPath, "-CliPath", CLI_FILE];
+  if (mode === "install") values.push("-Install");
+  if (mode === "uninstall") values.push("-Uninstall");
+  return values;
+}
+
+async function tray() {
+  if (!requireWindows()) return;
+  if (!await loadState()) { console.error("Not paired. Pair this computer before starting the tray."); process.exitCode = 2; return; }
+  const child = spawn("powershell.exe", trayArgs("tray"), { detached: true, stdio: "ignore", windowsHide: true });
+  child.unref();
+  console.log("Heny Connect is running in the system tray.");
+}
+
+async function install() {
+  if (!requireWindows()) return;
+  if (!await loadState()) { console.error("Not paired. Pair this computer before installing the tray."); process.exitCode = 2; return; }
+  const result = spawnSync("powershell.exe", trayArgs("install"), { encoding: "utf8", windowsHide: true });
+  if (result.stdout) process.stdout.write(result.stdout);
+  if (result.status !== 0) { if (result.stderr) process.stderr.write(result.stderr); process.exitCode = result.status || 1; }
+}
+
+async function uninstall() {
+  if (!requireWindows()) return;
+  const result = spawnSync("powershell.exe", trayArgs("uninstall"), { encoding: "utf8", windowsHide: true });
+  if (result.stdout) process.stdout.write(result.stdout);
+  if (result.status !== 0) { if (result.stderr) process.stderr.write(result.stderr); process.exitCode = result.status || 1; }
+}
+
+const commands = { pair, run, status, heartbeat: heartbeatOnce, tray, install, uninstall, help: async () => console.log("Commands: pair --code <6 digits> --server <url> [--run] | run [--every 30] [--beats N] | status | tray | install | uninstall") };
+try {
+  await (commands[command] || commands.help)();
+} catch (err) {
+  console.error(err.message);
+  process.exitCode = 1;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xevy/heny-connect",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Pair a computer with Heny and keep its device presence online.",
   "type": "module",
   "bin": {
@@ -8,6 +8,7 @@
   },
   "files": [
     "bin",
+    "windows",
     "README.md"
   ],
   "scripts": {

package/windows/heny-connect-tray.ps1

@@ -0,0 +1,306 @@
+param(
+  [Parameter(Mandatory = $true)]
+  [string] $NodePath,
+
+  [Parameter(Mandatory = $true)]
+  [string] $CliPath,
+
+  [string] $IconPath,
+  [switch] $Install,
+  [switch] $Uninstall
+)
+
+$ErrorActionPreference = "Stop"
+$taskName = "HenyConnect"
+$powershellPath = Join-Path $env:SystemRoot "System32\WindowsPowerShell\v1.0\powershell.exe"
+if (-not $IconPath) { $IconPath = Join-Path $PSScriptRoot "heny.ico" }
+
+function Get-TaskArguments {
+  return "-NoLogo -NoProfile -STA -ExecutionPolicy Bypass -WindowStyle Hidden -File `"$PSCommandPath`" -NodePath `"$NodePath`" -CliPath `"$CliPath`" -IconPath `"$IconPath`""
+}
+
+function Install-HenyTask {
+  param([bool] $StartNow)
+
+  $existing = Get-ScheduledTask -TaskName $taskName -ErrorAction SilentlyContinue
+  if ($existing) {
+    Stop-ScheduledTask -TaskName $taskName -ErrorAction SilentlyContinue
+    Unregister-ScheduledTask -TaskName $taskName -Confirm:$false
+  }
+
+  $currentUser = [Security.Principal.WindowsIdentity]::GetCurrent().Name
+  $action = New-ScheduledTaskAction -Execute $powershellPath -Argument (Get-TaskArguments)
+  $trigger = New-ScheduledTaskTrigger -AtLogOn -User $currentUser
+  $principal = New-ScheduledTaskPrincipal -UserId $currentUser -LogonType Interactive -RunLevel Limited
+  $settings = New-ScheduledTaskSettingsSet -AllowStartIfOnBatteries -DontStopIfGoingOnBatteries -ExecutionTimeLimit ([TimeSpan]::Zero) -MultipleInstances IgnoreNew -StartWhenAvailable
+  Register-ScheduledTask -TaskName $taskName -Action $action -Trigger $trigger -Principal $principal -Settings $settings | Out-Null
+  if ($StartNow) { Start-ScheduledTask -TaskName $taskName }
+}
+
+if ($Uninstall) {
+  $existing = Get-ScheduledTask -TaskName $taskName -ErrorAction SilentlyContinue
+  if ($existing) {
+    Stop-ScheduledTask -TaskName $taskName -ErrorAction SilentlyContinue
+    Unregister-ScheduledTask -TaskName $taskName -Confirm:$false
+  }
+  Write-Output "Heny Connect start-at-sign-in removed. Quit the tray to end the current session."
+  exit 0
+}
+
+if ($Install) {
+  Install-HenyTask -StartNow $true
+  Write-Output "Heny Connect installed. It is running in the system tray and will start at sign-in."
+  exit 0
+}
+
+$createdNew = $false
+$mutex = [Threading.Mutex]::new($true, "Local\HenyConnectTray", [ref] $createdNew)
+if (-not $createdNew) {
+  $mutex.Dispose()
+  exit 0
+}
+
+Add-Type -AssemblyName System.Windows.Forms
+Add-Type -AssemblyName System.Drawing
+[Windows.Forms.Application]::EnableVisualStyles()
+
+$iconAvailable = [Drawing.Icon]::new($IconPath)
+$iconPaused = [Drawing.Icon]::new((Join-Path $PSScriptRoot "heny-paused.ico"))
+$iconError = [Drawing.Icon]::new((Join-Path $PSScriptRoot "heny-error.ico"))
+$notifyIcon = [Windows.Forms.NotifyIcon]::new()
+$notifyIcon.Icon = $iconPaused
+$notifyIcon.Text = "Heny Connect - Connecting"
+$notifyIcon.Visible = $true
+
+$menu = [Windows.Forms.ContextMenuStrip]::new()
+$titleItem = [Windows.Forms.ToolStripMenuItem]::new("Heny Connect")
+$titleItem.Enabled = $false
+$titleItem.Font = [Drawing.Font]::new($titleItem.Font, [Drawing.FontStyle]::Bold)
+$statusItem = [Windows.Forms.ToolStripMenuItem]::new("Status: Connecting...")
+$statusItem.Enabled = $false
+$openItem = [Windows.Forms.ToolStripMenuItem]::new("Open Heny")
+$pauseItem = [Windows.Forms.ToolStripMenuItem]::new("Pause")
+$reconnectItem = [Windows.Forms.ToolStripMenuItem]::new("Reconnect now")
+$startItem = [Windows.Forms.ToolStripMenuItem]::new("Start at sign-in")
+$startItem.CheckOnClick = $false
+$startItem.Checked = [bool] (Get-ScheduledTask -TaskName $taskName -ErrorAction SilentlyContinue)
+$quitItem = [Windows.Forms.ToolStripMenuItem]::new("Quit")
+
+$null = $menu.Items.Add($titleItem)
+$null = $menu.Items.Add($statusItem)
+$null = $menu.Items.Add([Windows.Forms.ToolStripSeparator]::new())
+$null = $menu.Items.Add($openItem)
+$null = $menu.Items.Add($pauseItem)
+$null = $menu.Items.Add($reconnectItem)
+$null = $menu.Items.Add($startItem)
+$null = $menu.Items.Add([Windows.Forms.ToolStripSeparator]::new())
+$null = $menu.Items.Add($quitItem)
+$notifyIcon.ContextMenuStrip = $menu
+
+$script:paused = $false
+$script:heartbeatProcess = $null
+$script:heartbeatState = $null
+$script:nextHeartbeat = Get-Date
+$script:lastStatus = "connecting"
+$script:exiting = $false
+
+function Set-HenyStatus {
+  param(
+    [ValidateSet("available", "paused", "connecting", "error")]
+    [string] $State,
+    [string] $Detail
+  )
+
+  $label = switch ($State) {
+    "available" { "Available" }
+    "paused" { "Paused" }
+    "connecting" { "Connecting..." }
+    "error" { "Connection error" }
+  }
+  $statusItem.Text = "Status: $label"
+  if ($Detail) { $statusItem.ToolTipText = $Detail }
+  $notifyIcon.Text = "Heny Connect - $label"
+  $notifyIcon.Icon = switch ($State) {
+    "available" { $iconAvailable }
+    "error" { $iconError }
+    default { $iconPaused }
+  }
+  $script:lastStatus = $State
+}
+
+function Get-HenyUrl {
+  try {
+    $homePath = if ($env:HENY_CONNECT_HOME) { $env:HENY_CONNECT_HOME } else { [Environment]::GetFolderPath("UserProfile") }
+    $statePath = Join-Path $homePath ".heny-connect.json"
+    $state = Get-Content -LiteralPath $statePath -Raw | ConvertFrom-Json
+    return ([string] $state.server).TrimEnd("/") + "/desktop"
+  } catch {
+    return "https://heny.vyte.dev/desktop"
+  }
+}
+
+function Stop-HeartbeatProcess {
+  if (-not $script:heartbeatProcess) { return }
+  try {
+    if (-not $script:heartbeatProcess.HasExited) {
+      $script:heartbeatProcess.Kill()
+      $script:heartbeatProcess.WaitForExit(2000) | Out-Null
+    }
+    $script:heartbeatProcess.Dispose()
+  } catch {}
+  $script:heartbeatProcess = $null
+  $script:heartbeatState = $null
+}
+
+function Start-Heartbeat {
+  param(
+    [ValidateSet("available", "offline")]
+    [string] $State = "available"
+  )
+
+  if ($script:heartbeatProcess) { return }
+  $detail = if ($State -eq "offline") { "$env:COMPUTERNAME`: paused in tray" } else { "$env:COMPUTERNAME`: browser ready" }
+  $startInfo = [Diagnostics.ProcessStartInfo]::new()
+  $startInfo.FileName = $NodePath
+  $startInfo.Arguments = "`"$CliPath`" heartbeat --state $State --detail `"$detail`""
+  $startInfo.UseShellExecute = $false
+  $startInfo.CreateNoWindow = $true
+  $startInfo.WindowStyle = [Diagnostics.ProcessWindowStyle]::Hidden
+  $startInfo.RedirectStandardOutput = $true
+  $startInfo.RedirectStandardError = $true
+  $process = [Diagnostics.Process]::new()
+  $process.StartInfo = $startInfo
+  try {
+    if (-not $process.Start()) { throw "The heartbeat process did not start." }
+    $script:heartbeatProcess = $process
+    $script:heartbeatState = $State
+    if ($State -eq "available") { Set-HenyStatus -State "connecting" -Detail "Contacting Heny" }
+  } catch {
+    $process.Dispose()
+    Set-HenyStatus -State "error" -Detail $_.Exception.Message
+    $script:nextHeartbeat = (Get-Date).AddSeconds(30)
+  }
+}
+
+function Complete-Heartbeat {
+  if (-not $script:heartbeatProcess -or -not $script:heartbeatProcess.HasExited) { return }
+  $process = $script:heartbeatProcess
+  $state = $script:heartbeatState
+  $stdout = $process.StandardOutput.ReadToEnd().Trim()
+  $stderr = $process.StandardError.ReadToEnd().Trim()
+  $exitCode = $process.ExitCode
+  $process.Dispose()
+  $script:heartbeatProcess = $null
+  $script:heartbeatState = $null
+  $script:nextHeartbeat = (Get-Date).AddSeconds(30)
+
+  if ($exitCode -eq 0) {
+    if ($state -eq "offline" -or $script:paused) {
+      Set-HenyStatus -State "paused" -Detail "Paused on this computer"
+    } else {
+      Set-HenyStatus -State "available" -Detail "Connected to Heny"
+    }
+    return
+  }
+
+  $detail = if ($stderr) { $stderr } elseif ($stdout) { $stdout } else { "Heartbeat exited with code $exitCode" }
+  $previousStatus = $script:lastStatus
+  Set-HenyStatus -State "error" -Detail $detail
+  if ($previousStatus -ne "error") {
+    $notifyIcon.ShowBalloonTip(4000, "Heny Connect", $detail, [Windows.Forms.ToolTipIcon]::Warning)
+  }
+}
+
+function Send-OfflineAndWait {
+  Stop-HeartbeatProcess
+  $startInfo = [Diagnostics.ProcessStartInfo]::new()
+  $startInfo.FileName = $NodePath
+  $startInfo.Arguments = "`"$CliPath`" heartbeat --state offline --detail `"$env:COMPUTERNAME`: tray closed`""
+  $startInfo.UseShellExecute = $false
+  $startInfo.CreateNoWindow = $true
+  $startInfo.WindowStyle = [Diagnostics.ProcessWindowStyle]::Hidden
+  $process = [Diagnostics.Process]::Start($startInfo)
+  if ($process) {
+    $process.WaitForExit(10000) | Out-Null
+    if (-not $process.HasExited) { $process.Kill() }
+    $process.Dispose()
+  }
+}
+
+$openHeny = {
+  Start-Process (Get-HenyUrl)
+}
+$openItem.Add_Click($openHeny)
+$notifyIcon.Add_DoubleClick($openHeny)
+
+$pauseItem.Add_Click({
+  if ($script:paused) {
+    $script:paused = $false
+    $pauseItem.Text = "Pause"
+    $script:nextHeartbeat = Get-Date
+    Set-HenyStatus -State "connecting" -Detail "Resuming"
+  } else {
+    $script:paused = $true
+    $pauseItem.Text = "Resume"
+    Stop-HeartbeatProcess
+    Start-Heartbeat -State "offline"
+  }
+})
+
+$reconnectItem.Add_Click({
+  $script:paused = $false
+  $pauseItem.Text = "Pause"
+  Stop-HeartbeatProcess
+  $script:nextHeartbeat = Get-Date
+  Set-HenyStatus -State "connecting" -Detail "Reconnecting"
+})
+
+$startItem.Add_Click({
+  try {
+    if ($startItem.Checked) {
+      Unregister-ScheduledTask -TaskName $taskName -Confirm:$false -ErrorAction SilentlyContinue
+      $startItem.Checked = $false
+      $notifyIcon.ShowBalloonTip(2500, "Heny Connect", "Start at sign-in is off.", [Windows.Forms.ToolTipIcon]::Info)
+    } else {
+      Install-HenyTask -StartNow $false
+      $startItem.Checked = $true
+      $notifyIcon.ShowBalloonTip(2500, "Heny Connect", "Start at sign-in is on.", [Windows.Forms.ToolTipIcon]::Info)
+    }
+  } catch {
+    Set-HenyStatus -State "error" -Detail $_.Exception.Message
+  }
+})
+
+$quitItem.Add_Click({
+  $script:exiting = $true
+  $timer.Stop()
+  Send-OfflineAndWait
+  $notifyIcon.Visible = $false
+  [Windows.Forms.Application]::ExitThread()
+})
+
+$timer = [Windows.Forms.Timer]::new()
+$timer.Interval = 1000
+$timer.Add_Tick({
+  Complete-Heartbeat
+  if (-not $script:paused -and -not $script:heartbeatProcess -and (Get-Date) -ge $script:nextHeartbeat) {
+    Start-Heartbeat -State "available"
+  }
+})
+$timer.Start()
+Start-Heartbeat -State "available"
+
+try {
+  [Windows.Forms.Application]::Run()
+} finally {
+  $timer.Stop()
+  Stop-HeartbeatProcess
+  $notifyIcon.Visible = $false
+  $notifyIcon.Dispose()
+  $menu.Dispose()
+  $iconAvailable.Dispose()
+  $iconPaused.Dispose()
+  $iconError.Dispose()
+  if ($createdNew) { $mutex.ReleaseMutex() }
+  $mutex.Dispose()
+}