tng 0.3.0 → 0.3.3

data/lib/tng/ui/go_ui_session.rb

@@ -0,0 +1,406 @@
+# frozen_string_literal: true
+
+require "json"
+require "tempfile"
+require "rbconfig"
+
+module Tng
+  module UI
+    # Ruby wrapper for calling go-ui binary
+    class GoUISession
+      def initialize
+        @binary_path = find_go_ui_binary
+        @running = false
+      end
+
+      def start
+        @running = true
+      end
+
+      def stop
+        @running = false
+      end
+
+      def running?
+        @running
+      end
+
+      # Show main menu and return selected choice
+      # Returns: "tests", "stats", "about", "exit"
+      def show_menu
+        output_file = Tempfile.new(["go_ui_menu", ".json"])
+        output_path = output_file.path
+        output_file.close
+
+        begin
+          system(@binary_path, "menu", "--output", output_path)
+
+          return "exit" unless File.exist?(output_path) && File.size(output_path) > 0
+
+          choice = File.read(output_path).strip
+          choice.empty? ? "exit" : choice
+        rescue StandardError
+          "exit"
+        ensure
+          File.unlink(output_path) if File.exist?(output_path)
+        end
+      end
+
+      # Show test type selection menu
+      # Returns: "controller", "model", "service", "other", "back"
+      def show_test_type_menu
+        output_file = Tempfile.new(["go_ui_test_type", ".json"])
+        output_path = output_file.path
+        output_file.close
+
+        begin
+          system(@binary_path, "rails-test-menu", "--output", output_path)
+
+          return "back" unless File.exist?(output_path) && File.size(output_path) > 0
+
+          choice = File.read(output_path).strip
+          choice.empty? ? "back" : choice
+        rescue StandardError
+          "back"
+        ensure
+          File.unlink(output_path) if File.exist?(output_path)
+        end
+      end
+
+      # Show searchable list view for any component
+      # @param title [String] List title (e.g., "Select Controller")
+      # @param items [Array<Hash>] List of items with :name and :path keys
+      # @return [String] Selected item name or "back"
+      def show_list_view(title, items)
+        data_json = JSON.generate({ title: title, items: items })
+
+        output_file = Tempfile.new(["go_ui_list", ".txt"])
+        output_path = output_file.path
+        output_file.close
+
+        begin
+          system(@binary_path, "list-view", "--data", data_json, "--output", output_path)
+
+          selected = File.read(output_path).strip
+          selected.empty? ? "back" : selected
+        ensure
+          File.unlink(output_path) if File.exist?(output_path)
+        end
+      end
+
+      # Show spinner while executing block
+      # @param message [String] Spinner message
+      # @yield Block that returns Hash with :success and :message keys
+      # @return [Hash] Result from block
+      def show_spinner(message, &block)
+        control_file = Tempfile.new(["go_ui_spinner", ".json"])
+        control_path = control_file.path
+        control_file.close
+
+        # Start spinner in background
+        pid = spawn(@binary_path, "spinner", "--message", message, "--control", control_path)
+
+        begin
+          result = yield
+
+          # Write completion status
+          status = {
+            status: result&.dig(:success) ? "success" : "error",
+            message: result&.dig(:message) || "Done!"
+          }
+          File.write(control_path, JSON.generate(status))
+
+          Process.wait(pid)
+          result
+        rescue StandardError => e
+          status = { status: "error", message: e.message }
+          File.write(control_path, JSON.generate(status))
+          Process.wait(pid)
+          raise
+        ensure
+          File.unlink(control_path) if File.exist?(control_path)
+        end
+      end
+
+      # Show progress bar with steps
+      # @param title [String] Progress title
+      # @yield [ProgressUpdater] Block receives updater and returns Hash with :message and :result
+      # @return [Hash] Result from block
+      def show_progress(title, &block)
+        control_file = Tempfile.new(["go_ui_progress", ".json"])
+        control_path = control_file.path
+        control_file.close
+
+        # Start progress in background (inherit TTY)
+        pid = spawn(@binary_path, "progress", "--title", title, "--control", control_path)
+
+        begin
+          updater = ProgressUpdater.new(control_path)
+          result = yield(updater)
+
+          # Write completion status
+          if result&.dig(:result, :error)
+            # Error already written by updater.error
+          else
+            completion = { type: "complete", message: result&.dig(:message) || "Done!" }
+            File.write(control_path, JSON.generate(completion))
+          end
+
+          Process.wait(pid)
+          result
+        rescue StandardError => e
+          error_data = { type: "error", message: "Error: #{e.message}" }
+          File.write(control_path, JSON.generate(error_data))
+          Process.wait(pid)
+          raise
+        ensure
+          File.unlink(control_path) if File.exist?(control_path)
+        end
+      end
+
+      # Show 'no items found' message
+      # @param item_type [String] Type of items (e.g., "controllers", "models")
+      def show_no_items(item_type)
+        system(@binary_path, "no-items", "--type", item_type)
+      end
+
+      # Show user statistics
+      # @param stats_data [Hash] Statistics data
+      def show_stats(stats_data)
+        stats_json = stats_data ? JSON.generate(stats_data) : "{}"
+        system(@binary_path, "stats", "--data", stats_json)
+      rescue StandardError => e
+        puts "Stats error: #{e.message}"
+      end
+
+      # Show about screen
+      def show_about
+        system(@binary_path, "about")
+      rescue StandardError => e
+        puts "About error: #{e.message}"
+      end
+
+      # Show configuration error
+      # @param missing [Array] List of missing configuration items
+      def show_config_error(missing)
+        data_json = JSON.generate({ missing: missing })
+        system(@binary_path, "config-error", "--data", data_json)
+      rescue StandardError => e
+        puts "Config error: #{e.message}"
+      end
+
+      # Show post-generation menu
+      # @param file_path [String] Generated test file path
+      # @param run_command [String] Command to run tests
+      # @return [String] Selected choice ("run_tests", "copy_command", "back")
+      def show_post_generation_menu(file_path, run_command)
+        data_json = JSON.generate({ file_path: file_path, run_command: run_command })
+
+        output_file = Tempfile.new(["go_ui_post_gen", ".txt"])
+        output_path = output_file.path
+        output_file.close
+
+        begin
+          system(@binary_path, "post-generation-menu", "--data", data_json, "--output", output_path)
+
+          choice = File.read(output_path).strip
+          choice.empty? ? "back" : choice
+        ensure
+          File.unlink(output_path) if File.exist?(output_path)
+        end
+      end
+
+      # Show clipboard success message
+      # @param command [String] Command that was copied
+      def show_clipboard_success(command)
+        system(@binary_path, "clipboard-success", "--command", command)
+      end
+
+      # Show test results
+      # @param title [String] Results title
+      # @param passed [Integer] Number of passed tests
+      # @param failed [Integer] Number of failed tests
+      # @param errors [Integer] Number of errors
+      # @param total [Integer] Total number of tests
+      # @param results [Array] Optional list of individual test results
+      def show_test_results(title, passed, failed, errors, total, results = [])
+        data = {
+          title: title,
+          passed: passed,
+          failed: failed,
+          errors: errors,
+          total: total,
+          results: results
+        }
+
+        data_json = JSON.generate(data)
+        system(@binary_path, "test-results", "--data", data_json)
+      rescue StandardError => e
+        puts "Test results error: #{e.message}"
+      end
+
+      # Show authentication error
+      # @param message [String] Error message to display
+      def show_auth_error(message = "Authentication failed")
+        system(@binary_path, "auth-error", "--message", message)
+      rescue StandardError => e
+        puts "Auth error: #{e.message}"
+      end
+
+      # Show welcome banner
+      # @param version [String] Version to display
+      def show_banner(version)
+        system(@binary_path, "banner", "--version", version)
+      rescue StandardError => e
+        puts "Banner error: #{e.message}"
+      end
+
+      # Show goodbye message
+      def show_goodbye
+        system(@binary_path, "goodbye")
+      rescue StandardError => e
+        puts "Goodbye error: #{e.message}"
+      end
+
+      # Show authentication warning
+      # @param missing_items [String] Description of missing items
+      # @return [String] "continue" or "cancel"
+      def show_auth_warning(missing_items = "")
+        output_file = Tempfile.new(["go_ui_auth", ".txt"])
+        output_path = output_file.path
+        output_file.close
+
+        data = { missing_items: missing_items }
+        data_json = JSON.generate(data)
+
+        pid = spawn(@binary_path, "auth-warning", "--data", data_json, out: output_path)
+        Process.wait(pid)
+
+        result = File.read(output_path).strip
+        result.empty? ? "cancel" : result
+      rescue StandardError => e
+        puts "Auth warning error: #{e.message}"
+        "cancel"
+      ensure
+        output_file&.unlink
+      end
+
+      # Show missing configuration
+      # @param missing_items [Array<String>] List of missing config items
+      def show_config_missing(missing_items)
+        data = { missing_items: Array(missing_items) }
+        data_json = JSON.generate(data)
+        system(@binary_path, "config-missing", "--data", data_json)
+      rescue StandardError => e
+        puts "Config missing error: #{e.message}"
+      end
+
+      # Show post-install message
+      # @param version [String] Version to display
+      def show_post_install(version)
+        system(@binary_path, "post-install", "--version", version)
+      rescue StandardError => e
+        puts "Post install error: #{e.message}"
+      end
+
+      # Show help
+      # @param version [String] Version to display
+      def show_help(version)
+        system(@binary_path, "help-box", "--version", version)
+      rescue StandardError => e
+        puts "Help error: #{e.message}"
+      end
+
+      # Show system status
+      # @param status [Hash] Status data from testng
+      def show_system_status(status)
+        data = {
+          status: status[:status].to_s,
+          message: status[:message],
+          local_version: status[:gem_version],
+          current_version: status[:current_version],
+          user_base_url: status[:user_base_url],
+          server_base_url: status[:server_base_url]
+        }
+        data_json = JSON.generate(data)
+        system(@binary_path, "system-status", "--data", data_json)
+      rescue StandardError => e
+        puts "System status error: #{e.message}"
+      end
+
+      private
+
+      def find_go_ui_binary
+        # Detect platform
+        platform = RUBY_PLATFORM
+        arch = RbConfig::CONFIG["host_cpu"]
+
+        # Normalize architecture
+        arch = case arch
+               when /x86_64|amd64/i then "amd64"
+               when /arm64|aarch64/i then "arm64"
+               else arch
+               end
+
+        # Determine binary name
+        binary_name = if platform.include?("darwin")
+                        "go-ui-darwin-#{arch}"
+                      elsif platform.include?("linux")
+                        "go-ui-linux-#{arch}"
+                      else
+                        raise "Unsupported platform: #{platform}"
+                      end
+
+        # Search paths
+        binary_paths = [
+          File.expand_path("../../../binaries/#{binary_name}", __dir__),
+          File.expand_path("../../../../binaries/#{binary_name}", __FILE__),
+          File.expand_path("binaries/#{binary_name}", Dir.pwd)
+        ]
+
+        binary_paths.each do |path|
+          return path if File.exist?(path) && File.executable?(path)
+        end
+
+        raise "go-ui binary not found. Searched: #{binary_paths.join(", ")}"
+      end
+    end
+
+    # Helper class for updating progress bar from within a block
+    class ProgressUpdater
+      def initialize(control_file)
+        @control_file = control_file
+        @step = 0
+      end
+
+      # Update progress with a new step
+      # @param message [String] Progress message
+      # @param percent [Integer, nil] Optional percentage (0-100)
+      # @param step_increment [Boolean] Whether to increment step counter (default: true)
+      # @param explicit_step [Integer, nil] Optional explicit step index
+      def update(message, percent = nil, step_increment: true, explicit_step: nil)
+        step_idx = if explicit_step
+                     explicit_step
+                   else
+                     step_increment ? @step : (@step > 0 ? @step - 1 : 0)
+                   end
+
+        data = { type: "step", step: step_idx, message: message }
+        data[:percent] = percent if percent
+
+        File.write(@control_file, JSON.generate(data))
+        @step += 1 if step_increment && explicit_step.nil?
+        sleep(0.1) # Give UI time to update
+      end
+
+      # Report an error
+      # @param message [String] Error message
+      def error(message)
+        data = { type: "error", message: message }
+        File.write(@control_file, JSON.generate(data))
+        sleep(0.1)
+      end
+    end
+  end
+end
+

