nastechai-desktop 18.1.0

package/src/store/composer-status.ts

@@ -0,0 +1,277 @@
+import { atom, computed } from 'nanostores'
+
+import { translateNow } from '@/i18n'
+import type { TodoItem, TodoStatus } from '@/lib/todos'
+
+import { $gateway } from './gateway'
+import { dispatchNativeNotification } from './native-notifications'
+import { $subagentsBySession, type SubagentProgress } from './subagents'
+import { $todosBySession } from './todos'
+
+/** Composer status stack feed — merged todos, subagents, background per session. */
+export type StatusItemState = 'done' | 'failed' | 'running'
+export type StatusItemType = 'background' | 'subagent' | 'todo'
+
+export interface ComposerStatusItem {
+  /** background: non-zero exit shown inline when failed. */
+  exitCode?: number
+  /** subagent: active tool label shown on the right. */
+  currentTool?: string
+  id: string
+  /** background process: captured stdout/stderr tail for the inline viewer. */
+  output?: string
+  /** subagent: its own stored session id — row click opens that session window
+   *  (livestreamed by the gateway's child-session mirror). */
+  sessionId?: string
+  state: StatusItemState
+  title: string
+  /** todo: the full four-state status driving the row's checkmark glyph. */
+  todoStatus?: TodoStatus
+  type: StatusItemType
+}
+
+// Writable source for background work, synced from the gateway's process
+// registry (`terminal(background=true)` spawns) via `process.list`.
+export const $backgroundStatusBySession = atom<Record<string, ComposerStatusItem[]>>({})
+
+// Rows the user X-ed away. The registry keeps finished processes around for a
+// while, so without this every refresh would resurrect a dismissed row.
+const dismissedBySession = new Map<string, Set<string>>()
+
+const subToItem = (s: SubagentProgress): ComposerStatusItem => ({
+  currentTool: s.currentTool,
+  id: s.id,
+  sessionId: s.sessionId,
+  state: 'running',
+  title: s.goal,
+  type: 'subagent'
+})
+
+const todoToItem = (t: TodoItem): ComposerStatusItem => ({
+  id: `todo:${t.id}`,
+  state: t.status === 'in_progress' ? 'running' : 'done',
+  title: t.content,
+  todoStatus: t.status,
+  type: 'todo'
+})
+
+// The single thing the stack reads: a typed, merged item list per session.
+export const $statusItemsBySession = computed(
+  [$subagentsBySession, $backgroundStatusBySession, $todosBySession],
+  (subs, background, todos) => {
+    const out: Record<string, ComposerStatusItem[]> = {}
+
+    const push = (sid: string, items: ComposerStatusItem[]) => {
+      if (items.length > 0) {
+        out[sid] = out[sid] ? [...out[sid], ...items] : items
+      }
+    }
+
+    for (const [sid, list] of Object.entries(todos)) {
+      push(sid, list.map(todoToItem))
+    }
+
+    for (const [sid, list] of Object.entries(subs)) {
+      push(sid, list.filter(s => s.status === 'running' || s.status === 'queued').map(subToItem))
+    }
+
+    for (const [sid, list] of Object.entries(background)) {
+      push(sid, list)
+    }
+
+    return out
+  }
+)
+
+// Fixed render order for the groups in the stack (top → bottom, above queue).
+const TYPE_ORDER: readonly StatusItemType[] = ['todo', 'subagent', 'background']
+
+export interface StatusGroup {
+  items: ComposerStatusItem[]
+  type: StatusItemType
+}
+
+export function groupStatusItems(items: readonly ComposerStatusItem[]): StatusGroup[] {
+  const byType = new Map<StatusItemType, ComposerStatusItem[]>()
+
+  for (const item of items) {
+    const list = byType.get(item.type)
+
+    if (list) {
+      list.push(item)
+    } else {
+      byType.set(item.type, [item])
+    }
+  }
+
+  return TYPE_ORDER.filter(type => byType.has(type)).map(type => ({ items: byType.get(type)!, type }))
+}
+
+const writeBackground = (sid: string, items: ComposerStatusItem[]) => {
+  const current = $backgroundStatusBySession.get()
+  const next = { ...current }
+
+  if (items.length > 0) {
+    next[sid] = items
+  } else {
+    delete next[sid]
+  }
+
+  $backgroundStatusBySession.set(next)
+}
+
+// `tui_gateway` process.list entry (tools/process_registry.list_sessions + output_tail).
+interface GatewayProcessEntry {
+  command?: string
+  exit_code?: number
+  output_tail?: string
+  session_id?: string
+  status?: string
+}
+
+const toBackgroundItem = (proc: GatewayProcessEntry): ComposerStatusItem => {
+  const exited = proc.status === 'exited'
+  const exitCode = typeof proc.exit_code === 'number' ? proc.exit_code : undefined
+
+  return {
+    exitCode,
+    id: proc.session_id ?? '',
+    output: proc.output_tail || undefined,
+    state: exited ? (exitCode ? 'failed' : 'done') : 'running',
+    title: (proc.command ?? '').split('\n')[0]!.trim() || 'background process',
+    type: 'background'
+  }
+}
+
+const sameItem = (a: ComposerStatusItem, b: ComposerStatusItem) =>
+  a.state === b.state && a.title === b.title && a.output === b.output && a.exitCode === b.exitCode
+
+/**
+ * Layout-stable sync of the registry snapshot into the store: existing rows
+ * keep their position (status flips happen in place, never reorder), new
+ * processes append, dismissed ids stay gone, and unchanged rows keep their
+ * object identity so memoised rows skip re-rendering.
+ */
+export function reconcileBackgroundProcesses(sid: string, procs: GatewayProcessEntry[]) {
+  const dismissed = dismissedBySession.get(sid)
+
+  const fresh = new Map(
+    procs
+      .filter(proc => proc.session_id && !dismissed?.has(proc.session_id))
+      .map(proc => [proc.session_id!, toBackgroundItem(proc)])
+  )
+
+  const prev = $backgroundStatusBySession.get()[sid] ?? []
+
+  // running → exited since the last snapshot = a background process just finished.
+  const prevState = new Map(prev.map(item => [item.id, item.state]))
+
+  for (const [id, item] of fresh) {
+    if (item.state !== 'running' && prevState.get(id) === 'running') {
+      dispatchNativeNotification({
+        body: item.title,
+        kind: 'backgroundDone',
+        sessionId: sid,
+        title: translateNow(
+          item.state === 'failed'
+            ? 'notifications.native.backgroundFailedTitle'
+            : 'notifications.native.backgroundDoneTitle'
+        )
+      })
+    }
+  }
+
+  const kept = prev.flatMap(old => {
+    const next = fresh.get(old.id)
+    fresh.delete(old.id)
+
+    return next ? [sameItem(old, next) ? old : next] : []
+  })
+
+  const next = [...kept, ...fresh.values()]
+
+  // Dismissals only need remembering while the registry still reports the id.
+  if (dismissed) {
+    const reported = new Set(procs.map(proc => proc.session_id))
+
+    for (const id of dismissed) {
+      if (!reported.has(id)) {
+        dismissed.delete(id)
+      }
+    }
+  }
+
+  if (next.length === prev.length && next.every((item, i) => item === prev[i])) {
+    return
+  }
+
+  writeBackground(sid, next)
+}
+
+/** Pull the session's live process snapshot from the gateway. */
+export async function refreshBackgroundProcesses(sid: string): Promise<void> {
+  const gateway = $gateway.get()
+
+  if (!sid || !gateway) {
+    return
+  }
+
+  try {
+    const result = await gateway.request<{ processes?: GatewayProcessEntry[] }>('process.list', { session_id: sid })
+
+    reconcileBackgroundProcesses(sid, result?.processes ?? [])
+  } catch {
+    // Transient socket loss — the next trigger (event or poll) retries.
+  }
+}
+
+/** X on a finished row: drop it now and keep it dropped across refreshes. */
+export function dismissBackgroundProcess(sid: string, id: string) {
+  const dismissed = dismissedBySession.get(sid) ?? new Set<string>()
+  dismissed.add(id)
+  dismissedBySession.set(sid, dismissed)
+
+  const list = $backgroundStatusBySession.get()[sid] ?? []
+
+  writeBackground(
+    sid,
+    list.filter(item => item.id !== id)
+  )
+}
+
+/** X on a running row: kill the process for real, then drop the row. */
+export function stopBackgroundProcess(sid: string, id: string) {
+  void $gateway
+    .get()
+    ?.request('process.kill', { process_id: id, session_id: sid })
+    .catch(() => undefined)
+  dismissBackgroundProcess(sid, id)
+}
+
+/**
+ * Rewind cleanup: a restore/edit discards the turns that spawned these
+ * processes, so they belong to an abandoned timeline. Kill the live ones and
+ * drop every row. Ids are marked dismissed so an in-flight `process.list` poll
+ * (kill is async) can't resurrect them; reconcile garbage-collects those once
+ * the registry stops reporting them.
+ */
+export function resetSessionBackground(sid: string) {
+  if (!sid) {
+    return
+  }
+
+  const gateway = $gateway.get()
+  const list = $backgroundStatusBySession.get()[sid] ?? []
+  const dismissed = dismissedBySession.get(sid) ?? new Set<string>()
+
+  for (const item of list) {
+    dismissed.add(item.id)
+
+    if (item.state === 'running') {
+      void gateway?.request('process.kill', { process_id: item.id, session_id: sid }).catch(() => undefined)
+    }
+  }
+
+  dismissedBySession.set(sid, dismissed)
+  writeBackground(sid, [])
+}

