nastechai-desktop 18.1.0

package/src/store/profile.ts

@@ -0,0 +1,365 @@
+import { atom, computed } from 'nanostores'
+
+import { getProfiles, setApiRequestProfile } from '@/nastech'
+import { queryClient } from '@/lib/query-client'
+import {
+  arraysEqual,
+  persistBoolean,
+  persistStringArray,
+  persistStringRecord,
+  storedBoolean,
+  storedStringArray,
+  storedStringRecord
+} from '@/lib/storage'
+import { $gateway, ensureGatewayForProfile } from '@/store/gateway'
+import type { ProfileInfo } from '@/types/nastech'
+
+// Canonical key for a profile: trimmed, empty → "default". Used everywhere we
+// compare a session's owning profile against the live gateway's profile.
+export function normalizeProfileKey(name: string | null | undefined): string {
+  const value = (name ?? '').trim()
+
+  return value || 'default'
+}
+
+// The profile the running local backend is actually scoped to (mirrors
+// /api/profiles/active `current`). "default" is the root ~/.NASTECH. This is the
+// display source of truth for the statusbar pill; the desktop's *stored*
+// preference (which may be unset) lives in the Electron main process.
+export const $activeProfile = atom<string>('default')
+
+// Cached profile list for the picker. Refreshed lazily; the dropdown also
+// re-fetches on open so a profile created elsewhere shows up.
+export const $profiles = atom<ProfileInfo[]>([])
+
+export function setActiveProfile(name: string): void {
+  $activeProfile.set(name || 'default')
+}
+
+// ── Rail order ─────────────────────────────────────────────────────────────
+// User-defined order for the named (non-default) profile squares in the rail.
+// Names absent from the list fall back to alphabetical, appended at the tail —
+// so a freshly created profile lands at the end until the user drags it.
+const PROFILE_ORDER_STORAGE_KEY = 'NASTECH.desktop.profileOrder'
+
+export const $profileOrder = atom<string[]>(storedStringArray(PROFILE_ORDER_STORAGE_KEY))
+
+$profileOrder.subscribe(value => persistStringArray(PROFILE_ORDER_STORAGE_KEY, [...value]))
+
+export function setProfileOrder(names: string[]): void {
+  if (!arraysEqual($profileOrder.get(), names)) {
+    $profileOrder.set(names)
+  }
+}
+
+// Sort items by the stored order; unordered names alphabetise at the tail.
+export function sortByProfileOrder<T extends { name: string }>(items: T[], order: string[]): T[] {
+  const rank = new Map(order.map((name, index) => [name, index]))
+
+  return [...items].sort((a, b) => {
+    const ra = rank.get(a.name)
+    const rb = rank.get(b.name)
+
+    if (ra != null && rb != null) {
+      return ra - rb
+    }
+
+    return ra != null ? -1 : rb != null ? 1 : a.name.localeCompare(b.name)
+  })
+}
+
+// ── Rail colors ────────────────────────────────────────────────────────────
+// Optional per-profile color override (long-press a rail square to pick). Absent
+// names fall back to the deterministic hue from profileColor(); a local-only
+// cosmetic preference, so single-profile users never touch it.
+const PROFILE_COLORS_STORAGE_KEY = 'NASTECH.desktop.profileColors'
+
+export const $profileColors = atom<Record<string, string>>(storedStringRecord(PROFILE_COLORS_STORAGE_KEY))
+
+$profileColors.subscribe(value => persistStringRecord(PROFILE_COLORS_STORAGE_KEY, value))
+
+// Set (or, with null, clear) a profile's color override.
+export function setProfileColor(name: string, color: null | string): void {
+  const key = normalizeProfileKey(name)
+  const next = { ...$profileColors.get() }
+
+  if (color) {
+    next[key] = color
+  } else {
+    delete next[key]
+  }
+
+  $profileColors.set(next)
+}
+
+interface ActiveProfileResponse {
+  active: string
+  current: string
+}
+
+// Pull the running backend's current profile + the available profile list.
+// Best-effort: failures (backend not up yet) leave the prior values intact.
+export async function refreshActiveProfile(): Promise<void> {
+  try {
+    const res = await window.NASTECHDesktop.api<ActiveProfileResponse>({ path: '/api/profiles/active' })
+
+    setActiveProfile(res.current || 'default')
+  } catch {
+    // Backend may not be ready; keep the last known value.
+  }
+
+  try {
+    const { profiles } = await getProfiles()
+    $profiles.set(profiles)
+  } catch {
+    // Leave the cached list in place.
+  }
+}
+
+// Persist the choice and relaunch the backend under the new NASTECH_HOME. The
+// main process reloads the window, so this normally never returns to the caller
+// (the renderer is torn down). We optimistically reflect the selection first so
+// the pill updates instantly if the reload is delayed.
+export async function switchProfile(name: string): Promise<void> {
+  if (!name || name === $activeProfile.get()) {
+    return
+  }
+
+  setActiveProfile(name)
+  await window.NASTECHDesktop.profile.set(name)
+}
+
+// ── Swap-minimal gateway routing ──────────────────────────────────────────
+// One live gateway at a time. When the user opens/sends a session whose profile
+// differs from the gateway's current profile, we lazily reconnect the single
+// gateway to that profile's backend (spawned on demand by the Electron pool).
+// A single-profile user never triggers a swap, so their path is unchanged.
+
+// The profile the live gateway WebSocket is currently connected to. Initialized
+// to the primary (window) backend's profile on boot.
+export const $activeGatewayProfile = atom<string>('default')
+
+// Profile for the NEXT new chat (chosen via the new-chat picker). null = primary
+// / default, so single-profile users are unaffected.
+export const $newChatProfile = atom<string | null>(null)
+
+// Bumped whenever the profile context actually changes (switch or create). The
+// chat controller subscribes and drops to a fresh new-session draft, so the
+// session you were in doesn't stay sticky across a profile switch.
+export const $freshSessionRequest = atom(0)
+
+function requestFreshSession(): void {
+  $freshSessionRequest.set($freshSessionRequest.get() + 1)
+}
+
+// Route profile-scoped REST settings (config/env/skills/tools/model/…) to the
+// profile the live gateway is currently on, and drop cached settings from the
+// previous profile so pages refetch against the right backend. Fires once
+// immediately (no real change → no invalidation), so single-profile users just
+// get "default" (→ the primary backend) with no extra fetches.
+let _lastRoutedProfile: string | null = null
+
+$activeGatewayProfile.subscribe(value => {
+  const key = normalizeProfileKey(value)
+  setApiRequestProfile(key)
+
+  if (_lastRoutedProfile !== null && _lastRoutedProfile !== key) {
+    // Profile-scoped settings + the unified session list are now stale.
+    void queryClient.invalidateQueries()
+  }
+
+  _lastRoutedProfile = key
+})
+
+// Target profile while a gateway swap is mid-flight (spawning/reconnecting that
+// profile's backend), else null. Drives the chat's "waking up <profile>" loader
+// so a lazy spawn doesn't read as a hang. Single-profile users never swap.
+export const $gatewaySwapTarget = atom<string | null>(null)
+
+let gatewaySwitch: Promise<void> | null = null
+
+// Make `profile`'s backend the active gateway, lazily opening its socket if it
+// isn't live yet. Unlike the old single-socket swap, background profiles keep
+// their sockets — so their sessions keep streaming concurrently. A null/empty
+// target means "no explicit profile" → keep the current gateway (a plain new
+// chat stays put; single-profile users never leave the primary).
+export async function ensureGatewayProfile(profile: string | null | undefined): Promise<void> {
+  if (profile == null || !String(profile).trim()) {
+    // "No explicit profile" = use the current gateway. But if an explicit swap
+    // (e.g. the user just picked a profile in the switcher) is still in flight,
+    // let it settle first so a new chat doesn't race session.create against a
+    // half-open socket and land on the wrong backend.
+    if (gatewaySwitch) {
+      await gatewaySwitch.catch(() => undefined)
+    }
+
+    return
+  }
+
+  const target = normalizeProfileKey(profile)
+
+  if (normalizeProfileKey($activeGatewayProfile.get()) === target && $gateway.get()) {
+    return
+  }
+
+  // Serialize concurrent activations so two rapid session switches don't race
+  // the active pointer.
+  if (gatewaySwitch) {
+    await gatewaySwitch.catch(() => undefined)
+
+    if (normalizeProfileKey($activeGatewayProfile.get()) === target && $gateway.get()) {
+      return
+    }
+  }
+
+  $gatewaySwapTarget.set(target)
+  gatewaySwitch = (async () => {
+    // ensureGatewayForProfile opens (or reuses) the target's socket and points
+    // the active gateway at it — without closing the profile you came from.
+    await ensureGatewayForProfile(target)
+    $activeGatewayProfile.set(target)
+  })()
+
+  try {
+    await gatewaySwitch
+  } finally {
+    gatewaySwitch = null
+    $gatewaySwapTarget.set(null)
+  }
+}
+
+// ── Sidebar profile scope (the "workspace switcher" model) ─────────────────
+// Mirrors how Slack/VS Code/Linear do multi-context: you're "in" one profile at
+// a time and the sidebar shows only that profile's sessions (clean rows, no
+// per-row tags). The lone exception is an explicit "All profiles" mode that
+// fans every profile's sessions into one grouped, browsable list.
+
+export const ALL_PROFILES = '__all__'
+
+const SHOW_ALL_PROFILES_STORAGE_KEY = 'NASTECH.desktop.showAllProfiles'
+
+// Opt-in unified view. When false, scope follows the live gateway profile, so
+// single-profile users (who never see the switcher) are completely unaffected.
+export const $showAllProfiles = atom<boolean>(storedBoolean(SHOW_ALL_PROFILES_STORAGE_KEY, false))
+
+$showAllProfiles.subscribe(value => persistBoolean(SHOW_ALL_PROFILES_STORAGE_KEY, value))
+
+// The profile context the sidebar is currently showing: a concrete profile key,
+// or ALL_PROFILES for the unified grouped view. Concrete scope is tied to the
+// gateway so opening/selecting a profile (which swaps the gateway) moves the
+// whole sidebar with it — a real context switch, not a separate filter to keep
+// in sync.
+export const $profileScope = computed([$showAllProfiles, $activeGatewayProfile], (showAll, gateway) =>
+  showAll ? ALL_PROFILES : normalizeProfileKey(gateway)
+)
+
+// Switch the active context to `name`: leave "All profiles" mode, point new
+// chats at it, and swap the single live gateway onto its backend (which moves
+// $activeGatewayProfile → name, so $profileScope follows).
+export function selectProfile(name: string): void {
+  const target = normalizeProfileKey(name)
+  // Switching profiles (or coming back from the all-profiles browse view) starts
+  // fresh; re-tapping the profile you're already in leaves your session be.
+  const switching = $showAllProfiles.get() || target !== normalizeProfileKey($activeGatewayProfile.get())
+  $showAllProfiles.set(false)
+  $newChatProfile.set(target)
+
+  if (switching) {
+    requestFreshSession()
+  }
+
+  void ensureGatewayProfile(target)
+}
+
+// Start a fresh session in `name` WITHOUT collapsing the "All profiles" browse
+// view. Unlike selectProfile, it leaves $showAllProfiles untouched, so the
+// unified sidebar stays put — used by the per-profile "+" in the all-profiles
+// session list, where switching scope would throw away the browse state the user
+// is in. Points new chats at the profile and opens its backend so the next
+// message lands in the right place.
+export function newSessionInProfile(name: string): void {
+  const target = normalizeProfileKey(name)
+  $newChatProfile.set(target)
+  requestFreshSession()
+  void ensureGatewayProfile(target)
+}
+
+export function setShowAllProfiles(value: boolean): void {
+  $showAllProfiles.set(value)
+}
+
+export function toggleShowAllProfiles(): void {
+  $showAllProfiles.set(!$showAllProfiles.get())
+}
+
+// ── Hotkey-driven profile switching ────────────────────────────────────────
+// Positional + relative navigation for the rail, used by the keybind runtime.
+// The ordered list is [default, ...named-in-rail-order]; switching is a no-op
+// when the slot is empty so unused ⌘N keys stay harmless.
+
+function orderedProfileKeys(): string[] {
+  const profiles = $profiles.get()
+
+  const named = sortByProfileOrder(
+    profiles.filter(profile => !profile.is_default),
+    $profileOrder.get()
+  ).map(profile => normalizeProfileKey(profile.name))
+
+  const hasDefault = profiles.some(profile => profile.is_default)
+
+  return hasDefault ? ['default', ...named] : named
+}
+
+// Switch to the default (root ~/.NASTECH) profile — bound to ⌘1.
+export function switchToDefaultProfile(): void {
+  const def = $profiles.get().find(profile => profile.is_default)
+
+  selectProfile(def ? def.name : 'default')
+}
+
+// Switch to the Nth named (non-default) profile in rail order (1-based).
+export function switchProfileToSlot(slot: number): void {
+  const named = sortByProfileOrder(
+    $profiles.get().filter(profile => !profile.is_default),
+    $profileOrder.get()
+  )
+
+  const target = named[slot - 1]
+
+  if (target) {
+    selectProfile(target.name)
+  }
+}
+
+// Step to the next/previous profile in the rail, wrapping around.
+export function cycleProfile(direction: 1 | -1): void {
+  const keys = orderedProfileKeys()
+
+  if (keys.length < 2) {
+    return
+  }
+
+  const current = $showAllProfiles.get() ? -1 : keys.indexOf(normalizeProfileKey($activeGatewayProfile.get()))
+  const start = current < 0 ? (direction === 1 ? -1 : 0) : current
+  const next = (start + direction + keys.length) % keys.length
+
+  selectProfile(keys[next])
+}
+
+// Bumped to ask the rail to open its "create profile" dialog (the dialog state
+// is local to the rail component; this lets a global hotkey trigger it).
+export const $profileCreateRequest = atom(0)
+
+export function requestProfileCreate(): void {
+  $profileCreateRequest.set($profileCreateRequest.get() + 1)
+}
+
+// Keepalive ping for the active pool backend so the main-process idle reaper
+// (which can't see the direct renderer↔backend WS) spares it. No-op for the
+// primary/default backend, which is never pooled.
+export function touchActiveGatewayBackend(): void {
+  // Always ping: the main process no-ops for non-pool (primary) backends, so we
+  // don't need to know which profile is primary from here.
+  const target = normalizeProfileKey($activeGatewayProfile.get())
+  void window.NASTECHDesktop?.touchBackend?.(target).catch(() => undefined)
+}

