tng 0.5.4 → 0.5.5

data/bin/tng

@@ -7,10 +7,13 @@ lib_path = File.expand_path("../lib", __dir__)
 $LOAD_PATH.unshift(lib_path) unless $LOAD_PATH.include?(lib_path)
 
 require "tng"
+require "json"
 require "tng/services/direct_generation"
 require "tng/services/extract_methods"
 require "tng/services/file_type_detector"
+require "tng/services/file_discovery"
 require "tng/services/fix_orchestrator"
+require "tng/services/installer"
 
 require "tty-screen"
 require "tty-option"
@@ -77,6 +80,9 @@ class CLI
     example ""
     example "Call Sites:", ""
     example "  bundle exec tng --file=app/services/payment_service.rb --method=process --callsites", ""
+    example ""
+    example "Initialize configuration:", ""
+    example "  bundle exec tng init", ""
   end
 
   option :file do
@@ -183,6 +189,11 @@ class CLI
   end
 
   def start
+    if ARGV.first == "init" || ARGV.first == "install"
+      run_init_command
+      return
+    end
+
     normalized_argv = preprocess_arguments(ARGV.dup)
 
     parse(normalized_argv)
@@ -192,9 +203,13 @@ class CLI
       return
     end
 
-    rails_loaded = load_rails_environment
+    rails_root = find_rails_root || Dir.pwd
+    @rails_project = Tng::Utils.rails_project?(rails_root)
+    @generic_mode = !@rails_project
 
-    initialize_config_and_clients if rails_loaded
+    rails_loaded = @rails_project ? load_rails_environment : false
+
+    initialize_config_and_clients if rails_loaded || @generic_mode
 
     if params[:fix]
       handle_fix_command
@@ -211,7 +226,7 @@ class CLI
     elsif params[:file]
       run_direct_generation
     else
-      if find_rails_root && !rails_loaded && !defined?(Rails)
+      if @rails_project && !rails_loaded && !defined?(Rails)
         puts @pastel.decorate("#{Tng::UI::Theme.icon(:error)} Failed to load Rails environment.", Tng::UI::Theme.color(:error))
         puts @pastel.decorate("Please ensure you're running this command from a Rails application directory.", Tng::UI::Theme.color(:warning))
         puts @pastel.decorate("Try running: bundle exec tng", Tng::UI::Theme.color(:muted))
@@ -344,7 +359,46 @@ class CLI
     @go_ui.stop
   end
 
+  def run_init_command
+    rails_root = find_rails_root
+    project_root = rails_root || Dir.pwd
+    rails_project = Tng::Utils.rails_project?(project_root)
+    target = if rails_project
+               File.join(project_root, "config", "initializers", "tng.rb")
+             else
+               File.join(project_root, "tng.rb")
+             end
+
+    if File.exist?(target)
+      puts "Configuration file already exists at #{target}"
+      print "Do you want to overwrite it? (y/n): "
+      answer = STDIN.gets&.strip&.downcase
+      if answer == "y" || answer == "yes"
+        _created, path = Tng::Services::Installer.install(
+          project_root: project_root,
+          rails_project: rails_project,
+          force: true
+        )
+        puts "TNG configuration file updated at #{path}"
+      else
+        puts "Skipping configuration file creation."
+      end
+    else
+      _created, path = Tng::Services::Installer.install(
+        project_root: project_root,
+        rails_project: rails_project,
+        force: false
+      )
+      puts "TNG configuration file created at #{path}"
+    end
+  end
+
   def handle_test_generation
+    if @generic_mode
+      handle_generic_test_generation
+      return
+    end
+
     choice = @go_ui.show_test_type_menu
 
     case choice
@@ -360,6 +414,11 @@ class CLI
   end
 
   def handle_clone_detection
+    if @generic_mode
+      handle_generic_clone_detection
+      return
+    end
+
     # Pick a file type first to narrow down
     choice = @go_ui.show_test_type_menu("clones")
     return if choice == "back"
@@ -496,6 +555,11 @@ class CLI
   end
 
   def handle_audit_method
+    if @generic_mode
+      handle_generic_audit
+      return
+    end
+
     choice = @go_ui.show_test_type_menu("audit")
 
     case choice
