egaki 0.6.0 → 0.7.1

package/src/vite/store.test.ts

@@ -0,0 +1,202 @@
+/**
+ * Tests for the centralized egaki client store.
+ * Pure state transitions — no React, no mocks, just data in and data out.
+ */
+
+import { describe, test, expect, beforeEach } from 'vitest'
+import {
+  egakiStore,
+  selectMediaDurations,
+  selectGenerationStatus,
+} from './store.ts'
+import {
+  reportSectionDuration,
+  clearSectionDuration,
+  resetSectionDurations,
+} from './media-duration-store.ts'
+
+beforeEach(() => {
+  egakiStore.setState({
+    modules: {},
+    sectionReports: new Map(),
+    serverGenerationStatus: null,
+    serverGenerationEntries: [],
+    serverGenerationErrors: [],
+  })
+})
+
+// ---------------------------------------------------------------------------
+// Modules
+// ---------------------------------------------------------------------------
+
+describe('modules', () => {
+  test('setState updates modules', () => {
+    const mods = { './foo.tsx': { default: () => null } }
+    egakiStore.setState({ modules: mods })
+    expect(egakiStore.getState().modules).toBe(mods)
+  })
+
+  test('setState replaces previous modules', () => {
+    egakiStore.setState({ modules: { './a.tsx': { default: 1 } } })
+    const next = { './b.tsx': { default: 2 } }
+    egakiStore.setState({ modules: next })
+    expect(egakiStore.getState().modules).toBe(next)
+    expect(egakiStore.getState().modules['./a.tsx']).toBeUndefined()
+  })
+})
+
+// ---------------------------------------------------------------------------
+// Section duration reports
+// ---------------------------------------------------------------------------
+
+describe('sectionReports', () => {
+  test('reportSectionDuration adds a report', () => {
+    reportSectionDuration(0, 'inst-1', 5.5)
+    const durations = selectMediaDurations(egakiStore.getState())
+    expect(durations).toMatchInlineSnapshot(`
+      {
+        "0": 5.5,
+      }
+    `)
+  })
+
+  test('per-section max is derived from multiple instances', () => {
+    reportSectionDuration(0, 'inst-1', 3)
+    reportSectionDuration(0, 'inst-2', 7)
+    reportSectionDuration(0, 'inst-3', 5)
+    const durations = selectMediaDurations(egakiStore.getState())
+    expect(durations['0']).toBe(7)
+  })
+
+  test('multiple sections tracked independently', () => {
+    reportSectionDuration(0, 'a', 2)
+    reportSectionDuration(1, 'b', 4)
+    reportSectionDuration(2, 'c', 6)
+    const durations = selectMediaDurations(egakiStore.getState())
+    expect(durations).toMatchInlineSnapshot(`
+      {
+        "0": 2,
+        "1": 4,
+        "2": 6,
+      }
+    `)
+  })
+
+  test('duplicate report with same value is a no-op', () => {
+    reportSectionDuration(0, 'inst-1', 5)
+    const before = egakiStore.getState().sectionReports
+    reportSectionDuration(0, 'inst-1', 5)
+    expect(egakiStore.getState().sectionReports).toBe(before)
+  })
+
+  test('clearSectionDuration removes an instance report', () => {
+    reportSectionDuration(0, 'a', 3)
+    reportSectionDuration(0, 'b', 7)
+    clearSectionDuration(0, 'b')
+    const durations = selectMediaDurations(egakiStore.getState())
+    expect(durations['0']).toBe(3)
+  })
+
+  test('clearSectionDuration removes section entry when last instance clears', () => {
+    reportSectionDuration(0, 'a', 3)
+    clearSectionDuration(0, 'a')
+    const durations = selectMediaDurations(egakiStore.getState())
+    expect(durations).toMatchInlineSnapshot(`{}`)
+  })
+
+  test('clearSectionDuration is a no-op for missing instance', () => {
+    reportSectionDuration(0, 'a', 3)
+    const before = egakiStore.getState().sectionReports
+    clearSectionDuration(0, 'nonexistent')
+    expect(egakiStore.getState().sectionReports).toBe(before)
+  })
+
+  test('resetSectionDurations clears all reports', () => {
+    reportSectionDuration(0, 'a', 3)
+    reportSectionDuration(1, 'b', 5)
+    resetSectionDurations()
+    expect(egakiStore.getState().sectionReports.size).toBe(0)
+    expect(selectMediaDurations(egakiStore.getState())).toMatchInlineSnapshot(`{}`)
+  })
+
+  test('resetSectionDurations is a no-op when already empty', () => {
+    const before = egakiStore.getState().sectionReports
+    resetSectionDurations()
+    expect(egakiStore.getState().sectionReports).toBe(before)
+  })
+
+  test('ignores invalid inputs', () => {
+    reportSectionDuration(-1, 'a', 5)
+    reportSectionDuration(0, 'a', NaN)
+    reportSectionDuration(0, 'a', 0)
+    reportSectionDuration(0, 'a', -1)
+    expect(egakiStore.getState().sectionReports.size).toBe(0)
+  })
+})
+
+// ---------------------------------------------------------------------------
+// Generation status (server-driven)
+// ---------------------------------------------------------------------------
+
+describe('serverGenerationStatus', () => {
+  test('setState tracks server generation status', () => {
+    egakiStore.setState({
+      serverGenerationStatus: { counts: { image: 1 }, total: 1 },
+    })
+    const status = selectGenerationStatus(egakiStore.getState())
+    expect(status).toMatchInlineSnapshot(`
+      {
+        "counts": {
+          "image": 1,
+        },
+        "total": 1,
+      }
+    `)
+  })
+
+  test('multiple generation types counted separately', () => {
+    egakiStore.setState({
+      serverGenerationStatus: { counts: { image: 2, video: 1, audio: 1 }, total: 4 },
+    })
+    const status = selectGenerationStatus(egakiStore.getState())
+    expect(status).toMatchInlineSnapshot(`
+      {
+        "counts": {
+          "audio": 1,
+          "image": 2,
+          "video": 1,
+        },
+        "total": 4,
+      }
+    `)
+  })
+
+  test('setting status to null clears generation tracking', () => {
+    egakiStore.setState({
+      serverGenerationStatus: { counts: { image: 1 }, total: 1 },
+    })
+    egakiStore.setState({ serverGenerationStatus: null })
+    const status = selectGenerationStatus(egakiStore.getState())
+    expect(status).toBeNull()
+  })
+
+  test('returns null when no generations active', () => {
+    expect(selectGenerationStatus(egakiStore.getState())).toBeNull()
+  })
+
+  test('tracks any namespace', () => {
+    egakiStore.setState({
+      serverGenerationStatus: { counts: { audio: 1, transcription: 2 }, total: 3 },
+    })
+    const status = selectGenerationStatus(egakiStore.getState())
+    expect(status).toMatchInlineSnapshot(`
+      {
+        "counts": {
+          "audio": 1,
+          "transcription": 2,
+        },
+        "total": 3,
+      }
+    `)
+  })
+})

