tng 0.2.1

data/lib/tng/services/testng.rb

@@ -0,0 +1,100 @@
+module Services
+  class Testng
+    STATS_PATH = "cli/tng_rails/stats"
+
+    def initialize(http_client)
+      @http_client = http_client
+    end
+
+    def check_system_status
+      response = @http_client.ping
+
+      if response.is_a?(HTTPX::ErrorResponse)
+        return {
+          status: :error,
+          message: "Unable to connect to TNG service: #{response.error&.message}",
+          error_type: :connection_error
+        }
+      end
+
+      case response.status
+      when 200
+        begin
+          data = JSON.parse(response.body)
+          current_version = data["current_version"]["gem_version"]
+          server_base_url = data["base_url"]
+          user_base_url = Tng::Services::UserAppConfig.base_url
+
+          if current_version > Tng::VERSION
+            return {
+              status: :version_mismatch,
+              message: "Version mismatch detected",
+              current_version: current_version,
+              gem_version: Tng::VERSION,
+              error_type: :version_mismatch
+            }
+          end
+
+          # Check for base URL mismatch
+          if server_base_url && user_base_url && server_base_url != user_base_url
+            return {
+              status: :base_url_mismatch,
+              message: "Base URL mismatch detected",
+              server_base_url: server_base_url,
+              user_base_url: user_base_url,
+              current_version: current_version,
+              gem_version: Tng::VERSION,
+              error_type: :base_url_mismatch
+            }
+          end
+
+          {
+            status: :ok,
+            message: data["message"] || "System is operational",
+            current_version: current_version,
+            gem_version: Tng::VERSION,
+            server_base_url: server_base_url
+          }
+        rescue JSON::ParserError => e
+          {
+            status: :error,
+            message: "Invalid response from server: #{e.message}",
+            error_type: :parse_error
+          }
+        end
+      else
+        {
+          status: :service_down,
+          message: "TestNG service is currently unavailable (HTTP #{response.status})",
+          error_type: :service_unavailable
+        }
+      end
+    end
+
+    def get_user_stats
+      headers = {
+        "Authorization" => "Bearer #{Tng::Services::UserAppConfig.api_key}",
+        "Content-Type" => "application/json",
+        "User-Agent" => "Tng/#{Tng::VERSION} Ruby/#{RUBY_VERSION}"
+      }
+
+      response = @http_client.get(STATS_PATH, headers: headers)
+
+      return if response.is_a?(HTTPX::ErrorResponse)
+
+      if response.status == 200
+        begin
+          stats_data = JSON.parse(response.body)
+          puts "✅ Statistics fetched successfully!"
+          stats_data
+        rescue JSON::ParserError => e
+          puts "❌ Failed to parse statistics response: #{e.message}"
+          nil
+        end
+      else
+        puts "❌ Failed to fetch statistics (#{response.status})"
+        nil
+      end
+    end
+  end
+end

data/lib/tng/services/user_app_config.rb

@@ -0,0 +1,76 @@
+# frozen_string_literal: true
+
+module Tng
+  module Services
+    class UserAppConfig
+      def self.api_key
+        Tng.api_key
+      end
+
+      def self.base_url
+        Tng.base_url
+      end
+
+      def self.configured?
+        Tng.api_key && Tng.base_url
+      end
+
+      def self.missing_config
+        missing = []
+        missing << "API key" unless api_key
+        missing << "Base URL" unless base_url
+        missing
+      end
+
+      def self.config_with_source
+        auth_entry_points_with_source = []
+        methods = Tng.authentication_methods
+        methods.each do |method|
+          auth_entry_points_with_source << {
+            method: method[:method],
+            file: method[:file_location],
+            auth_type: method[:auth_type],
+            source: extract_method_source(method[:file_location], method[:method])
+          }
+        end
+        Tng.config[:authentication_entry_points_with_source] = auth_entry_points_with_source
+        Tng.config
+      end
+
+      def self.find_method_in_ast(node, method_name)
+        return unless node.is_a?(Prism::Node)
+
+        return node if node.is_a?(Prism::DefNode) && node.name == method_name.to_sym
+
+        node.child_nodes.each do |child|
+          result = find_method_in_ast(child, method_name)
+          return result if result
+        end
+
+        nil
+      end
+
+      def self.extract_method_source(file_path, method_name)
+        return unless file_path && method_name
+
+        full_path = Rails.root.join(file_path)
+        return unless File.exist?(full_path)
+
+        file_content = File.read(full_path)
+        result = Prism.parse(file_content)
+
+        method_node = find_method_in_ast(result.value, method_name)
+        return unless method_node
+
+        start_line = method_node.location.start_line - 1
+        end_line = method_node.location.end_line - 1
+
+        lines = file_content.lines
+        method_source = lines[start_line..end_line].join
+        method_source.strip
+      rescue StandardError
+        nil
+      end
+    end
+  end
+end