@@ -563,32 +627,40 @@ class CLI
   end
 
   def run_audit_for_controller_method(controller, method_info)
-    result = nil
+    audit_result = nil
     @go_ui.show_progress("Auditing #{controller[:name]}##{method_info[:name]}") do |progress|
       progress.update("Analyzing method context...", 25)
       progress.update("Running logical analysis...", 50)
       progress.update("Detecting issues and behaviours...", 75)
 
-      result = Tng::Services::TestGenerator.new(@http_client).run_audit_for_controller_method(
+      audit_result = Tng::Services::TestGenerator.new(@http_client).run_audit_for_controller_method(
         controller, method_info, progress: progress
       )
 
       progress.update("Processing results...", 100)
 
-      if result&.dig(:error)
-        { message: result[:message] || "Audit failed", error: result[:error] }
+      if audit_result&.dig(:error)
+        { message: audit_result[:message] || "Audit failed", error: audit_result[:error] }
       else
         { message: "Audit complete!" }
       end
     end
 
-    return @go_ui.show_auth_error(result[:message] || "Audit failed") if result&.dig(:error)
+    return @go_ui.show_auth_error(audit_result[:message] || "Audit failed") if audit_result&.dig(:error)
 
-    display_audit_results(result)
+    display_audit_results(audit_result)
   end
 
   def display_audit_results(result)
-    audit_data = result[:audit_results]
+    audit_data = result.is_a?(Hash) ? (result[:audit_results] || result["audit_results"] || result[:result] || result["result"] || result) : result
+    if audit_data.is_a?(String)
+      begin
+        audit_data = JSON.parse(audit_data)
+      rescue JSON::ParserError
+        audit_data = nil
+      end
+    end
+    return if audit_data.nil?
 
     # Send entire audit_data to go-ui (includes issues, behaviours, method_name, class_name, method_source_with_lines)
     @go_ui.show_audit_results(audit_data, "issues")
@@ -784,29 +856,29 @@ class CLI
   end
 
   def run_audit_for_model_method(model, method_info)
-    result = nil
+    audit_result = nil
     @go_ui.show_progress("Auditing #{model[:name]}##{method_info[:name]}") do |progress|
       progress.update("Parsing method details...", 25)
       progress.update("Analyzing method context...", 25)
       progress.update("Running logical analysis...", 50)
       progress.update("Detecting issues and behaviours...", 75)
 
-      result = Tng::Services::TestGenerator.new(@http_client).run_audit_for_model_method(
+      audit_result = Tng::Services::TestGenerator.new(@http_client).run_audit_for_model_method(
         model, method_info, progress: progress
       )
 
       progress.update("Processing results...", 100)
 
-      if result&.dig(:error)
-        { message: result[:message] || "Audit failed", error: result[:error] }
+      if audit_result&.dig(:error)
+        { message: audit_result[:message] || "Audit failed", error: audit_result[:error] }
       else
         { message: "Audit complete!" }
       end
     end
 
-    return @go_ui.show_auth_error(result[:message] || "Audit failed") if result&.dig(:error)
+    return @go_ui.show_auth_error(audit_result[:message] || "Audit failed") if audit_result&.dig(:error)
 
-    display_audit_results(result)
+    display_audit_results(audit_result)
   end
 
   def audit_service_method
@@ -860,29 +932,29 @@ class CLI
   end
 
   def run_audit_for_service_method(service, method_info)
-    result = nil
+    audit_result = nil
     @go_ui.show_progress("Auditing #{service[:name]}##{method_info[:name]}") do |progress|
       progress.update("Parsing method details...", 25)
       progress.update("Analyzing method context...", 25)
       progress.update("Running logical analysis...", 50)
       progress.update("Detecting issues and behaviours...", 75)
 
-      result = Tng::Services::TestGenerator.new(@http_client).run_audit_for_service_method(
+      audit_result = Tng::Services::TestGenerator.new(@http_client).run_audit_for_service_method(
         service, method_info, progress: progress
       )
 
       progress.update("Processing results...", 100)
 
-      if result&.dig(:error)
-        { message: result[:message] || "Audit failed", error: result[:error] }
+      if audit_result&.dig(:error)
+        { message: audit_result[:message] || "Audit failed", error: audit_result[:error] }
       else
         { message: "Audit complete!" }
       end
     end
 
-    return @go_ui.show_auth_error(result[:message] || "Audit failed") if result&.dig(:error)
+    return @go_ui.show_auth_error(audit_result[:message] || "Audit failed") if audit_result&.dig(:error)
 
-    display_audit_results(result)
+    display_audit_results(audit_result)
   end
 
   def audit_other_method
@@ -934,29 +1006,29 @@ class CLI
   end
 
   def run_audit_for_other_method(other_file, method_info)
-    result = nil
+    audit_result = nil
     @go_ui.show_progress("Auditing #{other_file[:name]}##{method_info[:name]}") do |progress|
       progress.update("Parsing method details...", 25)
       progress.update("Analyzing method context...", 25)
       progress.update("Running logical analysis...", 50)
       progress.update("Detecting issues and behaviours...", 75)
 
-      result = Tng::Services::TestGenerator.new(@http_client).run_audit_for_other_method(
+      audit_result = Tng::Services::TestGenerator.new(@http_client).run_audit_for_other_method(
         other_file, method_info, progress: progress
       )
 
       progress.update("Processing results...", 100)
 
-      if result&.dig(:error)
-        { message: result[:message] || "Audit failed", error: result[:error] }
+      if audit_result&.dig(:error)
+        { message: audit_result[:message] || "Audit failed", error: audit_result[:error] }
       else
         { message: "Audit complete!" }
       end
     end
 
-    return @go_ui.show_auth_error(result[:message] || "Audit failed") if result&.dig(:error)
+    return @go_ui.show_auth_error(audit_result[:message] || "Audit failed") if audit_result&.dig(:error)
 
-    display_audit_results(result)
+    display_audit_results(audit_result)
   end
 
   def generate_controller_tests
@@ -1467,6 +1539,11 @@ class CLI
   def initialize_config_and_clients
     @config_initialized = true
 
+    if @generic_mode
+      config_path = File.join(Dir.pwd, "tng.rb")
+      load config_path if File.exist?(config_path)
+    end
+
     unless Tng::Services::UserAppConfig.configured?
       missing = Tng::Services::UserAppConfig.missing_config
       @go_ui.show_config_missing(missing)
@@ -1493,6 +1570,11 @@ class CLI
   end
 
   def handle_trace_method
+    if @generic_mode
+      handle_generic_trace
+      return
+    end
+
     choice = @go_ui.show_test_type_menu("trace")
 
     case choice
@@ -1508,6 +1590,11 @@ class CLI
   end
 
   def handle_impact_method
+    if @generic_mode
+      handle_generic_impact
+      return
+    end
+
     choice = @go_ui.show_test_type_menu("impact")
 
     case choice
@@ -1523,6 +1610,11 @@ class CLI
   end
 
   def handle_xray_method
+    if @generic_mode
+      handle_generic_xray
+      return
+    end
+
     choice = @go_ui.show_test_type_menu("xray")
 
     case choice
@@ -1614,16 +1706,7 @@ class CLI
 
     @go_ui.show_spinner("Tracing #{method_info[:name]}...") do
       begin
-        project_root = Dir.pwd
-        path = File.expand_path(file_info[:path])
-
-        class_name = method_info[:class_name] || file_info[:name]
-        trace_json = Tng::Analyzer::Context.analyze_symbolic_trace(
-          project_root,
-          path,
-          method_info[:name],
-          class_name
-        )
+        trace_json = fetch_trace_json(file_info, method_info, v2: false)
 
         f = Tempfile.new(['trace', '.json'])
         f.write(JSON.generate(trace_json))
@@ -1642,21 +1725,101 @@ class CLI
     end
   end
 
-  def run_impact_for_method(file_info, method_info)
+  def run_trace_for_method_v2(file_info, method_info)
     result_path = nil
 
-    @go_ui.show_spinner("Running regression check for #{method_info[:name]}...") do
+    @go_ui.show_spinner("Tracing #{method_info[:name]} (v2)...") do
       begin
-        project_root = Dir.pwd
-        path = File.expand_path(file_info[:path])
+        trace_json = fetch_trace_json(file_info, method_info, v2: true)
 
-        class_name = method_info[:class_name] || file_info[:name]
-        impact_json = Tng::Analyzer::Context.analyze_impact(
+        f = Tempfile.new(['trace_v2', '.json'])
+        f.write(JSON.generate(trace_json))
+        f.close
+        result_path = f.path
+
+        { success: true, message: "Trace v2 generated" }
+      rescue => e
+        { success: false, message: e.message }
+      end
+    end
+
+    if result_path
+      @go_ui.show_trace_results(result_path)
+      File.unlink(result_path) if File.exist?(result_path)
+    end
+  end
+
+  def fetch_trace_json(file_info, method_info, v2:)
+    project_root = Dir.pwd
+    path = File.expand_path(file_info[:path])
+    # For direct trace runs, avoid forcing a class name from file path.
+    # In many codebases (e.g. service/agent folders), path-derived constants
+    # do not match the real Ruby namespace and would hide valid methods.
+    class_name = method_info[:class_name]
+
+    if @generic_mode
+      if v2
+        Tng::Analyzer::Context.analyze_symbolic_trace_ruby_v2(
           project_root,
           path,
           method_info[:name],
           class_name
         )
+      else
+        Tng::Analyzer::Context.analyze_symbolic_trace_ruby(
+          project_root,
+          path,
+          method_info[:name],
+          class_name
+        )
+      end
+    elsif v2
+      Tng::Analyzer::Context.analyze_symbolic_trace_v2(
+        project_root,
+        path,
+        method_info[:name],
+        class_name
+      )
+    else
+      Tng::Analyzer::Context.analyze_symbolic_trace(
+        project_root,
+        path,
+        method_info[:name],
+        class_name
+      )
+    end
+  end
+
+  def run_impact_for_method(file_info, method_info)
+    result_path = nil
+
+    @go_ui.show_spinner("Running regression check for #{method_info[:name]}...") do
+      begin
+        project_root = Dir.pwd
+        path = File.expand_path(file_info[:path])
+
+        class_name =
+          if @generic_mode
+            method_info[:class_name]
+          else
+            method_info[:class_name] || file_info[:name]
+          end
+        impact_json =
+          if @generic_mode
+            Tng::Analyzer::Context.analyze_impact_ruby(
+              project_root,
+              path,
+              method_info[:name],
+              class_name
+            )
+          else
+            Tng::Analyzer::Context.analyze_impact(
+              project_root,
+              path,
+              method_info[:name],
+              class_name
+            )
+          end
 
         f = Tempfile.new(['impact', '.json'])
         f.write(JSON.generate(impact_json))
@@ -1696,9 +1859,43 @@ class CLI
     file_info = { path: full_path, name: namespaced_name }
     method_info = { name: method_name, class_name: params[:class_name] }
 
+    if params[:json]
+      puts JSON.generate(fetch_trace_json(file_info, method_info, v2: false))
+      return
+    end
+
     run_trace_for_method(file_info, method_info)
   end
 
+  def run_direct_trace_v2
+    file_path = params[:file]
+    method_name = params[:method]
+
+    unless method_name
+      puts @pastel.decorate("Please specify a method name with --method or -m", Tng::UI::Theme.color(:error))
+      return
+    end
+
+    full_path = File.expand_path(file_path)
+    unless File.exist?(full_path)
+      puts @pastel.decorate("File not found: #{full_path}", Tng::UI::Theme.color(:error))
+      return
+    end
+    return unless ensure_not_ignored(full_path)
+
+    relative_path = full_path.gsub("#{Dir.pwd}/", "")
+    namespaced_name = relative_path.sub(/\.rb\z/, "").split("/").map(&:camelize).join("::")
+    file_info = { path: full_path, name: namespaced_name }
+    method_info = { name: method_name, class_name: params[:class_name] }
+
+    if params[:json]
+      puts JSON.generate(fetch_trace_json(file_info, method_info, v2: true))
+      return
+    end
+
+    run_trace_for_method_v2(file_info, method_info)
+  end
+
   def run_direct_call_sites
     file_path = params[:file]
     method_name = params[:method]
@@ -1805,20 +2002,27 @@ class CLI
 
     @go_ui.show_spinner("Generating X-Ray for #{method_info[:name]}...") do
       begin
-        type = Tng::Services::FileTypeDetector.detect_type(file_info[:path])
-        file_object = Tng::Services::FileTypeDetector.build_file_object(file_info[:path], type)
         generator = Tng::Services::TestGenerator.new(@http_client)
 
-        xray_result = case type
-                      when "controller"
-                        generator.run_xray_for_controller_method(file_object, method_info)
-                      when "model"
-                        generator.run_xray_for_model_method(file_object, method_info)
-                      when "service"
-                        generator.run_xray_for_service_method(file_object, method_info)
-                      else
-                        generator.run_xray_for_other_method(file_object, method_info)
-                      end
+        if @generic_mode
+          method_info = method_info.merge(test_type: method_info[:test_type] || "other")
+          file_object = generic_file_object(file_info)
+          xray_result = generator.run_xray_for_ruby_method(file_object, method_info)
+        else
+          type = Tng::Services::FileTypeDetector.detect_type(file_info[:path])
+          file_object = Tng::Services::FileTypeDetector.build_file_object(file_info[:path], type)
+
+          xray_result = case type
+                        when "controller"
+                          generator.run_xray_for_controller_method(file_object, method_info)
+                        when "model"
+                          generator.run_xray_for_model_method(file_object, method_info)
+                        when "service"
+                          generator.run_xray_for_service_method(file_object, method_info)
+                        else
+                          generator.run_xray_for_other_method(file_object, method_info)
+                        end
+        end
 
         if xray_result&.dig(:error)
           { success: false, message: xray_result[:message] || "X-Ray failed" }
@@ -1996,11 +2200,196 @@ class CLI
          next unless m_choice
          m_choice = m_choice.merge(class_name: selected_class || choice[:name])
 
-         yield(choice, m_choice)
+      yield(choice, m_choice)
+    end
+  end
+
+  def generic_file_object(file_info)
+    name = file_info[:name].sub(/\.rb\z/, "")
+    {
+      name: name.split("/").map(&:camelize).join("::"),
+      path: file_info[:path],
+      type: "other"
+    }
+  end
+
+  def list_generic_files
+    root = Dir.pwd
+    files = Tng::Services::FileDiscovery.list_ruby_files(root)
+    files.map do |path|
+      {
+        name: path.gsub(%r{\A#{Regexp.escape(root)}/?}, ""),
+        path: path
+      }
+    end
+  end
+
+  def select_generic_file(action_name)
+    files = nil
+    @go_ui.show_spinner("Analyzing files...") do
+      files = list_generic_files
+      { success: true, message: "Found #{files.length} files" }
+    end
+
+    return @go_ui.show_no_items("ruby files") if files.empty?
+
+    items = files.map { |f| { name: f[:name], path: f[:path] } }
+    selected = @go_ui.show_list_view("Select File to #{action_name}", items)
+    return if selected == "back"
+
+    files.find { |f| f[:name] == selected }
+  end
+
+  def select_generic_method(file_info, action_name)
+    file_object = generic_file_object(file_info)
+    methods = extract_other_methods(file_object)
+    if methods.empty?
+      @go_ui.show_no_items("methods in #{file_info[:name]}")
+      return
+    end
+
+    items = methods.map { |m| { name: m[:name], path: file_info[:name] } }
+    selected = @go_ui.show_list_view("Select Method to #{action_name}", items)
+    return if selected == "back"
+
+    methods.find { |m| m[:name] == selected }
+  end
+
+  def select_generic_method_and_type(action_name)
+    file_info = select_generic_file(action_name)
+    return if file_info.nil?
+
+    method_info = select_generic_method(file_info, action_name)
+    return if method_info.nil?
+
+    test_type = @go_ui.show_ruby_test_type_menu
+    return if test_type == "back"
+
+    [file_info, method_info.merge(test_type: test_type)]
+  end
+
+  def handle_generic_test_generation
+    selection = select_generic_method_and_type("Generate Tests")
+    return if selection.nil?
+
+    file_info, method_info = selection
+    file_object = generic_file_object(file_info)
+
+    @go_ui.display_info(@pastel.bright_white("🎯 Generating test for #{file_info[:name]}##{method_info[:name]}..."))
+
+    result = @go_ui.show_progress("Processing...") do |progress|
+      Tng::Services::TestGenerator.new(@http_client).run_for_ruby_method(
+        file_object,
+        method_info,
+        progress: progress
+      )
+    end
+
+    if result&.dig(:error)
+      @go_ui.display_error(@pastel.red("❌ Test generation failed: #{result[:message]}"))
+      return
+    end
+
+    show_post_generation_menu(result) if result && result[:file_path]
+  end
+
+  def handle_generic_audit
+    selection = select_generic_method_and_type("Audit")
+    return if selection.nil?
+
+    file_info, method_info = selection
+    file_object = generic_file_object(file_info)
+
+    audit_result = nil
+    result = @go_ui.show_progress("Auditing #{file_info[:name]}##{method_info[:name]}") do |progress|
+      progress.update("Analyzing method context...", 25)
+      progress.update("Running logical analysis...", 50)
+      progress.update("Detecting issues and behaviours...", 75)
+
+      audit_result = Tng::Services::TestGenerator.new(@http_client).run_audit_for_ruby_method(
+        file_object,
+        method_info,
+        progress: progress
+      )
+
+      progress.update("Processing results...", 100)
+      audit_result&.dig(:error) ? { message: audit_result[:message] || "Audit failed", error: audit_result[:error] } : { message: "Audit complete!" }
+    end
+
+    return @go_ui.show_auth_error(audit_result[:message] || "Audit failed") if audit_result&.dig(:error)
+
+    display_audit_results(audit_result || result)
+  end
+
+  def handle_generic_xray
+    selection = select_generic_method_and_type("X-Ray")
+    return if selection.nil?
+
+    file_info, method_info = selection
+    file_object = generic_file_object(file_info)
+
+    xray_result = nil
+    @go_ui.show_spinner("Generating X-Ray for #{method_info[:name]}...") do
+      xray_result = Tng::Services::TestGenerator.new(@http_client).run_xray_for_ruby_method(
+        file_object,
+        method_info
+      )
+
+      if xray_result&.dig(:error)
+        { success: false, message: xray_result[:message] || "X-Ray failed" }
+      else
+        { success: true, message: "X-Ray generated" }
       end
+    end
+
+    return @go_ui.display_error(@pastel.red("❌ X-Ray failed: #{xray_result[:message]}")) if xray_result&.dig(:error)
+
+    @go_ui.show_xray_results(xray_result, method_info) if xray_result
+  end
+
+  def handle_generic_trace
+    file_info = select_generic_file("Trace")
+    return if file_info.nil?
+
+    method_info = select_generic_method(file_info, "Trace")
+    return if method_info.nil?
+
+    run_trace_for_method(file_info, method_info)
+  end
+
+  def handle_generic_impact
+    file_info = select_generic_file("Regression Check")
+    return if file_info.nil?
+
+    method_info = select_generic_method(file_info, "Regression Check")
+    return if method_info.nil?
+
+    run_impact_for_method(file_info, method_info)
+  end
+
+  def handle_generic_clone_detection
+    file_info = select_generic_file("Check Duplicates")
+    return if file_info.nil?
+
+    level = @go_ui.show_clone_menu
+    return if level == "back"
+
+    run_clone_detection_for_file(file_info, level)
+  end
+
+  def handle_generic_dead_code
+    file_info = select_generic_file("Analyze Dead Code")
+    return if file_info.nil?
+
+    run_dead_code_detection_interactive(file_info[:path])
   end
 
   def handle_dead_code_detection_menu
+    if @generic_mode
+      handle_generic_dead_code
+      return
+    end
+
     # Pick a file type first to narrow down
     choice = @go_ui.show_test_type_menu("deadcode")
     return if choice == "back"