tng 0.2.3 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 905722688ad1966dbe0fbc488425daaa15124c1181a136df59a6a92b99871751
-  data.tar.gz: 61b791a6c204dbddeec4e6d936bcf294e5a334ad2e836dd715b042580bd55f23
+  metadata.gz: 98da639352758b3b24da35177573d67e7e51860b85121eb0037926601d141027
+  data.tar.gz: 7f289b904f527e8f055e0b4586dc0242d678f4b590112ad16b028ae50d793581
 SHA512:
-  metadata.gz: ee626a2d20b0d20ce0b412ee564e83f72326f2434a0065e66848a6f44b6d89e1d98453a6c311bff01ff551aa8634c60afdf586e5a5b1f538cd09ad63a3ff28e1
-  data.tar.gz: 1f7ae4397fd988a72e13642f2d5a5a824d5694e7a8991c5b33073962b33b03fdf48b2c976130ab14f03839265a8a683a4141b2288d7aeb2461692fbc061e27a0
+  metadata.gz: 5ddc1271b830f67a918c8998397df4afa28e7fb679192eae280a95d3d4aa47a724f013c7aacd08f7f143063e1fcafe0cac732d6b866c9308582057125bd15c21
+  data.tar.gz: ad5fb8e2a558f3b2f46cfe646c9abb1b8bf2861d8761ec2bf5c980fe7832068f45b53418e01c4f1bcdfc1ab91556a890d678405a10ade4e24d57c16dda09d59d

data/README.md

@@ -315,6 +315,33 @@ TNG focuses on precise, method-level test generation:
 
 Use `bundle exec tng --help` for more options.
 
+### Direct Mode (for Developers)
+
+For experienced developers who want quick, command-line test generation:
+
+```bash
+# Generate test for specific method using file path
+bundle exec tng app/controllers/users_controller.rb index
+
+# Short form using shortcuts
+bundle exec tng f=users_controller.rb m=show
+
+# Full parameter names
+bundle exec tng --file=app/models/user.rb --method=validate
+
+# Works from subdirectories (auto-detection)
+bundle exec tng users_controller.rb index
+
+# Mixed parameter order
+bundle exec tng m=index f=app/controllers/users_controller.rb
+```
+
+**Direct Mode Features:**
+- **Automatic file type detection** - No need to specify `-t controller/model/service`
+- **Flexible file path resolution** - Works with full paths or just filenames
+- **Multiple syntax formats** - Use `f=` shortcuts or `--file=` full parameters
+- **Smart path finding** - Automatically searches Rails directories
+
 ### Debug Mode
 
 For troubleshooting configuration or generation issues:
@@ -324,7 +351,7 @@ For troubleshooting configuration or generation issues:
 DEBUG=1 bundle exec tng
 
 # Direct mode with debug
-DEBUG=1 bundle exec tng -t c -f ping
+DEBUG=1 bundle exec tng app/controllers/users_controller.rb index
 ```
 
 ### Help and Options
@@ -400,7 +427,7 @@ bundle exec tng -h                  # Short help
 Enable debug mode to get detailed information:
 
 ```bash
