tng 0.2.9 → 0.3.3

data/bin/tng

@@ -11,30 +11,13 @@ require "tng/services/direct_generation"
 require "tng/services/extract_methods"
 require "tng/services/file_type_detector"
 
-require "tty-prompt"
-require "tty-spinner"
-require "tty-box"
-require "tty-table"
-require "tty-progressbar"
 require "tty-screen"
 require "tty-option"
 require "pastel"
-require "tng/ui/show_help"
-require "tng/ui/post_install_box"
 require "tng/ui/theme"
-require "tng/ui/system_status_display"
-require "tng/ui/configuration_display"
-require "tng/ui/authentication_warning_display"
-require "tng/ui/display_banner"
-require "tng/ui/user_stats_display"
-require "tng/ui/about_display"
 require "tng/services/testng"
-require "tng/ui/goodbye_display"
+require "tng/ui/go_ui_session"
 require "tng/api/http_client"
-require "tng/ui/controller_test_flow_display"
-require "tng/ui/model_test_flow_display"
-require "tng/ui/service_test_flow_display"
-require "tng/ui/other_test_flow_display"
 require "tng/analyzers/controller"
 require "tng/analyzers/model"
 require "tng/analyzers/service"
@@ -81,8 +64,8 @@ class CLI
   end
 
   def initialize
-    @prompt = TTY::Prompt.new
     @pastel = Pastel.new
+    @go_ui = Tng::UI::GoUISession.new
     @terminal_width = begin
       TTY::Screen.width
     rescue StandardError
@@ -99,8 +82,7 @@ class CLI
     parse(normalized_argv)
 
     if params[:help]
-      help_display = ShowHelp.new(@pastel, Tng::VERSION)
-      puts help_display.render
+      @go_ui.show_help(Tng::VERSION)
       return
     end
 
@@ -121,8 +103,10 @@ class CLI
       return unless check_configuration
 
       check_system_status
+      return unless check_authentication
+
       clear_screen
-      puts DisplayBanner.new(@pastel, Tng::VERSION).render
+      @go_ui.show_banner(Tng::VERSION)
       main_menu
     end
   end
@@ -179,309 +163,193 @@ class CLI
   end
 
   def main_menu
-    loop do
-      clear_screen
-
-      puts DisplayBanner.new(@pastel, Tng::VERSION).render
+    @go_ui.start
 
-      begin
-        choice = @prompt.select(
-          center_text("Select an option:"),
-          cycle: true,
-          per_page: 5,
-          symbols: { marker: "→" }
-        ) do |menu|
-          menu.choice "Generate tests", :tests
-          menu.choice "User stats", :stats
-          menu.choice "About", :about
-          menu.choice "Exit", :exit, key: "q"
-        end
+    loop do
+      choice = @go_ui.show_menu
 
-        handle_choice(choice)
-      rescue TTY::Reader::InputInterrupt, Interrupt
-        choice = :exit
+      case choice
+      when "tests"
+        handle_test_generation
+      when "stats"
+        user_stats
+      when "about"
+        show_about
+      when "exit"
+        show_goodbye
+        break
+      else
+        break
       end
-
-      break if choice == :exit
     end
-  end
 
-  def handle_choice(choice)
-    case choice
-    when :tests
-      handle_test_generation
-    when :stats
-      user_stats
-    when :about
-      show_about
-    when :exit
-      show_goodbye
-    end
+    @go_ui.stop
   end
 
   def handle_test_generation
-    choice = @prompt.select(
-      center_text("What would you like to test?"),
-      cycle: true,
-      per_page: 5,
-      symbols: { marker: "→" }
-    ) do |menu|
-      menu.choice "Controller", :controller
-      menu.choice "Model", :model
-      menu.choice "Service", :service
-      menu.choice "Other", :other
-      menu.choice @pastel.decorate("#{Tng::UI::Theme.icon(:back)} Back", Tng::UI::Theme.color(:info)), :back
-    end
+    choice = @go_ui.show_test_type_menu
 
     case choice
-    when :controller
+    when "controller"
       generate_controller_tests
-    when :model
+    when "model"
       generate_model_tests
-    when :service
+    when "service"
       generate_service_tests
-    when :other
+    when "other"
       generate_other_tests
     end
   end
 
   def generate_controller_tests
-    header = @pastel.decorate("Scanning for controllers...", Tng::UI::Theme.color(:primary))
-    puts center_text(header)
-
-    spinner = TTY::Spinner.new(
-      center_text("[:spinner] Analyzing controllers..."),
-      format: :dots,
-      success_mark: @pastel.decorate(Tng::UI::Theme.icon(:success), Tng::UI::Theme.color(:success)),
-      error_mark: @pastel.decorate(Tng::UI::Theme.icon(:error), Tng::UI::Theme.color(:error))
-    )
-    spinner.auto_spin
-    controllers = Tng::Analyzers::Controller.files_in_dir("app/controllers").map do |file|
-      relative_path = file[:path].gsub(%r{^.*app/controllers/}, "").gsub(".rb", "")
-      namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+    controllers = nil
 
