@brimveyn/aimux 1.1.0

package/src/state/types.ts

@@ -0,0 +1,367 @@
+export type BuiltinAssistantId = 'claude' | 'codex' | 'opencode' | 'terminal'
+
+export type AssistantId = BuiltinAssistantId | (string & {})
+
+export type TabStatus = 'starting' | 'running' | 'disconnected' | 'exited' | 'error'
+
+export type TabActivity = 'busy' | 'idle'
+
+export type FocusMode = 'navigation' | 'terminal-input' | 'modal' | 'command-edit' | 'layout'
+
+export type ModalType =
+  | 'new-tab'
+  | 'session-picker'
+  | 'session-name'
+  | 'create-session'
+  | 'rename-tab'
+  | 'snippet-picker'
+  | 'snippet-editor'
+  | 'theme-picker'
+  | 'help'
+  | 'split-picker'
+  | null
+
+export interface TerminalSpan {
+  text: string
+  fg?: string
+  bg?: string
+  bold?: boolean
+  italic?: boolean
+  underline?: boolean
+  cursor?: boolean
+}
+
+export interface TerminalLine {
+  spans: TerminalSpan[]
+}
+
+export interface TerminalSnapshot {
+  lines: TerminalLine[]
+  viewportY: number
+  baseY: number
+  cursorVisible: boolean
+}
+
+export interface TerminalModeState {
+  mouseTrackingMode: 'none' | 'x10' | 'vt200' | 'drag' | 'any'
+  sendFocusMode: boolean
+  alternateScrollMode: boolean
+  isAlternateBuffer: boolean
+  bracketedPasteMode: boolean
+}
+
+export interface PersistedTabSnapshot {
+  id: string
+  assistant: AssistantId
+  title: string
+  command: string
+  status: Exclude<TabStatus, 'disconnected'>
+  buffer: string
+  viewport?: TerminalSnapshot
+  terminalModes: TerminalModeState
+  errorMessage?: string
+  exitCode?: number
+}
+
+export interface WorkspaceSnapshotV1 {
+  version: 1
+  savedAt: string
+  activeTabId: string | null
+  sidebar: {
+    visible: boolean
+    width: number
+    gitPanelVisible?: boolean
+    gitPanelRatio?: number
+  }
+  tabs: PersistedTabSnapshot[]
+  layoutTree?: import('./layout-tree').LayoutNode
+  layoutTrees?: Record<string, import('./layout-tree').LayoutNode>
+  tabGroupMap?: Record<string, string>
+}
+
+export interface SessionRecord {
+  id: string
+  name: string
+  projectPath?: string
+  createdAt: string
+  updatedAt: string
+  lastOpenedAt: string
+  workspaceSnapshot?: WorkspaceSnapshotV1
+}
+
+export interface TabSession {
+  id: string
+  assistant: AssistantId
+  title: string
+  status: TabStatus
+  activity?: TabActivity
+  buffer: string
+  viewport?: TerminalSnapshot
+  terminalModes: TerminalModeState
+  command: string
+  errorMessage?: string
+  exitCode?: number
+}
+
+export interface SidebarState {
+  visible: boolean
+  width: number
+  minWidth: number
+  maxWidth: number
+  gitPanelVisible: boolean
+  gitPanelRatio: number
+}
+
+export type GitFileStatus = 'M' | 'A' | 'D' | 'R' | 'C' | 'U' | '?'
+
+export type GitFileSection = 'staged' | 'unstaged' | 'untracked'
+
+export interface GitFileEntry {
+  path: string
+  renamedFrom?: string
+  section: GitFileSection
+  status: GitFileStatus
+  added: number | null
+  removed: number | null
+}
+
+export type GitPanelError = 'not-a-repo' | 'unknown'
+
+export interface GitPanelState {
+  branch: string | null
+  ahead: number
+  behind: number
+  files: GitFileEntry[]
+  error: GitPanelError | null
+}
+
+interface ModalBase {
+  selectedIndex: number
+  editBuffer: string | null
+  sessionTargetId: string | null
+}
+
+export interface ModalClosed extends ModalBase {
+  type: null
+  editBuffer: null
+  sessionTargetId: null
+}
+
+export interface ModalNewTab extends ModalBase {
+  type: 'new-tab'
+}
+
+export interface ModalSessionPicker extends ModalBase {
+  type: 'session-picker'
+}
+
+export interface ModalSessionName extends ModalBase {
+  type: 'session-name'
+}
+
+export interface ModalRenameTab extends ModalBase {
+  type: 'rename-tab'
+}
+
+export interface ModalSnippetPicker extends ModalBase {
+  type: 'snippet-picker'
+}
+
+export interface ModalThemePicker extends ModalBase {
+  type: 'theme-picker'
+}
+
+export interface ModalHelp extends ModalBase {
+  type: 'help'
+}
+
+export interface ModalSplitPicker extends ModalBase {
+  type: 'split-picker'
+  splitDirection: import('./layout-tree').SplitDirection
+}
+
+export interface ModalCreateSession extends ModalBase {
+  type: 'create-session'
+  directoryResults: DirectoryResult[]
+  pendingProjectPath: string | null
+  activeField: 'directory' | 'name'
+  nameBuffer: string
+}
+
+export interface ModalSnippetEditor extends ModalBase {
+  type: 'snippet-editor'
+  activeField: 'name' | 'content'
+  contentBuffer: string
+}
+
+export type DirectoryResultType = 'git-repo' | 'worktree' | 'workspace'
+
+export interface DirectoryResult {
+  path: string
+  type: DirectoryResultType
+}
+
+export type ModalState =
+  | ModalClosed
+  | ModalNewTab
+  | ModalSessionPicker
+  | ModalSessionName
+  | ModalRenameTab
+  | ModalSnippetPicker
+  | ModalThemePicker
+  | ModalHelp
+  | ModalSplitPicker
+  | ModalCreateSession
+  | ModalSnippetEditor
+
+export interface LayoutState {
+  terminalCols: number
+  terminalRows: number
+}
+
+export interface SnippetRecord {
+  id: string
+  name: string
+  content: string
+}
+
+export interface AppState {
+  tabs: TabSession[]
+  activeTabId: string | null
+  layoutTrees: Record<string, import('./layout-tree').LayoutNode>
+  tabGroupMap: Record<string, string>
+  sessions: SessionRecord[]
+  currentSessionId: string | null
+  snippets: SnippetRecord[]
+  focusMode: FocusMode
+  sidebar: SidebarState
+  modal: ModalState
+  layout: LayoutState
+  customCommands: Record<AssistantId, string>
+  gitPanel: GitPanelState
+  /** Chord prefix the sequence resolver is currently waiting on, or null when idle. */
+  pendingChords: string[] | null
+}
+
+// -- Modal actions --
+export type ModalAction =
+  | { type: 'open-new-tab-modal' }
+  | { type: 'open-help-modal' }
+  | { type: 'open-split-picker'; direction: import('./layout-tree').SplitDirection }
+  | { type: 'open-session-picker' }
+  | { type: 'open-session-name-modal'; sessionTargetId?: string; initialName?: string }
+  | { type: 'close-modal' }
+  | { type: 'move-modal-selection'; delta: number }
+  | { type: 'begin-command-edit' }
+  | { type: 'update-command-edit'; char: string }
+  | { type: 'commit-command-edit' }
+  | { type: 'cancel-command-edit' }
+  | { type: 'open-create-session-modal' }
+  | { type: 'set-directory-results'; results: DirectoryResult[] }
+  | { type: 'switch-create-session-field' }
+  | { type: 'select-directory' }
+  | { type: 'begin-session-filter' }
+  | { type: 'open-rename-tab-modal' }
+  | { type: 'open-snippet-picker' }
+  | { type: 'open-snippet-editor'; snippetId?: string }
+  | { type: 'begin-snippet-filter' }
+  | { type: 'open-theme-picker' }
+
+// -- Session actions --
+export type SessionAction =
+  | { type: 'load-session'; sessionId: string; workspaceSnapshot?: WorkspaceSnapshotV1 }
+  | { type: 'set-sessions'; sessions: SessionRecord[] }
+  | { type: 'create-session-record'; session: SessionRecord }
+  | { type: 'rename-session-record'; sessionId: string; name: string }
+  | { type: 'delete-session-record'; sessionId: string }
+
+// -- Tab actions --
+export type TabAction =
+  | { type: 'add-tab'; tab: TabSession }
+  | {
+      type: 'hydrate-workspace'
+      tabs: TabSession[]
+      activeTabId: string | null
+      layoutTree?: import('./layout-tree').LayoutNode | null
+      layoutTrees?: Record<string, import('./layout-tree').LayoutNode>
+      tabGroupMap?: Record<string, string>
+    }
+  | { type: 'close-tab'; tabId: string }
+  | { type: 'close-active-tab' }
+  | { type: 'set-active-tab'; tabId: string }
+  | { type: 'move-active-tab'; delta: number }
+  | { type: 'reorder-active-tab'; delta: number }
+  | { type: 'reset-tab-session'; tabId: string }
+  | { type: 'rename-tab'; tabId: string; title: string }
+  | { type: 'append-tab-buffer'; tabId: string; chunk: string }
+  | {
+      type: 'replace-tab-viewport'
+      tabId: string
+      viewport: TerminalSnapshot
+      terminalModes: TerminalModeState
+    }
+  | { type: 'set-tab-activity'; tabId: string; activity?: TabActivity }
+  | { type: 'set-tab-status'; tabId: string; status: TabStatus; exitCode?: number }
+  | { type: 'set-tab-error'; tabId: string; message: string }
+
+// -- Layout actions --
+export type LayoutAction =
+  | {
+      type: 'split-pane'
+      direction: import('./layout-tree').SplitDirection
+      newTab: TabSession
+    }
+  | { type: 'close-pane'; tabId: string }
+  | {
+      type: 'focus-pane-direction'
+      direction: 'left' | 'right' | 'up' | 'down'
+    }
+  | {
+      type: 'resize-pane'
+      tabId: string
+      delta: number
+      axis?: import('./layout-tree').SplitDirection
+    }
+  | {
+      type: 'set-split-ratio'
+      tabId: string
+      ratio: number
+      axis?: import('./layout-tree').SplitDirection
+    }
+
+// -- UI actions --
+export type UIAction =
+  | { type: 'toggle-sidebar' }
+  | { type: 'resize-sidebar'; delta: number }
+  | { type: 'set-focus-mode'; focusMode: FocusMode }
+  | { type: 'set-terminal-size'; cols: number; rows: number }
+  | { type: 'toggle-git-panel' }
+  | { type: 'resize-git-panel'; delta: number }
+  | { type: 'set-pending-chords'; chords: string[] | null }
+
+// -- Git panel actions --
+export interface GitRefreshPayload {
+  branch: string | null
+  ahead: number
+  behind: number
+  files: GitFileEntry[]
+}
+
+export type GitPanelAction =
+  | { type: 'git-refresh-success'; payload: GitRefreshPayload }
+  | { type: 'git-refresh-error'; kind: GitPanelError }
+  | { type: 'git-panel-reset' }
+
+// -- Data actions --
+export type DataAction =
+  | { type: 'set-snippets'; snippets: SnippetRecord[] }
+  | { type: 'delete-snippet'; snippetId: string }
+
+export type AppAction =
+  | ModalAction
+  | SessionAction
+  | TabAction
+  | LayoutAction
+  | UIAction
+  | DataAction
+  | GitPanelAction