data/lib/tng/ui/about_display.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+require "tty-box"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+class AboutDisplay
+  def initialize(pastel, prompt, version)
+    @pastel = pastel
+    @prompt = prompt
+    @version = version
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def display
+    about_content = [
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:secondary),
+                          "Tng is an LLM-powered test generation tool for Rails applications."),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:accent), "Features:"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "#{Tng::UI::Theme.icon(:bullet)} 20+ Rails file types supported"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "#{Tng::UI::Theme.icon(:bullet)} Core: Controllers, Models, Services"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "#{Tng::UI::Theme.icon(:bullet)} Extended: Jobs, Helpers, Lib, Policies, Presenters, Mailers, GraphQL + more"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted), "#{Tng::UI::Theme.icon(:bullet)} LLM-powered test suggestions"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted), "#{Tng::UI::Theme.icon(:bullet)} Test coverage analysis"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:accent), "Technology:"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "#{Tng::UI::Theme.icon(:bullet)} Static code analysis with Prism"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "#{Tng::UI::Theme.icon(:bullet)} Intelligent pattern recognition"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted), "#{Tng::UI::Theme.icon(:bullet)} Rails-specific optimizations"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:muted), "Version: #{@version}"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "Built with #{Tng::UI::Theme.icon(:heart)} for Rails developers")
+    ].join("\n")
+
+    box_width = Tng::UI::Theme.calculate_box_width(@terminal_width)
+    style = Tng::UI::Theme.box_style(:default)
+
+    about_box = TTY::Box.frame(
+      title: { top_left: " About Tng " },
+      style: style[:style],
+      padding: style[:padding],
+      width: box_width
+    ) do
+      about_content
+    end
+
+    puts Tng::UI::Theme.center_box(about_box, box_width, @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/authentication_warning_display.rb

@@ -0,0 +1,172 @@
+# frozen_string_literal: true
+
+require "tty-box"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+class AuthenticationWarningDisplay
+  def initialize(pastel)
+    @pastel = pastel
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def display_if_missing
+    return false unless authentication_config_missing?
+
+    display_warning
+    handle_user_choice
+  end
+
+  def display_warning
+    warning_content = [
+      @pastel.public_send(Tng::UI::Theme.color(:error)).bold("Authentication Configuration Missing"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:primary), "TNG needs authentication setup to generate proper tests."),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:accent), "Missing: ") + build_missing_items_summary,
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:secondary), "Quick fix:"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "#{Tng::UI::Theme.icon(:bullet)} Edit config/initializers/tng.rb"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "#{Tng::UI::Theme.icon(:bullet)} Uncomment authentication_methods array"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted), "#{Tng::UI::Theme.icon(:bullet)} Add your auth method details"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:success), "Press Enter to continue anyway"),
+      @pastel.public_send(Tng::UI::Theme.color(:error), "Press Esc to cancel and fix configuration")
+    ].join("\n")
+
+    box_width = Tng::UI::Theme.calculate_box_width(@terminal_width)
+    style = Tng::UI::Theme.box_style(:warning)
+
+    warning_box = TTY::Box.frame(
+      title: { top_left: " Auth Warning " },
+      style: style[:style],
+      padding: style[:padding],
+      width: box_width
+    ) do
+      warning_content
+    end
+
+    puts Tng::UI::Theme.center_box(warning_box, box_width, @terminal_width)
+    puts
+  end
+
+  private
+
+  def handle_user_choice
+    require "io/console"
+
+    loop do
+      key = $stdin.getch
+      case key
+      when "\r", "\n" # Enter key
+        return false # Continue anyway
+      when "\e" # Escape key
+        return true # Cancel (authentication missing)
+      when "\u0003" # Ctrl+C
+        puts "\n#{@pastel.public_send(Tng::UI::Theme.color(:warning), "Operation cancelled.")}"
+        exit(0)
+      end
+    end
+  rescue StandardError
+    # Fallback if getch is not available
+    puts @pastel.public_send(Tng::UI::Theme.color(:muted), "Press Enter to continue or Ctrl+C to cancel...")
+    gets
+    false
+  end
+
+  def build_missing_items_summary
+    issues = []
+
+    issues << "empty auth methods" if Tng.authentication_enabled && authentication_methods_empty?
+
+    issues << "invalid auth config" if !authentication_methods_empty? && !authentication_methods_valid?
+
+    issues << "auth disabled" unless Tng.authentication_enabled
+
+    issues.empty? ? "unknown issue" : issues.join(", ")
+  end
+
+  def authentication_config_missing?
+    # Check if authentication is enabled but methods are not configured
+    return true if Tng.authentication_enabled && authentication_methods_empty?
+
+    # Check if authentication methods exist but are invalid
+    return true if !authentication_methods_empty? && !authentication_methods_valid?
+
+    false
+  end
+
+  def authentication_methods_empty?
+    methods = Tng.authentication_methods
+    methods.nil? || methods.empty?
+  end
+
+  def authentication_methods_valid?
+    methods = Tng.authentication_methods
+    return false if methods.nil? || methods.empty?
+
+    methods.all? do |method|
+      method.is_a?(Hash) &&
+        method.key?(:method) && !method[:method].to_s.strip.empty? &&
+        method.key?(:file_location) && !method[:file_location].to_s.strip.empty? &&
+        method.key?(:auth_type) && !method[:auth_type].to_s.strip.empty?
+    end
+  end
+
+  def build_missing_items_list
+    issues = []
+
+    if Tng.authentication_enabled && authentication_methods_empty?
+      issues << @pastel.public_send(Tng::UI::Theme.color(:error), "  • authentication_methods array is empty")
+    end
+
+    if !authentication_methods_empty? && !authentication_methods_valid?
+      issues << @pastel.public_send(Tng::UI::Theme.color(:error), "  • authentication_methods contains invalid entries")
+
+      Tng.authentication_methods.each_with_index do |method, index|
+        next if method.is_a?(Hash)
+
+        issues << @pastel.public_send(Tng::UI::Theme.color(:muted), "    - Entry #{index + 1}: not a valid hash")
+        next
+      end
+
+      Tng.authentication_methods.each_with_index do |method, index|
+        next unless method.is_a?(Hash)
+
+        if !method.key?(:method) || method[:method].to_s.strip.empty?
+          issues << @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                        "    - Entry #{index + 1}: missing or empty 'method'")
+        end
+
+        if !method.key?(:file_location) || method[:file_location].to_s.strip.empty?
+          issues << @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                        "    - Entry #{index + 1}: missing or empty 'file_location'")
+        end
+
+        if !method.key?(:auth_type) || method[:auth_type].to_s.strip.empty?
+          issues << @pastel.public_send(Tng::UI::Theme.color(:muted),
+                                        "    - Entry #{index + 1}: missing or empty 'auth_type'")
+        end
+      end
+    end
+
+    unless Tng.authentication_enabled
+      issues << @pastel.public_send(Tng::UI::Theme.color(:warning),
+                                    "  • authentication_enabled is set to false")
+    end
+
+    if issues.empty?
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "  • Configuration appears valid")
+    else
+      issues.join("\n")
+    end
+  end
+end