-DEBUG=1 bundle exec tng -t c -f your_controller
+DEBUG=1 bundle exec tng app/controllers/your_controller.rb method_name
 ```
 
 Debug output includes:

data/Rakefile

@@ -19,6 +19,27 @@ RbSys::ExtensionTask.new("tng", GEMSPEC) do |ext|
   ext.lib_dir = "lib/tng"
 end
 
+# Create a task for the binaries/tng.bundle file that rb_sys expects
+file "binaries/tng.bundle" do
+  FileUtils.mkdir_p("binaries")
+  # Copy from lib/tng if it exists, otherwise create empty file to satisfy dependency
+  if File.exist?("lib/tng/tng.bundle")
+    FileUtils.cp("lib/tng/tng.bundle", "binaries/tng.bundle")
+  else
+    # Create a placeholder file that will be overwritten by redo task
+    FileUtils.touch("binaries/tng.bundle")
+  end
+end
+
+# Ensure staging directories exist before compilation
+task :prepare_staging do
+  staging_dir = "tmp/arm64-darwin24/stage/binaries"
+  FileUtils.mkdir_p(staging_dir) unless Dir.exist?(staging_dir)
+end
+
+# Make compile depend on staging preparation
+task compile: :prepare_staging
+
 task default: %i[compile test rubocop]
 
 desc "Recompile Rust code and copy the binary for packaging"
@@ -32,13 +53,26 @@ task :redo do
   source_file = File.join("lib", "tng", binary_name)
   destination_dir = "binaries"
 
-  if File.exist?(source_file)
-    FileUtils.mkdir_p(destination_dir)
-    FileUtils.cp(source_file, File.join(destination_dir, binary_name))
-    puts "✅ Copied #{binary_name} to #{destination_dir}/"
-  else
-    puts "❌ Error: Compiled binary not found at #{source_file}"
-    puts "Make sure 'rake compile' ran successfully."
+  retries = 0
+  max_retries = 5
+
+  while retries < max_retries
+    if File.exist?(source_file)
+      FileUtils.mkdir_p(destination_dir)
+      FileUtils.cp(source_file, File.join(destination_dir, binary_name))
+      puts "✅ Copied #{binary_name} to #{destination_dir}/"
+      break
+    else
+      retries += 1
+      if retries < max_retries
+        puts "⏳ Waiting for compiled binary... (attempt #{retries}/#{max_retries})"
+        sleep(0.5)
+      else
+        puts "❌ Error: Compiled binary not found at #{source_file}"
+        puts "Make sure 'rake compile' ran successfully."
+        exit 1
+      end
+    end
   end
 end
 

data/bin/tng

@@ -9,6 +9,7 @@ $LOAD_PATH.unshift(lib_path) unless $LOAD_PATH.include?(lib_path)
 require "tng"
 require "tng/services/direct_generation"
 require "tng/services/extract_methods"
+require "tng/services/file_type_detector"
 
 require "tty-prompt"
 require "tty-spinner"
@@ -54,19 +55,17 @@ class CLI
     example "  • Browse and search controllers/models/services", ""
     example "  • Select specific methods from filtered lists", ""
     example "  • Generate tests for individual methods", ""
-  end
-
-  option :type do
-    short "-t"
-    long "--type=TYPE"
-    desc "Test type (controller, model, service, other)"
-    permit %w[controller model service other c m mo s se o]
+    example ""
+    example "Direct mode (automatic file type detection):", ""
+    example "  bundle exec tng app/controllers/users_controller.rb index", ""
+    example "  bundle exec tng f=users_controller.rb m=show", ""
+    example "  bundle exec tng --file=app/models/user.rb --method=validate", ""
   end
 
   option :file do
     short "-f"
     long "--file=FILE"
-    desc "File name (without .rb extension)"
+    desc "File path (with or without .rb extension)"
   end
 
   option :method do
@@ -109,7 +108,7 @@ class CLI
 
     initialize_config_and_clients if rails_loaded
 
-    if params[:type] && params[:file]
+    if params[:file]
       run_direct_generation
     else
       if find_rails_root && !rails_loaded && !defined?(Rails)
@@ -130,35 +129,56 @@ class CLI
 
   def preprocess_arguments(argv)
     normalized = []
-
-    argv.each do |arg|
+    positional_args = []
+    
+    argv.each_with_index do |arg, index|
       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)
+      when /^(?:--)?(file|f)=(.+)$/
+        normalized << "--file=#{$2}"
+      when /^(?:--)?(method|m)=(.+)$/
+        normalized << "--method=#{$2}"
+      when /^(help|h)=(.+)$/
+        normalized << "--help=#{$2}"
+      when /^--file$/, /^-f$/
         normalized << arg
-      when /^-[tfmh]$/ # Handle single flags like -t, -f, -m, -h (keep as is)
+      when /^--method$/, /^-m$/
         normalized << arg
-      when /^--\w+$/ # Handle long flags like --help (keep as is)
+      when /^--help$/, /^-h$/
         normalized << arg
-      else
-        # Regular arguments or values
+      when /^--\w+=.+$/
         normalized << arg
+      else
+        prev_arg = argv[index - 1] if index > 0
+        if prev_arg && (prev_arg.match?(/^(-f|--file)$/) || prev_arg.match?(/^(-m|--method)$/))
+          normalized << arg
+        else
+          positional_args << arg
+        end
       end
     end
-
+    
+    if positional_args.length >= 2
+      has_file = normalized.any? { |a| a.match?(/^--file/) }
+      has_method = normalized.any? { |a| a.match?(/^--method/) }
+      
+      unless has_file
+        normalized << "--file=#{positional_args[0]}"
+      end
+      unless has_method
+        normalized << "--method=#{positional_args[1]}"
+      end
+    elsif positional_args.length == 1
+      arg = positional_args[0]
+      has_file = normalized.any? { |a| a.match?(/^--file/) }
+      has_method = normalized.any? { |a| a.match?(/^--method/) }
+      
+      if !has_file && (arg.end_with?('.rb') || arg.include?('/'))
+        normalized << "--file=#{arg}"
+      elsif !has_method && !arg.include?('/')
+        normalized << "--method=#{arg}"
+      end
+    end
+    
     normalized
   end
 
@@ -452,7 +472,7 @@ class CLI
     # Show completion message with test count and timing
     if result && result[:test_count]
       count_msg = result[:test_count] == 1 ? "1 test" : "#{result[:test_count]} tests"
-      time_msg = result[:generation_time] ? " in #{format_generation_time(result[:generation_time])}" : ""
+      time_msg = result[:generation_time] ? " in #{Tng::Utils.format_generation_time(result[:generation_time])}" : ""
       completion_msg = @pastel.decorate(
         "#{Tng::UI::Theme.icon(:success)} Generated #{count_msg} for controller method#{time_msg}!", Tng::UI::Theme.color(:success)
       )
@@ -548,7 +568,7 @@ class CLI
     # Show completion message with test count and timing
     if result && result[:test_count]
       count_msg = result[:test_count] == 1 ? "1 test" : "#{result[:test_count]} tests"
-      time_msg = result[:generation_time] ? " in #{format_generation_time(result[:generation_time])}" : ""
+      time_msg = result[:generation_time] ? " in #{Tng::Utils.format_generation_time(result[:generation_time])}" : ""
       completion_msg = @pastel.decorate(
         "#{Tng::UI::Theme.icon(:success)} Generated #{count_msg} for model method#{time_msg}!", Tng::UI::Theme.color(:success)
       )
@@ -607,7 +627,7 @@ class CLI
     # Show completion message with test count and timing
     if result && result[:test_count]
       count_msg = result[:test_count] == 1 ? "1 test" : "#{result[:test_count]} tests"
-      time_msg = result[:generation_time] ? " in #{format_generation_time(result[:generation_time])}" : ""
+      time_msg = result[:generation_time] ? " in #{Tng::Utils.format_generation_time(result[:generation_time])}" : ""
       completion_msg = @pastel.decorate(
         "#{Tng::UI::Theme.icon(:success)} Generated #{count_msg} for service method#{time_msg}!", Tng::UI::Theme.color(:success)
       )
@@ -710,7 +730,7 @@ class CLI
     # Show completion message with test count and timing
     if result && result[:test_count]
       count_msg = result[:test_count] == 1 ? "1 test" : "#{result[:test_count]} tests"
-      time_msg = result[:generation_time] ? " in #{format_generation_time(result[:generation_time])}" : ""
+      time_msg = result[:generation_time] ? " in #{Tng::Utils.format_generation_time(result[:generation_time])}" : ""
       completion_msg = @pastel.decorate(
         "#{Tng::UI::Theme.icon(:success)} Generated #{count_msg} for other method#{time_msg}!", Tng::UI::Theme.color(:success)
       )
@@ -733,7 +753,7 @@ class CLI
   end
 
   def check_system_status
-    unless params[:type] && params[:file]
+    unless params[:file]
       puts @pastel.decorate("#{Tng::UI::Theme.icon(:info)} Checking system status...",
                             Tng::UI::Theme.color(:muted))
     end
@@ -801,7 +821,7 @@ class CLI
   end
 
   def show_post_generation_menu(result)
-    is_direct_mode = params[:type] && params[:file]
+    is_direct_mode = params[:file]
 
     if is_direct_mode
       puts
@@ -820,6 +840,7 @@ class CLI
       return
     end
 
+    # Show menu regardless of test results
     loop do
       puts
       header = @pastel.decorate("#{Tng::UI::Theme.icon(:rocket)} Test Generated Successfully!", Tng::UI::Theme.color(:primary))
@@ -838,6 +859,8 @@ class CLI
         cycle: true,
         symbols: { marker: Tng::UI::Theme.icon(:marker) }
       ) do |menu|
+        menu.choice @pastel.decorate("#{Tng::UI::Theme.icon(:run)} Run tests", Tng::UI::Theme.color(:success)),
+                    :run_tests
         menu.choice @pastel.decorate("#{Tng::UI::Theme.icon(:config)} Copy run command", Tng::UI::Theme.color(:warning)),
                     :copy_command
         menu.choice @pastel.decorate("#{Tng::UI::Theme.icon(:back)} Back to main menu", Tng::UI::Theme.color(:info)),
@@ -845,6 +868,8 @@ class CLI
       end
 
       case choice
+      when :run_tests
+        run_and_show_test_results(result[:run_command])
       when :copy_command
         copy_to_clipboard(result[:run_command])
       when :back
@@ -853,6 +878,99 @@ class CLI
     end
   end
 
+  def run_and_show_test_results(command)
+    puts
+    header = @pastel.decorate("#{Tng::UI::Theme.icon(:run)} Running tests...", Tng::UI::Theme.color(:primary))
+    puts center_text(header)
+
+    begin
+      # Check if test file exists before running
+      test_file = extract_test_file_from_command(command)
+      unless test_file && File.exist?(test_file)
+        error_msg = @pastel.decorate("#{Tng::UI::Theme.icon(:error)} Test file not found!", Tng::UI::Theme.color(:error))
+        puts center_text(error_msg)
+
+        if test_file
+          file_msg = @pastel.decorate("File: #{test_file}", Tng::UI::Theme.color(:muted))
+          puts center_text(file_msg)
+        end
+
+        suggestion_msg = @pastel.decorate("The test file may have been moved or deleted.", Tng::UI::Theme.color(:warning))
+        puts center_text(suggestion_msg)
+
+        @prompt.keypress(center_text(@pastel.decorate("Press any key to continue...", Tng::UI::Theme.color(:muted))))
+        return
+      end
+
+      # Modify command to get cleaner output
+      if command.include?("rspec")
+        # Use JSON format for RSpec - much cleaner parsing
+        json_command = command.sub(/rspec/, "rspec --format json --out /dev/null")
+        output = `#{json_command} 2>&1`
+        exit_code = $?.exitstatus
+
+        # Check for common errors before parsing
+        return if handle_test_execution_errors(output, exit_code)
+
+        Tng::Utils.parse_rspec_json_results(output, exit_code, @pastel, @terminal_width)
+      else
+        # For Rails test, use current approach but with cleaner output
+        output = `#{command} 2>&1`
+        exit_code = $?.exitstatus
+
+        # Check for common errors before parsing
+        return if handle_test_execution_errors(output, exit_code)
+
+        Tng::Utils.parse_minitest_results(output, exit_code, @pastel, @terminal_width)
+      end
+    rescue StandardError => e
+      error_msg = @pastel.decorate("#{Tng::UI::Theme.icon(:error)} Error running tests!", Tng::UI::Theme.color(:error))
+      puts center_text(error_msg)
+
+      details_msg = @pastel.decorate("#{e.message}", Tng::UI::Theme.color(:muted))
+      puts center_text(details_msg)
+    end
+
+    @prompt.keypress(center_text(@pastel.decorate("Press any key to continue...", Tng::UI::Theme.color(:muted))))
+  end
+
+  def handle_test_execution_errors(output, exit_code)
+    # Return true if we handled an error (caller should return early)
+    # Return false if no error was handled (caller should continue)
+
+    # If tests failed to run (non-zero exit code or common error patterns)
+    if !exit_code.zero? || output.include?("LoadError") || output.include?("cannot load such file") ||
+       (output.include?("bundler") && output.include?("Could not find")) ||
+       (output.include?("bin/rails") && (output.include?("require") || output.include?("load")))
+
+      error_msg = @pastel.decorate("#{Tng::UI::Theme.icon(:error)} Tests failed to run", Tng::UI::Theme.color(:error))
+      puts center_text(error_msg)
+
+      @prompt.keypress(center_text(@pastel.decorate("Press any key to continue...", Tng::UI::Theme.color(:muted))))
+      return true
+    end
+
+    false
+  end
+
+  def extract_test_file_from_command(command)
+    # Extract test file path from commands like:
+    # "bundle exec rails test test/controllers/foo_test.rb"
+    # "bundle exec rspec spec/controllers/foo_spec.rb"
+
+    if command.include?("rails test")
+      # For Rails test commands
+      match = command.match(/rails test\s+(.+?)(?:\s|$)/)
+      match[1] if match
+    elsif command.include?("rspec")
+      # For RSpec commands
+      match = command.match(/rspec\s+(.+?)(?:\s|$)/)
+      match[1] if match
+    else
+      nil
+    end
+  end
+
   def initialize_config_and_clients
     @config_initialized = true
 
