kward 0.66.0 → 0.67.0

data/lib/kward/prompt_interface.rb

@@ -4,8 +4,7 @@ require "tty-cursor"
 require "tty-reader"
 require "tty-screen"
 require_relative "ansi"
-require_relative "resources/pixel_logo"
-require_relative "resources/avatar_kward_logo"
+require_relative "prompt_interface/banner"
 
 module Kward
   class PromptInterface
@@ -16,11 +15,8 @@ module Kward
     FOOTER_REFRESH_INTERVAL = 1.0
     COMPOSER_MAX_INPUT_ROWS = 6
     TRANSCRIPT_BUFFER_LIMIT = 200_000
-    BANNER_LOGO_WIDTH = 32
-    BANNER_LOGO_PIXEL_HEIGHT = 32
-    BANNER_MIN_LOGO_HEIGHT = 4
-    BANNER_LOGO_PIXELS = Kward::Resources::AvatarKwardLogo::PIXELS
-    BANNER_MESSAGE = "State your business.".freeze
+    BANNER_LOGO_PIXELS = Banner::LOGO_PIXELS
+    BANNER_MESSAGE = Banner::MESSAGE
     KEYBOARD_PROTOCOL_ENABLE = "\e[>1u".freeze
     KEYBOARD_PROTOCOL_RESTORE = "\e[<u".freeze
     BRACKETED_PASTE_ENABLE = "\e[?2004h".freeze
@@ -65,10 +61,13 @@ module Kward
       @assistant_label = "Assistant"
       @stream_block = nil
       @rendered_rows = 0
+      @last_composer_rows = []
       @cursor_rendered_row = 0
       @stream_col = 0
       @stream_pending_wrap = false
       @transcript_buffer = +""
+      @transcript_display_rows_cache_width = nil
+      @transcript_display_rows_cache = nil
       @visual_banner_count = 0
       @transcript_viewport_rows = 0
       @restoring_transcript = false
@@ -80,6 +79,7 @@ module Kward
       @history = []
       @history_index = nil
       @history_draft = nil
+      @prefill_input = nil
       @slash_commands = normalize_slash_commands(slash_commands)
       @slash_selection_index = 0
       @slash_overlay_dismissed_input = nil
@@ -90,15 +90,14 @@ module Kward
       @reserved_rows = 0
       @color_enabled = ANSI.enabled?(output)
       @cursor_visible = true
+      @synchronized_output_depth = 0
       @overlay_settings = normalize_overlay_settings(overlay_settings)
       @footer = footer
       @composer_status = composer_status
       @busy_help = busy_help
       @attachment_badges = attachment_badges
       @attachment_parser = attachment_parser
-      @banner_message = banner_message.to_s
-      @banner_logo_pixels = banner_pixels
-      @banner_logo_cache = {}
+      @banner = Banner.new(message: banner_message, pixels: banner_pixels, screen_height: method(:screen_height))
     end
 
     def start
@@ -140,11 +139,13 @@ module Kward
           next
         end
 
-        clear_prompt_for_output_locked
-        write_transcript_text_locked(text)
-        write_transcript_text_locked("\n") unless text.end_with?("\n")
-        @stream_block = nil
-        render_prompt_after_output_locked
+        with_synchronized_output_locked do
+          clear_prompt_for_output_locked
+          write_transcript_text_locked(text)
+          write_transcript_text_locked("\n") unless text.end_with?("\n")
+          @stream_block = nil
+          render_prompt_after_output_locked
+        end
         @output_io.flush
       end
     end
@@ -153,12 +154,14 @@ module Kward
       @mutex.synchronize do
         return if @restoring_transcript
 
-        clear_prompt_for_output_locked
-        text = message.to_s
-        write_visual_transcript_text_locked(text)
-        write_visual_transcript_text_locked("\n") unless text.end_with?("\n")
-        @stream_block = nil
-        render_prompt_after_output_locked
+        with_synchronized_output_locked do
+          clear_prompt_for_output_locked
+          text = message.to_s
+          write_visual_transcript_text_locked(text)
+          write_visual_transcript_text_locked("\n") unless text.end_with?("\n")
+          @stream_block = nil
+          render_prompt_after_output_locked
+        end
         @output_io.flush
       end
     end