data/lib/tng/ui/configuration_display.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require "tty-box"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+class ConfigurationDisplay
+  def initialize(pastel)
+    @pastel = pastel
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def display_missing_config(missing_config)
+    config_content = [
+      @pastel.public_send(Tng::UI::Theme.color(:error)).bold("Configuration Required"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:accent), "Missing required configuration:"),
+      missing_config.map do |key|
+        @pastel.public_send(Tng::UI::Theme.color(:secondary), "#{Tng::UI::Theme.icon(:bullet)} #{key}")
+      end.join("\n"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:secondary), "Quick fix:"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "#{Tng::UI::Theme.icon(:bullet)} Edit config/initializers/tng.rb"),
+      @pastel.public_send(Tng::UI::Theme.color(:muted),
+                          "#{Tng::UI::Theme.icon(:bullet)} Set your API key and other required values"),
+      "",
+      @pastel.public_send(Tng::UI::Theme.color(:success), "Press Enter to continue after fixing"),
+      @pastel.public_send(Tng::UI::Theme.color(:error), "Press Ctrl+C to exit")
+    ].join("\n")
+
+    box_width = Tng::UI::Theme.calculate_box_width(@terminal_width)
+    style = Tng::UI::Theme.box_style(:error)
+
+    config_box = TTY::Box.frame(
+      title: { top_left: " Config Required " },
+      style: style[:style],
+      padding: style[:padding],
+      width: box_width
+    ) do
+      config_content
+    end
+
+    puts Tng::UI::Theme.center_box(config_box, box_width, @terminal_width)
+    puts
+  end
+end

