ruby_coded 0.1.1 → 0.2.1

data/lib/ruby_coded/chat/codex_models.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module RubyCoded
+  module Chat
+    # Local catalog of models available through the ChatGPT Codex backend.
+    # These models are not listed in RubyLLM.models because they use a
+    # different API (Responses API via chatgpt.com/backend-api).
+    module CodexModels
+      CodexModel = Struct.new(:id, :display_name, :context_window, :max_output, keyword_init: true) do
+        def to_s
+          id
+        end
+      end
+
+      MODELS = [
+        CodexModel.new(id: "gpt-5.4", display_name: "GPT 5.4 (Recommended)",
+                       context_window: 272_000, max_output: 128_000),
+        CodexModel.new(id: "gpt-5.4-mini", display_name: "GPT 5.4 Mini",
+                       context_window: 272_000, max_output: 128_000),
+        CodexModel.new(id: "gpt-5.3-codex-spark", display_name: "GPT 5.3 Codex Spark (Pro only)",
+                       context_window: 272_000, max_output: 128_000),
+        CodexModel.new(id: "gpt-5.2-codex", display_name: "GPT 5.2 Codex",
+                       context_window: 272_000, max_output: 128_000),
+        CodexModel.new(id: "gpt-5.2", display_name: "GPT 5.2",
+                       context_window: 272_000, max_output: 128_000)
+      ].freeze
+
+      def self.all
+        MODELS
+      end
+
+      def self.find(id)
+        MODELS.find { |m| m.id == id }
+      end
+
+      def self.codex_model?(id)
+        MODELS.any? { |m| m.id == id }
+      end
+    end
+  end
+end

data/lib/ruby_coded/chat/command_handler/login_commands.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module RubyCoded
+  module Chat
+    class CommandHandler
+      # Slash command for authenticating with AI providers from within the TUI.
+      module LoginCommands
+        private
+
+        def cmd_login(rest)
+          provider_name = rest&.strip&.downcase
+
+          if provider_name && !provider_name.empty?
+            return show_login_usage unless valid_providers.include?(provider_name)
+
+            @state.enter_login_flow!(provider: provider_name.to_sym)
+          else
+            @state.enter_login_flow!
+          end
+        end
+
+        def valid_providers
+          RubyCoded::Auth::AuthManager::PROVIDERS.keys.map(&:to_s)
+        end
+
+        def show_login_usage
+          providers = valid_providers.join(", ")
+          @state.add_message(:system, "Usage: /login [#{providers}]")
+        end
+      end
+    end
+  end
+end

data/lib/ruby_coded/chat/command_handler/model_commands.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require_relative "../model_filter"
+require_relative "../codex_models"
 
 module RubyCoded
   module Chat
@@ -46,7 +47,7 @@ module RubyCoded
 
         def open_model_selector(show_all: false)
           models = fetch_models_for_authenticated_providers
