tng 0.2.1

data/bin/tng

@@ -0,0 +1,888 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+$0 = "tng"
+
+lib_path = File.expand_path("../lib", __dir__)
+$LOAD_PATH.unshift(lib_path) unless $LOAD_PATH.include?(lib_path)
+
+require "tng"
+require "tng/services/direct_generation"
+require "tng/services/extract_methods"
+
+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/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"
+require "tng/analyzers/other"
+require "tng/services/test_generator"
+require "tng/services/user_app_config"
+require "tng/utils"
+
+class CLI
+  include Tng::Services::ExtractMethods
+  include TTY::Option
+  include Tng::Utils
+
+  usage do
+    program "tng"
+    desc "LLM-Powered Rails Test Generator"
+    example "Interactive mode (method-specific test generation):", "bundle exec tng"
+    example "  • Browse and search controllers/models/services", ""
+    example "  • Select specific methods from filtered lists", ""
+    example "  • Generate tests for individual methods", ""
+  end
+
+  option :type do
+    short "-t"
+    long "--type=TYPE"
+    desc "Test type (controller, model, service, other)"
+    permit %w[controller model service other c m mo s se o]
+  end
+
+  option :file do
+    short "-f"
+    long "--file=FILE"
+    desc "File name (without .rb extension)"
+  end
+
+  option :method do
+    short "-m"
+    long "--method=METHOD"
+    desc "Method name (for per-method tests)"
+  end
+
+  flag :help do
+    short "-h"
+    long "--help"
+    desc "Show help message"
+  end
+
+  def initialize
+    @prompt = TTY::Prompt.new
+    @pastel = Pastel.new
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+    @config_initialized = false
+    @http_client = nil
+    @testng = nil
+  end
+
+  def start
+    normalized_argv = preprocess_arguments(ARGV.dup)
+
+    parse(normalized_argv)
+
+    if params[:help]
+      help_display = ShowHelp.new(@pastel, Tng::VERSION)
+      puts help_display.render
+      return
+    end
+
+    rails_loaded = load_rails_environment
+
+    initialize_config_and_clients if rails_loaded
+
+    if params[:type] && params[:file]
+      run_direct_generation
+    else
+      if find_rails_root && !rails_loaded && !defined?(Rails)
+        puts @pastel.decorate("#{Tng::UI::Theme.icon(:error)} Failed to load Rails environment.", Tng::UI::Theme.color(:error))
+        puts @pastel.decorate("Please ensure you're running this command from a Rails application directory.", Tng::UI::Theme.color(:warning))
+        puts @pastel.decorate("Try running: bundle exec tng", Tng::UI::Theme.color(:muted))
+        return
+      end
+
+      return unless check_configuration
+
+      check_system_status
+      clear_screen
+      puts DisplayBanner.new(@pastel, Tng::VERSION).render
+      main_menu
+    end
+  end
+
+  def preprocess_arguments(argv)
+    normalized = []
+
+    argv.each do |arg|
+      case arg
+      when /^(type|t)=(.+)$/ # Handle type=value or t=value (without dash)
+        normalized << "--type=#{::Regexp.last_match(2)}"
+      when /^(file|f)=(.+)$/ # Handle file=value or f=value (without dash)
+        normalized << "--file=#{::Regexp.last_match(2)}"
+      when /^(method|m)=(.+)$/ # Handle method=value or m=value (without dash)
+        normalized << "--method=#{::Regexp.last_match(2)}"
+      when /^(help|h)=(.+)$/ # Handle help=value or h=value (without dash)
+        normalized << "--help=#{::Regexp.last_match(2)}"
+      when /^-([tf])=(.+)$/ # Handle -t=value or -f=value (convert to long form)
+        key = ::Regexp.last_match(1) == "t" ? "type" : "file"
+        normalized << "--#{key}=#{::Regexp.last_match(2)}"
+      when /^-([mh])=(.+)$/ # Handle -m=value or -h=value
+        key = ::Regexp.last_match(1) == "m" ? "method" : "help"
+        normalized << "--#{key}=#{::Regexp.last_match(2)}"
+      when /^--\w+=.+$/ # Handle --key=value (already correct)
+        normalized << arg
+      when /^-[tfmh]$/ # Handle single flags like -t, -f, -m, -h (keep as is)
+        normalized << arg
+      when /^--\w+$/ # Handle long flags like --help (keep as is)
+        normalized << arg
+      else
+        # Regular arguments or values
+        normalized << arg
+      end
+    end
+
+    normalized
+  end
+
+  def main_menu
+    loop do
+      clear_screen
+
+      puts DisplayBanner.new(@pastel, Tng::VERSION).render
+
+      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
+
+        handle_choice(choice)
+      rescue TTY::Reader::InputInterrupt, Interrupt
+        choice = :exit
+      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
+  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
+
+    case choice
+    when :controller
+      generate_controller_tests
+    when :model
+      generate_model_tests
+    when :service
+      generate_service_tests
+    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("::")
+
+      {
+        name: namespaced_name,
+        path: file[:path]
+      }
+    end
+
+    flow_display = ControllerTestFlowDisplay.new(@prompt, @pastel)
+
+    if controllers.empty?
+      spinner.error
+      flow_display.show_no_controllers_message
+      return
+    end
+
+    success_msg = @pastel.decorate("Found #{controllers.length} controllers", Tng::UI::Theme.color(:success))
+    spinner.success(center_text(success_msg))
+
+    controller_choice = flow_display.select_controller(controllers)
+    return if controller_choice == :back
+
+    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 = 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]
+      }
+    end
+
+    flow_display = ModelTestFlowDisplay.new(@prompt, @pastel)
+
+    if models.empty?
+      spinner.error
+      flow_display.show_no_models_message
+      return
+    end
+
+    success_msg = @pastel.decorate("Found #{models.length} models", Tng::UI::Theme.color(:success))
+    spinner.success(center_text(success_msg))
+
+    model_choice = flow_display.select_model(models)
+    return if model_choice == :back
+
+    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 = 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]
+      }
+    end
+
+    flow_display = ServiceTestFlowDisplay.new(@prompt, @pastel)
+
+    if services.empty?
+      spinner.error
+      flow_display.show_no_services_message
+      return
+    end
+
+    success_msg = @pastel.decorate("Found #{services.length} services", Tng::UI::Theme.color(:success))
+    spinner.success(center_text(success_msg))
+
+    service_choice = flow_display.select_service(services)
+    return if service_choice == :back
+
+    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 = 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]
+      }
+    end
+
+    flow_display = OtherTestFlowDisplay.new(@prompt, @pastel)
+
+    if other_files.empty?
+      spinner.error
+      flow_display.show_no_other_files_message
+      return
+    end
+
+    success_msg = @pastel.decorate("Found #{other_files.length} other files", Tng::UI::Theme.color(:success))
+    spinner.success(center_text(success_msg))
+
+    other_choice = flow_display.select_other_file(other_files)
+    return if other_choice == :back
+
+    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)
+      return
+    end
+
+    method_choice = flow_display.select_controller_method(controller, methods)
+
+    return if method_choice == :back
+
+    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
+    )
+
+    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 #{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)
+
+    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))))
+    end
+  end
+
+  def show_controller_test_options(controller_choice)
+    return unless check_authentication_configuration
+
+    show_controller_method_selection(controller_choice)
+  end
+
+  def show_model_test_options(model_choice)
+    return unless check_authentication_configuration
+
+    show_model_method_selection(model_choice)
+  end
+
+  def show_service_test_options(service_choice)
+    return unless check_authentication_configuration
+
+    show_service_method_selection(service_choice)
+  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)
+      return
+    end
+
+    method_choice = flow_display.select_model_method(model, methods)
+
+    return if method_choice == :back
+
+    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
+    )
+
+    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 #{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)
+
+    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))))
+    end
+  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
+    )
+
+    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 #{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)
+
+    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))))
+    end
+  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)
+      return
+    end
+
+    method_choice = flow_display.select_service_method(service, methods)
+
+    return if method_choice == :back
+
+    generate_test_for_service_method(service, method_choice)
+  end
+
+  def show_other_test_options(other_choice)
+    return unless check_authentication_configuration
+
+    show_other_method_selection(other_choice)
+  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)
+      return
+    end
+
+    method_choice = flow_display.select_other_method(other_file, methods)
+
+    return if method_choice == :back
+
+    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
+    )
+
+    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 #{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))
+    end
+    puts center_text(completion_msg)
+
+    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))))
+    end
+  end
+
+  # STATS
+  def user_stats
+    stats_data = @testng.get_user_stats
+    UserStatsDisplay.new(@pastel, @prompt).display(stats_data)
+  end
+
+  def check_system_status
+    unless params[:type] && params[:file]
+      puts @pastel.decorate("#{Tng::UI::Theme.icon(:info)} Checking system status...",
+                            Tng::UI::Theme.color(:muted))
+    end
+
+    status = @testng.check_system_status
+    status_ok = SystemStatusDisplay.new(@pastel, params).display(status)
+
+    return if status_ok
+
+    puts
+    puts @pastel.decorate("Press any key to exit...", Tng::UI::Theme.color(:muted))
+    $stdin.getch
+    exit(1)
+  end
+
+  def check_configuration
+    return true if Tng::Services::UserAppConfig.configured?
+
+    missing = Tng::Services::UserAppConfig.missing_config
+
+    ConfigurationDisplay.new(@pastel).display_missing_config(missing)
+    false
+  end
+
+  def check_authentication_configuration
+    auth_warning = AuthenticationWarningDisplay.new(@pastel)
+    auth_missing = auth_warning.display_if_missing
+
+    return false if auth_missing
+
+    true
+  end
+
+  # ABOUT
+  def show_about
+    AboutDisplay.new(@pastel, @prompt, Tng::VERSION).display
+  end
+
+  # GOODBYE
+  def show_goodbye
+    clear_screen
+    GoodbyeDisplay.new(@pastel).display
+  end
+
+  def run_direct_generation
+    return unless check_configuration
+
+    return unless check_authentication_configuration
+
+    status = @testng.check_system_status
+    unless SystemStatusDisplay.new(@pastel, params).display(status)
+      puts @pastel.red("❌ System status check failed")
+      return
+    end
+
+    direct_generator = Tng::Services::DirectGeneration.new(
+      @pastel,
+      @testng,
+      @http_client,
+      params,
+      method(:show_post_generation_menu)
+    )
+
+    direct_generator.run
+  end
+
+  def show_post_generation_menu(result)
+    is_direct_mode = params[:type] && params[:file]
+
+    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(:warning))
+      ].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)))
+      return
+    end
+
+    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(:warning))
+      ].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(: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
+
+      case choice
+      when :copy_command
+        copy_to_clipboard(result[:run_command])
+      when :back
+        break
+      end
+    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)
+      exit 1
+    end
+
+    @http_client = Tng::HttpClient.new(
+      Tng::Services::UserAppConfig.base_url,
+      Tng::Services::UserAppConfig.api_key
+    )
+    @testng = Services::Testng.new(@http_client)
+  end
+
+  private
+
+  def format_generation_time(seconds)
+    if seconds < 1
+      "#{(seconds * 1000).round}ms"
+    elsif seconds < 60
+      "#{seconds.round(1)}s"
+    else
+      minutes = (seconds / 60).floor
+      remaining_seconds = (seconds % 60).round
+      "#{minutes}m #{remaining_seconds}s"
+    end
+  end
+end
+
+cli = CLI.new
+cli.start