@jaychang1989/dsh-webchat 0.3.0 → 0.4.0

package/src/routes.ts

@@ -1,307 +0,0 @@
-/**
- * The /api/dsh-webchat route family: engine state, browser login control,
- * chat operations (new chat / send / stop / switch model), transcript
- * export, and the harness transfer that seeds a new session with a web
- * transcript. Every route carries the same loopback-only trust fence the
- * dsh-ssh plugin uses — these endpoints drive a browser and create sessions,
- * so LAN-exposed deployments must not serve them.
- */
-
-import type { IncomingMessage, ServerResponse } from 'node:http'
-import type { Context } from '@deepseek-ai/cordis'
-import type { WebRoute } from '@deepseek-ai/dsh-host-webserver'
-import type { DeepSeekWebEngine } from './engine/engine.ts'
-import type { TranscriptStore } from './store.ts'
-import type { WebChatTranscript } from './protocol.ts'
-import { exportTranscriptFile, transferToHarnessSession } from './transfer.ts'
-import type { DistillConfig } from './transfer.ts'
-
-/** Cap on JSON request bodies (chat ops are small). */
-const MAX_JSON_BODY_BYTES = 64 * 1024
-
-/** Loopback-only trust fence (mirrors dsh-ssh). */
-function isLoopbackRequest(req: IncomingMessage): boolean {
-  const host = req.headers.host ?? ''
-  const address = req.socket.remoteAddress ?? ''
-  const loopbackHost = host.startsWith('127.0.0.1') || host.startsWith('localhost') || host.startsWith('[::1]')
-  const loopbackAddr = address === '127.0.0.1' || address === '::1' || address === '::ffff:127.0.0.1' || address === undefined
-  return loopbackHost && loopbackAddr
-}
-
-/** One JSON response. */
-function writeJson(res: ServerResponse, status: number, body: unknown): void {
-  const payload = JSON.stringify(body)
-  res.writeHead(status, { 'content-type': 'application/json; charset=utf-8', 'referrer-policy': 'no-referrer' })
-  res.end(payload)
-}
-
-/** Read a JSON request body. */
-async function readJsonBody(req: IncomingMessage): Promise<Record<string, unknown> | undefined> {
-  const chunks: Buffer[] = []
-  let size = 0
-  for await (const chunk of req) {
-    const buffer = chunk as Buffer
-    size += buffer.length
-    if (size > MAX_JSON_BODY_BYTES) return undefined
-    chunks.push(buffer)
-  }
-  try {
-    const parsed: unknown = JSON.parse(Buffer.concat(chunks).toString('utf8'))
-    return typeof parsed === 'object' && parsed !== null ? parsed as Record<string, unknown> : undefined
-  } catch {
-    return undefined
-  }
-}
-
-function stringField(body: Record<string, unknown> | undefined, name: string): string | undefined {
-  const value = body?.[name]
-  return typeof value === 'string' && value !== '' ? value : undefined
-}
-
-/** Route family dependencies. */
-export interface WebChatRoutesDeps {
-  ctx: Context
-  engine: DeepSeekWebEngine
-  store: TranscriptStore
-  distill: DistillConfig
-}
-
-/** Build every /api/dsh-webchat route. */
-export function makeRoutes(deps: WebChatRoutesDeps): WebRoute[] {
-  const { ctx, engine, store, distill } = deps
-
-  const guard = (req: IncomingMessage, res: ServerResponse): boolean => {
-    if (isLoopbackRequest(req)) return true
-    writeJson(res, 403, { ok: false, error: 'loopback only' })
-    return false
-  }
-
-  const stateView = async (): Promise<Record<string, unknown>> => {
-    const status = await engine.status()
-    return {
-      ok: true,
-      engine: status.engine,
-      engineError: status.engineError,
-      loggedIn: status.loggedIn,
-      pageUrl: status.pageUrl,
-      deepThink: status.deepThink,
-      search: status.search,
-      busy: status.busy,
-      lastError: status.lastError,
-      lastErrorCode: status.lastErrorCode,
-      activeChatId: store.activeChat()?.id,
-      chats: store.list(),
-    }
-  }
-
-  return [
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/state',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        writeJson(res, 200, await stateView())
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/open-login',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const result = await engine.openLoginWindow()
-        writeJson(res, result.ok ? 200 : 500, { ok: result.ok, error: result.error })
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/close-browser',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        await engine.disposeBrowser()
-        writeJson(res, 200, { ok: true })
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/new-chat',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const result = await engine.newChat()
-        writeJson(res, result.ok ? 200 : 500, result)
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/send',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const body = await readJsonBody(req)
-        const text = stringField(body, 'text')
-        if (text === undefined) {
-          writeJson(res, 400, { ok: false, error: '缺少 text 字段' })
-          return
-        }
-        // GUI sends resolve immediately; the reply streams in the background
-        // into the transcript and the panel polls /state for live updates.
-        const images = Array.isArray(body?.['images'])
-          ? (body['images'] as unknown[]).filter(value => typeof value === 'string').map(value => value as string)
-          : undefined
-        const result = await engine.send(text, false, images)
-        writeJson(res, result.ok ? 200 : 500, result)
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/stop',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        await engine.stop()
-        writeJson(res, 200, { ok: true })
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/deep-think',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const body = await readJsonBody(req)
-        const enabled = typeof body?.['enabled'] === 'boolean' ? body['enabled'] : undefined
-        if (enabled === undefined) {
-          writeJson(res, 400, { ok: false, error: '缺少 enabled 字段' })
-          return
-        }
-        const result = await engine.setDeepThink(enabled)
-        writeJson(res, result.ok ? 200 : 500, result)
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/search',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const body = await readJsonBody(req)
-        const enabled = typeof body?.['enabled'] === 'boolean' ? body['enabled'] : undefined
-        if (enabled === undefined) {
-          writeJson(res, 400, { ok: false, error: '缺少 enabled 字段' })
-          return
-        }
-        const result = await engine.setSearch(enabled)
-        writeJson(res, result.ok ? 200 : 500, result)
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/transfer',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const body = await readJsonBody(req)
-        const chatId = stringField(body, 'chatId') ?? store.activeChat()?.id
-        const cwd = stringField(body, 'cwd')
-        const workspaceId = stringField(body, 'workspaceId')
-        const targetSessionId = stringField(body, 'targetSessionId')
-        const mode = body?.['mode'] === 'raw' ? 'raw' : body?.['mode'] === 'distill' ? 'distill' : undefined
-        const transcript: WebChatTranscript | undefined = chatId === undefined ? undefined : store.getChat(chatId)
-        if (transcript === undefined) {
-          writeJson(res, 404, { ok: false, error: '找不到该对话记录' })
-          return
-        }
-        try {
-          const workspace = workspaceId === undefined ? undefined : { workspaceId }
-          const { sessionId, distilled, attached, workspaceId: attachedWorkspaceId } = await transferToHarnessSession(ctx, { transcript, cwd, workspace, targetSessionId }, distill, mode)
-          writeJson(res, 200, { ok: true, sessionId, distilled, attached, continued: targetSessionId !== undefined, workspaceId: attachedWorkspaceId })
-        } catch (error) {
-          writeJson(res, 500, { ok: false, error: String(error) })
-        }
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/export',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const body = await readJsonBody(req)
-        const chatId = stringField(body, 'chatId') ?? store.activeChat()?.id
-        const cwd = stringField(body, 'cwd')
-        const transcript: WebChatTranscript | undefined = chatId === undefined ? undefined : store.getChat(chatId)
-        if (transcript === undefined) {
-          writeJson(res, 404, { ok: false, error: '找不到该对话记录' })
-          return
-        }
-        try {
-          const { filePath } = exportTranscriptFile({ transcript, cwd })
-          writeJson(res, 200, { ok: true, filePath })
-        } catch (error) {
-          writeJson(res, 500, { ok: false, error: String(error) })
-        }
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/web-chats',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const web = await engine.listWebConversations()
-        const localTitles = new Set(store.titles())
-        const missing = web.filter(item => !localTitles.has(item.title)).map(item => item.title)
-        writeJson(res, 200, { ok: true, web, missing })
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/recover',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const body = await readJsonBody(req)
-        const title = stringField(body, 'title')
-        if (title === undefined) {
-          writeJson(res, 400, { ok: false, error: '缺少 title 字段' })
-          return
-        }
-        const result = await engine.recoverWebConversation(title)
-        writeJson(res, result.ok ? 200 : 500, result)
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/rename',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const body = await readJsonBody(req)
-        const chatId = stringField(body, 'chatId')
-        const title = stringField(body, 'title')
-        if (chatId === undefined || title === undefined) {
-          writeJson(res, 400, { ok: false, error: '缺少 chatId 或 title 字段' })
-          return
-        }
-        const renamed = store.renameChat(chatId, title)
-        if (renamed === undefined) writeJson(res, 404, { ok: false, error: '找不到该对话记录' })
-        else writeJson(res, 200, { ok: true })
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/delete',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const body = await readJsonBody(req)
-        const chatId = stringField(body, 'chatId')
-        if (chatId === undefined) {
-          writeJson(res, 400, { ok: false, error: '缺少 chatId 字段' })
-          return
-        }
-        const deleted = store.deleteChat(chatId)
-        if (!deleted) writeJson(res, 404, { ok: false, error: '找不到该对话记录' })
-        else writeJson(res, 200, { ok: true })
-      },
-    },
-    {
-      kind: 'exact',
-      path: '/api/dsh-webchat/clear',
-      handler: async (req, res) => {
-        if (!guard(req, res)) return
-        const count = store.clearAllChats()
-        writeJson(res, 200, { ok: true, count })
-      },
-    },
-  ]
-}