data/lib/tng/utils.rb

@@ -20,26 +20,6 @@ module Tng
       end.join("\n")
     end
 
-    def copy_to_clipboard(text)
-      # Try to copy to clipboard
-      if system("which pbcopy > /dev/null 2>&1")
-        system("echo '#{text}' | pbcopy")
-        success_msg = @pastel.green("✅ Command copied to clipboard!")
-      elsif system("which xclip > /dev/null 2>&1")
-        system("echo '#{text}' | xclip -selection clipboard")
-        success_msg = @pastel.green("✅ Command copied to clipboard!")
-      else
-        success_msg = @pastel.yellow("📋 Copy this command:")
-        puts center_text(success_msg)
-        puts center_text(@pastel.bright_white(text))
-        @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
-        return
-      end
-
-      puts center_text(success_msg)
-      @prompt.keypress(center_text(@pastel.dim("Press any key to continue...")))
-    end
-
     def load_rails_environment
       # Use bundler environment to avoid gem conflicts
       require "bundler/setup"
@@ -101,22 +81,17 @@ module Tng
       parsed_response = JSON.parse(test_content)
 
       if parsed_response["error"]
-        puts "❌ API responded with an error: #{parsed_response["error"]}"
-        return
+        return { error: :generation_failed, message: parsed_response["error"] }
       end
       # Validate required fields
       unless parsed_response["file_content"]
