tavern-harness 0.1.0

package/src/core/turnLoop.ts

@@ -0,0 +1,211 @@
+import type { ChatParticipant, ChatSession, TurnOrderMode } from '../types/models';
+import { translate } from './i18n';
+import { escapeRegExp } from './strings';
+
+// ============================================================
+// 魔数命令（与 App 一致：/new 与 /pass）
+// ============================================================
+
+interface MagicCommand {
+  text: string;
+  description: string;
+}
+
+const MAGIC_COMMANDS: MagicCommand[] = [
+  { text: '/new', description: '' },
+  { text: '/pass', description: '' },
+];
+
+/** 获取命令描述（每次读取当前语言） */
+function commandDescription(text: string): string {
+  return text === '/new' ? translate('chat.cmdNew') : translate('chat.cmdPass');
+}
+
+/** 精确匹配 /new 或 /pass（有附件时不生效） */
+export function parseMagicCommand(text: string, hasAttachments: boolean): { text: string } | null {
+  if (hasAttachments) return null;
+  const trimmed = text.trim();
+  const cmd = MAGIC_COMMANDS.find((c) => c.text === trimmed);
+  return cmd ? { text: cmd.text } : null;
+}
+
+export function suggestMagicCommands(input: string): MagicCommand[] {
+  if (!input.startsWith('/') || input.includes('\n')) return [];
+  return MAGIC_COMMANDS.filter((c) => c.text.startsWith(input)).map((c) => ({
+    text: c.text,
+    description: commandDescription(c.text),
+  }));
+}
+
+// ============================================================
+// 回合循环（纯队列数学，与 App 的 TurnLoop.kt 一致）
+// ============================================================
+
+/** 初始化发言队列：全体成员（含玩家）按座位排序；RANDOM 洗牌 */
+export function initializeTurnQueue(participants: ChatParticipant[], mode: TurnOrderMode): string[] {
+  const sorted = [...participants].sort((a, b) => a.seatOrder - b.seatOrder);
+  const ids = sorted.map((p) => String(p.participantId));
+  if (mode === 'RANDOM') {
+    return shuffle(ids);
+  }
+  return ids;
+}
+
+function shuffle<T>(arr: T[]): T[] {
+  const a = [...arr];
+  for (let i = a.length - 1; i > 0; i--) {
+    const j = Math.floor(Math.random() * (i + 1));
+    [a[i], a[j]] = [a[j], a[i]];
+  }
+  return a;
+}
+
+/** 从文本中提取被 @ 提及的参与者 id（最长名称优先匹配） */
+export function mentionedParticipantIds(
+  text: string,
+  participants: ChatParticipant[],
+  speakerParticipantId: number
+): number[] {
+  const mentionables = participants
+    .filter((p) => p.participantId !== speakerParticipantId)
+    .flatMap((p) => p.kind === 'PLAYER'
+      ? ['User', '用户', '用戶'].map((name) => ({ id: p.participantId, name }))
+      : [{ id: p.participantId, name: p.displayName }])
+    .sort((a, b) => b.name.length - a.name.length);
+  const hits: Array<{ id: number; index: number; length: number }> = [];
+  for (const p of mentionables) {
+    // @名字 后跟空白或标点（中英文）
+    const re = new RegExp(`@${escapeRegExp(p.name)}(?=\\s|[，。！？,.!?]|$)`, 'g');
+    for (const m of text.matchAll(re)) {
+      hits.push({ id: p.id, index: m.index ?? 0, length: m[0].length });
+    }
+  }
+  hits.sort((a, b) => (a.index - b.index) || (b.length - a.length));
+  const found: number[] = [];
+  const occupied: Array<{ start: number; end: number }> = [];
+  for (const h of hits) {
+    const start = h.index;
+    const end = h.index + h.length;
+    const overlap = occupied.some((r) => !(end <= r.start || start >= r.end));
+    if (overlap || found.includes(h.id)) continue;
+    occupied.push({ start, end });
+    found.push(h.id);
+  }
+  return found;
+}
+
+/**
+ * 完成一轮发言：发言者（或 /pass 跳过的玩家）移出队列。
+ * 被 @ 点名者统一插入到「当前发言者之后」（即剩余队列队首）：
+ * - 若原本仍在本轮队列中，先移除原位置再插入（避免重复）；
+ * - 若原本已不在本轮队列中，则直接插入。
+ * 多个 @ 按文本出现顺序排列（后出现的排在前者后面）。
+ */
+export function completeTurn(
+  queue: string[],
+  speakerParticipantId: number,
+  mentionedIds: number[]
+): string[] {
+  const rest = queue.filter((id) => id !== String(speakerParticipantId));
+  const mentioned = [...new Set(mentionedIds.map(String))];
+  let out = [...rest];
+  for (const id of mentioned) {
+    out = out.filter((q) => q !== id);
+  }
+  return [...mentioned, ...out];
+}
+
+/** 玩家 /pass 跳过：将玩家移出队列，不修改任何对话历史 */
+export function passPlayer(queue: string[], playerParticipantId: number): string[] {
+  return queue.filter((id) => id !== String(playerParticipantId));
+}
+
+/**
+ * 把某位发言者重新放回队列（供「重新生成/编辑消息」后联动发言队列使用）：
+ * - 若该卷仍在本轮队列中 → 移回队首（优先补发，模拟“重讲一遍”）
+ * - 若该卷已被消费（不在队列、说明本轮已轮过）→ 插入队首，相当于进入新一轮补发
+ * 无论哪种情况，都保留其余成员相对顺序，仅把目标提到队首。
+ */
+export function requeueSpeaker(queue: string[], speakerParticipantId: number): string[] {
+  const id = String(speakerParticipantId);
+  const rest = queue.filter((q) => q !== id);
+  return [id, ...rest];
+}
+
+/** 刷新队列：为空时 loopIndex++ 并重新初始化；返回是否刚开启新循环 */
+export function refreshQueue(
+  session: Pick<ChatSession, 'turnOrderMode' | 'loopIndex' | 'turnQueueJson'>,
+  participants: ChatParticipant[]
+): { queue: string[]; loopIndex: number; loopStarted: boolean } {
+  let queue = safeParseQueue(session.turnQueueJson);
+  const validIds = new Set(participants.map((p) => String(p.participantId)));
+  queue = queue.filter((id) => validIds.has(id));
+  let loopIndex = session.loopIndex;
+  let loopStarted = false;
+  if (queue.length === 0) {
+    loopIndex += 1;
+    queue = initializeTurnQueue(participants, session.turnOrderMode);
+    loopStarted = true;
+  }
+  return { queue, loopIndex, loopStarted };
+}
+
+export function safeParseQueue(json: string): string[] {
+  try {
+    const v = JSON.parse(json || '[]');
+    return Array.isArray(v) ? v.map(String) : [];
+  } catch {
+    return [];
+  }
+}
+
+export function queueJson(queue: string[]): string {
+  return JSON.stringify(queue);
+}
+
+// ============================================================
+// 循环历史（完整初始顺序）辅助
+// ============================================================
+
+/** 解析循环历史 JSON（二维数组），失败返回空数组 */
+export function safeParseQueueHistory(json: string): string[][] {
+  try {
+    const v = JSON.parse(json || '[]');
+    if (Array.isArray(v) && v.every((x) => Array.isArray(x))) {
+      return v.map((arr) => arr.map(String));
+    }
+    return [];
+  } catch {
+    return [];
+  }
+}
+
+/** 序列化循环历史 → JSON */
+export function queueHistoryJson(history: string[][]): string {
+  return JSON.stringify(history);
+}
+
+/**
+ * 计算用于左侧「发言队列」展示的队列（含补全逻辑）：
+ * - 队列非空 → 不解构，原样返回（含当前发言者）
+ * - 队列为空（循环结束）→ 模拟下一循环的初始顺序（不加 loopIndex，仅展示）
+ */
+export function effectiveDisplayQueue(
+  session: Pick<ChatSession, 'mode' | 'turnOrderMode' | 'loopIndex' | 'turnQueueJson'>,
+  participants: ChatParticipant[]
+): string[] {
+  if (session.mode !== 'GROUP') return [];
+  const { queue } = refreshQueue(session, participants);
+  if (queue.length > 0) return queue;
+  return initializeTurnQueue(participants, session.turnOrderMode);
+}
+
+/** 流式输出中的发言者标签：玩家 → 用户，NPC → 显示名 */
+export function speakerLabel(p: ChatParticipant): string {
+  return p.kind === 'PLAYER' ? translate('common.user') : p.displayName;
+}
+
+/** 一个实用的 AI 工具调用 id 回退生成 */
+export function fallbackToolCallId(index: number): string {
+  return `call-${Date.now()}-${index}`;
+}