@@ -868,20 +986,6 @@ class CLI
     )
     @testng = Services::Testng.new(@http_client)
   end
-
-  private
-
-  def format_generation_time(seconds)
-    if seconds < 1
-      "#{(seconds * 1000).round}ms"
-    elsif seconds < 60
-      "#{seconds.round(1)}s"
-    else
-      minutes = (seconds / 60).floor
-      remaining_seconds = (seconds % 60).round
-      "#{minutes}m #{remaining_seconds}s"
-    end
-  end
 end
 
 cli = CLI.new

data/lib/tng/analyzers/controller.rb

@@ -48,37 +48,6 @@ module Tng
         Tng::Analyzer::Controller.model_info_for_controller(file_path)
       end
 
-      def self.find_test_file_for_controller(controller_path)
-        controller_name = File.basename(controller_path, ".rb")
-        testing_framework = Tng.testing_framework
-        return [] if testing_framework.nil?
-
-        paths = if testing_framework.downcase == "rspec"
-                  [
-                    "spec/controllers/#{controller_name}_spec.rb",
-                    "spec/requests/#{controller_name}_spec.rb"
-                  ]
-                else
-                  [
-                    "test/controllers/#{controller_name}_test.rb",
-                    "test/functional/#{controller_name}_test.rb"
-                  ]
-                end
-
-        paths.select { |path| File.exist?(path) }
-      end
-
-      def self.read_test_file_for_controller(controller_path)
-        files = find_test_file_for_controller(controller_path)
-
-        files.map do |file_path|
-          {
-            path: file_path,
-            content: File.read(file_path)
-          }
-        end
-      end
-
       def self.methods_for_controller(controller_name)
         raise "controller_name is required" if controller_name.nil?
 

