@mseep/claudian 2.0.25

package/src/features/chat/state/ChatState.ts

@@ -0,0 +1,436 @@
+import type { UsageInfo } from '../../../core/types';
+import type {
+  ChatMessage,
+  ChatStateCallbacks,
+  ChatStateData,
+  PendingToolCall,
+  QueuedMessage,
+  ThinkingBlockState,
+  TodoItem,
+  WriteEditState,
+} from './types';
+
+function createInitialState(): ChatStateData {
+  return {
+    messages: [],
+    isStreaming: false,
+    cancelRequested: false,
+    streamGeneration: 0,
+    isCreatingConversation: false,
+    isSwitchingConversation: false,
+    hasPendingConversationSave: false,
+    currentConversationId: null,
+    queuedMessage: null,
+    currentContentEl: null,
+    currentTextEl: null,
+    currentTextContent: '',
+    currentThinkingState: null,
+    thinkingEl: null,
+    queueIndicatorEl: null,
+    thinkingIndicatorTimeout: null,
+    toolCallElements: new Map(),
+    writeEditStates: new Map(),
+    pendingTools: new Map(),
+    usage: null,
+    ignoreUsageUpdates: false,
+    currentTodos: null,
+    needsAttention: false,
+    autoScrollEnabled: true, // Default; controllers will override based on settings
+    responseStartTime: null,
+    flavorTimerInterval: null,
+    pendingNewSessionPlan: null,
+    planFilePath: null,
+    prePlanPermissionMode: null,
+  };
+}
+
+export class ChatState {
+  private state: ChatStateData;
+  private _callbacks: ChatStateCallbacks;
+  private thinkingIndicatorTimeoutWindow: Window | null = null;
+  private flavorTimerIntervalWindow: Window | null = null;
+
+  constructor(callbacks: ChatStateCallbacks = {}) {
+    this.state = createInitialState();
+    this._callbacks = callbacks;
+  }
+
+  get callbacks(): ChatStateCallbacks {
+    return this._callbacks;
+  }
+
+  set callbacks(value: ChatStateCallbacks) {
+    this._callbacks = value;
+  }
+
+  // ============================================
+  // Messages
+  // ============================================
+
+  get messages(): ChatMessage[] {
+    return [...this.state.messages];
+  }
+
+  set messages(value: ChatMessage[]) {
+    this.state.messages = value;
+    this._callbacks.onMessagesChanged?.();
+  }
+
+  addMessage(msg: ChatMessage): void {
+    this.state.messages.push(msg);
+    this._callbacks.onMessagesChanged?.();
+  }
+
+  clearMessages(): void {
+    this.state.messages = [];
+    this._callbacks.onMessagesChanged?.();
+  }
+
+  truncateAt(messageId: string): number {
+    const idx = this.state.messages.findIndex(m => m.id === messageId);
+    if (idx === -1) return 0;
+    const removed = this.state.messages.length - idx;
+    this.state.messages = this.state.messages.slice(0, idx);
+    this._callbacks.onMessagesChanged?.();
+    return removed;
+  }
+
+  // ============================================
+  // Streaming Control
+  // ============================================
+
+  get isStreaming(): boolean {
+    return this.state.isStreaming;
+  }
+
+  set isStreaming(value: boolean) {
+    this.state.isStreaming = value;
+    this._callbacks.onStreamingStateChanged?.(value);
+  }
+
+  get cancelRequested(): boolean {
+    return this.state.cancelRequested;
+  }
+
+  set cancelRequested(value: boolean) {
+    this.state.cancelRequested = value;
+  }
+
+  get streamGeneration(): number {
+    return this.state.streamGeneration;
+  }
+
+  bumpStreamGeneration(): number {
+    this.state.streamGeneration += 1;
+    return this.state.streamGeneration;
+  }
+
+  get isCreatingConversation(): boolean {
+    return this.state.isCreatingConversation;
+  }
+
+  set isCreatingConversation(value: boolean) {
+    this.state.isCreatingConversation = value;
+  }
+
+  get isSwitchingConversation(): boolean {
+    return this.state.isSwitchingConversation;
+  }
+
+  set isSwitchingConversation(value: boolean) {
+    this.state.isSwitchingConversation = value;
+  }
+
+  get hasPendingConversationSave(): boolean {
+    return this.state.hasPendingConversationSave;
+  }
+
+  set hasPendingConversationSave(value: boolean) {
+    this.state.hasPendingConversationSave = value;
+  }
+
+  // ============================================
+  // Conversation
+  // ============================================
+
+  get currentConversationId(): string | null {
+    return this.state.currentConversationId;
+  }
+
+  set currentConversationId(value: string | null) {
+    this.state.currentConversationId = value;
+    this._callbacks.onConversationChanged?.(value);
+  }
+
+  // ============================================
+  // Queued Message
+  // ============================================
+
+  get queuedMessage(): QueuedMessage | null {
+    return this.state.queuedMessage;
+  }
+
+  set queuedMessage(value: QueuedMessage | null) {
+    this.state.queuedMessage = value;
+  }
+
+  // ============================================
+  // Streaming DOM State
+  // ============================================
+
+  get currentContentEl(): HTMLElement | null {
+    return this.state.currentContentEl;
+  }
+
+  set currentContentEl(value: HTMLElement | null) {
+    this.state.currentContentEl = value;
+  }
+
+  get currentTextEl(): HTMLElement | null {
+    return this.state.currentTextEl;
+  }
+
+  set currentTextEl(value: HTMLElement | null) {
+    this.state.currentTextEl = value;
+  }
+
+  get currentTextContent(): string {
+    return this.state.currentTextContent;
+  }
+
+  set currentTextContent(value: string) {
+    this.state.currentTextContent = value;
+  }
+
+  get currentThinkingState(): ThinkingBlockState | null {
+    return this.state.currentThinkingState;
+  }
+
+  set currentThinkingState(value: ThinkingBlockState | null) {
+    this.state.currentThinkingState = value;
+  }
+
+  get thinkingEl(): HTMLElement | null {
+    return this.state.thinkingEl;
+  }
+
+  set thinkingEl(value: HTMLElement | null) {
+    this.state.thinkingEl = value;
+  }
+
+  get queueIndicatorEl(): HTMLElement | null {
+    return this.state.queueIndicatorEl;
+  }
+
+  set queueIndicatorEl(value: HTMLElement | null) {
+    this.state.queueIndicatorEl = value;
+  }
+
+  get thinkingIndicatorTimeout(): number | null {
+    return this.state.thinkingIndicatorTimeout;
+  }
+
+  set thinkingIndicatorTimeout(value: number | null) {
+    this.state.thinkingIndicatorTimeout = value;
+    this.thinkingIndicatorTimeoutWindow = value === null ? null : this.getDefaultTimerWindow();
+  }
+
+  // ============================================
+  // Tool Tracking Maps (mutable references)
+  // ============================================
+
+  get toolCallElements(): Map<string, HTMLElement> {
+    return this.state.toolCallElements;
+  }
+
+  get writeEditStates(): Map<string, WriteEditState> {
+    return this.state.writeEditStates;
+  }
+
+  get pendingTools(): Map<string, PendingToolCall> {
+    return this.state.pendingTools;
+  }
+
+  // ============================================
+  // Usage State
+  // ============================================
+
+  get usage(): UsageInfo | null {
+    return this.state.usage;
+  }
+
+  set usage(value: UsageInfo | null) {
+    this.state.usage = value;
+    this._callbacks.onUsageChanged?.(value);
+  }
+
+  get ignoreUsageUpdates(): boolean {
+    return this.state.ignoreUsageUpdates;
+  }
+
+  set ignoreUsageUpdates(value: boolean) {
+    this.state.ignoreUsageUpdates = value;
+  }
+
+  // ============================================
+  // Current Todos (for persistent bottom panel)
+  // ============================================
+
+  get currentTodos(): TodoItem[] | null {
+    return this.state.currentTodos ? [...this.state.currentTodos] : null;
+  }
+
+  set currentTodos(value: TodoItem[] | null) {
+    // Normalize empty arrays to null for consistency
+    const normalizedValue = (value && value.length > 0) ? value : null;
+    this.state.currentTodos = normalizedValue;
+    this._callbacks.onTodosChanged?.(normalizedValue);
+  }
+
+  // ============================================
+  // Attention State (approval pending, error, etc.)
+  // ============================================
+
+  get needsAttention(): boolean {
+    return this.state.needsAttention;
+  }
+
+  set needsAttention(value: boolean) {
+    this.state.needsAttention = value;
+    this._callbacks.onAttentionChanged?.(value);
+  }
+
+  // ============================================
+  // Auto-Scroll Control
+  // ============================================
+
+  get autoScrollEnabled(): boolean {
+    return this.state.autoScrollEnabled;
+  }
+
+  set autoScrollEnabled(value: boolean) {
+    const changed = this.state.autoScrollEnabled !== value;
+    this.state.autoScrollEnabled = value;
+    if (changed) {
+      this._callbacks.onAutoScrollChanged?.(value);
+    }
+  }
+
+  // ============================================
+  // Response Timer State
+  // ============================================
+
+  get responseStartTime(): number | null {
+    return this.state.responseStartTime;
+  }
+
+  set responseStartTime(value: number | null) {
+    this.state.responseStartTime = value;
+  }
+
+  get flavorTimerInterval(): number | null {
+    return this.state.flavorTimerInterval;
+  }
+
+  set flavorTimerInterval(value: number | null) {
+    this.state.flavorTimerInterval = value;
+    this.flavorTimerIntervalWindow = value === null ? null : this.getDefaultTimerWindow();
+  }
+
+  get pendingNewSessionPlan(): string | null {
+    return this.state.pendingNewSessionPlan;
+  }
+
+  set pendingNewSessionPlan(value: string | null) {
+    this.state.pendingNewSessionPlan = value;
+  }
+
+  get planFilePath(): string | null {
+    return this.state.planFilePath;
+  }
+
+  set planFilePath(value: string | null) {
+    this.state.planFilePath = value;
+  }
+
+  get prePlanPermissionMode(): string | null {
+    return this.state.prePlanPermissionMode;
+  }
+
+  set prePlanPermissionMode(value: string | null) {
+    this.state.prePlanPermissionMode = value;
+  }
+
+  // ============================================
+  // Reset Methods
+  // ============================================
+
+  setThinkingIndicatorTimeout(value: number | null, ownerWindow: Window | null): void {
+    this.state.thinkingIndicatorTimeout = value;
+    this.thinkingIndicatorTimeoutWindow = value === null ? null : ownerWindow;
+  }
+
+  clearThinkingIndicatorTimeout(fallbackWindow: Window | null = null): void {
+    if (this.state.thinkingIndicatorTimeout) {
+      const ownerWindow = this.thinkingIndicatorTimeoutWindow ?? fallbackWindow ?? this.getDefaultTimerWindow();
+      ownerWindow?.clearTimeout(this.state.thinkingIndicatorTimeout);
+      this.state.thinkingIndicatorTimeout = null;
+      this.thinkingIndicatorTimeoutWindow = null;
+    }
+  }
+
+  setFlavorTimerInterval(value: number | null, ownerWindow: Window | null): void {
+    this.state.flavorTimerInterval = value;
+    this.flavorTimerIntervalWindow = value === null ? null : ownerWindow;
+  }
+
+  clearFlavorTimerInterval(): void {
+    if (this.state.flavorTimerInterval) {
+      const ownerWindow = this.flavorTimerIntervalWindow ?? this.getDefaultTimerWindow();
+      ownerWindow?.clearInterval(this.state.flavorTimerInterval);
+      this.state.flavorTimerInterval = null;
+      this.flavorTimerIntervalWindow = null;
+    }
+  }
+
+  resetStreamingState(): void {
+    this.state.currentContentEl = null;
+    this.state.currentTextEl = null;
+    this.state.currentTextContent = '';
+    this.state.currentThinkingState = null;
+    this.state.isStreaming = false;
+    this.state.cancelRequested = false;
+    // Clear thinking indicator timeout
+    this.clearThinkingIndicatorTimeout();
+    // Clear response timer
+    this.clearFlavorTimerInterval();
+    this.state.responseStartTime = null;
+  }
+
+  clearMaps(): void {
+    this.state.toolCallElements.clear();
+    this.state.writeEditStates.clear();
+    this.state.pendingTools.clear();
+  }
+
+  resetForNewConversation(): void {
+    this.clearMessages();
+    this.resetStreamingState();
+    this.clearMaps();
+    this.state.queuedMessage = null;
+    this.usage = null;
+    this.currentTodos = null;
+    this.autoScrollEnabled = true;
+  }
+
+  getPersistedMessages(): ChatMessage[] {
+    // Return messages as-is - image data is single source of truth
+    return this.state.messages;
+  }
+
+  private getDefaultTimerWindow(): Window | null {
+    return typeof window === 'undefined' ? null : window;
+  }
+}
+
+export { createInitialState };

