tng 0.2.1

data/lib/tng/ui/model_test_flow_display.rb

@@ -0,0 +1,80 @@
+# frozen_string_literal: true
+
+require "tty-prompt"
+require "tty-spinner"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+class ModelTestFlowDisplay
+  def initialize(prompt, pastel)
+    @prompt = prompt
+    @pastel = pastel
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def select_model(models)
+    header = @pastel.public_send(Tng::UI::Theme.color(:primary)).bold("#{Tng::UI::Theme.icon(:config)} Select model to test:")
+    puts Tng::UI::Theme.center_text(header, @terminal_width)
+
+    @prompt.select(
+      "",
+      cycle: true,
+      per_page: 12,
+      filter: true,
+      symbols: { marker: Tng::UI::Theme.icon(:marker) }
+    ) do |menu|
+      models.each do |model|
+        display_name = "#{model[:name]} #{@pastel.public_send(Tng::UI::Theme.color(:muted), "(#{model[:path]})")}"
+        menu.choice display_name, model
+      end
+      menu.choice @pastel.public_send(Tng::UI::Theme.color(:secondary), "#{Tng::UI::Theme.icon(:back)}  Back"), :back
+    end
+  end
+
+  def show_no_models_message
+    puts
+    puts Tng::UI::Theme.center_text(
+      @pastel.public_send(Tng::UI::Theme.color(:warning),
+                          "#{Tng::UI::Theme.icon(:warning)} No models found in app/models"), @terminal_width
+    )
+    puts Tng::UI::Theme.center_text(
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "Make sure you're in a Rails project with models"), @terminal_width
+    )
+    puts
+  end
+
+  def select_model_method(model, methods)
+    header = @pastel.public_send(Tng::UI::Theme.color(:primary)).bold("#{Tng::UI::Theme.icon(:rocket)} Select method to test in #{model[:name]}")
+    puts Tng::UI::Theme.center_text(header, @terminal_width)
+
+    @prompt.select(
+      "",
+      cycle: true,
+      per_page: 10,
+      filter: true,
+      symbols: { marker: Tng::UI::Theme.icon(:marker) }
+    ) do |menu|
+      methods.each do |method|
+        menu.choice method[:name], method
+      end
+      menu.choice @pastel.public_send(Tng::UI::Theme.color(:secondary), "#{Tng::UI::Theme.icon(:back)}  Back"), :back
+    end
+  end
+
+  def show_no_methods_message(model)
+    error_msg = "#{@pastel.public_send(Tng::UI::Theme.color(:error)).bold("#{Tng::UI::Theme.icon(:error)} No methods found in #{model[:name]}")}\n#{@pastel.public_send(
+      Tng::UI::Theme.color(:muted), "Model may be empty or have syntax errors"
+    )}"
+    puts Tng::UI::Theme.center_text(error_msg, @terminal_width)
+    @prompt.keypress(Tng::UI::Theme.center_text(
+                       @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                           "Press any key to continue..."), @terminal_width
+                     ))
+  end
+end

data/lib/tng/ui/other_test_flow_display.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require "tty-prompt"
+require "tty-spinner"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+class OtherTestFlowDisplay
+  def initialize(prompt, pastel)
+    @prompt = prompt
+    @pastel = pastel
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def select_other_file(other_files)
+    header = @pastel.public_send(Tng::UI::Theme.color(:primary)).bold("#{Tng::UI::Theme.icon(:config)} Select file to test:")
+    puts Tng::UI::Theme.center_text(header, @terminal_width)
+
+    @prompt.select(
+      "",
+      cycle: true,
+      per_page: 12,
+      filter: true,
+      symbols: { marker: Tng::UI::Theme.icon(:marker) }
+    ) do |menu|
+      other_files.each do |file|
+        display_name = "#{file[:name]} #{@pastel.public_send(Tng::UI::Theme.color(:muted), "(#{file[:path]})")}"
+        menu.choice display_name, file
+      end
+      menu.choice @pastel.public_send(Tng::UI::Theme.color(:secondary), "#{Tng::UI::Theme.icon(:back)}  Back"), :back
+    end
+  end
+
+  def show_no_other_files_message
+    error_msg = "#{@pastel.public_send(Tng::UI::Theme.color(:error)).bold("#{Tng::UI::Theme.icon(:error)} No other files found in your application")}\n#{@pastel.public_send(
+      Tng::UI::Theme.color(:muted), "Make sure you have files in supported directories (app/jobs, app/helpers, lib/, app/policies, app/presenters, app/mailers, app/graphql, etc.)"
+    )}"
+    puts Tng::UI::Theme.center_text(error_msg, @terminal_width)
+    @prompt.keypress(Tng::UI::Theme.center_text(
+                       @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                           "Press any key to continue..."), @terminal_width
+                     ))
+  end
+
+  def select_other_method(other_file, methods)
+    header = @pastel.public_send(Tng::UI::Theme.color(:primary)).bold("#{Tng::UI::Theme.icon(:rocket)} Select method to test in #{other_file[:name]}")
+    puts Tng::UI::Theme.center_text(header, @terminal_width)
+
+    @prompt.select(
+      "",
+      cycle: true,
+      per_page: 10,
+      filter: true,
+      symbols: { marker: Tng::UI::Theme.icon(:marker) }
+    ) do |menu|
+      methods.each do |method|
+        menu.choice method[:name], method
+      end
+      menu.choice @pastel.public_send(Tng::UI::Theme.color(:secondary), "#{Tng::UI::Theme.icon(:back)}  Back"), :back
+    end
+  end
+
+  def show_no_methods_message(other_file)
+    error_msg = "#{@pastel.public_send(Tng::UI::Theme.color(:error)).bold("#{Tng::UI::Theme.icon(:error)} No methods found in #{other_file[:name]}")}\n#{@pastel.public_send(
+      Tng::UI::Theme.color(:muted), "File may be empty or have syntax errors"
+    )}"
+    puts Tng::UI::Theme.center_text(error_msg, @terminal_width)
+    @prompt.keypress(Tng::UI::Theme.center_text(
+                       @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                           "Press any key to continue..."), @terminal_width
+                     ))
+  end
+end

