anima-core 0.1.0 → 0.2.0

data/lib/tui/screens/chat.rb

@@ -11,29 +11,32 @@ module TUI
       ROLE_ASSISTANT = "assistant"
       ROLE_LABELS = {ROLE_USER => "You", ROLE_ASSISTANT => "Anima"}.freeze
 
-      attr_reader :input, :message_collector, :session
+      SCROLL_STEP = 1
+      MOUSE_SCROLL_STEP = 2
 
-      def initialize(message_collector: nil, persister: nil, session: nil, shell_session: nil)
-        @message_collector = message_collector || Events::Subscribers::MessageCollector.new
+      attr_reader :input, :message_store, :scroll_offset, :session_info
+
+      # @param cable_client [TUI::CableClient] WebSocket client connected to the brain
+      # @param message_store [TUI::MessageStore, nil] injectable for testing
+      def initialize(cable_client:, message_store: nil)
+        @cable_client = cable_client
+        @message_store = message_store || MessageStore.new
         @input = ""
         @loading = false
-        @client = nil
-        @submit_thread = nil
-
-        @session = session || Session.order(id: :desc).first || Session.create!
-        load_session_messages
-        @persister = persister || Events::Subscribers::Persister.new(@session)
-        @shell_session = shell_session || ShellSession.new(session_id: @session.id)
-
-        Events::Bus.subscribe(@message_collector)
-        Events::Bus.subscribe(@persister)
+        @scroll_offset = 0
+        @auto_scroll = true
+        @visible_height = 0
+        @max_scroll = 0
+        @session_info = {id: cable_client.session_id, message_count: 0}
       end
 
       def messages
-        @message_collector.messages
+        @message_store.messages
       end
 
       def render(frame, area, tui)
+        process_incoming_messages
+
         chat_area, input_area = tui.split(
           area,
           direction: :vertical,
@@ -47,7 +50,10 @@ module TUI
         render_input(frame, input_area, tui)
       end
 
+      # Scrolling bypasses the loading guard so users can read chat history during LLM calls
       def handle_event(event)
+        return handle_mouse_event(event) if event.mouse?
+        return handle_scroll_key(event) if scroll_key?(event)
         return false if @loading
 
         if event.enter?
@@ -64,23 +70,15 @@ module TUI
         end
       end
 
+      # 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.
       def new_session
-        @submit_thread&.join
-        @shell_session&.finalize
-        @session = Session.create!
-        @persister.session = @session
-        @message_collector.clear
-        @input = ""
-        @loading = false
-        @shell_session = ShellSession.new(session_id: @session.id)
-        @registry = nil
+        @cable_client.create_session
       end
 
       def finalize
-        @submit_thread&.join
-        @shell_session&.finalize
-        Events::Bus.unsubscribe(@message_collector)
-        Events::Bus.unsubscribe(@persister)
       end
 
       def loading?
@@ -89,6 +87,63 @@ module TUI
 
       private
 
+      # Drains the WebSocket message queue and feeds events to the message store
+      def process_incoming_messages
+        @cable_client.drain_messages.each do |msg|
+          action = msg["action"]
+          type = msg["type"]
+
+          case action
+          when "session_changed"
+            handle_session_changed(msg)
+          when "sessions_list"
+            @sessions_list = msg["sessions"]
+          when "error"
+            # Silently ignored — no user-facing error display yet
+          else
+            case type
+            when "connection"
+              handle_connection_status(msg)
+            when "user_message"
+              @message_store.process_event(msg)
+              @session_info[:message_count] += 1
+              @loading = true
+            when "agent_message"
+              @message_store.process_event(msg)
+              @session_info[:message_count] += 1
+              @loading = false
+            else
+              @message_store.process_event(msg)
+            end
+          end
+        end
+      end
+
+      # Reacts to connection lifecycle changes from the WebSocket client.
+      # Clears stale state on (re)subscription so fresh history from the server
+      # replaces any messages displayed before the disconnect.
+      def handle_connection_status(msg)
+        case msg["status"]
+        when "subscribed"
+          @message_store.clear
+          @loading = false
+          @session_info[:message_count] = 0
+        when "disconnected", "failed"
+          @loading = false
+        end
+      end
+
+      def handle_session_changed(msg)
+        new_id = msg["session_id"]
+        @cable_client.update_session_id(new_id)
+        @message_store.clear
+        @session_info = {id: new_id, message_count: msg["message_count"] || 0}
+        @input = ""
+        @loading = false
+        @scroll_offset = 0
+        @auto_scroll = true
+      end
+
       def render_messages(frame, area, tui)
         lines = build_message_lines(tui)
 
@@ -104,9 +159,21 @@ module TUI
           ])
         end
 