data/lib/tng/analyzers/model.rb

@@ -14,37 +14,6 @@ module Tng
         Tng::Analyzer::Model.parse_model_file(file_path)
       end
 
-      def self.read_test_file_for_model(model_path)
-        files = find_test_file_for_model(model_path)
-
-        files.map do |file_path|
-          {
-            path: file_path,
-            content: File.read(file_path)
-          }
-        end
-      end
-
-      def self.find_test_file_for_model(model_path)
-        model_name = File.basename(model_path, ".rb")
-        testing_framework = Tng.testing_framework
-        return [] if testing_framework.nil?
-
-        paths = if testing_framework.downcase == "rspec"
-                  [
-                    "spec/models/#{model_name}_spec.rb",
-                    "spec/requests/#{model_name}_spec.rb"
-                  ]
-                else
-                  [
-                    "test/models/#{model_name}_test.rb",
-                    "test/functional/#{model_name}_test.rb"
-                  ]
-                end
-
-        paths.select { |path| File.exist?(path) }
-      end
-
       def self.model_connections(model)
         raise "model is required" if model.nil?
 

data/lib/tng/analyzers/other.rb

@@ -50,33 +50,6 @@ module Tng
         Tng::Analyzer::Service.parse_service_file(file_path)
       end
 
-      def self.read_test_file_for_other(other_path)
-        files = find_test_file_for_other(other_path)
-
-        files.map do |file_path|
-          {
-            path: file_path,
-            content: File.read(file_path)
-          }
-        end
-      end
-
-      def self.find_test_file_for_other(other_path)
-        # Extract the file type and name for test file discovery
-        file_type = determine_file_type(other_path)
-        file_name = File.basename(other_path, ".rb")
-        testing_framework = Tng.testing_framework
-        return [] if testing_framework.nil?
-
-        paths = if testing_framework.downcase == "rspec"
-                  build_rspec_test_paths(file_type, file_name)
-                else
-                  build_minitest_test_paths(file_type, file_name)
-                end
-
-        paths.select { |path| File.exist?(path) }
-      end
-
       def self.methods_for_other(other_name, file_path)
         raise "other_name is required" if other_name.nil?
         raise "file_path is required" if file_path.nil?

data/lib/tng/analyzers/service.rb

@@ -24,37 +24,6 @@ module Tng
         Tng::Analyzer::Service.parse_service_file(file_path)
       end
 
-      def self.read_test_file_for_service(service_path)
-        files = find_test_file_for_service(service_path)
-
-        files.map do |file_path|
-          {
-            path: file_path,
-            content: File.read(file_path)
-          }
-        end
-      end
-
-      def self.find_test_file_for_service(service_path)
-        service_name = File.basename(service_path, ".rb")
-        testing_framework = Tng.testing_framework
-        return [] if testing_framework.nil?
-
-        paths = if testing_framework.downcase == "rspec"
-                  [
-                    "spec/services/#{service_name}_spec.rb",
-                    "spec/service/#{service_name}_spec.rb"
-                  ]
-                else
-                  [
-                    "test/services/#{service_name}_test.rb",
-                    "test/service/#{service_name}_test.rb"
-                  ]
-                end
-
-        paths.select { |path| File.exist?(path) }
-      end
-
       def self.methods_for_service(service_name)
         raise "service_name is required" if service_name.nil?