Class: TUI::Screens::Chat

Inherits:
Object
  • Object
show all
Includes:
Formatting
Defined in:
lib/tui/screens/chat.rb

Constant Summary collapse

MIN_INPUT_HEIGHT = 
3
PRINTABLE_CHAR = 
/\A[[:print:]]\z/
ROLE_USER = 
"user"
ROLE_ASSISTANT = 
"assistant"
ROLE_LABELS = 
{ROLE_USER => "You", ROLE_ASSISTANT => "Anima"}.freeze
SCROLL_STEP = 
1
MOUSE_SCROLL_STEP = 
2
TOOL_ICON = 
"\u{1F527}"
CLOCK_ICON = 
"\u{1F552}"
CHECKMARK = 
"\u2713"
VIEWPORT_BACK_BUFFER =

Viewport virtualization tuning

3
VIEWPORT_OVERFLOW_MULTIPLIER =

entries before scroll target for upward scroll margin

2
VIEWPORT_BOTTOM_THRESHOLD =

build this many viewports worth of lines

10
ROLE_COLORS =

entries from end before we include all trailing

{"user" => "green", "assistant" => "cyan"}.freeze
VIEW_MODES =

Intentionally duplicated from Session::VIEW_MODES to keep the TUI independent of Rails. Must stay in sync when adding new modes.

%w[basic verbose debug].freeze

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from Formatting

#format_ns_timestamp, #format_token_label

Constructor Details

#initialize(cable_client:, message_store: nil, perf_logger: nil) ⇒ Chat

Returns a new instance of Chat.

Parameters:



54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
 
# File 'lib/tui/screens/chat.rb', line 54

def initialize(cable_client:, message_store: nil, perf_logger: nil)
  @cable_client = cable_client
  @message_store = message_store || MessageStore.new
  @perf_logger = perf_logger || PerformanceLogger.new(enabled: false)
  @input_buffer = InputBuffer.new
  @flash = Flash.new
  @loading = false
  @scroll_offset = 0
  @auto_scroll = true
  @visible_height = 0
  @max_scroll = 0
  @input_scroll_offset = 0
  @view_mode = "basic"
  @session_info = {id: cable_client.session_id || 0, message_count: 0, active_skills: [], active_workflow: nil, goals: [], children: []}
  @sessions_list = nil
  @parent_session_id = nil
  @authentication_required = false
  @token_save_result = nil
  @chat_focused = false
  @input_history = []
  @history_index = nil
  @saved_input = nil
  # Viewport virtualization: only renders messages visible in the scroll
  # window. Heights are estimated for all entries (cheap string math),
  # but Line objects are only built for the visible range + buffer.
  @height_map = HeightMap.new
  @height_map_version = -1
  @height_map_width = nil
  @height_map_loading = nil
  @viewport = viewport_cache_empty
end

Instance Attribute Details

#authentication_required ⇒ Object (readonly)

Returns the value of attribute authentication_required.



46
47
48
 
# File 'lib/tui/screens/chat.rb', line 46

def authentication_required
  @authentication_required
end

#chat_focused ⇒ Object (readonly)

Returns the value of attribute chat_focused.



46
47
48
 
# File 'lib/tui/screens/chat.rb', line 46

def chat_focused
  @chat_focused
end

#hud_hint ⇒ Object

Returns the value of attribute hud_hint.



49
50
51
 
# File 'lib/tui/screens/chat.rb', line 49

def hud_hint
  @hud_hint
end

#message_store ⇒ Object (readonly)

Returns the value of attribute message_store.



46
47
48
 
# File 'lib/tui/screens/chat.rb', line 46

def message_store
  @message_store
end

#parent_session_id ⇒ Object (readonly)

Returns the value of attribute parent_session_id.



46
47
48
 
# File 'lib/tui/screens/chat.rb', line 46

def parent_session_id
  @parent_session_id
end

#scroll_offset ⇒ Object (readonly)

Returns the value of attribute scroll_offset.



46
47
48
 
# File 'lib/tui/screens/chat.rb', line 46

def scroll_offset
  @scroll_offset
end

#session_info ⇒ Object (readonly)

Returns the value of attribute session_info.



46
47
48
 
# File 'lib/tui/screens/chat.rb', line 46

def session_info
  @session_info
end

#sessions_list ⇒ Object (readonly)

Returns the value of attribute sessions_list.



46
47
48
 
# File 'lib/tui/screens/chat.rb', line 46

def sessions_list
  @sessions_list
end

#token_save_result ⇒ Object (readonly)

Returns the value of attribute token_save_result.



46
47
48
 
# File 'lib/tui/screens/chat.rb', line 46

def token_save_result
  @token_save_result
end

#view_mode ⇒ Object (readonly)

Returns the value of attribute view_mode.



46
47
48
 
# File 'lib/tui/screens/chat.rb', line 46

def view_mode
  @view_mode
end

Instance Method Details

#clear_authentication_required ⇒ void

This method returns an undefined value.

Clears the authentication_required flag after the App has consumed it.



218
219
220
 
# File 'lib/tui/screens/chat.rb', line 218

def clear_authentication_required
  @authentication_required = false
end

#clear_input ⇒ void

This method returns an undefined value.

Clears the input buffer. Used when Escape is pressed with non-empty input.



212
213
214
 
# File 'lib/tui/screens/chat.rb', line 212

def clear_input
  @input_buffer.clear
end

#consume_token_save_result ⇒ Hash?