-      {
-        name: namespaced_name,
-        path: file[:path]
-      }
-    end
+    @go_ui.show_spinner("Analyzing controllers...") do
+      controllers = Tng::Analyzers::Controller.files_in_dir("app/controllers").map do |file|
+        relative_path = file[:path].gsub(%r{^.*app/controllers/}, "").gsub(".rb", "")
+        namespaced_name = relative_path.split("/").map(&:camelize).join("::")
 
-    flow_display = ControllerTestFlowDisplay.new(@prompt, @pastel)
+        { name: namespaced_name, path: file[:path] }
+      end
+      { success: true, message: "Found #{controllers.length} controllers" }
+    end
 
     if controllers.empty?
-      spinner.error
-      flow_display.show_no_controllers_message
+      @go_ui.show_no_items("controllers")
       return
     end
 
-    success_msg = @pastel.decorate("Found #{controllers.length} controllers", Tng::UI::Theme.color(:success))
-    spinner.success(center_text(success_msg))
+    items = controllers.map { |c| { name: c[:name], path: c[:path] } }
+    selected_name = @go_ui.show_list_view("Select Controller", items)
+    return if selected_name == "back"
 
-    controller_choice = flow_display.select_controller(controllers)
-    return if controller_choice == :back
+    controller_choice = controllers.find { |c| c[:name] == selected_name }
+    return unless controller_choice
 
     show_controller_test_options(controller_choice)
   end
 
   def generate_model_tests
-    header = @pastel.decorate("Scanning for models...", Tng::UI::Theme.color(:primary))
-    puts center_text(header)
-
-    spinner = TTY::Spinner.new(
-      center_text("[:spinner] Analyzing models..."),
-      format: :dots,
-      success_mark: @pastel.decorate(Tng::UI::Theme.icon(:success), Tng::UI::Theme.color(:success)),
-      error_mark: @pastel.decorate(Tng::UI::Theme.icon(:error), Tng::UI::Theme.color(:error))
-    )
-    spinner.auto_spin
+    models = nil
 
-    models = Tng::Analyzers::Model.files_in_dir("app/models").map do |file|
-      relative_path = file[:path].gsub(%r{^.*app/models/}, "").gsub(".rb", "")
-      namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+    @go_ui.show_spinner("Analyzing models...") do
+      models = Tng::Analyzers::Model.files_in_dir("app/models").map do |file|
+        relative_path = file[:path].gsub(%r{^.*app/models/}, "").gsub(".rb", "")
+        namespaced_name = relative_path.split("/").map(&:camelize).join("::")
 
-      {
-        name: namespaced_name,
-        path: file[:path]
-      }
+        { name: namespaced_name, path: file[:path] }
+      end
+      { success: true, message: "Found #{models.length} models" }
     end
 
-    flow_display = ModelTestFlowDisplay.new(@prompt, @pastel)
-
     if models.empty?
-      spinner.error
-      flow_display.show_no_models_message
+      @go_ui.show_no_items("models")
       return
     end
 
-    success_msg = @pastel.decorate("Found #{models.length} models", Tng::UI::Theme.color(:success))
-    spinner.success(center_text(success_msg))
+    items = models.map { |m| { name: m[:name], path: m[:path] } }
+    selected_name = @go_ui.show_list_view("Select Model", items)
+    return if selected_name == "back"
 
-    model_choice = flow_display.select_model(models)
-    return if model_choice == :back
+    model_choice = models.find { |m| m[:name] == selected_name }
+    return unless model_choice
 
     show_model_test_options(model_choice)
   end
 
   def generate_service_tests
-    header = @pastel.decorate("Scanning for services...", Tng::UI::Theme.color(:primary))
-    puts center_text(header)
-
-    spinner = TTY::Spinner.new(
-      center_text("[:spinner] Analyzing services..."),
-      format: :dots,
-      success_mark: @pastel.decorate(Tng::UI::Theme.icon(:success), Tng::UI::Theme.color(:success)),
-      error_mark: @pastel.decorate(Tng::UI::Theme.icon(:error), Tng::UI::Theme.color(:error))
-    )
-    spinner.auto_spin
+    services = nil
 
-    services = Tng::Analyzers::Service.files_in_dir.map do |file|
-      relative_path = file[:path].gsub(%r{^.*app/services?/}, "").gsub(".rb", "")
-      namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+    @go_ui.show_spinner("Analyzing services...") do
+      services = Tng::Analyzers::Service.files_in_dir.map do |file|
+        relative_path = file[:path].gsub(%r{^.*app/services?/}, "").gsub(".rb", "")
+        namespaced_name = relative_path.split("/").map(&:camelize).join("::")
 
-      {
-        name: namespaced_name,
-        path: file[:path]
-      }
+        { name: namespaced_name, path: file[:path] }
+      end
+      { success: true, message: "Found #{services.length} services" }
     end
 
-    flow_display = ServiceTestFlowDisplay.new(@prompt, @pastel)
-
     if services.empty?
-      spinner.error
-      flow_display.show_no_services_message
+      @go_ui.show_no_items("services")
       return
     end
 
-    success_msg = @pastel.decorate("Found #{services.length} services", Tng::UI::Theme.color(:success))
-    spinner.success(center_text(success_msg))
+    items = services.map { |s| { name: s[:name], path: s[:path] } }
+    selected_name = @go_ui.show_list_view("Select Service", items)
+    return if selected_name == "back"
 