data/lib/tng/ui/post_install_box.rb

@@ -0,0 +1,80 @@
+require "tty-box"
+require "pastel"
+require_relative "theme"
+
+class PostInstallBox
+  attr_reader :version, :pastel
+
+  def initialize(version)
+    @version = version
+    @pastel = Pastel.new
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def content
+    [
+      pastel.public_send(Tng::UI::Theme.color(:success)).bold("#{Tng::UI::Theme.icon(:success)} Tng installed successfully!"),
+      "",
+      pastel.public_send(Tng::UI::Theme.color(:warning)).bold("#{Tng::UI::Theme.icon(:config)} SETUP REQUIRED"),
+      "",
+      pastel.public_send(Tng::UI::Theme.color(:primary), "1. Generate configuration:"),
+      pastel.public_send(Tng::UI::Theme.color(:secondary), "   rails g tng:install"),
+      "",
+      pastel.public_send(Tng::UI::Theme.color(:primary), "2. Edit configuration file:"),
+      pastel.public_send(Tng::UI::Theme.color(:secondary), "   config/initializers/tng.rb"),
+      "",
+      pastel.public_send(Tng::UI::Theme.color(:primary), "3. Add your license key:"),
+      pastel.public_send(Tng::UI::Theme.color(:secondary), "   config.api_key = 'your-license-key-here'"),
+      "",
+      pastel.public_send(Tng::UI::Theme.color(:primary),
+                         "#{Tng::UI::Theme.icon(:config)} Check documentation for the correct authentication setup"),
+      "",
+      pastel.public_send(Tng::UI::Theme.color(:accent)).bold("#{Tng::UI::Theme.icon(:rocket)} Usage:"),
+      "",
+      pastel.public_send(Tng::UI::Theme.color(:primary), "Interactive mode:"),
+      pastel.public_send(Tng::UI::Theme.color(:success),
+                         "#{Tng::UI::Theme.icon(:bullet)} bundle exec tng") + pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                                                 "  - Interactive CLI with method selection"),
+      "",
+      pastel.public_send(Tng::UI::Theme.color(:primary), "Features:"),
+      pastel.public_send(Tng::UI::Theme.color(:success),
+                         "#{Tng::UI::Theme.icon(:bullet)} Test 20+ file types: Controllers, Models, Services + Jobs, Helpers, Lib, Policies, Presenters, Mailers, GraphQL, and more"),
+      pastel.public_send(Tng::UI::Theme.color(:success),
+                         "#{Tng::UI::Theme.icon(:bullet)} Select specific methods to test"),
+      pastel.public_send(Tng::UI::Theme.color(:success),
+                         "#{Tng::UI::Theme.icon(:bullet)} Search and filter through methods"),
+      "",
+      pastel.public_send(Tng::UI::Theme.color(:primary), "Help:"),
+      pastel.public_send(Tng::UI::Theme.color(:success),
+                         "#{Tng::UI::Theme.icon(:bullet)} bundle exec tng --help") + pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                                                        "  - Show help information"),
+      "",
+      pastel.public_send(Tng::UI::Theme.color(:muted),
+                         "#{Tng::UI::Theme.icon(:lightbulb)} Generate tests for individual methods with precision")
+    ].join("\n")
+  end
+
+  def render
+    # Use a reasonable width that won't cause issues
+    box_width = [@terminal_width, 70].min
+
+    style = Tng::UI::Theme.box_style(:default)
+    TTY::Box.frame(
+      title: { top_left: " TNG ", bottom_right: " v#{@version} " },
+      style: style[:style],
+      padding: style[:padding],
+      align: :left,
+      width: box_width
+    ) do
+      content
+    end
+  rescue StandardError
+    "TNG v#{@version} installed successfully!\n" +
+      "Run 'rails g tng:install' to get started.\n" +
+      "Use 'bundle exec tng --help' for usage information."
+  end
+end