package/src/store/prompts.test.ts

@@ -0,0 +1,121 @@
+import { afterEach, beforeEach, describe, expect, it } from 'vitest'
+
+import {
+  $approvalRequest,
+  $secretRequest,
+  $sudoRequest,
+  clearAllPrompts,
+  clearApprovalRequest,
+  clearSecretRequest,
+  clearSudoRequest,
+  setApprovalRequest,
+  setSecretRequest,
+  setSudoRequest
+} from './prompts'
+import { $activeSessionId } from './session'
+
+// Prompts are parked per-session; the exported $*Request views are scoped to the
+// active session, so each test focuses the session it's asserting on.
+beforeEach(() => {
+  $activeSessionId.set('s1')
+})
+
+afterEach(() => {
+  clearAllPrompts()
+  $activeSessionId.set(null)
+})
+
+describe('approval prompt store', () => {
+  it('holds the active session-keyed approval request', () => {
+    setApprovalRequest({ command: 'rm -rf /tmp/x', description: 'recursive delete', sessionId: 's1' })
+
+    expect($approvalRequest.get()).toEqual({
+      command: 'rm -rf /tmp/x',
+      description: 'recursive delete',
+      sessionId: 's1'
+    })
+  })
+
+  it('parks a background session prompt out of the active view', () => {
+    setApprovalRequest({ command: 'x', description: 'd', sessionId: 's2' })
+
+    // Not visible while s1 is focused …
+    expect($approvalRequest.get()).toBeNull()
+
+    // … but surfaces once the user switches to the session that raised it.
+    $activeSessionId.set('s2')
+    expect($approvalRequest.get()?.sessionId).toBe('s2')
+  })
+
+  it('clears the active session prompt', () => {
+    setApprovalRequest({ command: 'x', description: 'd', sessionId: 's1' })
+    clearApprovalRequest('s1')
+
+    expect($approvalRequest.get()).toBeNull()
+  })
+})
+
+describe('sudo prompt store', () => {
+  it('clears only when the request id matches the in-flight prompt', () => {
+    setSudoRequest({ requestId: 'abc', sessionId: 's1' })
+
+    // A stale clear for a different request must NOT drop the live prompt —
+    // otherwise a late response to a prior sudo ask would dismiss the current
+    // one and leave the agent blocked.
+    clearSudoRequest('s1', 'stale')
+    expect($sudoRequest.get()).toEqual({ requestId: 'abc', sessionId: 's1' })
+
+    clearSudoRequest('s1', 'abc')
+    expect($sudoRequest.get()).toBeNull()
+  })
+
+  it('clears unconditionally when no request id is given', () => {
+    setSudoRequest({ requestId: 'abc', sessionId: 's1' })
+    clearSudoRequest('s1')
+
+    expect($sudoRequest.get()).toBeNull()
+  })
+})
+
+describe('secret prompt store', () => {
+  it('carries env var and prompt, and clears on id match', () => {
+    setSecretRequest({ requestId: 'r1', envVar: 'OPENAI_API_KEY', prompt: 'Paste your key', sessionId: 's1' })
+
+    expect($secretRequest.get()).toEqual({
+      requestId: 'r1',
+      envVar: 'OPENAI_API_KEY',
+      prompt: 'Paste your key',
+      sessionId: 's1'
+    })
+
+    clearSecretRequest('s1', 'mismatch')
+    expect($secretRequest.get()).not.toBeNull()
+
+    clearSecretRequest('s1', 'r1')
+    expect($secretRequest.get()).toBeNull()
+  })
+})
+
+describe('clearAllPrompts', () => {
+  it('drops every kind for one session at once (turn end / interrupt)', () => {
+    setApprovalRequest({ command: 'x', description: 'd', sessionId: 's1' })
+    setSudoRequest({ requestId: 'abc', sessionId: 's1' })
+    setSecretRequest({ requestId: 'r1', envVar: 'E', prompt: 'p', sessionId: 's1' })
+
+    clearAllPrompts('s1')
+
+    expect($approvalRequest.get()).toBeNull()
+    expect($sudoRequest.get()).toBeNull()
+    expect($secretRequest.get()).toBeNull()
+  })
+
+  it('leaves other sessions parked prompts intact', () => {
+    setApprovalRequest({ command: 'x', description: 'd', sessionId: 's1' })
+    setApprovalRequest({ command: 'y', description: 'e', sessionId: 's2' })
+
+    clearAllPrompts('s1')
+
+    $activeSessionId.set('s2')
+    expect($approvalRequest.get()?.command).toBe('y')
+  })
+})