-    service_choice = flow_display.select_service(services)
-    return if service_choice == :back
+    service_choice = services.find { |s| s[:name] == selected_name }
+    return unless service_choice
 
     show_service_test_options(service_choice)
   end
 
   def generate_other_tests
-    header = @pastel.decorate("Scanning for other files...", Tng::UI::Theme.color(:primary))
-    puts center_text(header)
-
-    spinner = TTY::Spinner.new(
-      center_text("[:spinner] Analyzing other files..."),
-      format: :dots,
-      success_mark: @pastel.decorate(Tng::UI::Theme.icon(:success), Tng::UI::Theme.color(:success)),
-      error_mark: @pastel.decorate(Tng::UI::Theme.icon(:error), Tng::UI::Theme.color(:error))
-    )
-    spinner.auto_spin
+    other_files = nil
 
-    other_files = Tng::Analyzers::Other.files_in_dir.map do |file|
-      relative_path = file[:relative_path].gsub(".rb", "")
-      namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+    @go_ui.show_spinner("Analyzing other files...") do
+      other_files = Tng::Analyzers::Other.files_in_dir.map do |file|
+        relative_path = file[:relative_path].gsub(".rb", "")
+        namespaced_name = relative_path.split("/").map(&:camelize).join("::")
 
-      {
-        name: namespaced_name,
-        path: file[:path],
-        type: file[:type]
-      }
+        { name: namespaced_name, path: file[:path], type: file[:type] }
+      end
+      { success: true, message: "Found #{other_files.length} other files" }
     end
 
-    flow_display = OtherTestFlowDisplay.new(@prompt, @pastel)
-
     if other_files.empty?
-      spinner.error
-      flow_display.show_no_other_files_message
+      @go_ui.show_no_items("other files")
       return
     end
 
-    success_msg = @pastel.decorate("Found #{other_files.length} other files", Tng::UI::Theme.color(:success))
-    spinner.success(center_text(success_msg))
+    items = other_files.map { |f| { name: f[:name], path: f[:path] } }
+    selected_name = @go_ui.show_list_view("Select File", items)
+    return if selected_name == "back"
 
-    other_choice = flow_display.select_other_file(other_files)
-    return if other_choice == :back
+    other_choice = other_files.find { |f| f[:name] == selected_name }
+    return unless other_choice
 
     show_other_test_options(other_choice)
   end
 
   def show_controller_method_selection(controller)
-    header = @pastel.decorate("#{Tng::UI::Theme.icon(:info)} Analyzing methods for #{controller[:name]}", Tng::UI::Theme.color(:primary))
-    puts center_text(header)
-
     methods = extract_controller_methods(controller)
-    flow_display = ControllerTestFlowDisplay.new(@prompt, @pastel)
 
     if methods.empty?
-      flow_display.show_no_methods_message(controller)
+      @go_ui.show_no_items("methods in #{controller[:name]}")
       return
     end
 
-    method_choice = flow_display.select_controller_method(controller, methods)
+    items = methods.map { |m| { name: m[:name], path: controller[:name] } }
+    selected_name = @go_ui.show_list_view("Select Method in #{controller[:name]}", items)
+    return if selected_name == "back"
 
-    return if method_choice == :back
+    method_choice = methods.find { |m| m[:name] == selected_name }
+    return unless method_choice
 
     generate_test_for_controller_method(controller, method_choice)
   end
 
   def generate_test_for_controller_method(controller, method_info)
-    header = @pastel.decorate(
-      "#{Tng::UI::Theme.icon(:marker)} Generating test for #{controller[:name]}##{method_info[:name]}", Tng::UI::Theme.color(:primary)
-    )
-    puts center_text(header)
-
-    analysis_msg = @pastel.decorate("Analyzing method structure...", Tng::UI::Theme.color(:muted))
-    puts center_text(analysis_msg)
-
-    progress_padding = [(@terminal_width - 56) / 2, 0].max # Account for progress bar width
-    progress_bar = TTY::ProgressBar.new(
-      "#{" " * progress_padding}[:bar] :percent :current/:total",
-      total: 5,
-      bar_format: :box,
-      complete: @pastel.decorate("█", Tng::UI::Theme.color(:success)),
-      incomplete: @pastel.decorate("░", Tng::UI::Theme.color(:muted)),
-      width: 40
-    )
+    result = nil
 