data/lib/tng/ui/service_test_flow_display.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require "tty-prompt"
+require "tty-spinner"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+class ServiceTestFlowDisplay
+  def initialize(prompt, pastel)
+    @prompt = prompt
+    @pastel = pastel
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def select_service(services)
+    header = @pastel.public_send(Tng::UI::Theme.color(:primary)).bold("#{Tng::UI::Theme.icon(:config)} Select service to test:")
+    puts Tng::UI::Theme.center_text(header, @terminal_width)
+
+    @prompt.select(
+      "",
+      cycle: true,
+      per_page: 12,
+      filter: true,
+      symbols: { marker: Tng::UI::Theme.icon(:marker) }
+    ) do |menu|
+      services.each do |service|
+        display_name = "#{service[:name]} #{@pastel.public_send(Tng::UI::Theme.color(:muted), "(#{service[:path]})")}"
+        menu.choice display_name, service
+      end
+      menu.choice @pastel.public_send(Tng::UI::Theme.color(:secondary), "#{Tng::UI::Theme.icon(:back)}  Back"), :back
+    end
+  end
+
+  def show_no_services_message
+    error_msg = "#{@pastel.public_send(Tng::UI::Theme.color(:error)).bold("#{Tng::UI::Theme.icon(:error)} No services found in your application")}\n#{@pastel.public_send(
+      Tng::UI::Theme.color(:muted), "Make sure you have services in app/services/ or app/service/"
+    )}"
+    puts Tng::UI::Theme.center_text(error_msg, @terminal_width)
+    @prompt.keypress(Tng::UI::Theme.center_text(
+                       @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                           "Press any key to continue..."), @terminal_width
+                     ))
+  end
+
+  def select_service_method(service, methods)
+    header = @pastel.public_send(Tng::UI::Theme.color(:primary)).bold("#{Tng::UI::Theme.icon(:rocket)} Select method to test for #{service[:name]}")
+    puts Tng::UI::Theme.center_text(header, @terminal_width)
+
+    @prompt.select(
+      "",
+      cycle: true,
+      per_page: 12,
+      filter: true,
+      symbols: { marker: Tng::UI::Theme.icon(:marker) }
+    ) do |menu|
+      methods.each do |method|
+        menu.choice "#{method[:name]}", method
+      end
+      menu.choice @pastel.public_send(Tng::UI::Theme.color(:secondary), "#{Tng::UI::Theme.icon(:back)}  Back"), :back
+    end
+  end
+
+  def show_no_methods_message(service)
+    error_msg = "#{@pastel.public_send(Tng::UI::Theme.color(:error)).bold("#{Tng::UI::Theme.icon(:error)} No methods found in #{service[:name]}")}\n#{@pastel.public_send(
+      Tng::UI::Theme.color(:muted), "The service might not have any public methods or could not be analyzed."
+    )}"
+    puts Tng::UI::Theme.center_text(error_msg, @terminal_width)
+    @prompt.keypress(Tng::UI::Theme.center_text(
+                       @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                           "Press any key to continue..."), @terminal_width
+                     ))
+  end
+end

data/lib/tng/ui/show_help.rb

