85 lines
2.6 KiB
TypeScript
85 lines
2.6 KiB
TypeScript
/**
|
||
* 会话存储（A1 阶段内存实现）。
|
||
*
|
||
* ADR-0003：个人会话只读，写操作必须在项目会话内——这里在创建时就把住：
|
||
* scope=project 必须带 projectId，scope=personal 不允许带。
|
||
* A3.7 落库后换成持久层；主键按 ADR-0007 用应用侧生成。
|
||
*/
|
||
import { randomUUID } from 'node:crypto';
|
||
|
||
import type { ChatMessage, SessionSummary } from '@aps/contracts';
|
||
|
||
import { badRequest, notFound } from './errors.js';
|
||
import { EventBus } from './events.js';
|
||
|
||
export type RunState = 'idle' | 'running' | 'cancelling' | 'cancelled';
|
||
|
||
export interface RunHandle {
|
||
runId: string;
|
||
abort: AbortController;
|
||
}
|
||
|
||
export interface SessionRecord {
|
||
summary: SessionSummary;
|
||
messages: ChatMessage[];
|
||
bus: EventBus;
|
||
state: RunState;
|
||
runs: Map<string, RunHandle>;
|
||
/** 幂等键 → 已受理的消息与运行（同一 key 重试不产生第二个 run） */
|
||
accepted: Map<string, { messageId: string; runId: string }>;
|
||
}
|
||
|
||
export interface CreateSessionInput {
|
||
scope: 'personal' | 'project';
|
||
projectId?: string | undefined;
|
||
title?: string | undefined;
|
||
}
|
||
|
||
export function newId(prefix: string): string {
|
||
return `${prefix}-${randomUUID()}`;
|
||
}
|
||
|
||
export class SessionStore {
|
||
private readonly sessions = new Map<string, SessionRecord>();
|
||
|
||
create(input: CreateSessionInput): SessionRecord {
|
||
if (input.scope === 'project' && !input.projectId) throw badRequest('项目会话必须带 projectId（ADR-0003）');
|
||
if (input.scope === 'personal' && input.projectId) throw badRequest('个人会话不能带 projectId');
|
||
|
||
const summary: SessionSummary = {
|
||
sessionId: newId('s'),
|
||
projectId: input.scope === 'project' ? (input.projectId ?? null) : null,
|
||
scope: input.scope,
|
||
status: 'active',
|
||
title: input.title ?? (input.scope === 'project' ? '新会话' : '个人会话'),
|
||
createdAt: Date.now(),
|
||
};
|
||
const record: SessionRecord = {
|
||
summary,
|
||
messages: [],
|
||
bus: new EventBus(),
|
||
state: 'idle',
|
||
runs: new Map(),
|
||
accepted: new Map(),
|
||
};
|
||
this.sessions.set(summary.sessionId, record);
|
||
return record;
|
||
}
|
||
|
||
get(sessionId: string): SessionRecord | undefined {
|
||
return this.sessions.get(sessionId);
|
||
}
|
||
|
||
require(sessionId: string): SessionRecord {
|
||
const record = this.get(sessionId);
|
||
if (!record) throw notFound(`会话 ${sessionId}`);
|
||
return record;
|
||
}
|
||
|
||
list(projectId?: string): SessionSummary[] {
|
||
return [...this.sessions.values()]
|
||
.filter((record) => (projectId ? record.summary.projectId === projectId : true))
|
||
.map((record) => record.summary)
|
||
.sort((a, b) => b.createdAt - a.createdAt);
|
||
}
|
||
} |