| 123456789101112131415161718192021222324252627282930313233343536 |
- import { appendFile } from 'node:fs/promises'
- import { join } from 'node:path'
- import type { Context } from 'cordis'
- import type { Session, SessionEvent } from '@deepseek-ai/dsh-session'
- export const name = 'session-jsonl'
- export const inject = ['sessions']
- /**
- * Minimal persistence plugin: buffers session events (write-behind) and
- * drains to a JSONL file at every `session/flush` checkpoint — the pattern a
- * real JSONL/sqlite persistence plugin would follow.
- */
- export function apply(ctx: Context) {
- const buffers = new Map<Session, SessionEvent[]>()
- const path = (session: Session) => join(import.meta.dirname, '..', `${session.id}.jsonl`)
- ctx.on('session/event', (session, event) => {
- let buffer = buffers.get(session)
- if (!buffer) buffers.set(session, buffer = [])
- buffer.push(event)
- })
- const flush = async (session: Session) => {
- const buffer = buffers.get(session)
- if (!buffer?.length) return
- const lines = buffer.splice(0).map(event => JSON.stringify(event) + '\n').join('')
- await appendFile(path(session), lines)
- }
- ctx.on('session/flush', flush)
- ctx.effect(() => () => {
- // drain remaining buffers on dispose
- for (const session of buffers.keys()) void flush(session)
- }, 'session-jsonl')
- }
|