@@ -168,6 +171,7 @@ module Kward
         clear_prompt_for_output_locked
         @output_io.print(SYNCHRONIZED_OUTPUT_ENABLE)
         @transcript_buffer = +""
+        invalidate_transcript_display_rows_cache
         @visual_banner_count = 0
         @transcript_viewport_rows = 0
         @stream_block = nil
@@ -181,7 +185,8 @@ module Kward
       @mutex.synchronize do
         @restoring_transcript = false
         @output_io.print(SYNCHRONIZED_OUTPUT_DISABLE)
-        redraw_screen_locked
+        width, height = screen_size
+        redraw_screen_locked(width: width, height: height)
         @output_io.flush
       end
     end
@@ -193,8 +198,9 @@ module Kward
         preserve_input = was_composing && !@busy && !@input.empty?
         @prompt_label = message.to_s
         unless preserve_input
-          @input = ""
-          @cursor = 0
+          @input = @prefill_input.to_s
+          @prefill_input = nil
+          @cursor = @input.length
           @attachments.clear
           reset_history_navigation
         end
@@ -235,7 +241,7 @@ module Kward
       answer.start_with?("y")
     end
 
-    def select(message, choices, title: "Sessions", custom: false)
+    def select(message, choices, title: "Sessions", custom: false, initial_index: 0)
       return nil if choices.empty? && !custom
 
       start
@@ -248,7 +254,9 @@ module Kward
         @asking = true
         @busy = false
         @queued_count = 0
-        @select_state = { choices: choices.map(&:to_s), selection_index: 0, title: title.to_s, custom: custom }
+        choice_labels = choices.map(&:to_s)
+        selection_index = choice_labels.empty? ? 0 : [[initial_index.to_i, 0].max, choice_labels.length - 1].min
+        @select_state = { choices: choice_labels, selection_index: selection_index, title: title.to_s, custom: custom }
         reset_history_navigation
         render_prompt_locked
       end
@@ -390,59 +398,54 @@ module Kward
 
     def print_visual_banner
       @mutex.synchronize do
-        rows = banner_rows(screen_width)
+        width, height = screen_size
+        rows = banner_rows(width)
         return if rows.empty?
 
-        prepare_transcript_output_locked
-        rows.each do |row|
-          write_visual_transcript_text_locked(row)
-          write_visual_transcript_text_locked("\n")
+        with_synchronized_output_locked do
+          prepare_transcript_output_locked
+          rows.each do |row|
+            write_visual_transcript_text_locked(row)
+            write_visual_transcript_text_locked("\n")
+          end
+          @visual_banner_count += 1
+          invalidate_transcript_display_rows_cache
+          remember_transcript_viewport_locked(height)
+          @stream_block = nil
+          restore_composer_cursor_locked
         end
-        @visual_banner_count += 1
-        remember_transcript_viewport_locked
-        @stream_block = nil
-        restore_composer_cursor_locked
         @output_io.flush
       end
     end
 
     def start_stream_block(label)
       @mutex.synchronize do