package/src/state/validation.ts

@@ -0,0 +1,154 @@
+import type {
+  SessionRecord,
+  SnippetRecord,
+  TerminalLine,
+  TerminalModeState,
+  TerminalSnapshot,
+  WorkspaceSnapshotV1,
+} from './types'
+
+function isObjectRecord(value: unknown): value is Record<string, unknown> {
+  return typeof value === 'object' && value !== null
+}
+
+function isString(value: unknown): value is string {
+  return typeof value === 'string'
+}
+
+function isBoolean(value: unknown): value is boolean {
+  return typeof value === 'boolean'
+}
+
+function isFiniteNumber(value: unknown): value is number {
+  return typeof value === 'number' && Number.isFinite(value)
+}
+
+function isTerminalLine(value: unknown): value is TerminalLine {
+  if (!isObjectRecord(value) || !Array.isArray(value.spans)) {
+    return false
+  }
+
+  return value.spans.every((span) => {
+    if (!isObjectRecord(span) || !isString(span.text)) {
+      return false
+    }
+
+    return (
+      (span.fg === undefined || isString(span.fg)) &&
+      (span.bg === undefined || isString(span.bg)) &&
+      (span.bold === undefined || isBoolean(span.bold)) &&
+      (span.italic === undefined || isBoolean(span.italic)) &&
+      (span.underline === undefined || isBoolean(span.underline)) &&
+      (span.cursor === undefined || isBoolean(span.cursor))
+    )
+  })
+}
+
+function isTerminalSnapshot(value: unknown): value is TerminalSnapshot {
+  return (
+    isObjectRecord(value) &&
+    Array.isArray(value.lines) &&
+    value.lines.every(isTerminalLine) &&
+    isFiniteNumber(value.viewportY) &&
+    isFiniteNumber(value.baseY) &&
+    isBoolean(value.cursorVisible)
+  )
+}
+
+function isTerminalModeState(value: unknown): value is TerminalModeState {
+  return (
+    isObjectRecord(value) &&
+    (value.mouseTrackingMode === 'none' ||
+      value.mouseTrackingMode === 'x10' ||
+      value.mouseTrackingMode === 'vt200' ||
+      value.mouseTrackingMode === 'drag' ||
+      value.mouseTrackingMode === 'any') &&
+    isBoolean(value.sendFocusMode) &&
+    isBoolean(value.alternateScrollMode) &&
+    isBoolean(value.isAlternateBuffer) &&
+    isBoolean(value.bracketedPasteMode)
+  )
+}
+
+function isLayoutNode(value: unknown): boolean {
+  if (!isObjectRecord(value)) return false
+  if (value.type === 'leaf') {
+    return isString(value.tabId) && value.tabId.length > 0
+  }
+  if (value.type === 'split') {
+    return (
+      (value.direction === 'horizontal' || value.direction === 'vertical') &&
+      isFiniteNumber(value.ratio) &&
+      value.ratio > 0 &&
+      value.ratio < 1 &&
+      isLayoutNode(value.first) &&
+      isLayoutNode(value.second)
+    )
+  }
+  return false
+}
+
+function isLayoutTreesMap(value: unknown): boolean {
+  if (!isObjectRecord(value)) return false
+  return Object.values(value).every(isLayoutNode)
+}
+
+function isStringRecord(value: unknown): boolean {
+  if (!isObjectRecord(value)) return false
+  return Object.values(value).every(isString)
+}
+
+export function isWorkspaceSnapshotV1(value: unknown): value is WorkspaceSnapshotV1 {
+  return (
+    isObjectRecord(value) &&
+    value.version === 1 &&
+    isString(value.savedAt) &&
+    (value.activeTabId === null || isString(value.activeTabId)) &&
+    isObjectRecord(value.sidebar) &&
+    isBoolean(value.sidebar.visible) &&
+    isFiniteNumber(value.sidebar.width) &&
+    (value.sidebar.gitPanelVisible === undefined || isBoolean(value.sidebar.gitPanelVisible)) &&
+    (value.sidebar.gitPanelRatio === undefined || isFiniteNumber(value.sidebar.gitPanelRatio)) &&
+    Array.isArray(value.tabs) &&
+    value.tabs.every(
+      (tab) =>
+        isObjectRecord(tab) &&
+        isString(tab.id) &&
+        isString(tab.assistant) &&
+        tab.assistant.length > 0 &&
+        isString(tab.title) &&
+        isString(tab.command) &&
+        (tab.status === 'starting' ||
+          tab.status === 'running' ||
+          tab.status === 'exited' ||
+          tab.status === 'error') &&
+        isString(tab.buffer) &&
+        isTerminalModeState(tab.terminalModes) &&
+        (tab.viewport === undefined || isTerminalSnapshot(tab.viewport)) &&
+        (tab.errorMessage === undefined || isString(tab.errorMessage)) &&
+        (tab.exitCode === undefined || isFiniteNumber(tab.exitCode))
+    ) &&
+    (value.layoutTree === undefined || isLayoutNode(value.layoutTree)) &&
+    (value.layoutTrees === undefined || isLayoutTreesMap(value.layoutTrees)) &&
+    (value.tabGroupMap === undefined || isStringRecord(value.tabGroupMap))
+  )
+}
+
+export function isSessionRecord(value: unknown): value is SessionRecord {
+  return (
+    isObjectRecord(value) &&
+    isString(value.id) &&
+    isString(value.name) &&
+    (value.projectPath === undefined || isString(value.projectPath)) &&
+    isString(value.createdAt) &&
+    isString(value.updatedAt) &&
+    isString(value.lastOpenedAt) &&
+    (value.workspaceSnapshot === undefined || isWorkspaceSnapshotV1(value.workspaceSnapshot))
+  )
+}
+
+export function isSnippetRecord(value: unknown): value is SnippetRecord {
+  return (
+    isObjectRecord(value) && isString(value.id) && isString(value.name) && isString(value.content)
+  )
+}

