tng 0.1.0

data/lib/tng/ui/display_banner.rb

@@ -0,0 +1,54 @@
+require "tty-box"
+require "pastel"
+require "tty-screen"
+
+class DisplayBanner
+  attr_reader :pastel, :version, :terminal_width
+
+  def initialize(pastel, version)
+    @pastel = pastel
+    @version = version
+    @terminal_width = begin
+                        TTY::Screen.width
+                      rescue StandardError
+                        80
+                      end
+  end
+
+  def render
+    banner_content = [
+      pastel.cyan.bold("Tng - LLM-Powered Rails Test Generator"),
+      pastel.dim("Version: #{version}"),
+      "",
+      pastel.yellow("Generate comprehensive tests for your Rails application"),
+      pastel.dim("Powered by LLM and static analysis")
+    ].join("\n")
+
+    box_width = 64
+    box_width = terminal_width if box_width > terminal_width
+
+    box = TTY::Box.frame(
+      title: { top_left: " Tng ", bottom_right: " Ready " },
+      style: {
+        fg: :bright_white,
+        border: { fg: :cyan },
+        title: { fg: :bright_cyan }
+      },
+      padding: [1, 2],
+      align: :center,
+      width: box_width
+    ) do
+      banner_content
+    end
+
+    "#{center_box(box, box_width)}\n"
+  end
+
+  private
+
+  def center_box(box_string, box_width)
+    padding = (terminal_width - box_width) / 2
+    padding = 0 if padding.negative?
+    box_string.lines.map { |line| (" " * padding) + line.chomp }.join("\n")
+  end
+end

data/lib/tng/ui/goodbye_display.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require "tty-box"
+require "pastel"
+require "tty-screen"
+
+class GoodbyeDisplay
+  def initialize(pastel)
+    @pastel = pastel
+    @terminal_width = begin
+                        TTY::Screen.width
+                      rescue StandardError
+                        80
+                      end
+  end
+
+  def display
+    box_width = 54
+    goodbye_box = TTY::Box.frame(
+      title: { top_left: " Thank You " },
+      style: {
+        fg: :bright_white,
+        border: { fg: :green },
+        title: { fg: :bright_green }
+      },
+      padding: [1, 2],
+      align: :center,
+      width: box_width
+    ) do
+      [
+        @pastel.green.bold("Thanks for using Tng!"),
+        "",
+        @pastel.bright_white("Happy testing!"),
+        "",
+        @pastel.dim("Your Rails tests are now supercharged with LLM")
+      ].join("\n")
+    end
+
+    puts center_box(goodbye_box, box_width)
+    puts
+  end
+
+  private
+
+  def center_box(box_string, box_width)
+    padding = (@terminal_width - box_width) / 2
+    padding = 0 if padding.negative?
+    box_string.lines.map { |line| (" " * padding) + line.chomp }.join("\n")
+  end
+end

data/lib/tng/ui/model_test_flow_display.rb