-        puts "❌ API response missing file_content field"
-        puts "📋 Response keys: #{parsed_response.keys.inspect}"
-        return
+        return { error: :invalid_response, message: "API response missing file_content" }
       end
 
       # Handle both possible field names for 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
+        return { error: :invalid_response, message: "API response missing file path" }
       end
 
       begin
@@ -126,9 +101,7 @@ module Tng
         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)

data/lib/tng/version.rb

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

data/lib/tng.rb

@@ -7,25 +7,38 @@ require_relative "tng/utils"
 require_relative "tng/api/http_client"
 require_relative "tng/ui/theme"
 require_relative "tng/ui/post_install_box"
-require_relative "tng/ui/authentication_warning_display"
+require_relative "tng/ui/go_ui_session"
 require_relative "tng/services/test_generator"
 
 require_relative "tng/railtie" if defined?(Rails)
 
 begin
+  require "rbconfig"
+
   platform = RUBY_PLATFORM
-  binary_name = if platform.include?("darwin") # macOS
-                  "tng.bundle"
-                elsif platform.include?("linux")
-                  "tng.so"
-                else
-                  raise "Unsupported platform: #{platform}"
-                end
+  arch = RbConfig::CONFIG["host_cpu"]
+
+  # Normalize architecture names
+  arch = case arch
+         when /x86_64|amd64/i then "x86_64"
+         when /arm64|aarch64/i then "arm64"
+         else arch
+         end
+
+  # Determine OS and extension
+  os = platform.include?("darwin") ? "darwin" : "linux"
+  ext = platform.include?("darwin") ? "bundle" : "so"
 