package/src/state/workspace-save.ts

@@ -0,0 +1,33 @@
+import type { AppState, SessionRecord } from './types'
+
+import { loadConfig, saveConfig } from '../config'
+import { saveSessionCatalog } from './session-catalog'
+import { serializeWorkspace } from './session-persistence'
+
+export function buildSessionsWithCurrentSnapshot(
+  sessions: SessionRecord[],
+  currentSessionId: string | null,
+  state: AppState
+): SessionRecord[] {
+  return sessions.map((session) =>
+    session.id === currentSessionId
+      ? {
+          ...session,
+          updatedAt: new Date().toISOString(),
+          workspaceSnapshot: serializeWorkspace(state),
+        }
+      : session
+  )
+}
+
+export function saveCurrentWorkspace(state: AppState): void {
+  saveConfig({
+    ...loadConfig(),
+    customCommands: state.customCommands,
+    gitPanelRatio: state.sidebar.gitPanelRatio,
+    gitPanelVisible: state.sidebar.gitPanelVisible,
+  })
+  saveSessionCatalog(
+    buildSessionsWithCurrentSnapshot(state.sessions, state.currentSessionId, state)
+  )
+}

package/src/ui/components/create-session-modal.tsx

@@ -0,0 +1,100 @@
+import type { DirectoryResult } from '../../state/types'
+
+import { abbreviatePath } from '../path-format'
+import { theme } from '../theme'
+import { uiTokens } from '../ui-tokens'
+import { InputField } from './input-field'
+import { ListItem } from './list-item'
+import { ModalShell } from './modal-shell'
+
+function getDirectoryResultIcon(result: DirectoryResult): string {
+  if (result.type === 'worktree') {
+    return '\u{e728}'
+  }
+
+  if (result.type === 'workspace') {
+    return '\u{f07c}'
+  }
+
+  return '\u{e702}'
+}
+
+function getDirectoryResultColor(result: DirectoryResult): string {
+  if (result.type === 'worktree') {
+    return theme.warning
+  }
+
+  if (result.type === 'workspace') {
+    return theme.accentAlt
+  }
+
+  return theme.accent
+}
+
+interface CreateSessionModalProps {
+  activeField: 'directory' | 'name'
+  directoryQuery: string
+  sessionName: string
+  results: DirectoryResult[]
+  selectedIndex: number
+  pendingProjectPath: string | null
+}
+
+export function CreateSessionModal({
+  activeField,
+  directoryQuery,
+  pendingProjectPath,
+  results,
+  selectedIndex,
+  sessionName,
+}: CreateSessionModalProps) {
+  const dirActive = activeField === 'directory'
+  const nameActive = activeField === 'name'
+
+  return (
+    <ModalShell
+      title="Create session"
+      help="Tab switch field. Ctrl+n/p nav. Esc cancel."
+      width={uiTokens.modalWidth.xl}
+    >
+      <box flexDirection="column">
+        <text fg={dirActive ? theme.text : theme.textMuted}>Search projects</text>
+        <InputField
+          active={dirActive}
+          value={
+            pendingProjectPath && !dirActive ? abbreviatePath(pendingProjectPath) : directoryQuery
+          }
+        />
+      </box>
+
+      {dirActive && results.length === 0 && directoryQuery.length > 0 ? (
+        <text fg={theme.textMuted}>No matches</text>
+      ) : null}
+
+      {dirActive
+        ? results.map((result, index) => {
+            const active = index === selectedIndex
+            return (
+              <ListItem
+                key={result.path}
+                active={active}
+                leading={
+                  <text fg={getDirectoryResultColor(result)}>{getDirectoryResultIcon(result)}</text>
+                }
+                title={
+                  <text fg={active ? theme.text : theme.textMuted}>
+                    {abbreviatePath(result.path)}
+                  </text>
+                }
+              />
+            )
+          })
+        : null}
+
+      <box flexDirection="column">
+        <text fg={nameActive ? theme.text : theme.textMuted}>Session name</text>
+        <InputField active={nameActive} value={sessionName} />
+      </box>
+    </ModalShell>
+  )
+}