tng 0.1.6 → 0.2.1

data/lib/tng/ui/show_help.rb

@@ -24,89 +24,32 @@ class ShowHelp
       @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Usage:"),
       @pastel.public_send(Tng::UI::Theme.color(:primary),
                           "  bundle exec tng") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                     "                           # Interactive mode"),
-      @pastel.public_send(Tng::UI::Theme.color(:primary),
-                          "  bundle exec tng --type=TYPE --file=FILE") + @pastel.public_send(
-                            Tng::UI::Theme.color(:muted), "   # Direct mode"
-                          ),
-      @pastel.public_send(Tng::UI::Theme.color(:primary),
-                          "  bundle exec tng -t TYPE -f FILE") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                                     "           # Short aliases"),
-      @pastel.public_send(Tng::UI::Theme.color(:primary),
-                          "  bundle exec tng -t=TYPE f=FILE") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                                    "            # Mixed format"),
+                                                                     "                           # Interactive mode only"),
       "",
-      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Examples:"),
-      @pastel.public_send(Tng::UI::Theme.color(:success),
-                          "  bundle exec tng -t c -f ping") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                                  "                    # Controller test"),
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Features:"),
       @pastel.public_send(Tng::UI::Theme.color(:success),
-                          "  bundle exec tng -t=c f=ping") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                                 "                     # Mixed format"),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Controllers, Models, Services"),
       @pastel.public_send(Tng::UI::Theme.color(:success),
-                          "  bundle exec tng -t m -f user") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                                  "                    # Model test"),
+                          "  #{Tng::UI::Theme.icon(:bullet)} 17+ other file types (Jobs, Helpers, Lib, Policies, Presenters, Mailers, GraphQL, etc.)"),
       @pastel.public_send(Tng::UI::Theme.color(:success),
-                          "  bundle exec tng -t s -f user_service") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                                         "              # Service test"),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Per-method test generation"),
       @pastel.public_send(Tng::UI::Theme.color(:success),
-                          "  bundle exec tng --type=controller --file=users_controller"),
-      @pastel.public_send(Tng::UI::Theme.color(:success),
-                          "  bundle exec tng -t c -f \"admin/users_controller\"") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                                                        " # Namespaced"),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Searchable method lists"),
       "",
       @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Options:"),
-      @pastel.public_send(Tng::UI::Theme.color(:primary),
-                          "  -t, --type=TYPE") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                     "     Test type (controller, model, service)"),
-      @pastel.public_send(Tng::UI::Theme.color(:primary),
-                          "  -f, --file=FILE") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                     "     File name (without .rb extension)"),
-      @pastel.public_send(Tng::UI::Theme.color(:primary),
-                          "  -m, --method=METHOD") + @pastel.public_send(Tng::UI::Theme.color(:muted),
-                                                                         " Method name (for per-method tests)"),
       @pastel.public_send(Tng::UI::Theme.color(:primary),
                           "  -h, --help") + @pastel.public_send(Tng::UI::Theme.color(:muted),
                                                                 "          Show this help message"),
       "",