+        inner_width = [area.width - 2, 1].max
+        @visible_height = [area.height - 2, 0].max
+
+        content_widget = tui.paragraph(text: lines, wrap: true, style: tui.style(fg: "white"))
+        content_height = content_widget.line_count(inner_width)
+
+        @max_scroll = [content_height - @visible_height, 0].max
+        @scroll_offset = @max_scroll if @auto_scroll
+        @scroll_offset = @scroll_offset.clamp(0, @max_scroll)
+
         widget = tui.paragraph(
           text: lines,
           wrap: true,
+          style: tui.style(fg: "white"),
+          scroll: [@scroll_offset, 0],
           block: tui.block(
             title: "Chat",
             borders: [:all],
@@ -115,6 +182,18 @@ module TUI
           )
         )
         frame.render_widget(widget, area)
+
+        if @max_scroll > 0
+          scrollbar = tui.scrollbar(
+            content_length: @max_scroll,
+            position: @scroll_offset,
+            orientation: :vertical_right,
+            thumb_style: {fg: "cyan"},
+            track_symbol: "│",
+            track_style: {fg: "dark_gray"}
+          )
+          frame.render_widget(scrollbar, area)
+        end
       end
 
       def build_message_lines(tui)
@@ -126,29 +205,38 @@ module TUI
           end
 
           label = ROLE_LABELS.fetch(msg[:role], msg[:role])
+          content_lines = msg[:content].to_s.split("\n", -1)
 
-          [
-            tui.line(spans: [
-              tui.span(content: "#{label}: ", style: role_style),
-              tui.span(content: msg[:content], style: tui.style(fg: "white"))
-            ]),
-            tui.line(spans: [tui.span(content: "", style: tui.style(fg: "white"))])
-          ]
+          lines = [tui.line(spans: [
+            tui.span(content: "#{label}: ", style: role_style),
+            tui.span(content: content_lines.first.to_s)
+          ])]
+          content_lines.drop(1).each { |text| lines << tui.line(spans: [tui.span(content: text)]) }
+          lines << tui.line(spans: [tui.span(content: "")])
         end
       end
 
       def render_input(frame, area, tui)
