tng 0.1.0

data/bin/tng

@@ -0,0 +1,1096 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Set a clean process name for the terminal
+$0 = "tng"
+
+# Add the lib directory to the load path for development
+lib_path = File.expand_path("../lib", __dir__)
+$LOAD_PATH.unshift(lib_path) unless $LOAD_PATH.include?(lib_path)
+
+# Load the main gem first to ensure the Rust extension is loaded
+require "tng"
+
+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/system_status_display"
+require "tng/ui/configuration_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/analyzers/controller"
+require "tng/analyzers/model"
+require "tng/analyzers/service"
+require "tng/services/test_generator"
+require "tng/services/user_app_config"
+require "tng/utils"
+
+class CLI
+  include TTY::Option
+  include Tng::Utils
+
+  usage do
+    program "tng"
+    desc "LLM-Powered Rails Test Generator"
+    example "Interactive mode (full UI with search & selection):", "bundle exec tng"
+    example "  • Browse and search controllers/models", ""
+    example "  • Select specific files from filtered lists", ""
+    example "  • Choose test generation options", ""
+    example "", ""
+    example "Direct generation (bypass UI):", ""
+    example "Standard format:", "bundle exec tng --type=controller --file=users_controller"
+    example "Short aliases:", "bundle exec tng -t c -f ping"
+    example "Mixed format:", "bundle exec tng -t=c f=ping"
+    example "All equivalent:", "bundle exec tng --type=model --file=user"
+    example "                ", "bundle exec tng -t m -f user"
+    example "                ", "bundle exec tng -t=m f=user"
+  end
+
+  option :type do
+    short "-t"
+    long "--type=TYPE"
+    desc "Test type (controller, model, service)"
+    permit %w[controller model service c m mo s se]
+  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
+    # Preprocess ARGV to handle mixed formats like: -t=value or f=value
+    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
+
+    # Load Rails environment if we're in a Rails application (only for non-help commands)
+    rails_loaded = load_rails_environment
+
+    # Initialize config and clients after Rails is loaded
+    initialize_config_and_clients if rails_loaded
+
+    if params[:type] && params[:file]
+      # Initialize config for direct generation if not already done
+      initialize_config_and_clients unless @config_initialized
+      run_direct_generation
+    else
+      # Check if we're in a Rails app and if Rails loaded properly
+      if find_rails_root && !rails_loaded && !defined?(Rails)
+        puts @pastel.red("❌ Failed to load Rails environment.")
+        puts @pastel.yellow("Please ensure you're running this command from a Rails application directory.")
+        puts @pastel.dim("Try running: bundle exec tng")
+        return
+      end
+
+      # Check if configuration is set up before proceeding
+      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
+
+      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)
+      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 @pastel.cyan("⬅️  Back"), :back
+    end
+
+    case choice
+    when :controller
+      generate_controller_tests
+    when :model
+      generate_model_tests
+    when :service
+      generate_service_tests
+    when :route
+      generate_route_tests
+    end
+  end
+
+  def generate_controller_tests
+    header = @pastel.bright_white.bold("Scanning for controllers...")
+    puts center_text(header)
+
+    spinner = TTY::Spinner.new(
+      center_text("[:spinner] Analyzing controllers..."),
+      format: :dots,
+      success_mark: @pastel.green("✅"),
+      error_mark: @pastel.red("❌")
+    )
+    spinner.auto_spin
+    controllers = Tng::Analyzers::Controller.files_in_dir("app/controllers").map do |file|
+      # Extract the full path relative to app/controllers to build proper namespace
+      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.green("Found #{controllers.length} controllers")
+    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.bright_white.bold("Scanning for models...")
+    puts center_text(header)
+
+    spinner = TTY::Spinner.new(
+      center_text("[:spinner] Analyzing models..."),
+      format: :dots,
+      success_mark: @pastel.green("✅"),
+      error_mark: @pastel.red("❌")
+    )
+    spinner.auto_spin
+
+    models = Tng::Analyzers::Model.files_in_dir("app/models").map do |file|
+      # Extract the full path relative to app/models to build proper namespace
+      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.green("Found #{models.length} models")
+    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.bright_white.bold("Scanning for services...")
+    puts center_text(header)
+
+    spinner = TTY::Spinner.new(
+      center_text("[:spinner] Analyzing services..."),
+      format: :dots,
+      success_mark: @pastel.green("✅"),
+      error_mark: @pastel.red("❌")
+    )
+    spinner.auto_spin
+
+    services = Tng::Analyzers::Service.files_in_dir.map do |file|
+      # Extract the full path relative to app/services or app/service to build proper namespace
+      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.green("Found #{services.length} services")
+    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_test_for_controller(controller)
+    header = @pastel.bright_white.bold("🎯 Generating tests for #{controller[:name]}")
+    puts center_text(header)
+
+    # Show controller analysis
+    analysis_msg = @pastel.dim("Analyzing controller structure...")
+    puts center_text(analysis_msg)
+
+    # Center the progress bar
+    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.green("█"),
+      incomplete: @pastel.dim("░"),
+      width: 40
+    )
+
+    step_msg = @pastel.dim("Parsing controller file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Extracting actions...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Analyzing dependencies...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Generating test code...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    # test for the controller
+    result = Tng::Services::TestGenerator.new(@http_client).run_for_controller(controller)
+    sleep(0.5)
+
+    step_msg = @pastel.dim("Writing test file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.2)
+
+    completion_msg = @pastel.green.bold("✅ Test generation completed!")
+    puts center_text(completion_msg)
+
+    # Show post-generation menu if test was generated successfully
+    if result && result[:file_path]
+      show_post_generation_menu(result)
+    else
+      @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
+    end
+  end
+
+  def show_controller_method_selection(controller)
+    header = @pastel.bright_white.bold("🔍 Analyzing methods for #{controller[:name]}")
+    puts center_text(header)
+
+    # Extract methods from controller
+    methods = extract_controller_methods(controller)
+
+    if methods.empty?
+      flow_display = ControllerTestFlowDisplay.new(@prompt, @pastel)
+      flow_display.show_no_methods_message(controller)
+      return
+    end
+
+    # Show method selection UI
+    flow_display = ControllerTestFlowDisplay.new(@prompt, @pastel)
+    method_choice = flow_display.select_controller_method(controller, methods)
+
+    return if method_choice == :back
+
+    # Generate test for selected method
+    generate_test_for_controller_method(controller, method_choice)
+  end
+
+  def extract_controller_methods(controller)
+    methods = Tng::Analyzers::Controller.methods_for_controller(controller[:name])
+    Array(methods)
+  rescue StandardError => e
+    puts center_text(@pastel.red("❌ Error analyzing controller: #{e.message}"))
+    []
+  end
+
+  def generate_test_for_controller_method(controller, method_info)
+    header = @pastel.bright_white.bold("🎯 Generating test for #{controller[:name]}##{method_info[:name]}")
+    puts center_text(header)
+
+    # Show method analysis
+    analysis_msg = @pastel.dim("Analyzing method structure...")
+    puts center_text(analysis_msg)
+
+    # Center the progress bar
+    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.green("█"),
+      incomplete: @pastel.dim("░"),
+      width: 40
+    )
+
+    step_msg = @pastel.dim("Parsing method details...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Analyzing method dependencies...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Extracting method context...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Generating method test code...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    # Generate test for the specific method
+    result = Tng::Services::TestGenerator.new(@http_client).run_for_controller_method(controller, method_info)
+    sleep(0.5)
+
+    step_msg = @pastel.dim("Writing test file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.2)
+
+    completion_msg = @pastel.green.bold("✅ Method test generation completed!")
+    puts center_text(completion_msg)
+
+    # Show post-generation menu if test was generated successfully
+    if result && result[:file_path]
+      show_post_generation_menu(result)
+    else
+      @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
+    end
+  end
+
+  def show_controller_test_options(controller_choice)
+    flow_display = ControllerTestFlowDisplay.new(@prompt, @pastel)
+    test_option = flow_display.select_test_option(controller_choice)
+
+    return if test_option == :back
+
+    case test_option
+    when :all_possible_tests
+      generate_test_for_controller(controller_choice)
+    when :per_method
+      show_controller_method_selection(controller_choice)
+    end
+  end
+
+  def show_model_test_options(model_choice)
+    flow_display = ModelTestFlowDisplay.new(@prompt, @pastel)
+    test_option = flow_display.select_test_option(model_choice)
+
+    return if test_option == :back
+
+    case test_option
+    when :all_possible_tests
+      generate_test_for_model(model_choice)
+    when :per_method
+      show_model_method_selection(model_choice)
+    end
+  end
+
+  def show_service_test_options(service_choice)
+    flow_display = ServiceTestFlowDisplay.new(@prompt, @pastel)
+    test_option = flow_display.select_test_option(service_choice)
+
+    return if test_option == :back
+
+    case test_option
+    when :all_possible_tests
+      generate_test_for_service(service_choice)
+    when :per_method
+      show_service_method_selection(service_choice)
+    end
+  end
+
+  def generate_test_for_model(model)
+    header = @pastel.bright_white.bold("🎯 Generating tests for #{model[:name]}")
+    puts center_text(header)
+
+    # Show model analysis
+    analysis_msg = @pastel.dim("Analyzing model structure...")
+    puts center_text(analysis_msg)
+
+    # Center the progress bar
+    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.green("█"),
+      incomplete: @pastel.dim("░"),
+      width: 40
+    )
+
+    step_msg = @pastel.dim("Parsing model file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Analyzing validations...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Analyzing associations...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Generating test file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    # test for the model
+    result = Tng::Services::TestGenerator.new(@http_client).run_for_model(model)
+    sleep(0.5)
+
+    step_msg = @pastel.dim("Writing test file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.2)
+    p result
+    completion_msg = @pastel.green.bold("✅ Test generation completed!")
+    puts center_text(completion_msg)
+
+    # Show post-generation menu if test was generated successfully
+    if result && result[:file_path]
+      show_post_generation_menu(result)
+    else
+      @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
+    end
+  end
+
+  def show_model_method_selection(model)
+    header = @pastel.bright_white.bold("🔍 Analyzing methods for #{model[:name]}")
+    puts center_text(header)
+
+    # Extract methods from model
+    methods = extract_model_methods(model)
+
+    if methods.empty?
+      flow_display = ModelTestFlowDisplay.new(@prompt, @pastel)
+      flow_display.show_no_methods_message(model)
+      return
+    end
+
+    # Show method selection UI
+    flow_display = ModelTestFlowDisplay.new(@prompt, @pastel)
+    method_choice = flow_display.select_model_method(model, methods)
+
+    return if method_choice == :back
+
+    # Generate test for selected method
+    generate_test_for_model_method(model, method_choice)
+  end
+
+  def extract_model_methods(model)
+    methods = Tng::Analyzers::Model.methods_for_model(model[:name])
+    Array(methods)
+  rescue StandardError => e
+    puts center_text(@pastel.red("❌ Error analyzing model: #{e.message}"))
+    []
+  end
+
+  def extract_service_methods(service)
+    methods = Tng::Analyzers::Service.methods_for_service(service[:name])
+    Array(methods)
+  rescue StandardError => e
+    puts center_text(@pastel.red("❌ Error analyzing service: #{e.message}"))
+    []
+  end
+
+  def generate_test_for_model_method(model, method_info)
+    header = @pastel.bright_white.bold("🎯 Generating test for #{model[:name]}##{method_info[:name]}")
+    puts center_text(header)
+
+    # Show method analysis
+    analysis_msg = @pastel.dim("Analyzing method structure...")
+    puts center_text(analysis_msg)
+
+    # Center the progress bar
+    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.green("█"),
+      incomplete: @pastel.dim("░"),
+      width: 40
+    )
+
+    step_msg = @pastel.dim("Parsing method details...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Analyzing method dependencies...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Generating method test code...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    # Generate test for the specific method
+    result = Tng::Services::TestGenerator.new(@http_client).run_for_model_method(model, method_info)
+    sleep(0.5)
+
+    step_msg = @pastel.dim("Writing test file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.2)
+
+    completion_msg = @pastel.green.bold("✅ Method test generation completed!")
+    puts center_text(completion_msg)
+
+    # Show post-generation menu if test was generated successfully
+    if result && result[:file_path]
+      show_post_generation_menu(result)
+    else
+      @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
+    end
+  end
+
+  def generate_test_for_service_method(service, method_info)
+    header = @pastel.bright_white.bold("🎯 Generating test for #{service[:name]}##{method_info[:name]}")
+    puts center_text(header)
+
+    # Show method analysis
+    analysis_msg = @pastel.dim("Analyzing method structure...")
+    puts center_text(analysis_msg)
+
+    # Center the progress bar
+    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.green("█"),
+      incomplete: @pastel.dim("░"),
+      width: 40
+    )
+
+    step_msg = @pastel.dim("Parsing method details...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Analyzing method dependencies...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Generating method test code...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    # Generate test for the specific method
+    result = Tng::Services::TestGenerator.new(@http_client).run_for_service_method(service, method_info)
+    sleep(0.5)
+
+    step_msg = @pastel.dim("Writing test file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.2)
+
+    completion_msg = @pastel.green.bold("✅ Method test generation completed!")
+    puts center_text(completion_msg)
+
+    # Show post-generation menu if test was generated successfully
+    if result && result[:file_path]
+      show_post_generation_menu(result)
+    else
+      @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
+    end
+  end
+
+  def show_service_method_selection(service)
+    header = @pastel.bright_white.bold("🔍 Analyzing methods for #{service[:name]}")
+    puts center_text(header)
+
+    # Extract methods from service
+    methods = extract_service_methods(service)
+
+    if methods.empty?
+      flow_display = ServiceTestFlowDisplay.new(@prompt, @pastel)
+      flow_display.show_no_methods_message(service)
+      return
+    end
+
+    # Show method selection UI
+    flow_display = ServiceTestFlowDisplay.new(@prompt, @pastel)
+    method_choice = flow_display.select_service_method(service, methods)
+
+    return if method_choice == :back
+
+    # Generate test for selected method
+    generate_test_for_service_method(service, method_choice)
+  end
+
+  def generate_test_for_service(service)
+    header = @pastel.bright_white.bold("🎯 Generating tests for #{service[:name]}")
+    puts center_text(header)
+
+    # Show service analysis
+    analysis_msg = @pastel.dim("Analyzing service structure...")
+    puts center_text(analysis_msg)
+
+    # Center the progress bar
+    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.green("█"),
+      incomplete: @pastel.dim("░"),
+      width: 40
+    )
+
+    step_msg = @pastel.dim("Parsing service file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Analyzing methods...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Analyzing dependencies...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    step_msg = @pastel.dim("Generating test file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.3)
+
+    # test for the service
+    result = Tng::Services::TestGenerator.new(@http_client).run_for_service(service)
+    sleep(0.5)
+
+    step_msg = @pastel.dim("Writing test file...")
+    puts center_text(step_msg)
+    progress_bar.advance(1)
+    sleep(0.2)
+
+    completion_msg = @pastel.green.bold("✅ Test generation completed!")
+    puts center_text(completion_msg)
+
+    # Show post-generation menu if test was generated successfully
+    if result && result[:file_path]
+      show_post_generation_menu(result)
+    else
+      @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
+    end
+  end
+
+  # STATS
+  def user_stats
+    stats_data = @testng.get_user_stats
+    UserStatsDisplay.new(@pastel, @prompt).display(stats_data)
+  end
+
+  def check_system_status
+    # Only show status check in interactive mode
+    puts @pastel.dim("🔍 Checking system status...") unless params[:type] && params[:file]
+
+    status = @testng.check_system_status
+    status_ok = SystemStatusDisplay.new(@pastel, params).display(status)
+
+    return if status_ok
+
+    puts
+    puts @pastel.dim("Press any key to exit...")
+    $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
+
+  # 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
+    # Normalize type aliases
+    type = normalize_type(params[:type])
+
+    unless %w[controller model service].include?(type)
+      puts @pastel.red("❌ Invalid type: #{params[:type]}")
+      puts @pastel.yellow("Supported types: controller, model, service")
+      return
+    end
+
+    # Check configuration first
+    return unless check_configuration
+
+    # Check system status (but don't show the spinner in direct mode)
+    status = @testng.check_system_status
+    unless SystemStatusDisplay.new(@pastel, params).display(status)
+      puts @pastel.red("❌ System status check failed")
+      return
+    end
+
+    case type
+    when "controller"
+      run_direct_controller_generation
+    when "model"
+      run_direct_model_generation
+    when "service"
+      run_direct_service_generation
+    end
+  end
+
+  def normalize_type(type_param)
+    case type_param&.downcase
+    when "c", "controller"
+      "controller"
+    when "m", "mo", "model"
+      "model"
+    when "s", "se", "service"
+      "service"
+    else
+      type_param&.downcase
+    end
+  end
+
+  def run_direct_controller_generation
+    file_name = params[:file]
+    file_name += "_controller" unless file_name.end_with?("_controller")
+
+    # Build the expected path
+    controller_path = File.join("app/controllers", "#{file_name}.rb")
+
+    # Check if file exists
+    unless File.exist?(controller_path)
+      puts @pastel.red("❌ Controller file not found: #{controller_path}")
+      return
+    end
+
+    # Extract the full path relative to app/controllers to build proper namespace
+    relative_path = controller_path.gsub(%r{^.*app/controllers/}, "").gsub(".rb", "")
+    namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+
+    controller = {
+      name: namespaced_name,
+      path: controller_path
+    }
+
+    puts @pastel.bright_white("🎯 Generating tests for #{controller[:name]}...")
+
+    # Generate the test directly without progress bars for cleaner output
+    result = Tng::Services::TestGenerator.new(@http_client).run_for_controller(controller)
+
+    if result && result[:file_path]
+      show_post_generation_menu(result)
+    else
+      puts @pastel.red("❌ Failed to generate test")
+    end
+  end
+
+  def run_direct_model_generation
+    file_name = params[:file]
+
+    # Build the expected path
+    model_path = File.join("app/models", "#{file_name}.rb")
+
+    # Check if file exists
+    unless File.exist?(model_path)
+      puts @pastel.red("❌ Model file not found: #{model_path}")
+      return
+    end
+
+    # Extract the full path relative to app/models to build proper namespace
+    relative_path = model_path.gsub(%r{^.*app/models/}, "").gsub(".rb", "")
+    namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+
+    model = {
+      name: namespaced_name,
+      path: model_path
+    }
+
+    puts @pastel.bright_white("🎯 Generating tests for #{model[:name]}...")
+
+    # Generate the test directly without progress bars for cleaner output
+    result = Tng::Services::TestGenerator.new(@http_client).run_for_model(model)
+
+    if result && result[:file_path]
+      show_post_generation_menu(result)
+    else
+      puts @pastel.red("❌ Failed to generate test")
+    end
+  end
+
+  def run_direct_service_generation
+    file_name = params[:file]
+
+    # Try different service directories
+    service_dirs = ["app/services", "app/service"]
+    service_path = nil
+
+    service_dirs.each do |dir|
+      potential_path = File.join(dir, "#{file_name}.rb")
+      if File.exist?(potential_path)
+        service_path = potential_path
+        break
+      end
+    end
+
+    # Check if file exists
+    unless service_path
+      puts @pastel.red("❌ Service file not found: #{file_name}.rb")
+      puts @pastel.yellow("Searched in: #{service_dirs.join(", ")}")
+      return
+    end
+
+    # Extract the full path relative to app/services or app/service to build proper namespace
+    relative_path = service_path.gsub(%r{^.*app/services?/}, "").gsub(".rb", "")
+    namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+
+    service = {
+      name: namespaced_name,
+      path: service_path
+    }
+
+    puts @pastel.bright_white("🎯 Generating tests for #{service[:name]}...")
+
+    # Generate the test directly without progress bars for cleaner output
+    result = Tng::Services::TestGenerator.new(@http_client).run_for_service(service)
+
+    if result && result[:file_path]
+      show_post_generation_menu(result)
+    else
+      puts @pastel.red("❌ Failed to generate test")
+    end
+  end
+
+  def show_post_generation_menu(result)
+    # Check if we're in direct mode
+    is_direct_mode = params[:type] && params[:file]
+
+    if is_direct_mode
+      # In direct mode, show info and exit
+      puts
+      header = @pastel.bright_white.bold("🎉 Test Generated Successfully!")
+      puts center_text(header)
+      puts
+
+      # Show file info
+      info_msg = [
+        @pastel.cyan("📁 File: #{result[:file_path]}"),
+        @pastel.yellow("🏃 Run: #{result[:run_command]}")
+      ].join("\n")
+      puts center_text(info_msg)
+      puts
+
+      # In direct mode, just show the success message and exit
+      puts center_text(@pastel.green("✅ Test generation completed successfully!"))
+      return
+    end
+
+    # Interactive mode - show menu loop
+    loop do
+      puts
+      header = @pastel.bright_white.bold("🎉 Test Generated Successfully!")
+      puts center_text(header)
+      puts
+
+      # Show file info
+      info_msg = [
+        @pastel.cyan("📁 File: #{result[:file_path]}"),
+        @pastel.yellow("🏃 Run: #{result[:run_command]}")
+      ].join("\n")
+      puts center_text(info_msg)
+      puts
+
+      choice = @prompt.select(
+        @pastel.bright_white("What would you like to do?"),
+        cycle: true,
+        symbols: { marker: "▶" }
+      ) do |menu|
+        menu.choice @pastel.yellow("📋 Copy run command"), :copy_command
+        menu.choice @pastel.cyan("⬅️  Back to main menu"), :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
+
+    # Check if configuration is properly set up
+    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
+end
+
+cli = CLI.new
+cli.start