feat(web): add plan mode controls

2026-07-24 12:55:23 +08:00
parent bc63b5fe00
commit 9a69feeec0
41 changed files with 1047 additions and 27 deletions
@@ -0,0 +1,58 @@
+import { Context } from 'cordis'
+import { describe, expect, it, vi } from 'vitest'
+import { SlotsService } from '@deepseek-ai/dsh-client-runtime/client'
+import type { SessionId } from '@deepseek-ai/dsh-client-runtime/client'
+import { PlanModeControl } from '../src/client/PlanModeControl.tsx'
+import type { PlanModeControlInjected } from '../src/client/index.ts'
+import { apply, inject } from '../src/client/index.ts'
+
+const SID = 's-plan' as SessionId
+
+async function bench() {
+  const ctx = new Context()
+  await ctx.plugin(SlotsService).await()
+  const slots = ctx.get('slots') as SlotsService
+  slots.register({
+    name: 'root',
+    children: { 'conversation.composer.controls': { kind: 'list', scope: 'session' } },
+  } as never, () => null)
+  const setPlanMode = vi.fn(() => Promise.resolve({ ok: true, value: { active: false, pending: true } }))
+  ctx.provide('sessions', { manager: { get: () => ({ setPlanMode }) } })
+  ctx.provide('conversation', {})
+  return { ctx, slots, setPlanMode }
+}
+
+describe('ui-plan browser apply', () => {
+  it('declares every service it binds', () => {
+    expect(inject).toEqual(['slots', 'sessions', 'conversation'])
+  })
+
+  it('fails loud when conversation did not declare the controls slot', async () => {
+    const ctx = new Context()
+    await ctx.plugin(SlotsService).await()
+    ctx.provide('sessions', {})
+    ctx.provide('conversation', {})
+    await expect(ctx.plugin({ inject: [...inject], apply }))
+      .rejects.toThrow(/slot "conversation.composer.controls" is not declared/)
+  })
+
+  it('registers the control, bridges host results, and unregisters on teardown', async () => {
+    const b = await bench()
+    const fiber = b.ctx.plugin({ inject: [...inject], apply })
+    await fiber.await()
+    const entry = b.slots.entries('conversation.composer.controls')[0]!
+    expect(entry.component).toBe(PlanModeControl)
+    expect(entry.options).toMatchObject({ id: 'plan-mode', order: 10 })
+    const injected = (entry.inject as unknown as (id: SessionId) => PlanModeControlInjected)(SID)
+    await expect(injected.setPlanMode(true)).resolves.toBeNull()
+    expect(b.setPlanMode).toHaveBeenCalledWith(true)
+
+    b.setPlanMode.mockResolvedValueOnce({
+      ok: false, error: { code: 'session-not-found', message: 'gone', details: {} },
+    } as never)
+    await expect(injected.setPlanMode(false)).resolves.toBe('gone（session-not-found）')
+
+    await fiber.dispose()
+    expect(b.slots.entries('conversation.composer.controls')).toHaveLength(0)
+  })
+})