package/src/store.ts

@@ -1,217 +0,0 @@
-/**
- * Local transcript store. Chats are persisted as one JSON file under the
- * plugin data dir (~/.dsh/dsh-webchat/transcripts.json by default) so web
- * conversations survive restarts and can be transferred into harness mode
- * at any time. Atomic writes (tmp + rename) keep a crash from corrupting
- * history.
- */
-
-import { mkdirSync, readFileSync, renameSync, writeFileSync } from 'node:fs'
-import { dirname, join } from 'node:path'
-import { randomUUID } from 'node:crypto'
-import type { WebChatMessage, WebChatTranscript } from './protocol.ts'
-
-/** Default plugin data directory (tests inject a sandbox root). */
-export function defaultDataDir(): string {
-  const home = process.env.DSH_HOME ?? process.env.HOME ?? '.'
-  return join(home, '.dsh', 'dsh-webchat')
-}
-
-export interface TranscriptStoreOptions {
-  /** Root data dir; the transcripts file lives directly under it. */
-  dataDir?: string
-}
-
-interface StoreFile {
-  version: 1
-  activeChatId?: string
-  chats: WebChatTranscript[]
-}
-
-/**
- * JSON-file transcript store. All mutations are synchronous and persisted
- * immediately (chats are small); the engine and routes use this single
- * instance so GUI and agent tools always see the same history.
- */
-export class TranscriptStore {
-  readonly dataDir: string
-  private readonly file: string
-  private chats: WebChatTranscript[]
-  private activeChatId: string | undefined
-
-  constructor(options: TranscriptStoreOptions = {}) {
-    this.dataDir = options.dataDir ?? defaultDataDir()
-    this.file = join(this.dataDir, 'transcripts.json')
-    const loaded = this.read()
-    this.chats = loaded.chats
-    this.activeChatId = loaded.activeChatId
-    if (this.activeChatId !== undefined && !this.chats.some(chat => chat.id === this.activeChatId)) {
-      this.activeChatId = this.chats.at(-1)?.id
-    }
-  }
-
-  private read(): StoreFile {
-    try {
-      const parsed = JSON.parse(readFileSync(this.file, 'utf8')) as Partial<StoreFile>
-      const chats = Array.isArray(parsed.chats) ? parsed.chats : []
-      return {
-        version: 1,
-        activeChatId: typeof parsed.activeChatId === 'string' ? parsed.activeChatId : undefined,
-        chats: chats.filter(chat => typeof chat?.id === 'string' && Array.isArray(chat.messages)),
-      }
-    } catch {
-      return { version: 1, chats: [] }
-    }
-  }
-
-  private persist(): void {
-    mkdirSync(this.dataDir, { recursive: true, mode: 0o700 })
-    const payload: StoreFile = { version: 1, activeChatId: this.activeChatId, chats: this.chats }
-    const tmp = `${this.file}.tmp`
-    writeFileSync(tmp, JSON.stringify(payload, null, 2), { mode: 0o600 })
-    renameSync(tmp, this.file)
-  }
-
-  /** Create a fresh chat and make it active. */
-  createChat(model: string): WebChatTranscript {
-    const now = Date.now()
-    const chat: WebChatTranscript = {
-      id: `chat-${now.toString(36)}-${randomUUID().slice(0, 6)}`,
-      title: '新的对话',
-      createdAt: now,
-      updatedAt: now,
-      model,
-      messages: [],
-      streaming: false,
-    }
-    this.chats.unshift(chat)
-    this.activeChatId = chat.id
-    this.persist()
-    return chat
-  }
-
-  /** All chats, newest first. */
-  list(): WebChatTranscript[] {
-    return [...this.chats]
-  }
-
-  /** Local transcript titles (for matching against the web sidebar). */
-  titles(): string[] {
-    return this.chats.map(chat => chat.title)
-  }
-
-  /**
-   * Import a recovered web conversation as a local transcript. Dedups by exact
-   * title (web conversations with the same title map to the existing chat), so
-   * re-syncing is idempotent. Returns the chat plus whether it was newly created.
-   */
-  importTranscript(input: { title: string; model: string; messages: WebChatMessage[] }): { chat: WebChatTranscript; created: boolean } {
-    const cleanTitle = input.title.trim().replace(/\s+/g, ' ').slice(0, 80) || '新的对话'
-    const existing = this.chats.find(chat => chat.title === cleanTitle)
-    if (existing !== undefined) {
-      this.activeChatId = existing.id
-      return { chat: existing, created: false }
-    }
-    const now = Date.now()
-    const chat: WebChatTranscript = {
-      id: `chat-${now.toString(36)}-${randomUUID().slice(0, 6)}`,
-      title: cleanTitle,
-      createdAt: now,
-      updatedAt: now,
-      model: input.model,
-      messages: input.messages,
-      streaming: false,
-    }
-    this.chats.unshift(chat)
-    this.activeChatId = chat.id
-    this.persist()
-    return { chat, created: true }
-  }
-
-  /** The active chat, or undefined when none exists yet. */
-  activeChat(): WebChatTranscript | undefined {
-    if (this.activeChatId === undefined) return undefined
-    return this.chats.find(chat => chat.id === this.activeChatId)
-  }
-
-  /** Read one chat by id. */
-  getChat(id: string): WebChatTranscript | undefined {
-    return this.chats.find(chat => chat.id === id)
-  }
-
-  /** Pick the active chat, creating one if none exists. */
-  ensureActiveChat(model: string): WebChatTranscript {
-    return this.activeChat() ?? this.createChat(model)
-  }
-
-  /** Set which chat is active. */
-  setActiveChat(id: string): boolean {
-    if (!this.chats.some(chat => chat.id === id)) return false
-    this.activeChatId = id
-    this.persist()
-    return true
-  }
-
-  /** Mutate the active (or named) chat and persist. */
-  private update(id: string, mutate: (chat: WebChatTranscript) => void): WebChatTranscript | undefined {
-    const chat = this.chats.find(candidate => candidate.id === id)
-    if (chat === undefined) return undefined
-    mutate(chat)
-    chat.updatedAt = Date.now()
-    this.persist()
-    return chat
-  }
-
-  /** Append a message to a chat. */
-  appendMessage(id: string, message: WebChatMessage): WebChatTranscript | undefined {
-    return this.update(id, chat => {
-      chat.messages.push(message)
-      if (message.role === 'assistant') chat.streaming = message.streaming ?? false
-    })
-  }
-
-  /** Replace (or insert) one message by id — used for streaming updates. */
-  upsertMessage(id: string, message: WebChatMessage): WebChatTranscript | undefined {
-    return this.update(id, chat => {
-      const index = chat.messages.findIndex(candidate => candidate.id === message.id)
-      if (index >= 0) chat.messages[index] = message
-      else chat.messages.push(message)
-      if (message.role === 'assistant') chat.streaming = message.streaming ?? false
-    })
-  }
-
-  /** Mark the chat's streaming flag (assistant reply started/stopped). */
-  setStreaming(id: string, streaming: boolean, model?: string): WebChatTranscript | undefined {
-    return this.update(id, chat => {
-      chat.streaming = streaming
-      if (model !== undefined) chat.model = model
-    })
-  }
-
-  /** Rename a chat (used to pin a meaningful title after the first exchange). */
-  renameChat(id: string, title: string): WebChatTranscript | undefined {
-    const clean = title.trim().replace(/\s+/g, ' ').slice(0, 80)
-    if (clean === '') return undefined
-    return this.update(id, chat => { chat.title = clean })
-  }
-
-  /** Delete one chat; a deleted active chat falls back to the newest remaining. */
-  deleteChat(id: string): boolean {
-    const before = this.chats.length
-    this.chats = this.chats.filter(chat => chat.id !== id)
-    if (this.chats.length === before) return false
-    if (this.activeChatId === id) this.activeChatId = this.chats.at(0)?.id
-    this.persist()
-    return true
-  }
-
-  /** Delete every chat; returns the number removed. */
-  clearAllChats(): number {
-    const count = this.chats.length
-    if (count === 0) return 0
-    this.chats = []
-    this.activeChatId = undefined
-    this.persist()
-    return count
-  }
-}