package/src/features/chat/state/types.ts

@@ -0,0 +1,138 @@
+import type { EditorView } from '@codemirror/view';
+
+import type { ChatRuntimeQueryOptions, ChatTurnRequest } from '../../../core/runtime/types';
+import type { TodoItem } from '../../../core/tools/todo';
+import type {
+  ChatMessage,
+  ImageAttachment,
+  SubagentInfo,
+  ToolCallInfo,
+  UsageInfo,
+} from '../../../core/types';
+import type { BrowserSelectionContext } from '../../../utils/browser';
+import type { CanvasSelectionContext } from '../../../utils/canvas';
+import type { EditorSelectionContext } from '../../../utils/editor';
+import type { ThinkingBlockState } from '../rendering/ThinkingBlockRenderer';
+import type { WriteEditState } from '../rendering/WriteEditRenderer';
+
+/** Queued message waiting to be sent after current streaming completes. */
+export interface QueuedMessage {
+  content: string;
+  images?: ImageAttachment[];
+  editorContext: EditorSelectionContext | null;
+  browserContext?: BrowserSelectionContext | null;
+  canvasContext: CanvasSelectionContext | null;
+  /** Provider-neutral turn snapshot captured at enqueue time. */
+  turnRequest?: ChatTurnRequest;
+}
+
+/** Pending tool call waiting to be rendered (buffered until input is complete). */
+export interface PendingToolCall {
+  toolCall: ToolCallInfo;
+  parentEl: HTMLElement | null;
+}
+
+/** Stored selection state from editor polling. */
+export interface StoredSelection {
+  notePath: string;
+  selectedText: string;
+  lineCount: number;
+  startLine?: number;
+  from?: number;
+  to?: number;
+  editorView?: EditorView;
+  domRanges?: Range[];
+}
+
+/** Centralized chat state data. */
+export interface ChatStateData {
+  // Message state
+  messages: ChatMessage[];
+
+  // Streaming control
+  isStreaming: boolean;
+  cancelRequested: boolean;
+  streamGeneration: number;
+  /** Guards against concurrent operations during conversation creation. */
+  isCreatingConversation: boolean;
+  /** Guards against concurrent operations during conversation switching. */
+  isSwitchingConversation: boolean;
+  /** Local tab state is ahead of persisted conversation metadata. */
+  hasPendingConversationSave: boolean;
+
+  // Conversation identity
+  currentConversationId: string | null;
+
+  // Queued message
+  queuedMessage: QueuedMessage | null;
+
+  // Active streaming DOM state
+  currentContentEl: HTMLElement | null;
+  currentTextEl: HTMLElement | null;
+  currentTextContent: string;
+  currentThinkingState: ThinkingBlockState | null;
+  thinkingEl: HTMLElement | null;
+  queueIndicatorEl: HTMLElement | null;
+  /** Debounce timeout for showing thinking indicator after inactivity. */
+  thinkingIndicatorTimeout: number | null;
+
+  // Tool tracking maps
+  toolCallElements: Map<string, HTMLElement>;
+  writeEditStates: Map<string, WriteEditState>;
+  /** Pending tool calls buffered until input is complete (for non-streaming-style render). */
+  pendingTools: Map<string, PendingToolCall>;
+
+  // Context window usage
+  usage: UsageInfo | null;
+  // Flag to ignore usage updates (during session reset)
+  ignoreUsageUpdates: boolean;
+
+  // Current todo items for the persistent bottom panel
+  currentTodos: TodoItem[] | null;
+
+  // Attention state (approval pending, error, etc.)
+  needsAttention: boolean;
+
+  // Auto-scroll control during streaming
+  autoScrollEnabled: boolean;
+
+  // Response timer state
+  responseStartTime: number | null;
+  flavorTimerInterval: number | null;
+
+  // Pending plan content for approve-new-session (auto-sends in new session after stream ends)
+  pendingNewSessionPlan: string | null;
+
+  // Plan file path captured from Write tool calls to provider plan directory during plan mode
+  planFilePath: string | null;
+
+  // Saved permission mode before entering plan mode (for Shift+Tab toggle restore)
+  prePlanPermissionMode: string | null;
+}
+
+/** Callbacks for ChatState changes. */
+export interface ChatStateCallbacks {
+  onMessagesChanged?: () => void;
+  onStreamingStateChanged?: (isStreaming: boolean) => void;
+  onConversationChanged?: (id: string | null) => void;
+  onUsageChanged?: (usage: UsageInfo | null) => void;
+  onTodosChanged?: (todos: TodoItem[] | null) => void;
+  onAttentionChanged?: (needsAttention: boolean) => void;
+  onAutoScrollChanged?: (enabled: boolean) => void;
+}
+
+/** Options for query execution. */
+export type QueryOptions = ChatRuntimeQueryOptions;
+
+// Re-export types that are used across the chat feature
+export type {
+  ChatMessage,
+  EditorSelectionContext,
+  ImageAttachment,
+  SubagentInfo,
+  ThinkingBlockState,
+  TodoItem,
+  ToolCallInfo,
+  UsageInfo,
+  WriteEditState,
+};