kward 0.71.0 → 0.73.0

data/lib/kward/prompt_interface.rb

@@ -1,21 +1,46 @@
+require "find"
 require "io/console"
+require "pathname"
+require "rbconfig"
 require "thread"
+require_relative "project_files"
+require_relative "prompt_history"
 require "tty-cursor"
 require "tty-reader"
 require "tty-screen"
 require_relative "ansi"
+require_relative "terminal_sequences"
+require_relative "terminal_keys"
+require_relative "editor_mode"
 require_relative "prompt_interface/banner"
 require_relative "prompt_interface/composer_state"
+require_relative "prompt_interface/editor/state"
 require_relative "prompt_interface/transcript_buffer"
 require_relative "prompt_interface/transcript_renderer"
 require_relative "prompt_interface/prompt_renderer"
 require_relative "prompt_interface/stream_state"
 require_relative "prompt_interface/slash_overlay"
+require_relative "prompt_interface/file_overlay"
+require_relative "prompt_interface/project_browser"
 require_relative "prompt_interface/selection_prompt"
 require_relative "prompt_interface/question_prompt"
+require_relative "prompt_interface/git_prompt"
 require_relative "prompt_interface/overlay_renderer"
+require_relative "prompt_interface/editor/renderer"
+require_relative "prompt_interface/editor/syntax_highlighter"
+require_relative "prompt_interface/editor/auto_close_pairs"
+require_relative "prompt_interface/editor/endwise"
+require_relative "prompt_interface/editor/auto_indent"
 require_relative "prompt_interface/composer_renderer"
 require_relative "prompt_interface/composer_controller"
+require_relative "prompt_interface/editor/modes/modern"
+require_relative "prompt_interface/editor/modes/emacs"
+require_relative "prompt_interface/editor/modes/vibe_insert_readline"
+require_relative "prompt_interface/editor/modes/vibe"
+require_relative "prompt_interface/editor/controller"
+require_relative "prompt_interface/interactive/controller"
+require_relative "prompt_interface/interactive/renderer"
+require_relative "prompt_interface/interactive/state"
 require_relative "prompt_interface/layout"
 require_relative "prompt_interface/screen"
 require_relative "prompt_interface/key_handler"
@@ -39,33 +64,51 @@ module Kward
     SPINNER_FRAMES = %w[⠋ ⠙ ⠹ ⠸ ⠼ ⠴ ⠦ ⠧ ⠇ ⠏].freeze
     SPINNER_INTERVAL = 0.1
     FOOTER_REFRESH_INTERVAL = 1.0
+    COMPOSER_STATUS_REFRESH_INTERVAL = 1.0
     COMPOSER_MAX_INPUT_ROWS = 6
     TRANSCRIPT_BUFFER_LIMIT = 200_000
     BANNER_MESSAGE = Banner::MESSAGE
 
     include SlashOverlay
+    include FileOverlay
+    include ProjectBrowser
     include SelectionPrompt
     include QuestionPrompt
+    include GitPrompt
     include OverlayRenderer
+    include EditorRenderer
+    include EditorSyntaxHighlighter
+    include EditorAutoClosePairs
+    include EditorEndwise
+    include EditorAutoIndent
     include ComposerRenderer
     include ComposerController
+    include ModernEditorMode
+    include EmacsEditorMode
+    include VibeInsertReadline
+    include VibeEditorMode
+    include EditorController
+    include InteractiveRenderer
+    include InteractiveState
     include Layout
     include Screen
     include KeyHandler
     include RuntimeState
     include TranscriptRenderer
     include PromptRenderer
