ei-tui 0.6.7 → 0.7.0

package/tui/src/components/RoomMessageList.tsx

@@ -1,4 +1,4 @@
-import { For, Show, createMemo, createSignal, createEffect, on, onCleanup } from "solid-js";
+import { For, Show, createMemo, createSignal, createEffect, on, onCleanup, onMount } from "solid-js";
 import { TextAttributes, type ScrollBoxRenderable } from "@opentui/core";
 import { useEi } from "../context/ei.js";
 import { useKeyboardNav } from "../context/keyboard.js";
@@ -27,9 +27,21 @@ function formatTime(timestamp: string): string {
 }
 
 export function RoomMessageList() {
-  const { roomMessages, roomActivePath, personas, activeRoomId, getRoom, getQuotes, quotesVersion } = useEi();
+  const { roomMessages, roomActivePath, personas, activeRoomId, getRoom, getQuotes, quotesVersion, getHuman } = useEi();
   const { registerMessageScroll } = useKeyboardNav();
 
+  const [humanDisplayName, setHumanDisplayName] = createSignal("Human");
+
+  onMount(() => {
+    void getHuman().then(human => {
+      const name =
+        human.settings?.name_display ||
+        human.facts?.find(f => f.name === "Nickname/Preferred Name")?.description ||
+        "Human";
+      setHumanDisplayName(name);
+    });
+  });
+
   const personaNameMap = createMemo(() => {
     const map = new Map<string, string>();
     for (const p of personas()) {
@@ -99,7 +111,7 @@ export function RoomMessageList() {
   });
 
   const getSpeakerName = (msg: RoomMessage): string => {
-    if (msg.role === "human") return "Human";
+    if (msg.role === "human") return humanDisplayName();
     if (msg.persona_id) return personaNameMap().get(msg.persona_id) ?? msg.persona_id;
     return "Persona";
   };

package/tui/src/context/ei.tsx

@@ -110,6 +110,8 @@ export interface EiContextValue {
   dismissWelcomeOverlay: () => void;
   deleteMessages: (personaId: string, messageIds: string[]) => Promise<void>;
   setMessageContextStatus: (personaId: string, messageId: string, status: ContextStatus) => Promise<void>;
+  deleteRoomMessages: (roomId: string, messageIds: string[]) => Promise<void>;
+  setRoomMessageContextStatus: (roomId: string, messageId: string, status: ContextStatus) => Promise<void>;
   recallPendingMessages: () => Promise<string>;
   getToolProviderList: () => ToolProvider[];
   getToolList: () => ToolDefinition[];
@@ -470,6 +472,22 @@ export const EiProvider: ParentComponent = (props) => {
     setStore("messages", store.messages.map(m => m.id === messageId ? { ...m, context_status: status } : m));
   };
 
+  const deleteRoomMessages = async (roomId: string, messageIds: string[]): Promise<void> => {
+    if (!processor) return;
+    processor.getStateManager().removeRoomMessages(roomId, messageIds);
+    if (roomId === store.activeRoomId) {
+      setStore("roomMessages", msgs => msgs.filter(m => !messageIds.includes(m.id)));
+    }
+  };
+
+  const setRoomMessageContextStatus = async (roomId: string, messageId: string, status: ContextStatus): Promise<void> => {
+    if (!processor) return;
+    processor.getStateManager().updateRoomMessage(roomId, messageId, { context_status: status });
+    if (roomId === store.activeRoomId) {
+      setStore("roomMessages", msgs => msgs.map(m => m.id === messageId ? { ...m, context_status: status } : m));
+    }
+  };
+
   const recallPendingMessages = async (): Promise<string> => {
     if (!processor) return "";
     const personaId = store.activePersonaId;
@@ -905,6 +923,8 @@ export const EiProvider: ParentComponent = (props) => {
     dismissWelcomeOverlay: () => setShowWelcomeOverlay(false),
     deleteMessages,
     setMessageContextStatus,
+    deleteRoomMessages,
+    setRoomMessageContextStatus,
     recallPendingMessages,
     getToolProviderList,
     getToolList,

package/tui/src/util/cyp-tree.ts

@@ -0,0 +1,62 @@
+import type { RoomMessage } from "../../../src/core/types.js";
+
+export interface CYPTreeData {
+  ordered: RoomMessage[];
+  numToId: Map<number, string>;
+  idToNum: Map<string, number>;
+  childrenMap: Map<string, RoomMessage[]>;
+}
+
+export function buildCYPTree(messages: RoomMessage[]): CYPTreeData {
+  const childrenMap = new Map<string, RoomMessage[]>();
+
+  for (const m of messages) {
+    if (m.parent_id !== null && m.parent_id !== undefined) {
+      const existing = childrenMap.get(m.parent_id);
+      if (existing) {
+        existing.push(m);
+      } else {
+        childrenMap.set(m.parent_id, [m]);
+      }
+    }
+  }
+
+  const root = messages.find((m) => m.parent_id === null);
+  const ordered: RoomMessage[] = [];
+  const numToId = new Map<number, string>();
+  const idToNum = new Map<string, number>();
+
+  if (!root) {
+    return { ordered, numToId, idToNum, childrenMap };
+  }
+
+  const queue: RoomMessage[] = [root];
+  while (queue.length > 0) {
+    const current = queue.shift()!;
+    ordered.push(current);
+    const num = ordered.length;
+    numToId.set(num, current.id);
+    idToNum.set(current.id, num);
+    for (const child of childrenMap.get(current.id) ?? []) {
+      queue.push(child);
+    }
+  }
+
+  return { ordered, numToId, idToNum, childrenMap };
+}
+
+export function getSubtreeIds(
+  rootId: string,
+  childrenMap: Map<string, RoomMessage[]>
+): Set<string> {
+  const result = new Set<string>();
+  const queue = [rootId];
+  while (queue.length > 0) {
+    const id = queue.shift()!;
+    result.add(id);
+    for (const child of childrenMap.get(id) ?? []) {
+      queue.push(child.id);
+    }
+  }
+  return result;
+}

package/tui/src/util/yaml-context.ts

@@ -1,5 +1,5 @@
 import YAML from "yaml";
-import type { Message } from "../../../src/core/types.js";
+import type { Message, RoomMessage } from "../../../src/core/types.js";
 import { ContextStatus } from "../../../src/core/types.js";
 
 interface EditableMessage {
@@ -63,4 +63,90 @@ export function contextFromYAML(yamlContent: string): ContextYAMLResult {
   return { messages, deletedMessageIds };
 }
 
+interface FfaEditableNode {
+  id: string;
+  role: "human" | "persona";
+  speaker?: string;
+  context_status: ContextStatus;
+  _delete?: boolean;
+  content?: string;
+  silence_reason?: string;
+  children?: FfaEditableNode[];
+}
+
+function buildNode(msg: RoomMessage, messages: RoomMessage[], speakerMap: Map<string, string>, isRoot = false): FfaEditableNode {
+  const children = messages
+    .filter((m) => m.parent_id === msg.id)
+    .map((child) => buildNode(child, messages, speakerMap));
+
+  const node: FfaEditableNode = {
+    id: msg.id,
+    role: msg.role === "human" ? "human" : "persona",
+    context_status: msg.context_status,
+    ...(!isRoot && { _delete: false }),
+  };
+
+  if (msg.role === "persona" && msg.persona_id) {
+    node.speaker = speakerMap.get(msg.persona_id) ?? msg.persona_id.slice(0, 8);
+  }
+
+  const text = getContent(msg);
+  if (text) node.content = text;
+  if (msg.silence_reason) node.silence_reason = msg.silence_reason;
+  if (children.length > 0) node.children = children;
+
+  return node;
+}
+
+export function ffaContextToYAML(
+  messages: RoomMessage[],
+  speakerMap: Map<string, string>
+): string {
+  const header = [
+    "# context_status: default | always | never",
+    "# _delete: true — removes this message and all its descendants",
+  ].join("\n");
+
+  const rootMsg = messages.find((m) => m.parent_id === null);
+  if (!rootMsg) return header + "\n[]";
 
+  return header + "\n" + YAML.stringify([buildNode(rootMsg, messages, speakerMap, true)], { lineWidth: 0 });
+}
+
+export interface FfaContextYAMLResult {
+  messages: Array<{ id: string; context_status: ContextStatus }>;
+  deletedMessageIds: string[];
+  implicitDeleteCount: number;
+}
+
+export function ffaContextFromYAML(yamlContent: string): FfaContextYAMLResult {
+  const data = YAML.parse(yamlContent) as FfaEditableNode[];
+
+  const deletedMessageIds: string[] = [];
+  const messages: Array<{ id: string; context_status: ContextStatus }> = [];
+  let implicitDeleteCount = 0;
+
+  function collectDeleted(node: FfaEditableNode, parentDeleted: boolean): void {
+    const selfDeleted = parentDeleted || !!node._delete;
+    if (selfDeleted) {
+      deletedMessageIds.push(node.id);
+      if (!node._delete) implicitDeleteCount++;
+    } else {
+      const normalized = (node.context_status ?? 'default').toString().toLowerCase() as ContextStatus;
+      messages.push({ id: node.id, context_status: normalized });
+    }
+    for (const child of node.children ?? []) {
+      collectDeleted(child, selfDeleted);
+    }
+  }
+
+  const nodes = data ?? [];
+  for (let i = 0; i < nodes.length; i++) {
+    const node = nodes[i];
+    const isRoot = i === 0;
+    if (isRoot && node._delete) continue; // root deletion not allowed — it anchors the room
+    collectDeleted(node, false);
+  }
+
+  return { messages, deletedMessageIds, implicitDeleteCount };
+}