@@ -0,0 +1,95 @@
+# frozen_string_literal: true
+
+require "tty-prompt"
+require "tty-spinner"
+require "pastel"
+require "tty-screen"
+
+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.bright_white.bold("📋 Select model to test:")
+    puts center_text(header)
+
+    @prompt.select(
+      "",
+      cycle: true,
+      per_page: 12,
+      filter: true,
+      symbols: { marker: "▶" }
+    ) do |menu|
+      models.each do |model|
+        display_name = "#{model[:name]} #{@pastel.dim("(#{model[:path]})")}"
+        menu.choice display_name, model
+      end
+      menu.choice @pastel.cyan("⬅️  Back"), :back
+    end
+  end
+
+  def select_test_option(model_choice)
+    header = @pastel.bright_white.bold("🎯 Test Generation Options for #{model_choice[:name]}")
+    puts center_text(header)
+    puts
+
+    @prompt.select(
+      @pastel.bright_white("Choose test generation type:"),
+      cycle: true,
+      symbols: { marker: "▶" }
+    ) do |menu|
+      menu.choice @pastel.green("Generate all possible tests"), :all_possible_tests
+      menu.choice @pastel.blue("Generate per method"), :per_method
+      menu.choice @pastel.cyan("⬅️  Back"), :back
+    end
+  end
+
+  def show_no_models_message
+    puts
+    puts center_text(@pastel.yellow("⚠️  No models found in app/models"))
+    puts center_text(@pastel.dim("Make sure you're in a Rails project with models"))
+    puts
+  end
+
+  def select_model_method(model, methods)
+    header = @pastel.bright_white.bold("🎯 Select method to test in #{model[:name]}")
+    puts center_text(header)
+
+    @prompt.select(
+      "",
+      cycle: true,
+      per_page: 10,
+      filter: true,
+      symbols: { marker: "▶" }
+    ) do |menu|
+      methods.each do |method|
+        menu.choice method[:name], method
+      end
+      menu.choice @pastel.cyan("⬅️  Back"), :back
+    end
+  end
+
+  def show_no_methods_message(model)
+    error_msg = "#{@pastel.red.bold("❌ No methods found in #{model[:name]}")}\n#{@pastel.dim("Model may be empty or have syntax errors")}"
+    puts center_text(error_msg)
+    @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
+  end
+
+  private
+
+  def center_text(text)
+    padding = [(@terminal_width - strip_ansi(text).length) / 2, 0].max
+    "#{" " * padding}#{text}"
+  end
+
+  def strip_ansi(text)
+    text.gsub(/\e\[([;\d]+)?m/, "")
+  end
+end

data/lib/tng/ui/post_install_box.rb

@@ -0,0 +1,73 @@
+require "tty-box"
+require "pastel"
+
+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.green.bold("✅ Tng installed successfully!"),
+      "",
+      pastel.yellow.bold("📋 SETUP REQUIRED"),
+      "",
+      pastel.bright_white("1. Generate configuration:"),
+      pastel.cyan("   rails g tng:install"),
+      "",
+      pastel.bright_white("2. Edit configuration file:"),
+      pastel.cyan("   config/initializers/tng.rb"),
+      "",
+      pastel.bright_white("3. Add your license key:"),
+      pastel.cyan("   config.api_key = 'your-license-key-here'"),
+      "",
+      pastel.bright_white("📖 Check documentation for the correct authentication setup"),
+      "",
+      pastel.magenta.bold("🚀 Usage:"),
+      "",
+      pastel.bright_white("Interactive mode:"),
+      pastel.green("• bundle exec tng") + pastel.dim("  - Full interactive CLI"),
+      "",
+      pastel.bright_white("Direct generation:"),
+      pastel.green("• bundle exec tng -t c -f ping") + pastel.dim("  - Short"),
+      pastel.green("• bundle exec tng -t=c f=ping") + pastel.dim("   - Mixed"),
+      pastel.green("• bundle exec tng --type=controller --file=ping"),
+      "",
+      pastel.bright_white("Help:"),
+      pastel.green("• bundle exec tng --help") + pastel.dim("  - Show all options"),
+      "",
+      pastel.dim("💡 Use -t c for controller, -t m for model")
+    ].join("\n")
+  end
+
+  def render
+    # Use a reasonable width that won't cause issues
+    box_width = [@terminal_width, 70].min
+
+    TTY::Box.frame(
+      title: { top_left: " TestNG ", bottom_right: " v#{@version} " },
+      style: {
+        fg: :bright_white,
+        border: { fg: :cyan },
+        title: { fg: :bright_cyan }
+      },
+      padding: [1, 2],
+      align: :left,
+      width: box_width
+    ) do
+      content
+    end
+  rescue StandardError => e
+    "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,89 @@
+# frozen_string_literal: true
+
+require "tty-prompt"
+require "tty-spinner"
+require "pastel"
+require "tty-screen"
+
+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.bright_white.bold("📋 Select service to test:")
+    puts center_text(header)
+
+    @prompt.select(
+      "",
+      cycle: true,
+      per_page: 12,
+      filter: true,
+      symbols: { marker: "▶" }
+    ) do |menu|
+      services.each do |service|
+        display_name = "#{service[:name]} #{@pastel.dim("(#{service[:path]})")}"
+        menu.choice display_name, service
+      end
+      menu.choice @pastel.cyan("⬅️  Back"), :back
+    end
+  end
+
+  def select_test_option(service_choice)
+    header = @pastel.bright_white.bold("🎯 Test Generation Options for #{service_choice[:name]}")
+    puts center_text(header)
+    puts
+
+    @prompt.select(
+      @pastel.bright_white("Choose test generation type:"),
+      cycle: true,
+      symbols: { marker: "▶" }
+    ) do |menu|
+      menu.choice @pastel.green("Generate all possible tests"), :all_possible_tests
+      menu.choice @pastel.blue("Generate per method"), :per_method
+      menu.choice @pastel.cyan("⬅️  Back"), :back
+    end
+  end
+
+  def show_no_services_message
+    error_msg = "#{@pastel.red.bold("❌ No services found in your application")}\n#{@pastel.dim("Make sure you have services in app/services/ or app/service/")}"
+    puts center_text(error_msg)
+    @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
+  end
+
+  def select_service_method(service, methods)
+    header = @pastel.bright_white.bold("🔍 Select method to test for #{service[:name]}")
+    puts center_text(header)
+
+    @prompt.select(
+      "",
+      cycle: true,
+      per_page: 12,
+      filter: true,
+      symbols: { marker: "▶" }
+    ) do |menu|
+      methods.each do |method|
+        menu.choice "#{method[:name]}", method
+      end
+      menu.choice @pastel.cyan("⬅️  Back"), :back
+    end
+  end
+
+  def show_no_methods_message(service)
+    error_msg = "#{@pastel.red.bold("❌ No methods found in #{service[:name]}")}\n#{@pastel.dim("The service might not have any public methods or could not be analyzed.")}"
+    puts center_text(error_msg)
+    @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
+  end
+
+  def center_text(text)
+    padding = (@terminal_width - @pastel.strip(text).length) / 2
+    padding = 0 if padding.negative?
+    (" " * padding) + text
+  end
+end

data/lib/tng/ui/show_help.rb

@@ -0,0 +1,89 @@
+require "tty-box"
+require "pastel"
+require "tty-screen"
+
+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.cyan.bold("TestNG - LLM-Powered Rails Test Generator"),
+      @pastel.dim("Version: #{@version}"),
+      "",
+      @pastel.yellow.bold("Usage:"),
+      @pastel.bright_white("  bundle exec tng") + @pastel.dim("                           # Interactive mode"),
+      @pastel.bright_white("  bundle exec tng --type=TYPE --file=FILE") + @pastel.dim("   # Direct mode"),
+      @pastel.bright_white("  bundle exec tng -t TYPE -f FILE") + @pastel.dim("           # Short aliases"),
+      @pastel.bright_white("  bundle exec tng -t=TYPE f=FILE") + @pastel.dim("            # Mixed format"),
+      "",
+      @pastel.yellow.bold("Examples:"),
+      @pastel.green("  bundle exec tng -t c -f ping") + @pastel.dim("                    # Controller test"),
+      @pastel.green("  bundle exec tng -t=c f=ping") + @pastel.dim("                     # Mixed format"),
+      @pastel.green("  bundle exec tng -t m -f user") + @pastel.dim("                    # Model test"),
+      @pastel.green("  bundle exec tng --type=controller --file=users_controller"),
+      @pastel.green("  bundle exec tng -t c -f \"admin/users_controller\"") + @pastel.dim(" # Namespaced"),
+      "",
+      @pastel.yellow.bold("Options:"),
+      @pastel.bright_white("  -t, --type=TYPE") + @pastel.dim("     Test type (controller, model)"),
+      @pastel.bright_white("  -f, --file=FILE") + @pastel.dim("     File name (without .rb extension)"),
+      @pastel.bright_white("  -m, --method=METHOD") + @pastel.dim(" Method name (for per-method tests)"),
+      @pastel.bright_white("  -h, --help") + @pastel.dim("          Show this help message"),
+      "",
+      @pastel.yellow.bold("Type Aliases:"),
+      @pastel.dim("  • Controllers: ") + @pastel.cyan("c, controller"),
+      @pastel.dim("  • Models: ") + @pastel.cyan("m, mo, model"),
+      "",
+      @pastel.yellow.bold("File Format:"),
+      @pastel.dim("  • Controllers: ") + @pastel.green("users_controller, api/v1/posts_controller"),
+      @pastel.dim("  • Models: ") + @pastel.green("user, blog/post"),
+      @pastel.dim("  • Don't include .rb extension"),
+      "",
+      @pastel.magenta.bold("Pro Tips:"),
+      @pastel.dim("  • Mix formats: ") + @pastel.cyan("-t=c f=ping") + @pastel.dim(" works perfectly"),
+      @pastel.dim("  • Use filtering in interactive mode for large projects"),
+      @pastel.dim("  • All formats are equivalent - use what feels natural"),
+      "",
+      @pastel.cyan.bold("Happy testing! ") + @pastel.yellow("★")
+    ].join("\n")
+  end
+
+  def render
+    # Use a reasonable width that matches other UI components
+    box_width = [@terminal_width - 4, 80].min
+    box_width = 60 if box_width < 60
+
+    box = TTY::Box.frame(
+      title: { top_left: " TNG Help ", bottom_right: " v#{@version} " },
+      style: {
+        fg: :bright_white,
+        border: { fg: :cyan },
+        title: { fg: :bright_cyan }
+      },
+      padding: [1, 2],
+      align: :left,
+      width: box_width
+    ) do
+      content
+    end
+
+    center_box(box, box_width)
+  end
+
+  private
+
+  def center_box(box_string, box_width)
+    padding = (@terminal_width - box_width) / 2
+    padding = 0 if padding.negative?
+    box_string.lines.map { |line| (" " * padding) + line.chomp }.join("\n")
+  end
+end

data/lib/tng/ui/system_status_display.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+class SystemStatusDisplay
+  def initialize(pastel, args)
+    @pastel = pastel
+    @args = args
+  end
+
+  def display(status)
+    case status[:status]
+    when :ok
+      puts @pastel.green("✅ System operational") unless @args[:type] && @args[:file]
+      true
+    when :version_mismatch
+      puts @pastel.red.bold("❌ Version Mismatch!")
+      puts @pastel.yellow("Your gem version: #{status[:gem_version]}")
+      puts @pastel.yellow("Required version: #{status[:current_version]}")
+      puts
+      puts @pastel.bright_white("Please update your gem:")
+      puts @pastel.cyan("  gem update tng")
+      puts @pastel.dim("  or")
+      puts @pastel.cyan("  bundle update tng")
+      false
+    when :base_url_mismatch
+      puts @pastel.red.bold("❌ Endpoint URL Mismatch!")
+      puts @pastel.yellow("Your configured URL: #{status[:user_base_url]}")
+      puts @pastel.yellow("Server's current URL: #{status[:server_base_url]}")
+      puts
+      puts @pastel.bright_white("Please update your configuration:")
+      puts @pastel.cyan("  Edit config/initializers/tng.rb")
+      puts @pastel.cyan("  Set: config.base_url = '#{status[:server_base_url]}'")
+      puts
+      puts @pastel.dim("This ensures you're connecting to the correct endpoint.")
+      false
+    when :service_down
+      puts @pastel.red.bold("❌ Service Unavailable")
+      puts @pastel.dim(status[:message])
+      puts
+      puts @pastel.yellow("The TNG service is currently down.")
+      puts @pastel.dim("Please try again later or contact support.")
+      false
+    when :error
+      puts @pastel.red.bold("❌ Connection Error")
+      puts @pastel.dim(status[:message])
+      puts
+      puts @pastel.yellow("Unable to connect to TNG service.")
+      puts @pastel.dim("Please check your internet connection and try again.")
+      false
+    else
+      puts @pastel.red.bold("❌ Unknown Error")
+      puts @pastel.dim(status[:message] || "An unknown error occurred")
+      false
+    end
+  end
+end

data/lib/tng/ui/user_stats_display.rb

@@ -0,0 +1,153 @@
+# frozen_string_literal: true
+
+require "tty-box"
+require "tty-progressbar"
+require "pastel"
+require "tty-screen"
+
+class UserStatsDisplay
+  def initialize(pastel, prompt)
+    @pastel = pastel
+    @prompt = prompt
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def display(stats_data)
+    header = @pastel.bright_white.bold("📊 User Statistics")
+    puts center_text(header)
+
+    if stats_data
+      display_stats_box(stats_data)
+      display_usage_progress(stats_data)
+      display_usage_info(stats_data)
+    else
+      display_error
+    end
+
+    puts
+    tip_msg = @pastel.dim("💡 Tip: Contact support if you need more test generations")
+    puts center_text(tip_msg)
+    @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
+  end
+
+  def display_stats_box(stats_data)
+    stats_content = [
+      @pastel.cyan.bold("Account Information"),
+      "",
+      @pastel.bright_white("Test Runs: ") + @pastel.green.bold(stats_data["runs"].to_s),
+      @pastel.bright_white("Max Runs: ") + @pastel.yellow.bold(stats_data["max_runs"].to_s),
+      @pastel.bright_white("Gem Version: ") + @pastel.magenta(stats_data["gem_version"] || "N/A"),
+      "",
+      @pastel.dim("Request ID: #{stats_data["request_id"]}")
+    ].join("\n")
+
+    box_width = 50
+    box = TTY::Box.frame(
+      title: { top_left: " Your TNG Stats " },
+      style: { fg: :bright_white, border: { fg: :green }, title: { fg: :bright_green } },
+      padding: [1, 2],
+      align: :left,
+      width: box_width
+    ) { stats_content }
+
+    puts center_box(box, box_width)
+  end
+
+  def display_usage_progress(stats_data)
+    runs = stats_data["runs"]
+    max_runs = stats_data["max_runs"]
+
+    usage_percent = max_runs > 0 ? (runs.to_f / max_runs * 100).round(1) : 0
+
+    bar_color = determine_bar_color(usage_percent)
+
+    puts
+    puts center_text(@pastel.bright_white.bold("Usage Overview"))
+    puts
+
+    progress_width = 40
+    padding = (@terminal_width - progress_width - 20) / 2
+    padding = 0 if padding.negative?
+
+    bar_format = "#{" " * padding}Usage: [:bar] :percent (:current/:total)"
+
+    bar = TTY::ProgressBar.new(bar_format,
+                               total: max_runs,
+                               width: progress_width,
+                               complete: bar_color,
+                               incomplete: @pastel.dim("░"),
+                               head: bar_color)
+
+    current_progress = 0
+    while current_progress < runs
+      step = [runs / 10, 1].max
+      current_progress = [current_progress + step, runs].min
+      bar.current = current_progress
+      bar.render
+      sleep(0.05) if runs > 10
+    end
+
+    bar.current = runs
+    bar.render
+
+    puts
+
+    status_msg = case usage_percent
+                 when 0..50
+                   @pastel.green("🟢 Good usage - plenty of runs remaining")
+                 when 51..80
+                   @pastel.yellow("🟡 Moderate usage - consider monitoring")
+                 when 81..95
+                   @pastel.cyan("🟠 High usage - approaching limit")
+                 else
+                   @pastel.red("🔴 Limit reached - contact support for more runs")
+                 end
+
+    puts center_text(status_msg)
+  end
+
+  def determine_bar_color(usage_percent)
+    case usage_percent
+    when 0..50
+      @pastel.on_green(" ")
+    when 51..80
+      @pastel.on_yellow(" ")
+    when 81..95
+      @pastel.on_cyan(" ")
+    else
+      @pastel.on_red(" ")
+    end
+  end
+
+  def display_usage_info(stats_data)
+    usage_remaining = stats_data["max_runs"] - stats_data["runs"]
+    usage_msg = if usage_remaining.positive?
+                  @pastel.green("✅ You have #{usage_remaining} test generations remaining")
+                else
+                  @pastel.red("⚠️  You have reached your test generation limit")
+                end
+    puts center_text(usage_msg)
+  end
+
+  def display_error
+    error_msg = @pastel.red.bold("❌ Failed to fetch user statistics")
+    puts center_text(error_msg)
+    puts center_text(@pastel.dim("Please check your API key and internet connection"))
+  end
+
+  def center_text(text)
+    padding = (@terminal_width - @pastel.strip(text).length) / 2
+    padding = 0 if padding.negative?
+    (" " * padding) + text
+  end
+
+  def center_box(box_string, box_width)
+    padding = (@terminal_width - box_width) / 2
+    padding = 0 if padding.negative?
+    box_string.lines.map { |line| (" " * padding) + line.chomp }.join("\n")
+  end
+end