-    KEYBOARD_PROTOCOL_ENABLE = "\e[>1u".freeze
-    KEYBOARD_PROTOCOL_RESTORE = "\e[<u".freeze
-    BRACKETED_PASTE_ENABLE = "\e[?2004h".freeze
-    BRACKETED_PASTE_RESTORE = "\e[?2004l".freeze
-    BRACKETED_PASTE_START = "\e[200~".freeze
-    BRACKETED_PASTE_END = "\e[201~".freeze
-    SYNCHRONIZED_OUTPUT_ENABLE = "\e[?2026h".freeze
-    SYNCHRONIZED_OUTPUT_DISABLE = "\e[?2026l".freeze
-    CURSOR_SHOW = "\e[?25h".freeze
-    CURSOR_HIDE = "\e[?25l".freeze
-    SHIFT_ENTER_SEQUENCES = ["\e[13;2u", "\e[13;2~", "\e[27;2;13~", "\e\r", "\e\n"].freeze
+    KEYBOARD_PROTOCOL_ENABLE = TerminalSequences::KEYBOARD_PROTOCOL_ENABLE
+    KEYBOARD_PROTOCOL_RESTORE = TerminalSequences::KEYBOARD_PROTOCOL_RESTORE
+    BRACKETED_PASTE_ENABLE = TerminalSequences::BRACKETED_PASTE_ENABLE
+    BRACKETED_PASTE_RESTORE = TerminalSequences::BRACKETED_PASTE_RESTORE
+    BRACKETED_PASTE_START = TerminalSequences::BRACKETED_PASTE_START
+    BRACKETED_PASTE_END = TerminalSequences::BRACKETED_PASTE_END
+    SYNCHRONIZED_OUTPUT_ENABLE = TerminalSequences::SYNCHRONIZED_OUTPUT_ENABLE
+    SYNCHRONIZED_OUTPUT_DISABLE = TerminalSequences::SYNCHRONIZED_OUTPUT_DISABLE
+    CURSOR_SHOW = TerminalSequences::CURSOR_SHOW
+    CURSOR_HIDE = TerminalSequences::CURSOR_HIDE
+    CURSOR_SHAPE_DEFAULT = TerminalSequences::CURSOR_SHAPE_DEFAULT
+    CURSOR_SHAPE_BAR = TerminalSequences::CURSOR_SHAPE_BAR
+    SHIFT_ENTER_SEQUENCES = TerminalKeys::SHIFT_ENTER
     EXIT_INPUT = :exit_input
     CANCEL_INPUT = :cancel_input
     SELECT_CANCEL = :select_cancel
@@ -82,12 +125,14 @@ module Kward
       end
     end
 
-    def initialize(input: $stdin, output: $stdout, slash_commands: [], overlay_settings: nil, footer: nil, composer_status: nil, busy_help: true, attachment_badges: nil, attachment_parser: nil, banner_message: nil)
+    def initialize(input: $stdin, output: $stdout, slash_commands: [], overlay_settings: nil, footer: nil, composer_status: nil, busy_help: true, attachment_badges: nil, attachment_parser: nil, banner_message: nil, tab_keybindings: nil, prompt_history: nil, editor_mode: nil, editor_mode_source: nil, editor_auto_indent: true, editor_auto_indent_source: nil, editor_auto_close_pairs: true, editor_auto_close_pairs_source: nil, editor_soft_wrap: true, editor_soft_wrap_source: nil, editor_bar_cursor: true, editor_bar_cursor_source: nil, editor_line_numbers: "absolute", editor_line_numbers_source: nil)
       @input_io = input
       @output_io = output
       @reader = TTY::Reader.new(input: input, output: output, interrupt: :error)
       @mutex = Mutex.new
+      @prompt_history = prompt_history
       @composer = ComposerState.new
+      load_history(@prompt_history.values) if @prompt_history
       self.composer_input = @composer.input
       self.composer_cursor = @composer.cursor
       @started = false
@@ -99,6 +144,8 @@ module Kward
       @spinner_frame_index = 0
       @last_spinner_tick = monotonic_now
       @last_footer_refresh = monotonic_now
+      @last_composer_status_refresh = 0.0
+      @cached_composer_status_text = nil
       @prompt_label = "You>"
       @assistant_label = "Assistant"
       @stream_state = StreamState.new
@@ -109,18 +156,33 @@ module Kward
       @transcript_viewport_rows = 0
       @restoring_transcript = false
       @pending_keys = []
+      @completion_provider = nil
       @original_console_mode = nil
       @raw_mode_active = false
       @slash_commands = normalize_slash_commands(slash_commands)
       @slash_selection_index = 0
       @slash_overlay_dismissed_input = nil
+      @slash_overlay_disabled = false
+      @file_selection_index = 0
+      @file_overlay_dismissed_token = nil
+      @file_open_dismissed_token = nil
+      @file_editor_open_status = nil
+      @file_mention_paths = nil
+      @project_browser_state = nil
+      @project_browser_restore_after_editor = false
+      @editor_state = nil
+      @interactive_state = nil
+      @last_interactive_tick = monotonic_now
       @select_state = nil
       @question_state = nil