-        cursor = @loading ? "" : "\u2588"
-        border_style = @loading ? {fg: "dark_gray"} : {fg: "green"}
-        text_style = @loading ? tui.style(fg: "dark_gray") : tui.style(fg: "white")
+        disabled = @loading || !connected?
+        cursor = disabled ? "" : "\u2588"
+        border_style = disabled ? {fg: "dark_gray"} : {fg: "green"}
+        text_style = disabled ? tui.style(fg: "dark_gray") : tui.style(fg: "white")
+
+        title = if @loading
+          "Waiting..."
+        elsif !connected?
+          "Disconnected"
+        else
+          "Input"
+        end
 
         widget = tui.paragraph(
           text: tui.line(spans: [
             tui.span(content: "> #{@input}#{cursor}", style: text_style)
           ]),
           block: tui.block(
-            title: @loading ? "Waiting..." : "Input",
-            titles: @loading ? [] : [
+            title: title,
+            titles: disabled ? [] : [
               {content: "Enter send", position: :bottom, alignment: :center}
             ],
             borders: [:all],
@@ -162,42 +250,65 @@ module TUI
       def submit_message
         text = @input.strip
         return if text.empty?
+        return unless @cable_client.status == :subscribed
 
-        Events::Bus.emit(Events::UserMessage.new(content: text, session_id: @session.id))
         @input = ""
-        @loading = true
-
-        @submit_thread = Thread.new do
-          @client ||= LLM::Client.new
-          @registry ||= build_tool_registry
-          viewport_messages = @session.messages_for_llm
-          response = @client.chat_with_tools(
-            viewport_messages,
-            registry: @registry,
-            session_id: @session.id
-          )
-          Events::Bus.emit(Events::AgentMessage.new(content: response, session_id: @session.id))
-        rescue => e
-          Events::Bus.emit(Events::AgentMessage.new(content: "Error: #{e.message}", session_id: @session.id))
-        ensure
-          @loading = false
-        end
+        @cable_client.speak(text)
       end
 
-      def build_tool_registry
-        registry = Tools::Registry.new(context: {shell_session: @shell_session})
-        registry.register(Tools::WebGet)
-        registry.register(Tools::Bash)
-        registry
+      # @return [Boolean] whether the event is an arrow or page key used for scrolling
+      def scroll_key?(event)
+        event.up? || event.down? || event.page_up? || event.page_down?
       end
 
-      def load_session_messages
-        @session.events.where(event_type: Events::Subscribers::MessageCollector::DISPLAYABLE_TYPES).each do |event|
-          @message_collector.messages_push({
-            role: Events::Subscribers::MessageCollector::ROLE_MAP.fetch(event.event_type),
-            content: event.payload["content"].to_s
-          })
+      # Dispatches scroll key events to {#scroll_up} or {#scroll_down}
+      # @return [true] always redraws after scrolling
+      def handle_scroll_key(event)
+        if event.up?
+          scroll_up(SCROLL_STEP)
+        elsif event.down?
+          scroll_down(SCROLL_STEP)
+        elsif event.page_up?
+          scroll_up(@visible_height)
+        elsif event.page_down?
+          scroll_down(@visible_height)
         end
+        true
+      end
+
+      # Handles mouse wheel scroll events; ignores other mouse events
+      # @return [Boolean] true if the event was a scroll wheel event
+      def handle_mouse_event(event)
+        if event.scroll_up?
+          scroll_up(MOUSE_SCROLL_STEP)
+          true
+        elsif event.scroll_down?
+          scroll_down(MOUSE_SCROLL_STEP)
+          true
+        else
+          false
+        end
+      end
+
+      # Scrolls the viewport up, clamping at the top.
+      # Disables auto-scroll when the user moves away from the bottom.
+      # @param lines [Integer] number of lines to scroll
+      def scroll_up(lines)
+        @scroll_offset = [@scroll_offset - lines, 0].max
+        @auto_scroll = @scroll_offset >= @max_scroll
+      end
+
+      # Scrolls the viewport down, clamping at max_scroll.
+      # Re-enables auto-scroll when the user reaches the bottom.
+      # @param lines [Integer] number of lines to scroll
+      def scroll_down(lines)
+        @scroll_offset = [@scroll_offset + lines, @max_scroll].min
+        @auto_scroll = @scroll_offset >= @max_scroll
+      end
+
+      # @return [Boolean] true when WebSocket is fully subscribed and ready
+      def connected?
+        @cable_client.status == :subscribed
       end
 
       def printable_char?(event)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: anima-core
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Yevhenii Hurin
@@ -9,6 +9,20 @@ bindir: exe
 cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: foreman
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.88'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.88'
 - !ruby/object:Gem::Dependency
   name: httparty
   requirement: !ruby/object:Gem::Requirement
@@ -23,6 +37,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.24'
+- !ruby/object:Gem::Dependency
+  name: puma
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
 - !ruby/object:Gem::Dependency
   name: rails
   requirement: !ruby/object:Gem::Requirement
@@ -51,6 +79,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.4'
+- !ruby/object:Gem::Dependency
+  name: solid_cable
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
 - !ruby/object:Gem::Dependency
   name: solid_queue
   requirement: !ruby/object:Gem::Requirement
@@ -79,6 +121,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: websocket-client-simple
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.8'
 email:
 - evgeny.gurin@gmail.com
 executables:
@@ -89,25 +145,43 @@ files:
 - ".mise.toml"
 - ".reek.yml"
 - CHANGELOG.md
+- Gemfile
 - LICENSE.txt
+- Procfile
+- Procfile.dev
 - README.md
 - Rakefile
+- anima-core.gemspec
+- app/channels/application_cable/channel.rb
+- app/channels/application_cable/connection.rb
+- app/channels/session_channel.rb
+- app/controllers/api/sessions_controller.rb
+- app/controllers/application_controller.rb
+- app/jobs/agent_request_job.rb
 - app/jobs/application_job.rb
 - app/jobs/count_event_tokens_job.rb
 - app/models/application_record.rb
 - app/models/event.rb
 - app/models/session.rb
+- bin/jobs
+- bin/rails
+- bin/rake
+- config.ru
 - config/application.rb
 - config/boot.rb
+- config/cable.yml
 - config/database.yml
 - config/environment.rb
 - config/environments/development.rb
 - config/environments/production.rb
 - config/environments/test.rb
+- config/initializers/event_subscribers.rb
 - config/initializers/inflections.rb
+- config/puma.rb
 - config/queue.yml
 - config/recurring.yml
 - config/routes.rb
+- db/cable_schema.rb
 - db/migrate/.keep
 - db/migrate/20260308124202_create_sessions.rb
 - db/migrate/20260308124203_create_events.rb
@@ -118,6 +192,7 @@ files:
 - db/queue_schema.rb
 - db/seeds.rb
 - exe/anima
+- lib/agent_loop.rb
 - lib/anima.rb
 - lib/anima/cli.rb
 - lib/anima/installer.rb
@@ -126,6 +201,7 @@ files:
 - lib/events/base.rb
 - lib/events/bus.rb
 - lib/events/subscriber.rb
+- lib/events/subscribers/action_cable_bridge.rb
 - lib/events/subscribers/message_collector.rb
 - lib/events/subscribers/persister.rb
 - lib/events/system_message.rb
@@ -140,6 +216,8 @@ files:
 - lib/tools/registry.rb
 - lib/tools/web_get.rb
 - lib/tui/app.rb
+- lib/tui/cable_client.rb
+- lib/tui/message_store.rb
 - lib/tui/screens/anthropic.rb
 - lib/tui/screens/chat.rb
 - lib/tui/screens/settings.rb
@@ -166,6 +244,5 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 rubygems_version: 3.6.9
 specification_version: 4
-summary: Ruby framework for building AI agents with desires, personality, and personal
-  growth
+summary: A personal AI agent with desires, personality, and personal growth
 test_files: []