37 lines
1.2 KiB
TypeScript
37 lines
1.2 KiB
TypeScript
|
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')
|
||
|
|
}
|