+      @question_prompt_active = false
+      @git_state = nil
       @last_width = screen_width
       @last_height = screen_height
       @reserved_rows = 0
       @color_enabled = ANSI.enabled?(output)
       @cursor_visible = true
+      @editor_bar_cursor_active = false
       @synchronized_output_depth = 0
       @overlay_settings = normalize_overlay_settings(overlay_settings)
       @footer = footer
@@ -129,6 +191,21 @@ module Kward
       @attachment_badges = attachment_badges
       @attachment_parser = attachment_parser
       @banner = Banner.new(message: banner_message, screen_height: method(:screen_height))
+      @tabs = []
+      @active_tab_index = 0
+      @tab_keybindings = normalize_tab_keybindings(tab_keybindings)
+      @editor_mode = normalize_editor_mode(editor_mode)
+      @editor_mode_source = editor_mode_source
+      @editor_auto_indent = editor_auto_indent != false
+      @editor_auto_indent_source = editor_auto_indent_source
+      @editor_auto_close_pairs = editor_auto_close_pairs != false
+      @editor_auto_close_pairs_source = editor_auto_close_pairs_source
+      @editor_soft_wrap = editor_soft_wrap != false
+      @editor_soft_wrap_source = editor_soft_wrap_source
+      @editor_bar_cursor = editor_bar_cursor != false
+      @editor_bar_cursor_source = editor_bar_cursor_source
+      @editor_line_numbers = normalize_editor_line_numbers(editor_line_numbers)
+      @editor_line_numbers_source = editor_line_numbers_source
     end
 
     def start(render: true)
@@ -138,6 +215,7 @@ module Kward
         enter_raw_mode_locked
         @started = true
         @asking = true
+        disable_editor_mouse_reporting(force: true) unless editor_active?
         @output_io.print(KEYBOARD_PROTOCOL_ENABLE)
         @output_io.print(BRACKETED_PASTE_ENABLE)
         render_prompt_locked if render
@@ -150,8 +228,10 @@ module Kward
 
         clear_prompt_for_output_locked
         restore_scroll_region_locked
+        disable_editor_mouse_reporting(force: true)
         @output_io.print(BRACKETED_PASTE_RESTORE)
         @output_io.print(KEYBOARD_PROTOCOL_RESTORE)
+        restore_editor_cursor_shape_locked
         set_cursor_visible_locked(true, force: true)
         @output_io.puts
         @output_io.flush
@@ -220,6 +300,79 @@ module Kward
       end
     end
 
+    def with_completion_provider(provider, slash_overlay: true)
+      previous_provider = @completion_provider
+      previous_slash_overlay_disabled = @slash_overlay_disabled
+      @completion_provider = provider
+      @slash_overlay_disabled = !slash_overlay
+      yield
+    ensure
+      @completion_provider = previous_provider
+      @slash_overlay_disabled = previous_slash_overlay_disabled
+    end
+
+    def with_prompt_history(history)
+      previous_history = @prompt_history
+      @prompt_history = history
+      load_history(@prompt_history.values) if @prompt_history
+      yield
+    ensure
+      @prompt_history = previous_history
+      load_history(@prompt_history.values) if @prompt_history
+    end
+
+    def editing_file?
+      @mutex.synchronize { editor_active? }
+    end
+
+    def edit_file(path, base_dir: Dir.pwd, allow_new: true)
+      start(render: false)
+      opened = @mutex.synchronize do
+        open_editor(path, allow_new: allow_new, base_dir: base_dir, restrict_to_workspace: false).tap do
+          render_prompt_locked
+        end
+      end
+      return false unless opened
+
+      run_editor
+    end
+
+    def scratchpad(language = :text)
+      start(render: false)
+      opened = @mutex.synchronize do
+        open_scratchpad(language).tap do
+          render_prompt_locked
+        end
+      end
+      return false unless opened
+
+      run_editor
+    end
+
+    def run_editor
+      loop do
+        key = read_key(nonblock: true)
+        action = nil
+        editor_open = @mutex.synchronize do
+          if key.nil?
+            resized = handle_resize_locked
+            footer_refreshed = tick_footer_locked
+            render_prompt_locked if resized || footer_refreshed
+          else
+            result = handle_key(key)
+            action = result if prompt_action_result?(result)
+            render_prompt_locked unless result.is_a?(String) || result == EXIT_INPUT || prompt_action_result?(result)
+          end
+          editor_active?
+        end
+        return action if action
+        break unless editor_open
+
+        sleep 0.02 if key.nil?
+      end
+      true
+    end
+
     def ask(message = "You>")
       was_composing = @started && @asking
       start