-      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Type Aliases:"),
+      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("How to Use:"),
       @pastel.public_send(Tng::UI::Theme.color(:muted),
-                          "  #{Tng::UI::Theme.icon(:bullet)} Controllers: ") + @pastel.public_send(Tng::UI::Theme.color(:secondary),
-                                                                                                   "c, controller"),
-      @pastel.public_send(Tng::UI::Theme.color(:muted),
-                          "  #{Tng::UI::Theme.icon(:bullet)} Models: ") + @pastel.public_send(
-                            Tng::UI::Theme.color(:secondary), "m, mo, model"
-                          ),
-      @pastel.public_send(Tng::UI::Theme.color(:muted),
-                          "  #{Tng::UI::Theme.icon(:bullet)} Services: ") + @pastel.public_send(
-                            Tng::UI::Theme.color(:secondary), "s, se, service"
-                          ),
-      "",
-      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("File Format:"),
-      @pastel.public_send(Tng::UI::Theme.color(:muted),
-                          "  #{Tng::UI::Theme.icon(:bullet)} Controllers: ") + @pastel.public_send(Tng::UI::Theme.color(:success),
-                                                                                                   "users_controller, api/v1/posts_controller"),
-      @pastel.public_send(Tng::UI::Theme.color(:muted),
-                          "  #{Tng::UI::Theme.icon(:bullet)} Models: ") + @pastel.public_send(
-                            Tng::UI::Theme.color(:success), "user, blog/post"
-                          ),
-      @pastel.public_send(Tng::UI::Theme.color(:muted),
-                          "  #{Tng::UI::Theme.icon(:bullet)} Services: ") + @pastel.public_send(
-                            Tng::UI::Theme.color(:success), "user_service, payment/stripe_service"
-                          ),
-      @pastel.public_send(Tng::UI::Theme.color(:muted),
-                          "  #{Tng::UI::Theme.icon(:bullet)} Don't include .rb extension"),
-      "",
-      @pastel.public_send(Tng::UI::Theme.color(:accent)).bold("Pro Tips:"),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Run 'bundle exec tng' to start the interactive interface"),
       @pastel.public_send(Tng::UI::Theme.color(:muted),
-                          "  #{Tng::UI::Theme.icon(:bullet)} Mix formats: ") + @pastel.public_send(Tng::UI::Theme.color(:secondary),
-                                                                                                   "-t=c f=ping") + @pastel.public_send(
-                                                                                                     Tng::UI::Theme.color(:muted), " works perfectly"
-                                                                                                   ),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Select Controller, Model, Service, or Other to test"),
       @pastel.public_send(Tng::UI::Theme.color(:muted),
-                          "  #{Tng::UI::Theme.icon(:bullet)} Use filtering in interactive mode for large projects"),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Choose a specific method from the list"),
       @pastel.public_send(Tng::UI::Theme.color(:muted),
-                          "  #{Tng::UI::Theme.icon(:bullet)} All formats are equivalent - use what feels natural"),
+                          "  #{Tng::UI::Theme.icon(:bullet)} Use search/filter to find methods quickly"),
       "",
       @pastel.public_send(Tng::UI::Theme.color(:secondary)).bold("Happy testing! ") + @pastel.public_send(
         Tng::UI::Theme.color(:accent), "★"

data/lib/tng/utils.rb

@@ -1,5 +1,7 @@
 require "yaml"
 require "json"
+require "fileutils"
+require "prism"
 
 module Tng
   module Utils
@@ -98,26 +100,39 @@ module Tng
       puts "📋 Raw API response: #{test_content[0..200]}..." if ENV["DEBUG"]
       parsed_response = JSON.parse(test_content)
 
+      if parsed_response["error"]
+        puts "❌ API responded with an error: #{parsed_response["error"]}"
+        return
+      end
       # Validate required fields
       unless parsed_response["file_content"]
         puts "❌ API response missing file_content field"
         puts "📋 Response keys: #{parsed_response.keys.inspect}"
-        return nil
+        return
       end
 
       # Handle both possible field names for file path
-      file_path = parsed_response["test_file_path"] || parsed_response["file_path"]
+      file_path = parsed_response["test_file_path"] || parsed_response["file_path"] || parsed_response["file_name"] || parsed_response["file"]
       unless file_path
         puts "❌ API response missing test_file_path or file_path field"
         puts "📋 Response keys: #{parsed_response.keys.inspect}"
-        return nil
+        return
       end
 
-      File.write(file_path, parsed_response["file_content"])
+      begin
+        File.write(file_path, parsed_response["file_content"])
+      rescue Errno::ENOENT
+        # Create directory if it doesn't exist
+        FileUtils.mkdir_p(File.dirname(file_path))
+        File.write(file_path, parsed_response["file_content"])
+      end
       puts "✅ Test generated successfully!"
       absolute_path = File.expand_path(file_path)
       puts "Please review the generated tests at \e]8;;file://#{absolute_path}\e\\#{file_path}\e]8;;\e\\"
 
+      # Count tests in the generated file
+      test_count = count_tests_in_file(file_path)
+
       # Determine run command based on test framework
       run_command = if file_path.include?("/spec/")
                       "bundle exec rspec #{file_path}"
@@ -130,7 +145,8 @@ module Tng
         file_path: file_path,
         absolute_path: absolute_path,
         run_command: run_command,
-        test_class_name: parsed_response["test_class_name"]
+        test_class_name: parsed_response["test_class_name"],
+        test_count: test_count
       }
     end
 