package/src/vite/store.ts

@@ -0,0 +1,115 @@
+/**
+ * Centralized client-side state store (zustand/vanilla).
+ *
+ * Pure vanilla store with no React imports. State slices:
+ *
+ * 1. **modules** — eagerly-imported user .tsx/.ts modules, updated via HMR
+ * 2. **sectionReports** — per-section effective media durations reported by
+ *    Audio/Video components (ephemeral, not persisted)
+ * 3. **serverGenerationStatus** — server-reported generation progress from
+ *    the /api/generation-progress SSE stream, for the toolbar status indicator
+ *
+ * React hooks live in store-hooks.ts ('use client') to keep this file
+ * importable from any environment (RSC, SSR, client) without pulling in React.
+ */
+
+import { createStore } from 'zustand/vanilla'
+import { subscribeWithSelector } from 'zustand/middleware'
+
+// ---------------------------------------------------------------------------
+// Types
+// ---------------------------------------------------------------------------
+
+/** Same as safe-mdx's EagerModules: Record<string, Record<string, any>> */
+type EagerModules = Record<string, Record<string, any>>
+
+export interface GenerationStatus {
+  /** Count of active generations per namespace (e.g. { image: 2, video: 1 }) */
+  counts: Record<string, number>
+  total: number
+}
+
+/** Individual generation entry from the server progress stream. */
+export interface GenerationProgressEntry {
+  key: string
+  namespace: string
+  label: string
+  model?: string
+  startedAt: number
+  elapsedMs: number
+  params: Record<string, unknown>
+}
+
+/** A generation that failed, reported via the progress stream. */
+export interface GenerationError {
+  key: string
+  namespace: string
+  label: string
+  model?: string
+  error: string
+  durationMs: number
+}
+
+export interface EgakiClientState {
+  /** User-imported modules from virtual:egaki-modules. Updated by HMR. */
+  modules: EagerModules
+
+  /**
+   * Per-instance section duration reports (ephemeral).
+   * sectionIndex → Map<instanceId, effectiveSeconds>
+   * Each Audio/Video component reports its effective playback duration.
+   */
+  sectionReports: Map<number, Map<string, number>>
+
+  /** Server-reported generation progress from /api/generation-progress SSE stream. */
+  serverGenerationStatus: GenerationStatus | null
+
+  /** Detailed generation entries from the server progress stream. */
+  serverGenerationEntries: GenerationProgressEntry[]
+
+  /** Recent generation errors from the server progress stream.
+   *  Errors auto-clear after ERROR_DISPLAY_DURATION_MS. */
+  serverGenerationErrors: GenerationError[]
+}
+
+// ---------------------------------------------------------------------------
+// Store
+// ---------------------------------------------------------------------------
+
+export const egakiStore = createStore<EgakiClientState>()(
+  subscribeWithSelector(() => ({
+    modules: {} as EagerModules,
+    sectionReports: new Map(),
+    serverGenerationStatus: null as GenerationStatus | null,
+    serverGenerationEntries: [] as GenerationProgressEntry[],
+    serverGenerationErrors: [] as GenerationError[],
+  })),
+)
+
+// ---------------------------------------------------------------------------
+// Selectors — derive values from state
+// ---------------------------------------------------------------------------
+
+/**
+ * Derive per-section max media durations.
+ * Returns Record<string, number> (section-index-as-string → seconds).
+ */
+export function selectMediaDurations(state: EgakiClientState): Record<string, number> {
+  const obj: Record<string, number> = {}
+  for (const [idx, reports] of state.sectionReports) {
+    if (reports.size === 0) continue
+    let max = 0
+    for (const dur of reports.values()) {
+      if (dur > max) max = dur
+    }
+    if (max > 0) obj[String(idx)] = max
+  }
+  return obj
+}
+
+/** Generation status from the server progress stream, or null when idle. */
+export function selectGenerationStatus(state: EgakiClientState): GenerationStatus | null {
+  return state.serverGenerationStatus
+}
+
+

package/src/vite/styles.css

@@ -0,0 +1,21 @@
+@import 'tailwindcss';
+@source ".";
+
+html,
+body,
+#root {
+  height: 100%;
+  margin: 0;
+  background: #000;
+  color: #fafafa;
+  overflow: clip;
+  scrollbar-width: thin;
+  scrollbar-color: rgba(255, 255, 255, 0.15) transparent;
+}
+
+/* Tweakpane sidebar: remove rounded corners and shadow so it sits flush */
+.egaki-sidebar .tp-rotv {
+  border-radius: 0;
+  box-shadow: none;
+  background-color: transparent;
+}