@@ -250,10 +403,10 @@ module Kward
             render_prompt_locked if resized || footer_refreshed
           else
             result = handle_key(key)
-            render_prompt_locked unless result.is_a?(String) || result == EXIT_INPUT
+            render_prompt_locked unless result.is_a?(String) || result == EXIT_INPUT || prompt_action_result?(result)
           end
         end
-        return result if result.is_a?(String)
+        return result if result.is_a?(String) || prompt_action_result?(result)
         return nil if result == EXIT_INPUT
 
         sleep 0.02 if key.nil?
@@ -323,7 +476,10 @@ module Kward
       start
       saved_state = nil
       answers = []
-      @mutex.synchronize { saved_state = begin_question_prompt_state }
+      @mutex.synchronize do
+        @question_prompt_active = true
+        saved_state = begin_question_prompt_state
+      end
 
       questions.each_with_index do |question, index|
         answer = ask_single_user_question(question, index + 1, questions.length)
@@ -342,7 +498,157 @@ module Kward
     end
 
     def modal_active?
-      @mutex.synchronize { !@question_state.nil? || !@select_state.nil? }
+      @mutex.synchronize { modal_active_locked? }
+    end
+
+    def interactive_active?
+      @mutex.synchronize { interactive_active_locked? }
+    end
+
+    def interactive_exited?
+      @mutex.synchronize do
+        return false unless @interactive_state
+
+        @interactive_state[:controller].exited?
+      end
+    end
+
+    def finish_interactive
+      @mutex.synchronize do
+        return unless @interactive_state
+
+        snapshot = @interactive_state[:snapshot]
+        @interactive_state = nil
+        restore_composer_snapshot_locked(snapshot)
+        redraw_screen_locked if @started
+        @output_io.flush
+      end
+    end
+
+    def with_terminal_handoff
+      start
+      input = nil
+      output = nil
+      @mutex.synchronize do
+        clear_prompt_for_output_locked
+        restore_scroll_region_locked
+        disable_editor_mouse_reporting(force: true)
+        @output_io.print(BRACKETED_PASTE_RESTORE)
+        @output_io.print(KEYBOARD_PROTOCOL_RESTORE)
+        restore_editor_cursor_shape_locked
+        set_cursor_visible_locked(true, force: true)
+        @output_io.flush
+        restore_console_mode_locked
+        input = @input_io
+        output = @output_io
+      end
+
+      yield(input, output)
+    ensure
+      @mutex.synchronize do
+        enter_raw_mode_locked
+        @output_io.print(KEYBOARD_PROTOCOL_ENABLE)
+        @output_io.print(BRACKETED_PASTE_ENABLE)
+        @last_composer_rows = []
+        render_prompt_locked if @started && @asking
+        @output_io.flush
+      end
+    end
+
+    def start_interactive(title:, rows:, fps:)
+      snapshot = composer_snapshot
+      controller = InteractiveController.new(width: interactive_canvas_width, height: rows, fps: fps)
+      start
+      @mutex.synchronize do
+        @interactive_state = {
+          title: title.to_s,
+          rows: rows,
+          controller: controller,
+          snapshot: snapshot
+        }
+        @last_interactive_tick = monotonic_now
+        @asking = true
+        @busy = false
+        @last_composer_rows = []
+        render_prompt_locked
+      end
+      controller
+    end
+
+    def update_tabs(labels:, active_index: 0)
+      @mutex.synchronize do
+        @tabs = Array(labels).map { |label| normalize_tab_label(label) }
+        @active_tab_index = active_index.to_i
+        render_prompt_locked if @started && @asking
+      end
+    end
+
+    def composer_snapshot
+      @mutex.synchronize do
+        {
+          composer: @composer,
+          prompt_label: @prompt_label
+        }
+      end
+    end
+
+    def tab_view_snapshot
+      @mutex.synchronize do
+        {
+          composer: @composer.dup,
+          prompt_label: @prompt_label.dup,
+          editor_state: @editor_state&.dup,
+          transcript_buffer: @transcript_buffer.dup,
+          transcript_viewport_rows: @transcript_viewport_rows,
+          stream_state: @stream_state.dup
+        }
+      end
+    end
+
+    def restore_composer_snapshot(snapshot)
+      @mutex.synchronize do
+        restore_composer_snapshot_locked(snapshot)
+        restore_editor_snapshot_locked(snapshot)
+        redraw_screen_locked if @started
+      end
+    end
+
+    def restore_tab_view_snapshot(snapshot)
+      @mutex.synchronize do
+        restore_composer_snapshot_locked(snapshot)
+        restore_editor_snapshot_locked(snapshot)
+        @transcript_buffer = snapshot[:transcript_buffer] || TranscriptBuffer.new(limit: TRANSCRIPT_BUFFER_LIMIT)
+        @transcript_viewport_rows = snapshot[:transcript_viewport_rows].to_i
+        @stream_state = snapshot[:stream_state] || StreamState.new
+        @last_composer_rows = []
+        redraw_screen_locked if @started
+      end
+    end
+
+    def restore_composer_snapshot_locked(snapshot)
+      @composer = snapshot[:composer] || new_composer_state_with_history
+      @prompt_label = snapshot[:prompt_label].to_s.empty? ? "You>" : snapshot[:prompt_label].to_s
+      self.composer_input = @composer.input
+      self.composer_cursor = @composer.cursor
+      @last_composer_rows = []
+    end
+
+    def new_composer_state_with_history
+      composer = ComposerState.new
+      composer.load_history(@prompt_history.values) if @prompt_history
+      composer
+    end
+
+    def restore_editor_snapshot_locked(snapshot)
+      editor_was_active = editor_active?
+      @editor_state = snapshot[:editor_state]&.dup
+      editor_is_active = editor_active?
+
+      if editor_is_active
+        enable_editor_mouse_reporting unless editor_was_active
+      else
+        disable_editor_mouse_reporting(force: true)
+      end
     end
 
     def update_overlay_settings(settings)