@@ -259,5 +275,51 @@ module Tng
     def self.load_all_active_record_data
       []
     end
+
+    def self.count_tests_in_file(file_path)
+      return 0 unless File.exist?(file_path)
+
+      begin
+        content = File.read(file_path)
+        result = Prism.parse(content)
+        return 0 unless result.success?
+
+        count_test_nodes(result.value)
+      rescue StandardError => e
+        puts "⚠️  Warning: Could not parse #{file_path}: #{e.message}" if ENV["DEBUG"]
+        0
+      end
+    end
+
+    private_class_method def self.count_test_nodes(node)
+      return 0 unless node.respond_to?(:child_nodes)
+
+      count = 0
+
+      # Check if current node is a test
+      count += 1 if test_node?(node)
+
+      # Recursively check child nodes
+      node.child_nodes.each do |child|
+        count += count_test_nodes(child) if child
+      end
+
+      count
+    end
+
+    private_class_method def self.test_node?(node)
+      case node
+      when Prism::DefNode
+        # Minitest: def test_something
+        node.name.to_s.start_with?("test_")
+      when Prism::CallNode
+        # RSpec: it "...", specify "..."
+        # Minitest: test "..."
+        method_name = node.name.to_s
+        %w[it specify test].include?(method_name)
+      else
+        false
+      end
+    end
   end
 end

data/lib/tng/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tng
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.2.1
 platform: ruby
 authors:
 - ralucab
@@ -214,9 +214,12 @@ files:
 - lib/tng.rb
 - lib/tng/analyzers/controller.rb
 - lib/tng/analyzers/model.rb
+- lib/tng/analyzers/other.rb
 - lib/tng/analyzers/service.rb
 - lib/tng/api/http_client.rb
 - lib/tng/railtie.rb
+- lib/tng/services/direct_generation.rb
+- lib/tng/services/extract_methods.rb
 - lib/tng/services/test_generator.rb
 - lib/tng/services/testng.rb
 - lib/tng/services/user_app_config.rb
@@ -227,6 +230,7 @@ files:
 - lib/tng/ui/display_banner.rb
 - lib/tng/ui/goodbye_display.rb
 - lib/tng/ui/model_test_flow_display.rb
+- lib/tng/ui/other_test_flow_display.rb
 - lib/tng/ui/post_install_box.rb
 - lib/tng/ui/service_test_flow_display.rb
 - lib/tng/ui/show_help.rb
@@ -259,16 +263,16 @@ post_install_message: "┌ TNG ────────────────
   \                                                                   │\n│  \U0001F680
   Usage:                                                          │\n│                                                                    │\n│
   \ Interactive mode:                                                 │\n│  • bundle
-  exec tng  - Full interactive CLI                         │\n│                                                                    │\n│
-  \ Direct generation:                                                │\n│  • bundle
-  exec tng -t c -f ping  - Short                           │\n│  • bundle exec tng
-  -t=c f=ping   - Mixed                           │\n│  • bundle exec tng --type=controller
-  --file=ping                   │\n│                                                                    │\n│
-  \ Help:                                                             │\n│  • bundle
-  exec tng --help  - Show all options                      │\n│                                                                    │\n│
-  \ \U0001F4A1 Use -t c for controller, -t m for model                         │\n│
-  \                                                                   │\n└────────────────────────────────────────────────────────────
-  v0.1.6 ┘\n"
+  exec tng  - Interactive CLI with method selection        │\n│                                                                    │\n│
+  \ Features:                                                         │\n│  • Test
+  20+ file types: Controllers, Models, Services + Jobs,      │\n│  Helpers, Lib, Policies,
+  Presenters, Mailers, GraphQL, and more    │\n│  • Select specific methods to test
+  \                                │\n│  • Search and filter through methods                               │\n│
+  \                                                                   │\n│  Help:
+  \                                                            │\n│  • bundle exec
+  tng --help  - Show help information                 │\n│                                                                    │\n│
+  \ \U0001F4A1 Generate tests for individual methods with precision            │\n└────────────────────────────────────────────────────────────
+  v0.2.1 ┘\n"
 rdoc_options: []
 require_paths:
 - lib