-        if @stream_block != label
-          prepare_transcript_output_locked unless @restoring_transcript
-          ensure_transcript_block_separator_locked
-          write_transcript_text_locked("#{colored("#{transcript_label(label)}>", label_color(label), :bold)}\n")
-          @stream_block = label
-        end
-        restore_composer_cursor_locked unless @restoring_transcript
-        @output_io.flush unless @restoring_transcript
+        write_stream_block_locked(label, "", finish: false)
       end
     end
 
     def write_delta(delta)
       @mutex.synchronize do
-        prepare_transcript_output_locked unless @restoring_transcript
-        write_transcript_text_locked(delta.to_s)
-        restore_composer_cursor_locked unless @restoring_transcript
-        @output_io.flush unless @restoring_transcript
+        write_stream_block_locked(nil, delta.to_s, finish: false)
       end
     end
 
     def finish_stream_block
       @mutex.synchronize do
-        prepare_transcript_output_locked unless @restoring_transcript
-        if @stream_block
-          write_transcript_text_locked("\n")
-        end
-        @stream_block = nil
-        restore_composer_cursor_locked unless @restoring_transcript
-        @output_io.flush unless @restoring_transcript
+        write_stream_block_locked(nil, "", finish: true)
+      end
+    end
+
+    def write_stream_block(label, delta, finish: false)
+      @mutex.synchronize do
+        write_stream_block_locked(label, delta.to_s, finish: finish)
       end
     end
 
     def redraw
       @mutex.synchronize do
-        redraw_screen_locked
+        width, height = screen_size
+        with_synchronized_output_locked { redraw_screen_locked(width: width, height: height) }
         @output_io.flush
       end
     end
@@ -450,12 +453,14 @@ module Kward
     def clear_transcript
       @mutex.synchronize do
         @transcript_buffer = +""
+        invalidate_transcript_display_rows_cache
         @visual_banner_count = 0
         @transcript_viewport_rows = 0
         @stream_block = nil
         @stream_col = 0
         @stream_pending_wrap = false
-        redraw_screen_locked
+        width, height = screen_size
+        with_synchronized_output_locked { redraw_screen_locked(width: width, height: height) }
         @output_io.flush
       end
     end
@@ -486,6 +491,39 @@ module Kward
       @raw_mode_active = false
     end
 
+    def write_stream_block_locked(label, delta, finish: false)
+      with_synchronized_output_locked do
+        prepare_transcript_output_locked unless @restoring_transcript
+        if label && @stream_block != label
+          ensure_transcript_block_separator_locked
+          write_transcript_text_locked("#{colored("#{transcript_label(label)}>", label_color(label), :bold)}\n")
+          @stream_block = label
+        end
+        write_transcript_text_locked(delta) unless delta.empty?
+        write_transcript_text_locked("\n") if finish && @stream_block
+        @stream_block = nil if finish
+        restore_composer_cursor_locked unless @restoring_transcript
+      end
+      @output_io.flush unless @restoring_transcript
+    end
+
+    def with_synchronized_output_locked
+      if @restoring_transcript || @synchronized_output_depth.positive?
+        yield
+        return
+      end
+
+      synchronized = true
+      @synchronized_output_depth += 1
+      @output_io.print(SYNCHRONIZED_OUTPUT_ENABLE)
+      yield
+    ensure
+      if synchronized
+        @synchronized_output_depth -= 1
+        @output_io.print(SYNCHRONIZED_OUTPUT_DISABLE) if @synchronized_output_depth.zero?
+      end
+    end
+
     def write_transcript_text_locked(text)
       append_transcript_buffer(text.to_s)
       remember_transcript_viewport_locked unless text.to_s.empty?
@@ -493,19 +531,26 @@ module Kward
     end
 
     def write_visual_transcript_text_locked(text)
+      width, height = screen_size
       output_text = terminal_newlines(text.to_s)
-      advance_pending_stream_wrap_locked(output_text)
+      advance_pending_stream_wrap_locked(output_text, width: width, height: height)
       @output_io.print(output_text)
-      update_stream_position(output_text)
+      update_stream_position(output_text, width: width)
     end
 
     def append_transcript_buffer(text)
       @transcript_buffer << ANSI.sanitize_transcript(text)
+      invalidate_transcript_display_rows_cache
       return if @transcript_buffer.length <= TRANSCRIPT_BUFFER_LIMIT
 
       @transcript_buffer = @transcript_buffer[-TRANSCRIPT_BUFFER_LIMIT, TRANSCRIPT_BUFFER_LIMIT]
     end
 
+    def invalidate_transcript_display_rows_cache
+      @transcript_display_rows_cache_width = nil
+      @transcript_display_rows_cache = nil
+    end
+
     def ensure_transcript_block_separator_locked
       return if @transcript_buffer.empty? || @transcript_buffer.end_with?("\n\n")
 
@@ -1618,6 +1663,12 @@ module Kward
       @cursor = @input.length
     end
 
+    def prefill_input(value)
+      @mutex.synchronize do
+        @prefill_input = value.to_s
+      end
+    end
+
     def reset_history_navigation
       @history_index = nil
       @history_draft = nil
@@ -1698,14 +1749,15 @@ module Kward
       return unless @started && @asking
 
       handle_resize_locked
-      rows, cursor_row, cursor_col = composer_layout(screen_width)
-      ensure_scroll_region_locked(rows.length)
+      width, height = screen_size
+      rows, cursor_row, cursor_col = composer_layout(width, height)
+      ensure_scroll_region_locked(rows.length, width: width, height: height)
       @rendered_rows = rows.length
-      render_composer_rows_locked(rows)
+      render_composer_rows_locked(rows, height: height)
       @cursor_rendered_row = cursor_row
-      @last_width = screen_width
-      @last_height = screen_height
-      move_to_screen(composer_top_row + cursor_row, cursor_col + 1)
+      @last_width = width
+      @last_height = height
+      move_to_screen(composer_top_row(height) + cursor_row, cursor_col + 1)
       render_cursor_visibility_locked
       @output_io.flush
     end
@@ -1716,26 +1768,29 @@ module Kward
 
     def clear_prompt_locked
       handle_resize_locked
-      clear_composer_region_locked
+      width, height = screen_size
+      clear_composer_region_locked(height: height)
       @rendered_rows = 0
       @cursor_rendered_row = 0
-      redraw_transcript_locked
+      redraw_transcript_locked(width: width, height: height)
     end
 
     def clear_prompt_for_output_locked
       handle_resize_locked
-      reserve_composer_region_locked if @started && @asking
-      clear_composer_region_locked
+      width, height = screen_size
+      reserve_composer_region_locked(width: width, height: height) if @started && @asking
+      clear_composer_region_locked(height: height)
       @rendered_rows = 0
       @cursor_rendered_row = 0
-      move_to_transcript_cursor_locked if @started
+      move_to_transcript_cursor_locked(width: width, height: height) if @started
     end
 
     def prepare_transcript_output_locked
       handle_resize_locked
+      width, height = screen_size
       hide_cursor_for_transcript_output_locked
-      reserve_composer_region_locked
-      move_to_transcript_cursor_locked
+      reserve_composer_region_locked(width: width, height: height)
+      move_to_transcript_cursor_locked(width: width, height: height)
     end
 
     def hide_cursor_for_transcript_output_locked
@@ -1747,8 +1802,9 @@ module Kward
     def restore_composer_cursor_locked
       return unless @started && @asking
 
-      _rows, cursor_row, cursor_col = composer_layout(screen_width)
-      move_to_screen(composer_top_row + cursor_row, cursor_col + 1)
+      width, height = screen_size
+      _rows, cursor_row, cursor_col = composer_layout(width, height)
+      move_to_screen(composer_top_row(height) + cursor_row, cursor_col + 1)
       render_cursor_visibility_locked
     end
 
@@ -1764,26 +1820,25 @@ module Kward
       @cursor_visible = visible
     end
 
-    def reserve_composer_region_locked
-      rows, = composer_layout(screen_width)
-      ensure_scroll_region_locked(rows.length)
+    def reserve_composer_region_locked(width: screen_width, height: screen_height)
+      rows, = composer_layout(width, height)
+      ensure_scroll_region_locked(rows.length, width: width, height: height)
     end
 
-    def ensure_scroll_region_locked(row_count, redraw_transcript: true)
-      new_reserved_rows = [[row_count, 1].max, [screen_height - 1, 1].max].min
-      return if @reserved_rows == new_reserved_rows && @last_height == screen_height
+    def ensure_scroll_region_locked(row_count, redraw_transcript: true, width: screen_width, height: screen_height)
+      new_reserved_rows = [[row_count, 1].max, [height - 1, 1].max].min
+      return if @reserved_rows == new_reserved_rows && @last_height == height
 
       old_reserved_rows = @reserved_rows
       rows_to_clear = [old_reserved_rows, new_reserved_rows].max
       @reserved_rows = new_reserved_rows
-      @output_io.print("\e[1;#{transcript_bottom_row}r")
-      clear_composer_region_locked(rows_to_clear)
-      redraw_transcript_locked if redraw_transcript && new_reserved_rows < old_reserved_rows
+      @output_io.print("\e[1;#{transcript_bottom_row(height)}r")
+      clear_composer_region_locked(rows_to_clear, height: height)
+      redraw_transcript_locked(width: width, height: height) if redraw_transcript && new_reserved_rows < old_reserved_rows
     end
 
     def handle_resize_locked
-      current_width = screen_width
-      current_height = screen_height
+      current_width, current_height = screen_size
       return false if current_width == @last_width && current_height == @last_height
 
       old_width = @last_width
@@ -1795,7 +1850,7 @@ module Kward
       @reserved_rows = 0
       @last_width = current_width
       @last_height = current_height
-      redraw_screen_locked
+      redraw_screen_locked(width: current_width, height: current_height)
       true
     end
 
@@ -1804,18 +1859,33 @@ module Kward
       @reserved_rows = 0
     end
 
-    def render_composer_rows_locked(rows)
-      clear_composer_region_locked
-      top = composer_top_row
-      rows.each_with_index do |row, index|
+    def render_composer_rows_locked(rows, height: screen_height)
+      top = composer_top_row(height)
+      max_rows = [@last_composer_rows.length, rows.length].max
+      rows_to_clear = [@reserved_rows - rows.length, 0].max
+
+      max_rows.times do |index|
+        row = rows[index]
+        previous = @last_composer_rows[index]
+        next if row == previous
+
         move_to_screen(top + index, 1)
-        @output_io.print(row) unless row.empty?
+        @output_io.print(TTY::Cursor.clear_line)
+        @output_io.print(row) unless row.to_s.empty?
       end
+
+      rows.length.upto(rows.length + rows_to_clear - 1) do |index|
+        move_to_screen(top + index, 1)
+        @output_io.print(TTY::Cursor.clear_line)
+      end
+
+      @last_composer_rows = rows.dup
     end
 
-    def clear_composer_region_locked(rows_to_clear = nil)
+    def clear_composer_region_locked(rows_to_clear = nil, height: screen_height)
       rows_to_clear ||= [@reserved_rows, @rendered_rows].max
-      clear_bottom_rows_locked(screen_height, rows_to_clear)
+      clear_bottom_rows_locked(height, rows_to_clear)
+      @last_composer_rows = []
     end
 
     def resize_prompt_clear_rows(old_width, current_width, old_reserved_rows)
@@ -1838,7 +1908,7 @@ module Kward
     def clear_bottom_rows_locked(height, rows_to_clear)
       return unless rows_to_clear.positive?
 
-      bottom = [height, screen_height].min
+      bottom = height
       top = [bottom - rows_to_clear + 1, 1].max
       clear_screen_rows_locked(top, bottom)
     end
@@ -1850,32 +1920,33 @@ module Kward
       end
     end
 
-    def redraw_screen_locked
+    def redraw_screen_locked(width: screen_width, height: screen_height)
       return unless @started
 
       restore_scroll_region_locked
       @output_io.print(TTY::Cursor.clear_screen)
       move_to_screen(1, 1)
       @reserved_rows = 0
-      rows, cursor_row, cursor_col = composer_layout(screen_width)
-      ensure_scroll_region_locked(rows.length, redraw_transcript: false)
-      redraw_transcript_locked
+      @last_composer_rows = []
+      rows, cursor_row, cursor_col = composer_layout(width, height)
+      ensure_scroll_region_locked(rows.length, redraw_transcript: false, width: width, height: height)
+      redraw_transcript_locked(width: width, height: height)
       @rendered_rows = @asking ? rows.length : 0
-      render_composer_rows_locked(rows) if @asking
+      render_composer_rows_locked(rows, height: height) if @asking
       @cursor_rendered_row = @asking ? cursor_row : 0
-      @last_width = screen_width
-      @last_height = screen_height
-      reset_stream_position_from_transcript_locked
+      @last_width = width
+      @last_height = height
+      reset_stream_position_from_transcript_locked(width)
       if @asking
-        move_to_screen(composer_top_row + cursor_row, cursor_col + 1)
+        move_to_screen(composer_top_row(height) + cursor_row, cursor_col + 1)
         render_cursor_visibility_locked
       end
     end
 
-    def redraw_transcript_locked
+    def redraw_transcript_locked(width: screen_width, height: screen_height)
       return unless transcript_renderable?
 
-      rows = transcript_viewport_rows(transcript_redraw_row_count, screen_width)
+      rows = transcript_viewport_rows(transcript_redraw_row_count(height), width)
       clear_screen_rows_locked(1, rows.length)
       return if rows.empty?
 
@@ -1895,12 +1966,12 @@ module Kward
       rows
     end
 
-    def transcript_redraw_row_count
-      [[@transcript_viewport_rows, transcript_bottom_row].max, screen_height].min
+    def transcript_redraw_row_count(height = screen_height)
+      [[@transcript_viewport_rows, transcript_bottom_row(height)].max, height].min
     end
 
-    def remember_transcript_viewport_locked
-      @transcript_viewport_rows = transcript_bottom_row
+    def remember_transcript_viewport_locked(height = screen_height)
+      @transcript_viewport_rows = transcript_bottom_row(height)
     end
 
     def transcript_renderable?
@@ -1908,11 +1979,14 @@ module Kward
     end
 
     def transcript_display_rows(width)
+      return @transcript_display_rows_cache if @transcript_display_rows_cache_width == width && @transcript_display_rows_cache
+
       rows = []
       @visual_banner_count.times { rows.concat(banner_rows(width)) }
       rows << "" if @visual_banner_count.positive? && @transcript_buffer.empty?
       rows.concat(transcript_text_display_rows(width))
-      rows
+      @transcript_display_rows_cache_width = width
+      @transcript_display_rows_cache = rows
     end
 
     def transcript_text_display_rows(width)
@@ -1922,8 +1996,7 @@ module Kward
       end
     end
 
-    def reset_stream_position_from_transcript_locked
-      width = screen_width
+    def reset_stream_position_from_transcript_locked(width = screen_width)
       rows = transcript_display_rows(width)
       last_length = rows.empty? ? 0 : ANSI.strip(rows.last).length
       if last_length >= width
@@ -1935,34 +2008,34 @@ module Kward
       end
     end
 
-    def move_to_transcript_cursor_locked
+    def move_to_transcript_cursor_locked(width: screen_width, height: screen_height)
       if @stream_pending_wrap
-        move_to_screen(transcript_bottom_row, screen_width)
+        move_to_screen(transcript_bottom_row(height), width)
       else
-        move_to_screen(transcript_bottom_row, [@stream_col + 1, screen_width].min)
+        move_to_screen(transcript_bottom_row(height), [@stream_col + 1, width].min)
       end
     end
 
-    def advance_pending_stream_wrap_locked(output_text)
+    def advance_pending_stream_wrap_locked(output_text, width: screen_width, height: screen_height)
       return unless @stream_pending_wrap
       return if output_text.empty? || output_text.start_with?("\r", "\n")
 
-      move_to_screen(transcript_bottom_row, screen_width)
+      move_to_screen(transcript_bottom_row(height), width)
       @output_io.print("\r\n")
       @stream_col = 0
       @stream_pending_wrap = false
     end
 
-    def composer_layout(width)
-      return compact_composer_layout(width) if screen_height < 4
-      return question_composer_layout(width) if @question_state
+    def composer_layout(width, height = screen_height)
+      return compact_composer_layout(width) if height < 4
+      return question_composer_layout(width, height) if @question_state
 
       content_width = [width - 4, 1].max
       input_layout_rows, input_cursor_row, input_cursor_col = input_layout(content_width)
       attachment_rows = attachment_badge_rows(content_width)
-      overlay_rows = active_overlay_rows(width)
+      overlay_rows = active_overlay_rows(width, height: height)
       footer_text = footer_text()
-      max_input_rows = max_visible_input_rows(attachment_rows.length, overlay_rows.length, footer_text.empty? ? 0 : 1)
+      max_input_rows = max_visible_input_rows(attachment_rows.length, overlay_rows.length, footer_text.empty? ? 0 : 1, height: height)
       visible_start = [[input_cursor_row - max_input_rows + 1, 0].max, [input_layout_rows.length - max_input_rows, 0].max].min
       visible_rows = input_layout_rows[visible_start, max_input_rows] || [""]
       rows = overlay_rows + [top_border(width)]
@@ -1975,75 +2048,28 @@ module Kward
       [rows, cursor_row, cursor_col]
     end
 
-    def question_composer_layout(width)
+    def question_composer_layout(width, height = screen_height)
       content_width = [width - 4, 1].max
-      overlay_rows = active_overlay_rows(width)
+      overlay_rows = active_overlay_rows(width, height: height)
       rows = overlay_rows + [top_border(width), box_content_row("", content_width), bottom_border(width)]
       return [rows, question_custom_cursor_row, question_custom_cursor_col(width)] if selected_question_choice&.fetch(:custom, false)
 
       [rows, overlay_rows.length + 1, 2]
     end
 
-    def active_overlay_rows(width)
+    def active_overlay_rows(width, height: screen_height)
       return question_overlay_rows(width) if @question_state
-      return selection_overlay_rows(width) if @select_state
+      return selection_overlay_rows(width, height: height) if @select_state
 
-      slash_overlay_rows(width)
+      slash_overlay_rows(width, height: height)
     end
 
     def banner_rows(width)
-      return [] unless banner_visible?
-
-      rows = []
-      if banner_image_visible?
-        rows.concat(centered_banner_image_rows(width))
-      end
-      rows << align_plain_row(@banner_message, width) unless @banner_message.empty?
-      rows << ""
-      rows
-    end
-
-    def banner_visible?
-      !@banner_message.empty? || banner_image_visible?
-    end
-
-    def banner_image_visible?
-      !banner_logo_rows.empty?
-    end
-
-    def centered_banner_image_rows(width)
-      logo_width, = banner_logo_dimensions(width)
-      padding = [[(width - logo_width) / 2, 0].max, width - 1].min
-      banner_logo_rows.map { |row| (" " * padding) + row }
+      @banner.rows(width)
     end
 
     def banner_logo_rows
-      logo_width, logo_height = banner_logo_dimensions(screen_width)
-      return [] unless @banner_logo_pixels && max_banner_logo_height >= BANNER_MIN_LOGO_HEIGHT
-
-      key = [logo_width, logo_height]
-      @banner_logo_cache[key] ||= Kward::PixelLogo.half_block_rows_from_pixels(@banner_logo_pixels, width: logo_width, pixel_height: logo_height)
-    end
-
-    def banner_logo_dimensions(width)
-      logo_width = [BANNER_LOGO_WIDTH, [width - 2, 1].max].min
-      logo_height = [BANNER_LOGO_PIXEL_HEIGHT, max_banner_logo_height * 2].min
-      [logo_width, logo_height]
-    end
-
-    def max_banner_logo_height
-      message_rows = @banner_message.empty? ? 0 : 1
-      blank_after_banner = 1
-      minimum_composer_rows = 3
-      transcript_row = 1
-      reserved_rows = message_rows + blank_after_banner + minimum_composer_rows + transcript_row
-      [screen_height - reserved_rows, 0].max
-    end
-
-    def align_plain_row(text, width)
-      plain_length = ANSI.strip(text).length
-      padding = [width - plain_length, 0].max / 2
-      (" " * padding) + text.to_s
+      @banner.logo_rows(screen_width)
     end
 
     def question_overlay_rows(width)
@@ -2060,10 +2086,10 @@ module Kward
       overlay_card_rows(title, lines, width)
     end
 
-    def slash_overlay_rows(width)
+    def slash_overlay_rows(width, height: screen_height)
       return [] unless slash_overlay_visible?
 
-      visible = visible_slash_overlay_matches(slash_overlay_matches)
+      visible = visible_slash_overlay_matches(slash_overlay_matches, height: height)
       start_index = visible[:start]
       lines = visible[:commands].each_with_index.map do |command, offset|
         index = start_index + offset
@@ -2074,13 +2100,13 @@ module Kward
       overlay_card_rows("Slash commands", lines, width)
     end
 
-    def visible_slash_overlay_matches(matches)
-      max_rows = [[screen_height - 7, 1].max, 8].min
+    def visible_slash_overlay_matches(matches, height: screen_height)
+      max_rows = [[height - 7, 1].max, 8].min
       start = [[@slash_selection_index - max_rows + 1, 0].max, [matches.length - max_rows, 0].max].min
       { start: start, commands: matches[start, max_rows] || [] }
     end
 
-    def selection_overlay_rows(width)
+    def selection_overlay_rows(width, height: screen_height)
       matches = selection_matches
       lines = [overlay_text_line("↑/↓ select · Enter open · Esc cancel", :muted), overlay_blank_line]
       if matches.empty?
@@ -2092,7 +2118,7 @@ module Kward
         return overlay_card_rows(selection_overlay_title, lines, width)
       end
 
-      visible = visible_selection_matches(matches)
+      visible = visible_selection_matches(matches, height: height)
       start_index = visible[:start]
       visible[:choices].each_with_index do |choice, offset|
         index = start_index + offset
@@ -2106,8 +2132,8 @@ module Kward
       title && !title.empty? ? title : "Sessions"
     end
 
-    def visible_selection_matches(matches)
-      max_rows = [[screen_height - 7, 1].max, 8].min
+    def visible_selection_matches(matches, height: screen_height)
+      max_rows = [[height - 7, 1].max, 8].min
       start = [[selection_index - max_rows + 1, 0].max, [matches.length - max_rows, 0].max].min
       { start: start, choices: matches[start, max_rows] || [] }
     end
@@ -2354,17 +2380,17 @@ module Kward
       []
     end
 
-    def max_visible_input_rows(attachment_count = 0, overlay_count = active_overlay_rows(screen_width).length, footer_count = footer_text.to_s.empty? ? 0 : 1)
+    def max_visible_input_rows(attachment_count = 0, overlay_count = active_overlay_rows(screen_width).length, footer_count = footer_text.to_s.empty? ? 0 : 1, height: screen_height)
       input_cap = [COMPOSER_MAX_INPUT_ROWS - attachment_count, 1].max
-      [[input_cap, screen_height - 3 - overlay_count - footer_count - attachment_count].min, 1].max
+      [[input_cap, height - 3 - overlay_count - footer_count - attachment_count].min, 1].max
     end
 
-    def composer_top_row
-      [screen_height - @reserved_rows + 1, 1].max
+    def composer_top_row(height = screen_height)
+      [height - @reserved_rows + 1, 1].max
     end
 
-    def transcript_bottom_row
-      [screen_height - @reserved_rows, 1].max
+    def transcript_bottom_row(height = screen_height)
+      [height - @reserved_rows, 1].max
     end
 
     def move_to_screen(row, col)
@@ -2385,8 +2411,7 @@ module Kward
       [before_cursor.count("\n"), (before_cursor.split("\n", -1).last || "").length]
     end
 
-    def update_stream_position(text)
-      width = screen_width
+    def update_stream_position(text, width: screen_width)
       ANSI.strip(text).each_char do |char|
         case char
         when "\n", "\r"
@@ -2426,6 +2451,10 @@ module Kward
       end
     end
 
+    def screen_size
+      [screen_width, screen_height]
+    end
+
     def screen_width
       [TTY::Screen.width, 1].max
     end