@@ -409,6 +715,22 @@ module Kward
     def poll_input
       key = read_key(nonblock: true)
       @mutex.synchronize do
+        if interactive_active_locked?
+          if key.nil?
+            resized = handle_resize_locked
+            ticked = tick_interactive_locked
+            render_prompt_locked if resized || ticked
+            return :interactive_exited if @interactive_state[:controller].exited?
+            return nil
+          end
+
+          route_interactive_key(key)
+          ticked = tick_interactive_locked
+          render_prompt_locked if ticked
+          return :interactive_exited if @interactive_state[:controller].exited?
+          return nil
+        end
+
         if key.nil?
           resized = handle_resize_locked
           spun = tick_spinner_locked
@@ -417,8 +739,13 @@ module Kward
           return nil
         end
 
+        if modal_active_locked?
+          queue_pending_keys(key)
+          return nil
+        end
+
         result = handle_key(key)
-        render_prompt_locked unless [EXIT_INPUT, CANCEL_INPUT].include?(result)
+        render_prompt_locked unless [EXIT_INPUT, CANCEL_INPUT].include?(result) || prompt_action_result?(result)
         [EXIT_INPUT, CANCEL_INPUT].include?(result) ? result : result
       end
     end
@@ -465,6 +792,17 @@ module Kward
       end
     end
 
+    def write_transcript_delta(delta)
+      @mutex.synchronize do
+        with_synchronized_output_locked do
+          prepare_transcript_output_locked unless @restoring_transcript
+          write_transcript_text_locked(delta.to_s)
+          restore_composer_cursor_locked unless @restoring_transcript
+        end
+        @output_io.flush unless @restoring_transcript
+      end
+    end
+
     def write_stream_block(label, delta, finish: false)
       @mutex.synchronize do
         write_stream_block_locked(label, delta.to_s, finish: finish)
@@ -479,6 +817,14 @@ module Kward
       end
     end
 
+    def refresh_composer_status
+      @mutex.synchronize do
+        @cached_composer_status_text = nil
+        @last_composer_status_refresh = 0.0
+        render_prompt_locked if @started && @asking
+      end
+    end
+
     def clear_transcript
       @mutex.synchronize do
         @transcript_buffer.clear
@@ -493,210 +839,8 @@ module Kward
 
     private
 
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
+    def modal_active_locked?
+      @question_prompt_active || !@question_state.nil? || !@select_state.nil? || !@git_state.nil?
+    end
   end
 end