@@ -0,0 +1,78 @@
+require "tty-box"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+class ShowHelp
+  attr_reader :pastel
+
+  def initialize(pastel, version)
+    @pastel = pastel
+    @version = version
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def content
+    [
+      @pastel.public_send(Tng::UI::Theme.color(:secondary)).bold("TNG - LLM-Powered Rails Test Generator"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted), "Version: #{@version}"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Usage:"),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "  bundle exec tng") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                     "                           # Interactive mode only"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Features:"),
+      @pastel.public_send(Tng::UI::Theme.color(:success),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Controllers, Models, Services"),
+      @pastel.public_send(Tng::UI::Theme.color(:success),
+                          "  #{Tng::UI::Theme.icon(:bullet)} 17+ other file types (Jobs, Helpers, Lib, Policies, Presenters, Mailers, GraphQL, etc.)"),
+      @pastel.public_send(Tng::UI::Theme.color(:success),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Per-method test generation"),
+      @pastel.public_send(Tng::UI::Theme.color(:success),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Searchable method lists"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Options:"),
+      @pastel.public_send(Tng::UI::Theme.color(:primary),
+                          "  -h, --help") + @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                "          Show this help message"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("How to Use:"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Run 'bundle exec tng' to start the interactive interface"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Select Controller, Model, Service, or Other to test"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Choose a specific method from the list"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Use search/filter to find methods quickly"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:secondary)).bold("Happy testing! ") + @pastel.public_send(
+        Tng::UI::Theme.color(:accent), "★"
+      )
+    ].join("\n")
+  end
+
+  def render
+    # Use a reasonable width that matches other UI components
+    box_width = Tng::UI::Theme.calculate_box_width(@terminal_width)
+    box_width = [box_width, 80].min # Allow wider for help content
+    style = Tng::UI::Theme.box_style(:default)
+
+    box = TTY::Box.frame(
+      title: { top_left: " TNG Help ", bottom_right: " v#{@version} " },
+      style: style[:style],
+      padding: style[:padding],
+      align: :left,
+      width: box_width
+    ) do
+      content
+    end
+
+    Tng::UI::Theme.center_box(box, box_width, @terminal_width)
+  end
+end

data/lib/tng/ui/system_status_display.rb

@@ -0,0 +1,128 @@
+# frozen_string_literal: true
+
+require "pastel"
+require_relative "theme"
+
+class SystemStatusDisplay
+  def initialize(pastel, args)
+    @pastel = pastel
+    @args = args
+  end
+
+  def display(status)
+    case status[:status]
+    when :ok
+      unless @args[:type] && @args[:file]
+        puts @pastel.public_send(Tng::UI::Theme.color(:success),
+                                 "#{Tng::UI::Theme.icon(:success)} System operational")
+      end
+      true
+    when :version_mismatch
+      display_status_message(
+        title: "Version Mismatch",
+        icon: Tng::UI::Theme.icon(:error),
+        color: Tng::UI::Theme.color(:error),
+        content: [
+          "Your gem version: #{status[:gem_version]}",
+          "Required version: #{status[:current_version]}",
+          "",
+          "Quick fix:",
+          "• gem update tng",
+          "• bundle update tng"
+        ]
+      )
+      false
+    when :base_url_mismatch
+      display_status_message(
+        title: "URL Mismatch",
+        icon: Tng::UI::Theme.icon(:error),
+        color: Tng::UI::Theme.color(:error),
+        content: [
+          "Your URL: #{status[:user_base_url]}",
+          "Server URL: #{status[:server_base_url]}",
+          "",
+          "Quick fix:",
+          "• Edit config/initializers/tng.rb",
+          "• Set: config.base_url = '#{status[:server_base_url]}'"
+        ]
+      )
+      false
+    when :service_down
+      display_status_message(
+        title: "Service Unavailable",
+        icon: Tng::UI::Theme.icon(:error),
+        color: Tng::UI::Theme.color(:error),
+        content: [
+          status[:message] || "TNG service is currently down",
+          "",
+          "Please try again later or contact support"
+        ]
+      )
+      false
+    when :error
+      display_status_message(
+        title: "Connection Error",
+        icon: Tng::UI::Theme.icon(:error),
+        color: Tng::UI::Theme.color(:error),
+        content: [
+          status[:message] || "Unable to connect to TNG service",
+          "",
+          "Please check your internet connection and try again"
+        ]
+      )
+      false
+    else
+      display_status_message(
+        title: "Unknown Error",
+        icon: Tng::UI::Theme.icon(:error),
+        color: Tng::UI::Theme.color(:error),
+        content: [status[:message] || "An unknown error occurred"]
+      )
+      false
+    end
+  end
+
+  private
+
+  def display_status_message(title:, icon:, color:, content:)
+    require "tty-box"
+    require "tty-screen"
+
+    terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+
+    formatted_content = content.map do |line|
+      case line
+      when /^•/
+        @pastel.public_send(Tng::UI::Theme.color(:muted), line)
+      when /^Quick fix:/
+        @pastel.public_send(Tng::UI::Theme.color(:secondary), line)
+      when ""
+        line
+      else
+        @pastel.public_send(Tng::UI::Theme.color(:warning), line)
+      end
+    end.join("\n")
+
+    box_width = Tng::UI::Theme.calculate_box_width(terminal_width)
+
+    status_box = TTY::Box.frame(
+      title: { top_left: " #{title} " },
+      style: {
+        fg: :bright_white,
+        border: { fg: color },
+        title: { fg: color == Tng::UI::Theme.color(:error) ? :bright_red : color }
+      },
+      padding: [1, 2],
+      width: box_width
+    ) do
+      formatted_content
+    end
+
+    puts Tng::UI::Theme.center_box(status_box, box_width, terminal_width)
+    puts
+  end
+end