-    step_msg = @pastel.decorate("Parsing method details...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.3)
-
-    step_msg = @pastel.decorate("Analyzing method dependencies...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.3)
-
-    step_msg = @pastel.decorate("Extracting method context...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.3)
-
-    step_msg = @pastel.decorate("Generating method test code...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    result = Tng::Services::TestGenerator.new(@http_client).run_for_controller_method(controller, method_info)
-    sleep(0.5)
-
-    step_msg = @pastel.decorate("Writing test file...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.2)
-
-    # Show completion message with test count and timing
-    if result && result[:test_count]
-      count_msg = result[:test_count] == 1 ? "1 test" : "#{result[:test_count]} tests"
-      time_msg = result[:generation_time] ? " in #{Tng::Utils.format_generation_time(result[:generation_time])}" : ""
-      completion_msg = @pastel.decorate(
-        "#{Tng::UI::Theme.icon(:success)} Generated #{count_msg} for controller method#{time_msg}!", Tng::UI::Theme.color(:success)
-      )
-    else
-      completion_msg = @pastel.decorate("#{Tng::UI::Theme.icon(:success)} Method test generation completed!", Tng::UI::Theme.color(:success))
-    end
-    puts center_text(completion_msg)
+    @go_ui.show_progress("Generating test for #{controller[:name]}##{method_info[:name]}") do |progress|
+      progress.update("Parsing method details...", 20)
+      progress.update("Analyzing method dependencies...", 40)
+      progress.update("Extracting method context...", 60)
+      progress.update("Generating method test code...", 80)
 
-    if result && result[:file_path]
-      show_post_generation_menu(result)
-    else
-      @prompt.keypress(center_text(@pastel.decorate("Press any key to continue...", Tng::UI::Theme.color(:muted))))
+      result = Tng::Services::TestGenerator.new(@http_client).run_for_controller_method(controller, method_info, progress: progress)
+
+      progress.update("Writing test file...", 100)
+
+      handle_generation_result(result)
     end
+
+    return @go_ui.show_auth_error(result[:message] || "Generation failed") if result&.dig(:error)
+
+    show_post_generation_menu(result) if result && result[:file_path]
   end
 
   def show_controller_test_options(controller_choice)
@@ -503,157 +371,77 @@ class CLI
   end
 
   def show_model_method_selection(model)
-    header = @pastel.decorate("#{Tng::UI::Theme.icon(:info)} Analyzing methods for #{model[:name]}", Tng::UI::Theme.color(:primary))
-    puts center_text(header)
-
     methods = extract_model_methods(model)
-    flow_display = ModelTestFlowDisplay.new(@prompt, @pastel)
 
     if methods.empty?
-      flow_display.show_no_methods_message(model)
+      @go_ui.show_no_items("methods in #{model[:name]}")
       return
     end
 
-    method_choice = flow_display.select_model_method(model, methods)
+    items = methods.map { |m| { name: m[:name], path: model[:name] } }
+    selected_name = @go_ui.show_list_view("Select Method in #{model[:name]}", items)
+    return if selected_name == "back"
 
-    return if method_choice == :back
+    method_choice = methods.find { |m| m[:name] == selected_name }
+    return unless method_choice
 
     generate_test_for_model_method(model, method_choice)
   end
 
   def generate_test_for_model_method(model, method_info)
-    header = @pastel.decorate(
-      "#{Tng::UI::Theme.icon(:marker)} Generating test for #{model[:name]}##{method_info[:name]}", Tng::UI::Theme.color(:primary)
-    )
-    puts center_text(header)
-
-    analysis_msg = @pastel.decorate("Analyzing method structure...", Tng::UI::Theme.color(:muted))
-    puts center_text(analysis_msg)
-
-    progress_padding = [(@terminal_width - 56) / 2, 0].max # Account for progress bar width
-    progress_bar = TTY::ProgressBar.new(
-      "#{" " * progress_padding}[:bar] :percent :current/:total",
-      total: 4,
-      bar_format: :box,
-      complete: @pastel.decorate("█", Tng::UI::Theme.color(:success)),
-      incomplete: @pastel.decorate("░", Tng::UI::Theme.color(:muted)),
-      width: 40
-    )
+    result = nil
 
-    step_msg = @pastel.decorate("Parsing method details...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.3)
-
-    step_msg = @pastel.decorate("Analyzing method dependencies...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.3)
-
-    step_msg = @pastel.decorate("Generating method test code...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    result = Tng::Services::TestGenerator.new(@http_client).run_for_model_method(model, method_info)
-    sleep(0.5)
-
-    step_msg = @pastel.decorate("Writing test file...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.2)
-
-    # Show completion message with test count and timing
-    if result && result[:test_count]
-      count_msg = result[:test_count] == 1 ? "1 test" : "#{result[:test_count]} tests"
-      time_msg = result[:generation_time] ? " in #{Tng::Utils.format_generation_time(result[:generation_time])}" : ""
-      completion_msg = @pastel.decorate(
-        "#{Tng::UI::Theme.icon(:success)} Generated #{count_msg} for model method#{time_msg}!", Tng::UI::Theme.color(:success)
-      )
-    else
-      completion_msg = @pastel.decorate("#{Tng::UI::Theme.icon(:success)} Method test generation completed!", Tng::UI::Theme.color(:success))
-    end
-    puts center_text(completion_msg)
+    @go_ui.show_progress("Generating test for #{model[:name]}##{method_info[:name]}") do |progress|
+      progress.update("Parsing method details...", 25)
+      progress.update("Analyzing method dependencies...", 50)
+      progress.update("Generating method test code...", 75)
 
-    if result && result[:file_path]
-      show_post_generation_menu(result)
-    else
-      @prompt.keypress(center_text(@pastel.decorate("Press any key to continue...", Tng::UI::Theme.color(:muted))))
+      result = Tng::Services::TestGenerator.new(@http_client).run_for_model_method(model, method_info, progress: progress)
+
+      progress.update("Writing test file...", 100)
+
+      handle_generation_result(result)
     end
+
+    return @go_ui.show_auth_error(result[:message] || "Generation failed") if result&.dig(:error)
+
+    show_post_generation_menu(result) if result && result[:file_path]
   end
 
   def generate_test_for_service_method(service, method_info)
-    header = @pastel.decorate(
-      "#{Tng::UI::Theme.icon(:marker)} Generating test for #{service[:name]}##{method_info[:name]}", Tng::UI::Theme.color(:primary)
-    )
-    puts center_text(header)
-
-    analysis_msg = @pastel.decorate("Analyzing method structure...", Tng::UI::Theme.color(:muted))
-    puts center_text(analysis_msg)
-
-    progress_padding = [(@terminal_width - 56) / 2, 0].max # Account for progress bar width
-    progress_bar = TTY::ProgressBar.new(
-      "#{" " * progress_padding}[:bar] :percent :current/:total",
-      total: 4,
-      bar_format: :box,
-      complete: @pastel.decorate("█", Tng::UI::Theme.color(:success)),
-      incomplete: @pastel.decorate("░", Tng::UI::Theme.color(:muted)),
-      width: 40
-    )
+    result = nil
 
-    step_msg = @pastel.decorate("Parsing method details...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.3)
-
-    step_msg = @pastel.decorate("Analyzing method dependencies...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.3)
-
-    step_msg = @pastel.decorate("Generating method test code...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    result = Tng::Services::TestGenerator.new(@http_client).run_for_service_method(service, method_info)
-    sleep(0.5)
-
-    step_msg = @pastel.decorate("Writing test file...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.2)
-
-    # Show completion message with test count and timing
-    if result && result[:test_count]
-      count_msg = result[:test_count] == 1 ? "1 test" : "#{result[:test_count]} tests"
-      time_msg = result[:generation_time] ? " in #{Tng::Utils.format_generation_time(result[:generation_time])}" : ""
-      completion_msg = @pastel.decorate(
-        "#{Tng::UI::Theme.icon(:success)} Generated #{count_msg} for service method#{time_msg}!", Tng::UI::Theme.color(:success)
-      )
-    else
-      completion_msg = @pastel.decorate("#{Tng::UI::Theme.icon(:success)} Method test generation completed!", Tng::UI::Theme.color(:success))
-    end
-    puts center_text(completion_msg)
+    @go_ui.show_progress("Generating test for #{service[:name]}##{method_info[:name]}") do |progress|
+      progress.update("Parsing method details...", 25)
+      progress.update("Analyzing method dependencies...", 50)
+      progress.update("Generating method test code...", 75)
 
-    if result && result[:file_path]
-      show_post_generation_menu(result)
-    else
-      @prompt.keypress(center_text(@pastel.decorate("Press any key to continue...", Tng::UI::Theme.color(:muted))))
+      result = Tng::Services::TestGenerator.new(@http_client).run_for_service_method(service, method_info, progress: progress)
+
+      progress.update("Writing test file...", 100)
+
+      handle_generation_result(result)
     end
+
+    return @go_ui.show_auth_error(result[:message] || "Generation failed") if result&.dig(:error)
+
+    show_post_generation_menu(result) if result && result[:file_path]
   end
 
   def show_service_method_selection(service)
-    header = @pastel.decorate("#{Tng::UI::Theme.icon(:info)} Analyzing methods for #{service[:name]}", Tng::UI::Theme.color(:primary))
-    puts center_text(header)
-
     methods = extract_service_methods(service)
-    flow_display = ServiceTestFlowDisplay.new(@prompt, @pastel)
 
     if methods.empty?
-      flow_display.show_no_methods_message(service)
+      @go_ui.show_no_items("methods in #{service[:name]}")
       return
     end
 
-    method_choice = flow_display.select_service_method(service, methods)
+    items = methods.map { |m| { name: m[:name], path: service[:name] } }
+    selected_name = @go_ui.show_list_view("Select Method in #{service[:name]}", items)
+    return if selected_name == "back"
 
-    return if method_choice == :back
+    method_choice = methods.find { |m| m[:name] == selected_name }
+    return unless method_choice
 
     generate_test_for_service_method(service, method_choice)
   end
@@ -665,87 +453,58 @@ class CLI
   end
 
   def show_other_method_selection(other_file)
-    header = @pastel.decorate("#{Tng::UI::Theme.icon(:info)} Analyzing methods for #{other_file[:name]}", Tng::UI::Theme.color(:primary))
-    puts center_text(header)
-
     methods = extract_other_methods(other_file)
-    flow_display = OtherTestFlowDisplay.new(@prompt, @pastel)
 
     if methods.empty?
-      flow_display.show_no_methods_message(other_file)
+      @go_ui.show_no_items("methods in #{other_file[:name]}")
       return
     end
 
-    method_choice = flow_display.select_other_method(other_file, methods)
+    items = methods.map { |m| { name: m[:name], path: other_file[:name] } }
+    selected_name = @go_ui.show_list_view("Select Method in #{other_file[:name]}", items)
+    return if selected_name == "back"
 
-    return if method_choice == :back
+    method_choice = methods.find { |m| m[:name] == selected_name }
+    return unless method_choice
 
     generate_test_for_other_method(other_file, method_choice)
   end
 
   def generate_test_for_other_method(other_file, method_info)
-    header = @pastel.decorate(
-      "#{Tng::UI::Theme.icon(:marker)} Generating test for #{other_file[:name]}##{method_info[:name]}", Tng::UI::Theme.color(:primary)
-    )
-    puts center_text(header)
-
-    analysis_msg = @pastel.decorate("Analyzing method structure...", Tng::UI::Theme.color(:muted))
-    puts center_text(analysis_msg)
-
-    progress_padding = [(@terminal_width - 56) / 2, 0].max # Account for progress bar width
-    progress_bar = TTY::ProgressBar.new(
-      "#{" " * progress_padding}[:bar] :percent :current/:total",
-      total: 4,
-      bar_format: :box,
-      complete: @pastel.decorate("█", Tng::UI::Theme.color(:success)),
-      incomplete: @pastel.decorate("░", Tng::UI::Theme.color(:muted)),
-      width: 40
-    )
+    result = nil
 
-    step_msg = @pastel.decorate("Parsing method details...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.3)
-
-    step_msg = @pastel.decorate("Analyzing method dependencies...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.3)
-
-    step_msg = @pastel.decorate("Generating method test code...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    result = Tng::Services::TestGenerator.new(@http_client).run_for_other_method(other_file, method_info)
-    sleep(0.5)
-
-    step_msg = @pastel.decorate("Writing test file...", Tng::UI::Theme.color(:muted))
-    puts center_text(step_msg)
-    progress_bar.advance(1)
-    sleep(0.2)
-
-    # Show completion message with test count and timing
-    if result && result[:test_count]
-      count_msg = result[:test_count] == 1 ? "1 test" : "#{result[:test_count]} tests"
-      time_msg = result[:generation_time] ? " in #{Tng::Utils.format_generation_time(result[:generation_time])}" : ""
-      completion_msg = @pastel.decorate(
-        "#{Tng::UI::Theme.icon(:success)} Generated #{count_msg} for other method#{time_msg}!", Tng::UI::Theme.color(:success)
-      )
-    else
-      completion_msg = @pastel.decorate("#{Tng::UI::Theme.icon(:success)} Method test generation completed!", Tng::UI::Theme.color(:success))
+    @go_ui.show_progress("Generating test for #{other_file[:name]}##{method_info[:name]}") do |progress|
+      progress.update("Parsing method details...", 25)
+      progress.update("Analyzing method dependencies...", 50)
+      progress.update("Generating method test code...", 75)
+
+      result = Tng::Services::TestGenerator.new(@http_client).run_for_other_method(other_file, method_info, progress: progress)
+
+      progress.update("Writing test file...", 100)
+
+      handle_generation_result(result)
     end
-    puts center_text(completion_msg)
 
-    if result && result[:file_path]
-      show_post_generation_menu(result)
+    return @go_ui.show_auth_error(result[:message] || "Generation failed") if result&.dig(:error)
+
+    show_post_generation_menu(result) if result && result[:file_path]
+  end
+
+  def handle_generation_result(result)
+    if result&.dig(:error)
+      { message: result[:message] || "Generation failed", error: result[:error], result: result }
+    elsif result && result[:test_count]
+      count_msg = result[:test_count] == 1 ? "1 test" : "#{result[:test_count]} tests"
+      { message: "Generated #{count_msg}!", result: result }
     else
-      @prompt.keypress(center_text(@pastel.decorate("Press any key to continue...", Tng::UI::Theme.color(:muted))))
+      { message: "Test generation completed!", result: result }
     end
   end
 
   # STATS
   def user_stats
     stats_data = @testng.get_user_stats
-    UserStatsDisplay.new(@pastel, @prompt).display(stats_data)
+    @go_ui.show_stats(stats_data)
   end
 
   def check_system_status
@@ -755,13 +514,17 @@ class CLI
     end
 
     status = @testng.check_system_status
-    status_ok = SystemStatusDisplay.new(@pastel, params).display(status)
 
-    return if status_ok
+    if status[:status] == :ok
+      unless params[:file]
+        puts @pastel.decorate("#{Tng::UI::Theme.icon(:success)} System operational",
+                              Tng::UI::Theme.color(:success))
+      end
+      return
+    end
 
-    puts
-    puts @pastel.decorate("Press any key to exit...", Tng::UI::Theme.color(:muted))
-    $stdin.getch
+    # Show error via Go UI
+    @go_ui.show_system_status(status)
     exit(1)
   end
 
@@ -770,28 +533,74 @@ class CLI
 
     missing = Tng::Services::UserAppConfig.missing_config
 
-    ConfigurationDisplay.new(@pastel).display_missing_config(missing)
+    @go_ui.show_config_missing(missing)
     false
   end
 
   def check_authentication_configuration
-    auth_warning = AuthenticationWarningDisplay.new(@pastel)
-    auth_missing = auth_warning.display_if_missing
+    return true unless authentication_config_missing?
 
-    return false if auth_missing
+    choice = @go_ui.show_auth_warning(build_auth_missing_summary)
+
+    # "continue" means user wants to proceed, "cancel" means abort
+    choice == "continue"
+  end
 
-    true
+  def authentication_config_missing?
+    return true if Tng.authentication_enabled && authentication_methods_empty?
+    return true if !authentication_methods_empty? && !authentication_methods_valid?
+
+    false
+  end
+
+  def authentication_methods_empty?
+    methods = Tng.authentication_methods
+    methods.nil? || methods.empty?
+  end
+
+  def authentication_methods_valid?
+    methods = Tng.authentication_methods
+    return false if methods.nil? || methods.empty?
+
+    methods.all? do |method|
+      method.is_a?(Hash) &&
+        method.key?(:method) && !method[:method].to_s.strip.empty? &&
+        method.key?(:file_location) && !method[:file_location].to_s.strip.empty? &&
+        method.key?(:auth_type) && !method[:auth_type].to_s.strip.empty?
+    end
+  end
+
+  def build_auth_missing_summary
+    issues = []
+    issues << "empty auth methods" if Tng.authentication_enabled && authentication_methods_empty?
+    issues << "invalid auth config" if !authentication_methods_empty? && !authentication_methods_valid?
+    issues << "auth disabled" unless Tng.authentication_enabled
+    issues.empty? ? "unknown issue" : issues.join(", ")
+  end
+
+  def check_authentication
+    puts @pastel.decorate("#{Tng::UI::Theme.icon(:info)} Verifying authentication...",
+                          Tng::UI::Theme.color(:muted))
+
+    result = @http_client.verify_auth
+
+    if result[:valid]
+      true
+    else
+      @go_ui.show_auth_error(result[:error])
+      false
+    end
   end
 
   # ABOUT
   def show_about
-    AboutDisplay.new(@pastel, @prompt, Tng::VERSION).display
+    @go_ui.show_about
   end
 
   # GOODBYE
   def show_goodbye
     clear_screen
-    GoodbyeDisplay.new(@pastel).display
+    @go_ui.show_goodbye
   end
 
   def run_direct_generation
@@ -800,8 +609,8 @@ class CLI
     return unless check_authentication_configuration
 
     status = @testng.check_system_status
-    unless SystemStatusDisplay.new(@pastel, params).display(status)
-      puts @pastel.red("❌ System status check failed")
+    unless status[:status] == :ok
+      @go_ui.show_system_status(status)
       return
     end
 
@@ -819,160 +628,57 @@ class CLI
   def show_post_generation_menu(result)
     is_direct_mode = params[:file]
 
+    # Direct mode - just print info and exit
     if is_direct_mode
-      puts
-      header = @pastel.decorate("#{Tng::UI::Theme.icon(:rocket)} Test Generated Successfully!", Tng::UI::Theme.color(:primary))
-      puts center_text(header)
-      puts
-
-      info_msg = [
-        @pastel.decorate("#{Tng::UI::Theme.icon(:config)} File: #{result[:file_path]}", Tng::UI::Theme.color(:info)),
-        @pastel.decorate("#{Tng::UI::Theme.icon(:marker)} Run: #{result[:run_command]}", Tng::UI::Theme.color(:secondary))
-      ].join("\n")
-      puts center_text(info_msg)
-      puts
-
-      puts center_text(@pastel.decorate("#{Tng::UI::Theme.icon(:success)} Test generation completed successfully!", Tng::UI::Theme.color(:success)))
+      puts "\n🚀 Test Generated Successfully!"
+      puts "📁 File: #{result[:file_path]}"
+      puts "▶️  Run: #{result[:run_command]}"
+      puts "✅ Test generation completed successfully!"
       return
     end
 
-    # Show menu regardless of test results
+    # Interactive mode - show Go UI menu
     loop do
-      puts
-      header = @pastel.decorate("#{Tng::UI::Theme.icon(:rocket)} Test Generated Successfully!", Tng::UI::Theme.color(:primary))
-      puts center_text(header)
-      puts
-
-      info_msg = [
-        @pastel.decorate("#{Tng::UI::Theme.icon(:config)} File: #{result[:file_path]}", Tng::UI::Theme.color(:info)),
-        @pastel.decorate("#{Tng::UI::Theme.icon(:marker)} Run: #{result[:run_command]}", Tng::UI::Theme.color(:secondary))
-      ].join("\n")
-      puts center_text(info_msg)
-      puts
-
-      choice = @prompt.select(
-        @pastel.decorate("What would you like to do?", Tng::UI::Theme.color(:primary)),
-        cycle: true,
-        symbols: { marker: Tng::UI::Theme.icon(:marker) }
-      ) do |menu|
-        menu.choice @pastel.decorate("#{Tng::UI::Theme.icon(:run)} Run tests", Tng::UI::Theme.color(:success)),
-                    :run_tests
-        menu.choice @pastel.decorate("#{Tng::UI::Theme.icon(:config)} Copy run command", Tng::UI::Theme.color(:warning)),
-                    :copy_command
-        menu.choice @pastel.decorate("#{Tng::UI::Theme.icon(:back)} Back to main menu", Tng::UI::Theme.color(:info)),
-                    :back
-      end
+      choice = @go_ui.show_post_generation_menu(result[:file_path], result[:run_command])
 
       case choice
-      when :run_tests
-        run_and_show_test_results(result[:run_command])
-      when :copy_command
+
+      when "copy_command"
         copy_to_clipboard(result[:run_command])
-      when :back
+      when "back"
         break
-      end
-    end
-  end
-
-  def run_and_show_test_results(command)
-    puts
-    header = @pastel.decorate("#{Tng::UI::Theme.icon(:run)} Running tests...", Tng::UI::Theme.color(:primary))
-    puts center_text(header)
-
-    begin
-      # Check if test file exists before running
-      test_file = extract_test_file_from_command(command)
-      unless test_file && File.exist?(test_file)
-        error_msg = @pastel.decorate("#{Tng::UI::Theme.icon(:error)} Test file not found!", Tng::UI::Theme.color(:error))
-        puts center_text(error_msg)
-
-        if test_file
-          file_msg = @pastel.decorate("File: #{test_file}", Tng::UI::Theme.color(:muted))
-          puts center_text(file_msg)
-        end
-
-        suggestion_msg = @pastel.decorate("The test file may have been moved or deleted.", Tng::UI::Theme.color(:warning))
-        puts center_text(suggestion_msg)
-
-        @prompt.keypress(center_text(@pastel.decorate("Press any key to continue...", Tng::UI::Theme.color(:muted))))
-        return
-      end
-
-      # Modify command to get cleaner output
-      if command.include?("rspec")
-        # Use JSON format for RSpec - much cleaner parsing
-        json_command = command.sub(/rspec/, "rspec --format json --out /dev/null")
-        output = `#{json_command} 2>&1`
-        exit_code = $?.exitstatus
-
-        # Check for common errors before parsing
-        return if handle_test_execution_errors(output, exit_code)
-
-        Tng::Utils.parse_rspec_json_results(output, exit_code, @pastel, @terminal_width)
       else
-        # For Rails test, use current approach but with cleaner output
-        output = `#{command} 2>&1`
-        exit_code = $?.exitstatus
-
-        # Check for common errors before parsing
-        return if handle_test_execution_errors(output, exit_code)
-
-        Tng::Utils.parse_minitest_results(output, exit_code, @pastel, @terminal_width)
+        break
       end
-    rescue StandardError => e
-      error_msg = @pastel.decorate("#{Tng::UI::Theme.icon(:error)} Error running tests!", Tng::UI::Theme.color(:error))
-      puts center_text(error_msg)
-
-      details_msg = @pastel.decorate("#{e.message}", Tng::UI::Theme.color(:muted))
-      puts center_text(details_msg)
     end
-
-    @prompt.keypress(center_text(@pastel.decorate("Press any key to continue...", Tng::UI::Theme.color(:muted))))
   end
 
-  def handle_test_execution_errors(output, exit_code)
-    # Return true if we handled an error (caller should return early)
-    # Return false if no error was handled (caller should continue)
 
-    # If tests failed to run (non-zero exit code or common error patterns)
-    if !exit_code.zero? || output.include?("LoadError") || output.include?("cannot load such file") ||
-       (output.include?("bundler") && output.include?("Could not find")) ||
-       (output.include?("bin/rails") && (output.include?("require") || output.include?("load")))
 
-      error_msg = @pastel.decorate("#{Tng::UI::Theme.icon(:error)} Tests failed to run", Tng::UI::Theme.color(:error))
-      puts center_text(error_msg)
-
-      @prompt.keypress(center_text(@pastel.decorate("Press any key to continue...", Tng::UI::Theme.color(:muted))))
-      return true
+  def copy_to_clipboard(command)
+    if RUBY_PLATFORM.include?("darwin")
+      IO.popen("pbcopy", "w") { |io| io.print command }
+      @go_ui.show_clipboard_success(command)
+    elsif RUBY_PLATFORM.include?("linux")
+      begin
+        IO.popen("xclip -selection clipboard", "w") { |io| io.print command }
+        @go_ui.show_clipboard_success(command)
+      rescue Errno::ENOENT
+        puts "\n📋 Copy this command:\n#{command}\n"
+      end
+    else
+      puts "\n📋 Copy this command:\n#{command}\n"
     end
-
-    false
   end
 
-  def extract_test_file_from_command(command)
-    # Extract test file path from commands like:
-    # "bundle exec rails test test/controllers/foo_test.rb"
-    # "bundle exec rspec spec/controllers/foo_spec.rb"
 
-    if command.include?("rails test")
-      # For Rails test commands
-      match = command.match(/rails test\s+(.+?)(?:\s|$)/)
-      match[1] if match
-    elsif command.include?("rspec")
-      # For RSpec commands
-      match = command.match(/rspec\s+(.+?)(?:\s|$)/)
-      match[1] if match
-    else
-      nil
-    end
-  end
 
   def initialize_config_and_clients
     @config_initialized = true
 
     unless Tng::Services::UserAppConfig.configured?
       missing = Tng::Services::UserAppConfig.missing_config
-      ConfigurationDisplay.new(@pastel).display_missing_config(missing)
+      @go_ui.show_config_missing(missing)
       exit 1
     end