package/src/store/composer.test.ts

@@ -0,0 +1,106 @@
+import { afterEach, describe, expect, it } from 'vitest'
+
+import {
+  $composerAttachments,
+  addComposerAttachment,
+  clearSessionDraft,
+  type ComposerAttachment,
+  removeComposerAttachment,
+  SESSION_DRAFTS_STORAGE_KEY,
+  stashSessionDraft,
+  takeSessionDraft,
+  updateComposerAttachment
+} from './composer'
+
+function attachment(overrides: Partial<ComposerAttachment> & Pick<ComposerAttachment, 'id'>): ComposerAttachment {
+  return { kind: 'file', label: 'doc.pdf', ...overrides }
+}
+
+describe('updateComposerAttachment', () => {
+  afterEach(() => {
+    $composerAttachments.set([])
+  })
+
+  it('replaces an existing attachment in place', () => {
+    addComposerAttachment(attachment({ id: 'file:a', uploadState: 'uploading' }))
+
+    const updated = updateComposerAttachment(attachment({ id: 'file:a', attachedSessionId: 'sess-1' }))
+
+    expect(updated).toBe(true)
+    const current = $composerAttachments.get()
+    expect(current).toHaveLength(1)
+    expect(current[0]?.attachedSessionId).toBe('sess-1')
+    expect(current[0]?.uploadState).toBeUndefined()
+  })
+
+  it('does NOT resurrect an attachment the user removed mid-upload', () => {
+    // Drop → eager upload starts → user removes the chip → upload resolves.
+    // The late success must not re-add the removed attachment.
+    addComposerAttachment(attachment({ id: 'file:a', uploadState: 'uploading' }))
+    removeComposerAttachment('file:a')
+
+    const updated = updateComposerAttachment(attachment({ id: 'file:a', attachedSessionId: 'sess-1' }))
+
+    expect(updated).toBe(false)
+    expect($composerAttachments.get()).toHaveLength(0)
+  })
+})
+
+describe('session drafts', () => {
+  afterEach(() => {
+    for (const scope of ['session-a', 'session-b', null]) {
+      clearSessionDraft(scope)
+    }
+
+    window.localStorage.clear()
+  })
+
+  it('keeps drafts isolated per session scope', () => {
+    stashSessionDraft('session-a', 'draft a', [])
+    stashSessionDraft('session-b', 'draft b', [attachment({ id: 'image:b', kind: 'image' })])
+
+    expect(takeSessionDraft('session-a')).toEqual({ attachments: [], text: 'draft a' })
+    expect(takeSessionDraft('session-b').text).toBe('draft b')
+    expect(takeSessionDraft('session-b').attachments.map(a => a.id)).toEqual(['image:b'])
+  })
+
+  it('scopes the unsaved new-session draft separately from real sessions', () => {
+    stashSessionDraft(null, 'new chat draft', [])
+    stashSessionDraft('session-a', 'session draft', [])
+
+    expect(takeSessionDraft(null).text).toBe('new chat draft')
+    expect(takeSessionDraft(undefined).text).toBe('new chat draft')
+    expect(takeSessionDraft('session-a').text).toBe('session draft')
+  })
+
+  it('persists draft text (not attachments) to localStorage', () => {
+    stashSessionDraft('session-a', 'survives reload', [attachment({ id: 'file:a' })])
+
+    const persisted = JSON.parse(window.localStorage.getItem(SESSION_DRAFTS_STORAGE_KEY) ?? '{}') as Record<string, string>
+
+    expect(persisted['session-a']).toBe('survives reload')
+  })
+
+  it('evicts empty drafts instead of leaving stale entries behind', () => {
+    stashSessionDraft('session-a', 'saved', [])
+    stashSessionDraft('session-a', '   ', [])
+
+    expect(takeSessionDraft('session-a')).toEqual({ attachments: [], text: '' })
+  })
+
+  it('clears a stashed draft after an accepted submit', () => {
+    stashSessionDraft('session-a', 'sent prompt', [attachment({ id: 'file:a' })])
+    clearSessionDraft('session-a')
+
+    expect(takeSessionDraft('session-a')).toEqual({ attachments: [], text: '' })
+  })
+
+  it('returns clones so callers cannot mutate the stash', () => {
+    stashSessionDraft('session-a', 'draft', [attachment({ id: 'file:a' })])
+
+    const taken = takeSessionDraft('session-a')
+    taken.attachments[0]!.label = 'mutated'
+
+    expect(takeSessionDraft('session-a').attachments[0]?.label).toBe('doc.pdf')
+  })
+})