Returns and clears the token save result for one-shot consumption by the App.

Returns:

  • (Hash, nil) —

    true or false, message: “…”, or nil



224
225
226
227
228
 
# File 'lib/tui/screens/chat.rb', line 224

def consume_token_save_result
  result = @token_save_result
  @token_save_result = nil
  result
end

#cursor_pos ⇒ Integer

Returns current cursor position (delegates to InputBuffer).

Returns:

  • (Integer) —

    current cursor position (delegates to InputBuffer)



96
97
98
 
# File 'lib/tui/screens/chat.rb', line 96

def cursor_pos
  @input_buffer.cursor_pos
end

#finalize ⇒ Object 



230
231
 
# File 'lib/tui/screens/chat.rb', line 230

def finalize
end

#focus_chat ⇒ void

This method returns an undefined value.

Switches focus to the chat pane for keyboard scrolling.



239
240
241
 
# File 'lib/tui/screens/chat.rb', line 239

def focus_chat
  @chat_focused = true
end

#handle_event(event) ⇒ Object

Dispatches keyboard, mouse, and paste events. Supports two focus modes: input mode (default) where arrows navigate the input buffer with bash-style history overflow, and chat-focused mode where arrows scroll the chat pane.

Page Up/Down and mouse scroll always control the chat pane regardless of focus mode.



127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
 
# File 'lib/tui/screens/chat.rb', line 127

def handle_event(event)
  return handle_mouse_event(event) if event.mouse?
  return handle_paste_event(event) if event.paste?
  return handle_scroll_key(event) if event.page_up? || event.page_down?

  # Dismiss flash on any keypress (flash auto-expires too)
  @flash.dismiss! if @flash.any?

  return handle_chat_focused_event(event) if @chat_focused

  if event.up?
    return true if @input_buffer.move_up
    return true if @input_buffer.text.empty? && recall_pending_message
    return navigate_history_back
  end

  if event.down?
    return true if @input_buffer.move_down
    return navigate_history_forward
  end

  if event.enter?
    submit_message
    true
  elsif event.backspace?
    reset_history_browsing
    @input_buffer.backspace
    true
  elsif event.delete?
    reset_history_browsing
    @input_buffer.delete
    true
  elsif event.left?
    @input_buffer.move_left
  elsif event.right?
    @input_buffer.move_right
  elsif event.home?
    @input_buffer.move_home
  elsif event.end?
    @input_buffer.move_end
  elsif printable_char?(event) && !@input_buffer.full?
    reset_history_browsing
    @input_buffer.insert(event.code)
  else
    false
  end
end

#input ⇒ String

Returns current input text (delegates to InputBuffer).

Returns:

  • (String) —

    current input text (delegates to InputBuffer)



91
92
93
 
# File 'lib/tui/screens/chat.rb', line 91

def input
  @input_buffer.text
end

#interrupt_execution ⇒ void

This method returns an undefined value.

Sends an interrupt request to the server to stop the current tool chain. Called when Escape is pressed with empty input during active processing.



205
206
207
 
# File 'lib/tui/screens/chat.rb', line 205

def interrupt_execution
  @cable_client.interrupt
end

#loading? ⇒ Boolean

Returns:

  • (Boolean) 


233
234
235
 
# File 'lib/tui/screens/chat.rb', line 233

def loading?
  @loading
end

#messages ⇒ Object 



86
87
88
 
# File 'lib/tui/screens/chat.rb', line 86

def messages
  @message_store.messages
end

#new_session ⇒ Object

Creates a new session through the WebSocket protocol. The brain creates the session, switches the channel stream, and sends a session_changed signal followed by (empty) history. The client-side state reset happens when session_changed is received.



179
180
181
 
# File 'lib/tui/screens/chat.rb', line 179

def new_session
  @cable_client.create_session
end

#render(frame, area, tui) ⇒ Object 



100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
 
# File 'lib/tui/screens/chat.rb', line 100

def render(frame, area, tui)
  process_incoming_messages

  input_height = calculate_input_height(tui, area.width, area.height)

  chat_area, input_area = tui.split(
    area,
    direction: :vertical,
    constraints: [
      tui.constraint_fill(1),
      tui.constraint_length(input_height)
    ]
  )

  render_messages(frame, chat_area, tui)
  render_flash(frame, chat_area, tui)

  render_input(frame, input_area, tui)
end

#switch_session(session_id) ⇒ Object

Switches to an existing session through the WebSocket protocol. The brain switches the channel stream and sends a session_changed signal followed by chat history.

Parameters:

  • session_id (Integer) —

    target session to switch to



188
189
190
 
# File 'lib/tui/screens/chat.rb', line 188

def switch_session(session_id)
  @cable_client.switch_session(session_id)
end

#switch_view_mode(mode) ⇒ Object

Sends an explicit view mode switch command to the server. The server broadcasts the mode change and re-transmits the viewport decorated in the new mode to all connected clients.

Parameters:

  • mode (String) —

    target view mode (“basic”, “verbose”, or “debug”)



197
198
199
 
# File 'lib/tui/screens/chat.rb', line 197

def switch_view_mode(mode)
  @cable_client.change_view_mode(mode)
end

#unfocus_chat ⇒ void

This method returns an undefined value.

Returns focus from the chat pane to the input field.



245
246
247
 
# File 'lib/tui/screens/chat.rb', line 245

def unfocus_chat
  @chat_focused = false
end