tng 0.4.4 → 0.4.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6eddc7e29e797e24c60704b80241673219df08aeaede138116921e5b971904d7
-  data.tar.gz: f6376250f73f39ae016cb90f82eaf7f67d1b396eafe92a3718f9e82d4a560322
+  metadata.gz: 70161861ae22db154e31792a86414ad76d14d56d6a5c7ea14de126e26bf2e7a5
+  data.tar.gz: ee5bc28ae08666fa789ff0ca21c67735340c01e850b11afcbf34e80fa0d80d3a
 SHA512:
-  metadata.gz: a835a99f7f8bd93b3cd50d22f4e100aa5f7ed06c67588806b99c86cdaf5a08c4e5bcda43f4234bc1c1491a4c4e2cde3f457d6b83a348224bfec6c820eb28c81a
-  data.tar.gz: 0d2a70f92c28099725e8e2de31a467b6ef1cbe6e3263a788a09ba2408e617bb0efce643eb537d8f9d56ad01656dd2f1a06e8c1daf7b8ba60ba6a90b02391c97d
+  metadata.gz: ebd0fb7dbcbbb795810ae2da51ef4333c0a3e2d226c0cb5f5c5f3de7a965899ef099f7e0874c12f1a040c6f47a8db220b3658d12d5cae8b86b4827d4ade12691
+  data.tar.gz: 87e2aee29fbd86bb25aa2519d1e602b96bfc45a7d9a63ffd3e91697a9dae5eb4503bdf30c0b345b8de7641b91ee947b8c6b2b48a5b3da3f78a03024bcb636e93

data/bin/tng

@@ -64,6 +64,11 @@ class CLI
     desc "Run in audit mode (find issues and behaviours instead of generating tests)"
   end
 
+  flag :trace do
+    long "--trace"
+    desc "Generate and visualize a symbolic trace"
+  end
+
   flag :json do
     long "--json"
     desc "Output results in JSON format"
@@ -116,6 +121,8 @@ class CLI
 
     if params[:fix]
       handle_fix_command
+    elsif params[:trace] && params[:file]
+      run_direct_trace
     elsif params[:file]
       run_direct_generation
     else
@@ -149,6 +156,8 @@ class CLI
         normalized << "--method=#{::Regexp.last_match(2)}"
       when /^(?:--)?(audit|a)$/
         normalized << "--audit"
+      when /^(?:--)?(trace)$/
+        normalized << "--trace"
       when /^(?:--)?(json)$/
         normalized << "--json"
       when /^(?:--)?(fix|x)$/
@@ -208,6 +217,8 @@ class CLI
         handle_test_generation
       when "audit"
         handle_audit_method
+      when "trace"
+        handle_trace_method
       when "stats"
         user_stats
       when "about"
@@ -1069,7 +1080,254 @@ class CLI
     orchestrator = Tng::Services::FixOrchestrator.new(@pastel, @http_client)
     orchestrator.run(file_path)
   end