package/src/store/prompts.ts

@@ -0,0 +1,115 @@
+import { atom, computed, type ReadableAtom } from 'nanostores'
+
+import { $activeSessionId } from './session'
+
+// Blocking interactive prompts the gateway raises mid-turn. Each maps to a
+// `*.request` event the Python side emits while it blocks the agent thread
+// waiting for a `*.respond` RPC. Without a renderer for these, the agent
+// silently stalls until its timeout (default 5 min) and the tool is BLOCKED.
+//
+// Like clarify, every prompt is parked under the runtime session id that raised
+// it (not one shared slot), so a *background* session running concurrently can
+// raise an approval/sudo/secret prompt and have it wait — surfaced via the
+// sidebar "needs input" badge — until the user switches to that chat. The
+// exported $*Request view is scoped to the active session, so a background
+// prompt never hijacks the foreground.
+
+const keyFor = (sessionId: string | null | undefined): string => sessionId ?? ''
+
+interface KeyedPrompt {
+  sessionId: string | null
+}
+
+interface PromptStore<T extends KeyedPrompt> {
+  $active: ReadableAtom<null | T>
+  clear: (sessionId?: string | null, requestId?: string) => void
+  reset: () => void
+  set: (request: T) => void
+}
+
+// One per-session prompt kind: a map keyed by session, plus an active-session
+// view for the overlays. `clear` drops one session's entry (a request-id
+// mismatch is a no-op so a stale resolve can't wipe a newer prompt); with no
+// session hint it drops every entry, optionally filtered by request id.
+function keyedPromptStore<T extends KeyedPrompt>(): PromptStore<T> {
+  const $all = atom<Record<string, T>>({})
+  const idOf = (value: T): string | undefined => (value as { requestId?: string }).requestId
+
+  return {
+    $active: computed([$all, $activeSessionId], (all, activeId) => all[keyFor(activeId)] ?? null),
+    reset: () => $all.set({}),
+    set: request => $all.set({ ...$all.get(), [keyFor(request.sessionId)]: request }),
+    clear(sessionId, requestId) {
+      const all = $all.get()
+
+      if (sessionId !== undefined) {
+        const key = keyFor(sessionId)
+        const current = all[key]
+
+        if (current && !(requestId && idOf(current) !== requestId)) {
+          const next = { ...all }
+          delete next[key]
+          $all.set(next)
+        }
+
+        return
+      }
+
+      const next = Object.fromEntries(Object.entries(all).filter(([, v]) => requestId && idOf(v) !== requestId))
+
+      if (Object.keys(next).length !== Object.keys(all).length) {
+        $all.set(next as Record<string, T>)
+      }
+    }
+  }
+}
+
+// Approval is session-keyed on the backend (one in-flight approval per session,
+// resolved via approval.respond {choice, session_id}). It carries no request_id,
+// unlike sudo/secret which are _block()-style request/response.
+export interface ApprovalRequest extends KeyedPrompt {
+  command: string
+  description: string
+}
+
+export interface SudoRequest extends KeyedPrompt {
+  requestId: string
+}
+
+export interface SecretRequest extends KeyedPrompt {
+  envVar: string
+  prompt: string
+  requestId: string
+}
+
+const approval = keyedPromptStore<ApprovalRequest>()
+const sudo = keyedPromptStore<SudoRequest>()
+const secret = keyedPromptStore<SecretRequest>()
+
+export const $approvalRequest = approval.$active
+export const setApprovalRequest = approval.set
+export const clearApprovalRequest = approval.clear
+
+export const $sudoRequest = sudo.$active
+export const setSudoRequest = sudo.set
+export const clearSudoRequest = sudo.clear
+
+export const $secretRequest = secret.$active
+export const setSecretRequest = secret.set
+export const clearSecretRequest = secret.clear
+
+// Drop in-flight prompts for `sessionId` (a turn ended) across all three kinds —
+// or every parked prompt when no session is given (global reset / tests).
+export function clearAllPrompts(sessionId?: string | null): void {
+  if (sessionId === undefined) {
+    approval.reset()
+    sudo.reset()
+    secret.reset()
+
+    return
+  }
+
+  approval.clear(sessionId)
+  sudo.clear(sessionId)
+  secret.clear(sessionId)
+}