package/src/store/composer.ts

@@ -0,0 +1,184 @@
+import { atom } from 'nanostores'
+
+import { triggerHaptic } from '@/lib/haptics'
+
+export interface ComposerAttachment {
+  id: string
+  kind: 'image' | 'file' | 'folder' | 'terminal' | 'url'
+  label: string
+  detail?: string
+  refText?: string
+  previewUrl?: string
+  path?: string
+  attachedSessionId?: string
+}
+
+export const $composerDraft = atom('')
+export const $composerAttachments = atom<ComposerAttachment[]>([])
+export const $composerTerminalSelections = atom<Record<string, string>>({})
+
+export function setComposerDraft(value: string) {
+  $composerDraft.set(value)
+}
+
+export function appendComposerDraft(value: string) {
+  const text = value.trim()
+
+  if (!text) {
+    return
+  }
+
+  const current = $composerDraft.get()
+  const separator = current && !current.endsWith('\n') ? '\n\n' : ''
+
+  $composerDraft.set(`${current}${separator}${text}`)
+}
+
+export function appendComposerInline(value: string) {
+  const text = value.trim()
+
+  if (!text) {
+    return
+  }
+
+  const current = $composerDraft.get().trimEnd()
+  const separator = current ? ' ' : ''
+
+  $composerDraft.set(`${current}${separator}${text}`)
+}
+
+export function clearComposerDraft() {
+  $composerDraft.set('')
+}
+
+export function addComposerAttachment(attachment: ComposerAttachment) {
+  const previous = $composerAttachments.get()
+  const next = upsertAttachment(previous, attachment)
+  $composerAttachments.set(next)
+
+  if (next.length > previous.length && attachment.kind !== 'url') {
+    triggerHaptic('selection')
+  }
+}
+
+export function removeComposerAttachment(id: string): ComposerAttachment | null {
+  const current = $composerAttachments.get()
+  const removed = current.find(attachment => attachment.id === id) || null
+  $composerAttachments.set(current.filter(attachment => attachment.id !== id))
+
+  return removed
+}
+
+export function clearComposerAttachments() {
+  $composerAttachments.set([])
+}
+
+const TERMINAL_REF_RE = /@terminal:(`[^`\n]+`|"[^"\n]+"|'[^'\n]+'|\S+)/g
+
+function unquoteRefValue(raw: string) {
+  const head = raw[0]
+  const tail = raw[raw.length - 1]
+  const quoted = (head === '`' && tail === '`') || (head === '"' && tail === '"') || (head === "'" && tail === "'")
+
+  return (quoted ? raw.slice(1, -1) : raw).replace(/[,.;!?]+$/, '').trim()
+}
+
+function terminalLabelsFromDraft(draft: string) {
+  const labels: string[] = []
+  const seen = new Set<string>()
+
+  for (const match of draft.matchAll(TERMINAL_REF_RE)) {
+    const label = unquoteRefValue(match[1] || '')
+
+    if (!label || seen.has(label)) {
+      continue
+    }
+
+    seen.add(label)
+    labels.push(label)
+  }
+
+  return labels
+}
+
+export function setComposerTerminalSelection(label: string, text: string) {
+  const nextLabel = label.trim()
+  const nextText = text.trim()
+
+  if (!nextLabel || !nextText) {
+    return
+  }
+
+  const current = $composerTerminalSelections.get()
+
+  if (current[nextLabel] === nextText) {
+    return
+  }
+
+  $composerTerminalSelections.set({
+    ...current,
+    [nextLabel]: nextText
+  })
+}
+
+export function reconcileComposerTerminalSelections(draft: string) {
+  const current = $composerTerminalSelections.get()
+  const labels = new Set(terminalLabelsFromDraft(draft))
+  let changed = false
+  const next: Record<string, string> = {}
+
+  for (const [label, text] of Object.entries(current)) {
+    if (!labels.has(label)) {
+      changed = true
+
+      continue
+    }
+
+    next[label] = text
+  }
+
+  if (changed) {
+    $composerTerminalSelections.set(next)
+  }
+}
+
+export function terminalContextBlocksFromDraft(draft: string) {
+  const labels = terminalLabelsFromDraft(draft)
+
+  if (labels.length === 0) {
+    return []
+  }
+
+  const selections = $composerTerminalSelections.get()
+
+  return labels.flatMap(label => {
+    const text = selections[label]?.trim()
+
+    if (!text) {
+      return []
+    }
+
+    return `\`\`\`terminal\n${text}\n\`\`\``
+  })
+}
+
+export function clearComposerTerminalSelections() {
+  if (Object.keys($composerTerminalSelections.get()).length === 0) {
+    return
+  }
+
+  $composerTerminalSelections.set({})
+}
+
+function upsertAttachment(attachments: ComposerAttachment[], attachment: ComposerAttachment) {
+  const index = attachments.findIndex(item => item.id === attachment.id)
+
+  if (index < 0) {
+    return [...attachments, attachment]
+  }
+
+  const next = [...attachments]
+  next[index] = attachment
+
+  return next
+}

package/src/store/cron.ts

@@ -0,0 +1,19 @@
+import { atom } from 'nanostores'
+
+import type { CronJob } from '@/types/nastech'
+
+// Cron *jobs* (not run sessions) power the sidebar "Cron jobs" section. Listing
+// the job — schedule, state, live next-run countdown — makes the job the
+// first-class entity; its runs (sessions) resolve under it in the cron detail.
+export const $cronJobs = atom<CronJob[]>([])
+export const setCronJobs = (jobs: CronJob[]) => $cronJobs.set(jobs)
+
+// In-place edit so the cron overlay's mutations (create/edit/delete/pause/…)
+// land in the same atom the sidebar renders — no stale list until the next poll.
+export const updateCronJobs = (fn: (jobs: CronJob[]) => CronJob[]) => $cronJobs.set(fn($cronJobs.get()))
+
+// One-shot focus target: clicking "Manage" on a job sets this, then opens the
+// cron overlay, which reads it once to select + scroll to that job. Cleared
+// after consumption so re-opening cron normally doesn't re-focus a stale job.
+export const $cronFocusJobId = atom<null | string>(null)
+export const setCronFocusJobId = (id: null | string) => $cronFocusJobId.set(id)