+
+  def handle_trace_method
+    choice = @go_ui.show_test_type_menu("trace")
+
+    case choice
+    when "controller"
+      trace_controller_method
+    when "model"
+      trace_model_method
+    when "service"
+      trace_service_method
+    when "other"
+      trace_other_method
+    end
+  end
+
+  def trace_controller_method
+    select_controller_and_method("Trace") do |controller, method_info|
+      run_trace_for_method(controller, method_info)
+    end
+  end
+
+  def trace_model_method
+    select_model_and_method("Trace") do |model, method_info|
+      run_trace_for_method(model, method_info)
+    end
+  end
+
+  def trace_service_method
+    select_service_and_method("Trace") do |service, method_info|
+      run_trace_for_method(service, method_info)
+    end
+  end
+
+  def trace_other_method
+    select_other_and_method("Trace") do |file, method_info|
+      run_trace_for_method(file, method_info)
+    end
+  end
+
+  def run_trace_for_method(file_info, method_info)
+    result_path = nil
+
+    @go_ui.show_spinner("Tracing #{method_info[:name]}...") do
+      begin
+        project_root = Dir.pwd
+        path = File.expand_path(file_info[:path])
+
+        trace_json = Tng::Analyzer::Context.analyze_symbolic_trace(
+          project_root,
+          path,
+          method_info[:name],
+          nil
+        )
+
+        f = Tempfile.new(['trace', '.json'])
+        f.write(JSON.generate(trace_json))
+        f.close
+        result_path = f.path
+
+        { success: true, message: "Trace 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 run_direct_trace
+    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
+
+    file_info = { path: full_path, name: File.basename(full_path) }
+    method_info = { name: method_name }
+
+    run_trace_for_method(file_info, method_info)
+  end
+
+  def select_controller_and_method(action_name)
+    controllers = nil
+    @go_ui.show_spinner("Analyzing controllers...") do
+       controllers = Tng::Analyzers::Controller.files_in_dir("app/controllers").map do |file|
+         relative_path = file[:path].gsub(%r{^.*app/controllers/}, "").gsub(".rb", "")
+         namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+         { name: namespaced_name, path: file[:path] }
+       end
+       { success: true, message: "Found #{controllers.length} controllers" }
+    end
+
+    if controllers.empty?
+       @go_ui.show_no_items("controllers")
+       return
+    end
+
+    items = controllers.map { |c| { name: c[:name], path: c[:path] } }
+
+    loop do
+       selected = @go_ui.show_list_view("Select Controller to #{action_name}", items)
+       return if selected == "back"
+
+       choice = controllers.find { |c| c[:name] == selected }
+       next unless choice
+
+       methods = extract_controller_methods(choice)
+       if methods.empty?
+          @go_ui.show_no_items("methods in #{choice[:name]}")
+          next
+       end
+
+       m_items = methods.map { |m| { name: m[:name], path: choice[:name] } }
+       m_selected = @go_ui.show_list_view("Select Method to #{action_name}", m_items)
+       next if m_selected == "back"
+
+       m_choice = methods.find { |m| m[:name] == m_selected }
+       next unless m_choice
+
+       yield(choice, m_choice)
+    end
+  end
+
+  def select_model_and_method(action_name)
+      models = nil
+      @go_ui.show_spinner("Analyzing models...") do
+        models = Tng::Analyzers::Model.files_in_dir("app/models").map do |file|
+          relative_path = file[:path].gsub(%r{^.*app/models/}, "").gsub(".rb", "")
+          namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+          { name: namespaced_name, path: file[:path] }
+        end
+        { success: true, message: "Found #{models.length} models" }
+      end
+
+      return @go_ui.show_no_items("models") if models.empty?
+
+      items = models.map { |m| { name: m[:name], path: m[:path] } }
+
+      loop do
+         selected = @go_ui.show_list_view("Select Model to #{action_name}", items)
+         return if selected == "back"
+
+         choice = models.find { |m| m[:name] == selected }
+         next unless choice
+
+         methods = extract_model_methods(choice)
+         if methods.empty?
+             @go_ui.show_no_items("methods in #{choice[:name]}")
+             next
+         end
+
+         m_items = methods.map { |m| { name: m[:name], path: choice[:name] } }
+         m_selected = @go_ui.show_list_view("Select Method to #{action_name}", m_items)
+         next if m_selected == "back"
+
+         m_choice = methods.find { |m| m[:name] == m_selected }
+         yield(choice, m_choice)
+      end
+  end
+
+  def select_service_and_method(action_name)
+      services = nil
+      @go_ui.show_spinner("Analyzing services...") do
+        services = Tng::Analyzers::Service.files_in_dir.map do |file|
+          relative_path = file[:path].gsub(%r{^.*app/services?/}, "").gsub(".rb", "")
+          namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+          { name: namespaced_name, path: file[:path] }
+        end
+        { success: true, message: "Found #{services.length} services" }
+      end
+
+      return @go_ui.show_no_items("services") if services.empty?
+
+      items = services.map { |s| { name: s[:name], path: s[:path] } }
+
+      loop do
+         selected = @go_ui.show_list_view("Select Service to #{action_name}", items)
+         return if selected == "back"
+
+         choice = services.find { |s| s[:name] == selected }
+         next unless choice
+
+         methods = extract_service_methods(choice)
+         if methods.empty?
+             @go_ui.show_no_items("methods in #{choice[:name]}")
+             next
+         end
+
+         m_items = methods.map { |m| { name: m[:name], path: choice[:name] } }
+         m_selected = @go_ui.show_list_view("Select Method to #{action_name}", m_items)
+         next if m_selected == "back"
+
+         m_choice = methods.find { |m| m[:name] == m_selected }
+         yield(choice, m_choice)
+      end
+  end
+
+  def select_other_and_method(action_name)
+      files = nil
+      @go_ui.show_spinner("Analyzing other files...") do
+        files = Tng::Analyzers::Other.files_in_dir.map do |file|
+          relative_path = file[:relative_path].gsub(".rb", "")
+          namespaced_name = relative_path.split("/").map(&:camelize).join("::")
+          { name: namespaced_name, path: file[:path], type: file[:type] }
+        end
+        { success: true, message: "Found #{files.length} other files" }
+      end
+
+      return @go_ui.show_no_items("other files") if files.empty?
+
+      items = files.map { |f| { name: f[:name], path: f[:path] } }
+
+      loop do
+         selected = @go_ui.show_list_view("Select File to #{action_name}", items)
+         return if selected == "back"
+
+         choice = files.find { |f| f[:name] == selected }
+         next unless choice
+
+         methods = extract_other_methods(choice)
+         if methods.empty?
+             @go_ui.show_no_items("methods in #{choice[:name]}")
+             next
+         end
+
+         m_items = methods.map { |m| { name: m[:name], path: choice[:name] } }
+         m_selected = @go_ui.show_list_view("Select Method to #{action_name}", m_items)
+         next if m_selected == "back"
+
+         m_choice = methods.find { |m| m[:name] == m_selected }
+         yield(choice, m_choice)
+      end
+  end
 end
 
+
+
 cli = CLI.new
 cli.start

data/lib/tng/services/direct_generation.rb

@@ -17,14 +17,10 @@ module Tng
         @go_ui = go_ui
       end
 
-      # ... (rest of methods)
-
       def display_audit_results(result)
         audit_results = result[:audit_results]
         return unless audit_results
 
-        # Use Go UI to display rich audit results
-        # "issues" type handles both issues and behaviours in the unified view
         @go_ui.show_audit_results(audit_results, "issues")
       end
 
@@ -52,8 +48,6 @@ module Tng
         generate_test_for_file(resolved_path, method_name, type)
       end
 
-      private
-
       def suggest_similar_files(file_path)
         base_name = File.basename(file_path, ".rb")
 
@@ -106,11 +100,14 @@ module Tng
 
         if result&.dig(:error)
           @go_ui.display_error(@pastel.red("❌ #{@params[:audit] ? "Audit" : "Test generation"} failed: #{result[:message]}"))
+        elsif result&.dig(:trace_generated)
+          if result[:file_path]
+            @go_ui.show_trace_results(result[:file_path])
+            File.unlink(result[:file_path]) if File.exist?(result[:file_path])
+          end
         elsif @params[:audit]
-          # Audit mode - display results inline
           display_audit_results(result)
         elsif result && result[:file_path]
-          # Test generation mode - show post-generation menu
           @show_post_generation_menu.call(result)
         else
           @go_ui.display_error(@pastel.red("❌ Failed to generate test"))
@@ -129,8 +126,9 @@ module Tng
       def generate_test_result(file_object, method_info, type, progress)
         generator = Tng::Services::TestGenerator.new(@http_client)
 
-        if @params[:audit]
-          # Audit mode - return issues and behaviours
+        if @params[:trace]
+          perform_trace_analysis(file_object, method_info, progress)
+        elsif @params[:audit]
           case type
           when "controller" then generator.run_audit_for_controller_method(file_object, method_info, progress: progress)
           when "model" then generator.run_audit_for_model_method(file_object, method_info, progress: progress)
@@ -138,7 +136,6 @@ module Tng
           else generator.run_audit_for_other_method(file_object, method_info, progress: progress)
           end
         else
-          # Test generation mode
           case type
           when "controller" then generator.run_for_controller_method(file_object, method_info, progress: progress)
           when "model" then generator.run_for_model_method(file_object, method_info, progress: progress)
@@ -147,6 +144,33 @@ module Tng
           end
         end
       end
+
+      def perform_trace_analysis(file_object, method_info, progress)
+        progress.update("Tracing method execution...", 50)
+
+        project_root = Dir.pwd
+        path = File.expand_path(file_object[:path])
+
+        begin
+          trace_json = Tng::Analyzer::Context.analyze_symbolic_trace(
+            project_root,
+            path,
+            method_info[:name],
+            nil
+          )
+
+          f = Tempfile.new(['trace', '.json'])
+          f.write(JSON.generate(trace_json))
+          f.close
+          result_path = f.path
+
+          progress.update("Trace generated!", 100)
+
+          { trace_generated: true, file_path: result_path }
+        rescue => e
+          { error: true, message: e.message }
+        end
+      end
     end
   end
 end

data/lib/tng/ui/go_ui_session.rb

@@ -253,6 +253,14 @@ module Tng
         File.unlink(input_path) if input_path && File.exist?(input_path)
       end
 
+      # Show symbolic trace results
+      # @param trace_file_path [String] Path to JSON trace file
+      def show_trace_results(trace_file_path)
+        system(@binary_path, "trace-results", "--file", trace_file_path)
+      rescue StandardError => e
+        puts "Trace results error: #{e.message}"
+      end
+
       # Show authentication error
       # @param message [String] Error message to display
       def show_auth_error(message = "Authentication failed")
@@ -320,11 +328,65 @@ module Tng
       # Show help
       # @param version [String] Version to display
       def show_help(version)
-        system(@binary_path, "help-box", "--version", version)
+        help_data = {
+          title: "TNG - LLM-Powered Rails Test Generator",
+          usages: [
+            { cmd: "bundle exec tng", desc: "Interactive mode" },
+            { cmd: "bundle exec tng --file=FILE --method=METHOD", desc: "Direct test generation" },
+            { cmd: "bundle exec tng --file=FILE --method=METHOD --audit", desc: "Direct audit mode" },
+            { cmd: "bundle exec tng --file=FILE --method=METHOD --trace", desc: "Symbolic trace mode" }
+          ],
+          features: [
+            "Controllers, Models, Services",
+            "17+ other file types (Jobs, Helpers, Lib, Policies, etc.)",
+            "Per-method test generation",
+            "Code audit for issues & behaviors",
+            "Symbolic execution traces",
+            "Searchable method lists"
+          ],
+          options: [
+            { flag: "-h, --help", desc: "Show this help message" },
+            { flag: "--file=PATH", desc: "Target file path (enables direct CLI mode)" },
+            { flag: "--method=NAME", desc: "Method name to generate test for" },
+            { flag: "--audit", desc: "Run audit mode instead of test generation" },
+            { flag: "--trace", desc: "Run symbolic trace mode" }
+          ],
+          how_to: [
+            "Run 'bundle exec tng' to start the interactive interface",
+            "Select Controller, Model, Service, or Other to test",
+            "Choose a specific method from the list",
+            "Use search/filter to find methods quickly"
+          ],
+          footer: "Happy testing! "
+        }
+
+        system(@binary_path, "help-box", "--version", version, "--data", JSON.generate(help_data))
       rescue StandardError => e
         puts "Help error: #{e.message}"
       end
 
+      # Display error message using standard output (fallback for simple messages)
+      def display_error(message)
+        system(@binary_path, "auth-error", "--message", message, "--title", "Error")
+      end
+
+      # Display warning message
+      def display_warning(message)
+        # Re-using auth-error for now but with different color/prefix if possible, or just puts if binary doesn't support generic warning
+        puts message
+      end
+
+      # Display info message
+      def display_info(message)
+        puts message
+      end
+
+      # Display list of items
+      def display_list(title, items)
+        puts title
+        items.each { |item| puts "  - #{item}" }
+      end
+
       # Show system status
       # @param status [Hash] Status data from testng
       def show_system_status(status)

data/lib/tng/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Tng
-  VERSION = "0.4.4"
+  VERSION = "0.4.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tng
 version: !ruby/object:Gem::Version
-  version: 0.4.4
+  version: 0.4.5
 platform: ruby
 authors:
 - ralucab
@@ -222,7 +222,7 @@ post_install_message: "┌ TNG ────────────────
   \                                                            │\n│  • bundle exec
   tng --help  - Show help information                 │\n│                                                                    │\n│
   \ \U0001F4A1 Generate tests for individual methods with precision            │\n└────────────────────────────────────────────────────────────
-  v0.4.4 ┘\n"
+  v0.4.5 ┘\n"
 rdoc_options: []
 require_paths:
 - lib