data/lib/tng/ui/controller_test_flow_display.rb

@@ -0,0 +1,79 @@
+# frozen_string_literal: true
+
+require "tty-prompt"
+require "tty-spinner"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+class ControllerTestFlowDisplay
+  def initialize(prompt, pastel)
+    @prompt = prompt
+    @pastel = pastel
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def select_controller(controllers)
+    header = @pastel.public_send(Tng::UI::Theme.color(:primary)).bold("#{Tng::UI::Theme.icon(:config)} Select controller 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|
+      controllers.each do |controller|
+        display_name = "#{controller[:name]} #{@pastel.public_send(Tng::UI::Theme.color(:muted),
+                                                                   "(#{controller[:path]})")}"
+        menu.choice display_name, controller
+      end
+      menu.choice @pastel.public_send(Tng::UI::Theme.color(:secondary), "#{Tng::UI::Theme.icon(:back)}  Back"), :back
+    end
+  end
+
+  def show_no_controllers_message
+    error_msg = "#{@pastel.public_send(Tng::UI::Theme.color(:error)).bold("#{Tng::UI::Theme.icon(:error)} No controllers found in your application")}\n#{@pastel.public_send(
+      Tng::UI::Theme.color(:muted), "Make sure you have controllers in app/controllers/"
+    )}"
+    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_controller_method(controller, methods)
+    header = @pastel.public_send(Tng::UI::Theme.color(:primary)).bold("#{Tng::UI::Theme.icon(:rocket)} Select method to test in #{controller[: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(controller)
+    error_msg = "#{@pastel.public_send(Tng::UI::Theme.color(:error)).bold("#{Tng::UI::Theme.icon(:error)} No methods found in #{controller[:name]}")}\n#{@pastel.public_send(
+      Tng::UI::Theme.color(:muted), "Controller 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/display_banner.rb

@@ -0,0 +1,44 @@
+require "tty-box"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+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.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), "Generate comprehensive tests for your Rails application"),
+      pastel.public_send(Tng::UI::Theme.color(:muted), "Powered by LLM and static analysis")
+    ].join("\n")
+
+    box_width = 64
+    box_width = terminal_width if box_width > terminal_width
+    style = Tng::UI::Theme.box_style(:default)
+
+    box = TTY::Box.frame(
+      title: { top_left: " Tng ", bottom_right: " Ready " },
+      style: style[:style],
+      padding: style[:padding],
+      align: :center,
+      width: box_width
+    ) do
+      banner_content
+    end
+
+    "#{Tng::UI::Theme.center_box(box, box_width, terminal_width)}\n"
+  end
+end

data/lib/tng/ui/goodbye_display.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require "tty-box"
+require "pastel"
+require "tty-screen"
+require_relative "theme"
+
+class GoodbyeDisplay
+  def initialize(pastel)
+    @pastel = pastel
+    @terminal_width = begin
+      TTY::Screen.width
+    rescue StandardError
+      80
+    end
+  end
+
+  def display
+    box_width = Tng::UI::Theme.calculate_box_width(@terminal_width)
+    style = Tng::UI::Theme.box_style(:success)
+
+    goodbye_box = TTY::Box.frame(
+      title: { top_left: " Thank You " },
+      style: style[:style],
+      padding: style[:padding],
+      align: :center,
+      width: box_width
+    ) do
+      [
+        @pastel.public_send(Tng::UI::Theme.color(:success)).bold("Thanks for using Tng!"),
+        "",
+        @pastel.public_send(Tng::UI::Theme.color(:secondary), "Happy testing!"),
+        "",
+        @pastel.public_send(Tng::UI::Theme.color(:muted), "Your Rails tests are now supercharged with LLM")
+      ].join("\n")
+    end
+
+    puts Tng::UI::Theme.center_box(goodbye_box, box_width, @terminal_width)
+    puts
+  end
+end