-          models = ModelFilter.filter(models) unless show_all
+          models = ModelFilter.filter(models) unless show_all || codex_oauth_active?
 
           if models.empty?
             @state.add_message(:system,
@@ -62,18 +63,23 @@ module RubyCoded
           return fetch_chat_models unless @credentials_store
 
           models = []
-
           Auth::AuthManager::PROVIDERS.each_key do |name|
-            next unless @credentials_store.retrieve(name)
-
-            provider_models = RubyLLM.models.by_provider(name).chat_models.to_a
-            models.concat(provider_models)
+            creds = @credentials_store.retrieve(name)
+            models.concat(models_for_provider(name, creds)) if creds
           end
           models
         rescue StandardError
           fetch_chat_models
         end
 
+        def models_for_provider(name, creds)
+          if name == :openai && creds["auth_method"] == "oauth"
+            CodexModels.all
+          else
+            RubyLLM.models.by_provider(name).chat_models.to_a
+          end
+        end
+
         def fetch_chat_models
           RubyLLM.models.chat_models.to_a
         rescue StandardError
@@ -85,6 +91,13 @@ module RubyCoded
 
           model.to_s
         end
+
+        def codex_oauth_active?
+          return false unless @credentials_store
+
+          creds = @credentials_store.retrieve(:openai)
+          creds && creds["auth_method"] == "oauth"
+        end
       end
     end
   end

data/lib/ruby_coded/chat/command_handler.rb

@@ -8,6 +8,7 @@ require_relative "command_handler/token_formatting"
 require_relative "command_handler/token_commands"
 require_relative "command_handler/agent_commands"
 require_relative "command_handler/plan_commands"
+require_relative "command_handler/login_commands"
 
 module RubyCoded
   module Chat
@@ -21,6 +22,7 @@ module RubyCoded
       include TokenCommands
       include AgentCommands
       include PlanCommands
+      include LoginCommands
 
       BASE_COMMANDS = {
         "/help" => :cmd_help,
@@ -31,16 +33,18 @@ module RubyCoded
         "/history" => :cmd_history,
         "/tokens" => :cmd_tokens,
         "/agent" => :cmd_agent,
-        "/plan" => :cmd_plan
+        "/plan" => :cmd_plan,
+        "/login" => :cmd_login
       }.freeze
 
       HELP_TEXT = File.read(File.join(__dir__, "help.txt")).freeze
 
-      def initialize(state, llm_bridge:, user_config: nil, credentials_store: nil)
+      def initialize(state, llm_bridge:, user_config: nil, credentials_store: nil, auth_manager: nil)
         @state = state
         @llm_bridge = llm_bridge
         @user_config = user_config
         @credentials_store = credentials_store
+        @auth_manager = auth_manager
         @commands = build_command_map
       end
 

data/lib/ruby_coded/chat/help.txt

@@ -14,6 +14,8 @@ Available commands:
   /plan off          Disable plan mode (warns if unsaved plan)
   /plan off --force  Disable plan mode discarding unsaved plan
   /plan save [file]  Save current plan to a markdown file
+  /login             Authenticate with an AI provider
+  /login <provider>  Authenticate directly with a specific provider (openai, anthropic)
   /exit, /quit       Exit the chat
 
 Tool confirmations (agent mode):

data/lib/ruby_coded/chat/input_handler/login_inputs.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+module RubyCoded
+  module Chat
+    class InputHandler
+      # Handles input events for the multi-step login wizard:
+      # provider selection, auth method selection, API key entry, and OAuth waiting.
+      module LoginInputs
+        private
+
+        def handle_login_mode(event)
+          return :quit if event.ctrl_c?
+
+          case @state.login_step
+          when :provider_select, :auth_method_select
+            handle_login_select(event)
+          when :api_key_input
+            handle_login_key_input(event)
+          when :oauth_waiting
+            handle_login_oauth_waiting(event)
+          end
+        end
+
+        def handle_login_select(event)
+          return :login_cancel if event.esc?
+          return login_select_confirmed if event.enter?
+
+          if event.up?
+            @state.login_select_up
+          elsif event.down?
+            @state.login_select_down
+          end
+          nil
+        end
+
+        def login_select_confirmed
+          @state.login_step == :provider_select ? :login_provider_selected : :login_method_selected
+        end
+
+        def handle_login_key_input(event)
+          return :login_cancel if event.esc?
+          return handle_login_key_enter if event.enter?
+          return @state.delete_last_login_key_char if event.backspace?
+
+          append_login_key_char(event)
+        end
+
+        def handle_login_key_enter
+          @state.login_key_buffer.strip.empty? ? nil : :login_key_submitted
+        end
+
+        def append_login_key_char(event)
+          char = event.to_s
+          @state.append_to_login_key(char) unless char.empty? || event.ctrl? || event.alt?
+          nil
+        end
+
+        def handle_login_oauth_waiting(event)
+          return :login_oauth_cancel if event.esc?
+
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/ruby_coded/chat/input_handler.rb

@@ -3,6 +3,7 @@
 require "ratatui_ruby"
 
 require_relative "input_handler/modal_inputs"
+require_relative "input_handler/login_inputs"
 require_relative "input_handler/normal_mode_input"
 
 module RubyCoded
@@ -10,6 +11,7 @@ module RubyCoded
     # This class is used to handle the input events for the chat
     class InputHandler
       include ModalInputs
+      include LoginInputs
       include NormalModeInput
 
       def initialize(state)
@@ -30,6 +32,7 @@ module RubyCoded
         return handle_tool_confirmation_mode(event) if @state.awaiting_tool_confirmation?
         return handle_plan_clarification_mode(event) if @state.plan_clarification?
         return handle_model_select_mode(event) if @state.model_select?
+        return handle_login_mode(event) if @state.login_active?
         return handle_streaming_mode(event) if @state.streaming?
 
         handle_normal_mode(event)

data/lib/ruby_coded/chat/renderer/chat_panel_input.rb

@@ -38,7 +38,7 @@ module RubyCoded
         end
 
         def input_locked?
-          @state.streaming? || @state.model_select? || @state.plan_clarification?
+          @state.streaming? || @state.model_select? || @state.plan_clarification? || @state.login_active?
         end
 
         def render_input_cursor(frame, area, prefix_len, scroll_offset)

data/lib/ruby_coded/chat/renderer/login_flow.rb

@@ -0,0 +1,105 @@
+# frozen_string_literal: true
+
+module RubyCoded
+  module Chat
+    class Renderer
+      # Renders a centered popup for the multi-step login wizard.
+      module LoginFlow
+        private
+
+        def render_login_flow(frame, area)
+          popup_area = login_centered_popup(area)
+          frame.render_widget(@tui.clear, popup_area)
+
+          case @state.login_step
+          when :provider_select    then render_login_select(frame, popup_area, title: "Select AI Provider")
+          when :auth_method_select then render_login_select(frame, popup_area, title: "Select Auth Method")
+          when :api_key_input      then render_login_api_key(frame, popup_area)
+          when :oauth_waiting      then render_login_oauth_waiting(frame, popup_area)
+          end
+        end
+
+        def render_login_select(frame, popup_area, title:)
+          hint_area, list_area = login_select_layout(popup_area)
+          render_login_hint(frame, hint_area)
+          render_login_list(frame, list_area, title)
+        end
+
+        def render_login_hint(frame, area)
+          widget = @tui.paragraph(
+            text: "↑↓ navigate, Enter select, Esc cancel",
+            block: @tui.block(title: "Login", borders: [:all])
+          )
+          frame.render_widget(widget, area)
+        end
+
+        def render_login_list(frame, area, title)
+          items = @state.login_items.map { |item| item[:label] }
+          widget = @tui.list(
+            items: items,
+            selected_index: @state.login_select_index,
+            highlight_style: @tui.style(bg: :blue, fg: :white, modifiers: [:bold]),
+            highlight_symbol: "> ",
+            scroll_padding: 2,
+            block: @tui.block(title: title, borders: [:all])
+          )
+          frame.render_widget(widget, area)
+        end
+
+        def render_login_api_key(frame, popup_area)
+          provider = @state.login_provider_module
+          info_area, input_area, error_area = login_api_key_layout(popup_area)
+
+          render_login_api_key_info(frame, info_area, provider)
+          render_login_api_key_input(frame, input_area)
+          render_login_api_key_error(frame, error_area)
+        end
+
+        def render_login_api_key_info(frame, area, provider)
+          console = provider.respond_to?(:console_url) ? provider.console_url : ""
+          text = "Generate your API key at:\n#{console}"
+          widget = @tui.paragraph(
+            text: text,
+            wrap: true,
+            block: @tui.block(title: "#{provider.display_name} API Key", borders: [:all])
+          )
+          frame.render_widget(widget, area)
+        end
+
+        def render_login_api_key_input(frame, area)
+          masked = "*" * @state.login_key_buffer.length
+          widget = @tui.paragraph(
+            text: "Key: #{masked}",
+            block: @tui.block(title: "Enter submit, Esc cancel", borders: [:all])
+          )
+          frame.render_widget(widget, area)
+        end
+
+        def render_login_api_key_error(frame, area)
+          error = @state.login_error
+          text = error || ""
+          style = error ? @tui.style(fg: :red) : @tui.style(fg: :dark_gray)
+          widget = @tui.paragraph(
+            text: text,
+            style: style,
+            wrap: true,
+            block: @tui.block(borders: [:all])
+          )
+          frame.render_widget(widget, area)
+        end
+
+        def render_login_oauth_waiting(frame, popup_area)
+          provider = @state.login_provider_module
+          widget = @tui.paragraph(
+            text: "Your browser has been opened for authentication.\n\n" \
+                  "Waiting for #{provider.display_name} callback...\n\n" \
+                  "Press Esc to cancel.",
+            wrap: true,
+            block: @tui.block(title: "Authenticating with #{provider.display_name}", borders: [:all])
+          )
+          frame.render_widget(widget, popup_area)
+        end
+      end
+    end
+  end
+end

data/lib/ruby_coded/chat/renderer/login_flow_layout.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+module RubyCoded
+  module Chat
+    class Renderer
+      # Layout helpers for centering the login flow popup.
+      module LoginFlowLayout
+        private
+
+        def login_centered_popup(area)
+          vertical = login_centered_vertical(area)
+          login_centered_horizontal(vertical[1])
+        end
+
+        def login_centered_vertical(area)
+          @tui.layout_split(
+            area,
+            direction: :vertical,
+            constraints: [
+              @tui.constraint_percentage(15),
+              @tui.constraint_percentage(70),
+              @tui.constraint_percentage(15)
+            ]
+          )
+        end
+
+        def login_centered_horizontal(area)
+          horizontal = @tui.layout_split(
+            area,
+            direction: :horizontal,
+            constraints: [
+              @tui.constraint_percentage(20),
+              @tui.constraint_percentage(60),
+              @tui.constraint_percentage(20)
+            ]
+          )
+          horizontal[1]
+        end
+
+        def login_select_layout(popup_area)
+          @tui.layout_split(
+            popup_area,
+            direction: :vertical,
+            constraints: [
+              @tui.constraint_length(3),
+              @tui.constraint_fill(1)
+            ]
+          )
+        end
+
+        def login_api_key_layout(popup_area)
+          @tui.layout_split(
+            popup_area,
+            direction: :vertical,
+            constraints: [
+              @tui.constraint_length(5),
+              @tui.constraint_length(3),
+              @tui.constraint_fill(1)
+            ]
+          )
+        end
+      end
+    end
+  end
+end

data/lib/ruby_coded/chat/renderer/status_bar.rb

@@ -15,7 +15,8 @@ module RubyCoded
 
         def status_bar_text(width)
           left = status_bar_left
-          right = "#{@state.model} | #{format_cost(@state.total_session_cost)} "
+          model_label = @state.codex_mode ? "#{@state.model} (ChatGPT)" : "#{@state.model} (API)"
+          right = "#{model_label} | #{format_cost(@state.total_session_cost)} "
           center_pad = [width - left.length - right.length, 1].max
           "#{left}#{" " * center_pad}#{right}"
         end

data/lib/ruby_coded/chat/renderer.rb

@@ -6,6 +6,8 @@ require_relative "renderer/chat_panel_input"
 require_relative "renderer/model_selector"
 require_relative "renderer/plan_clarifier_layout"
 require_relative "renderer/plan_clarifier"
+require_relative "renderer/login_flow_layout"
+require_relative "renderer/login_flow"
 require_relative "renderer/status_bar"
 
 module RubyCoded
@@ -18,6 +20,8 @@ module RubyCoded
       include ModelSelector
       include PlanClarifierLayout
       include PlanClarifier
+      include LoginFlowLayout
+      include LoginFlow
       include StatusBar
 
       def initialize(tui, state)
@@ -33,14 +37,19 @@ module RubyCoded
           render_chat_panel(frame, chat_area)
           render_status_bar(frame, status_area)
           render_input_panel(frame, input_area)
-          render_model_selector(frame, chat_area) if @state.model_select?
-          render_plan_clarifier(frame, chat_area) if @state.plan_clarification?
-          render_plugin_overlays(frame, chat_area, input_area)
+          render_overlays(frame, chat_area, input_area)
         end
       end
 
       private
 
+      def render_overlays(frame, chat_area, input_area)
+        render_model_selector(frame, chat_area) if @state.model_select?
+        render_plan_clarifier(frame, chat_area) if @state.plan_clarification?
+        render_login_flow(frame, chat_area) if @state.login_active?
+        render_plugin_overlays(frame, chat_area, input_area)
+      end
+
       # Calls each plugin's render method in registration order.
       def render_plugin_overlays(frame, chat_area, input_area)
         RubyCoded.plugin_registry.render_configs.each do |config|

data/lib/ruby_coded/chat/state/login_flow.rb

@@ -0,0 +1,117 @@
+# frozen_string_literal: true
+
+require_relative "login_flow_steps"
+
+module RubyCoded
+  module Chat
+    class State
+      # Manages the multi-step login wizard state within the TUI.
+      # Steps: :provider_select -> :auth_method_select -> :api_key_input / :oauth_waiting
+      module LoginFlow
+        include LoginFlowSteps
+
+        attr_reader :login_step, :login_provider, :login_auth_method,
+                    :login_items, :login_select_index,
+                    :login_key_buffer, :login_key_cursor,
+                    :login_error, :login_oauth_result
+
+        def init_login_flow
+          reset_login_state
+        end
+
+        def login_active?
+          @mode == :login
+        end
+
+        def enter_login_flow!(provider: nil)
+          reset_login_state
+          return enter_login_step_provider_select! unless provider
+
+          enter_login_for_provider!(provider)
+        end
+
+        def login_select_up
+          return if @login_items.empty?
+
+          @login_select_index = (@login_select_index - 1) % @login_items.size
+          mark_dirty!
+        end
+
+        def login_select_down
+          return if @login_items.empty?
+
+          @login_select_index = (@login_select_index + 1) % @login_items.size
+          mark_dirty!
+        end
+
+        def login_selected_item
+          @login_items[@login_select_index]
+        end
+
+        def login_advance_to_auth_method!(provider_name)
+          @login_provider = provider_name
+          enter_login_step_auth_method!(provider_name)
+        end
+
+        def login_advance_to_api_key!(provider_name, auth_method = :api_key)
+          @login_provider = provider_name
+          @login_auth_method = auth_method
+          enter_login_step_api_key!(provider_name)
+        end
+
+        def login_advance_to_oauth!(provider_name)
+          @login_provider = provider_name
+          @login_auth_method = :oauth
+          @login_step = :oauth_waiting
+          @login_items = []
+          @login_select_index = 0
+          @login_error = nil
+          @mode = :login
+          mark_dirty!
+        end
+
+        def append_to_login_key(text)
+          @login_key_buffer.insert(@login_key_cursor, text)
+          @login_key_cursor += text.length
+          @login_error = nil
+          mark_dirty!
+        end
+
+        def delete_last_login_key_char
+          return if @login_key_cursor <= 0
+
+          @login_key_buffer.slice!(@login_key_cursor - 1)
+          @login_key_cursor -= 1
+          @login_error = nil
+          mark_dirty!
+        end
+
+        def login_set_oauth_result!(result)
+          @mutex.synchronize do
+            @login_oauth_result = result
+            @dirty = true
+          end
+        end
+
+        def login_clear_oauth_result!
+          @login_oauth_result = nil
+        end
+
+        def login_set_error!(msg)
+          @login_error = msg
+          mark_dirty!
+        end
+
+        def exit_login_flow!
+          @mode = :chat
+          reset_login_state
+          mark_dirty!
+        end
+
+        def login_provider_module
+          providers_map[@login_provider]
+        end
+      end
+    end
+  end
+end

data/lib/ruby_coded/chat/state/login_flow_steps.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+
+module RubyCoded
+  module Chat
+    class State
+      # Private helpers for LoginFlow: state reset, provider lookup,
+      # and step-specific initialization.
+      module LoginFlowSteps
+        private
+
+        def reset_login_state
+          @login_step = nil
+          @login_provider = nil
+          @login_auth_method = nil
+          @login_items = []
+          @login_select_index = 0
+          @login_key_buffer = String.new
+          @login_key_cursor = 0
+          @login_error = nil
+          @login_oauth_result = nil
+        end
+
+        def providers_map
+          RubyCoded::Auth::AuthManager::PROVIDERS
+        end
+
+        def enter_login_for_provider!(provider)
+          methods = providers_map[provider].auth_methods
+          if methods.size == 1
+            enter_login_step_api_key!(provider)
+          else
+            enter_login_step_auth_method!(provider)
+          end
+        end
+
+        def enter_login_step_provider_select!
+          @login_step = :provider_select
+          @login_items = providers_map.map { |key, prov| { key: key, label: prov.display_name } }
+          @login_select_index = 0
+          @mode = :login
+          mark_dirty!
+        end
+
+        def enter_login_step_auth_method!(provider_name)
+          @login_provider = provider_name
+          provider = providers_map[provider_name]
+          @login_step = :auth_method_select
+          @login_items = provider.auth_methods
+          @login_select_index = 0
+          @mode = :login
+          mark_dirty!
+        end
+
+        def enter_login_step_api_key!(provider_name)
+          @login_provider = provider_name
+          @login_auth_method = :api_key
+          @login_step = :api_key_input
+          @login_items = []
+          @login_select_index = 0
+          @login_key_buffer = String.new
+          @login_key_cursor = 0
+          @login_error = nil
+          @mode = :login
+          mark_dirty!
+        end
+      end
+    end
+  end
+end