+  # Try loading in order of preference
   binary_paths = [
-    File.expand_path("../binaries/#{binary_name}", __dir__), # From gem root
-    File.expand_path("../../binaries/#{binary_name}", __FILE__), # Alternative path
-    File.expand_path("binaries/#{binary_name}", __dir__) # Direct from lib
+    # Development: simple name (rake dev output)
+    File.expand_path("tng/tng.#{ext}", __dir__),
+    File.expand_path("../binaries/tng.#{ext}", __dir__),
+    # Production: arch-specific names
+    File.expand_path("../binaries/tng-#{os}-#{arch}.#{ext}", __dir__),
+    # Fallback: current dir
+    File.expand_path("binaries/tng.#{ext}", Dir.pwd),
+    File.expand_path("binaries/tng-#{os}-#{arch}.#{ext}", Dir.pwd)
   ]
 
   loaded = false

data/tng.gemspec

@@ -12,13 +12,13 @@ Gem::Specification.new do |spec|
   spec.description = "TNG (Test Next Generation) is a Rails gem that automatically generates comprehensive test files by analyzing your Ruby code using static analysis and AI. It supports models, controllers, and services with intelligent test case generation."
   spec.homepage = "https://tng.sh/"
   spec.required_ruby_version = ">= 3.1.0"
-  spec.required_rubygems_version = ">= 3.3.11"
+  spec.required_rubygems_version = ">= 3.0"
 
   spec.metadata["allowed_push_host"] = "https://rubygems.org"
   spec.metadata["homepage_uri"] = spec.homepage
-  spec.metadata["source_code_uri"] = "https://github.com/tng-sh/tng-rails"
+  spec.metadata["source_code_uri"] = "https://github.com/tng-sh/tng-rails-public"
   spec.license = "Nonstandard"
-  spec.metadata["license_uri"] = "https://github.com/tng-sh/tng-rails/blob/master/LICENSE.md"
+  spec.metadata["license_uri"] = "https://github.com/tng-sh/tng-rails-public/blob/main/LICENSE.md"
 
   # Package pre-compiled binaries and exclude the Rust source code.
   spec.files = Dir.glob("lib/**/*.rb") +
@@ -31,20 +31,15 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
   spec.extensions = []
 
+  spec.add_dependency "activesupport", ">= 6.0", "< 9.0"
   spec.add_dependency "httpx", "~> 1.5"
   spec.add_dependency "openssl", "~> 3.3"
   spec.add_dependency "pastel", "~> 0.8.0"
   spec.add_dependency "prism", "~> 1.4.0"
   spec.add_dependency "rb_sys", "~> 0.9.91"
   spec.add_dependency "tty-box", "~> 0.7"
-  spec.add_dependency "tty-file", "~> 0.10"
   spec.add_dependency "tty-option", "~> 0.3"
-  spec.add_dependency "tty-progressbar", "~> 0.18.3"
-  spec.add_dependency "tty-prompt", "~> 0.23"
-  spec.add_dependency "tty-reader", "~> 0.9"
   spec.add_dependency "tty-screen", "~> 0.8"
-  spec.add_dependency "tty-spinner", "~> 0.9.3"
-  spec.add_dependency "tty-table", "~> 0.12"
 
   spec.post_install_message = begin
     require_relative "lib/tng/ui/post_install_box"